Fix code generation for Events

eclipse-esmf · Oct 13, 2023 · cb4a5d6 · cb4a5d6
1 parent 27fc2e8
commit cb4a5d6
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/...erator/src/main/java/org/eclipse/esmf/aspectmodel/java/pojo/AspectModelJavaGenerator.java b/...erator/src/main/java/org/eclipse/esmf/aspectmodel/java/pojo/AspectModelJavaGenerator.java
@@ -20,10 +20,10 @@
 import org.eclipse.esmf.aspectmodel.java.JavaCodeGenerationConfig;
 import org.eclipse.esmf.aspectmodel.java.JavaGenerator;
 import org.eclipse.esmf.aspectmodel.java.QualifiedName;
-
 import org.eclipse.esmf.characteristic.Enumeration;
 import org.eclipse.esmf.metamodel.Aspect;
 import org.eclipse.esmf.metamodel.ComplexType;
+import org.eclipse.esmf.metamodel.Event;
 
 /**
  * Generates Java Domain classes for an Aspect model and all its contained elements.
@@ -35,9 +35,12 @@ public AspectModelJavaGenerator( final Aspect aspect, final JavaCodeGenerationCo
 
    @Override
    protected Stream<Artifact<QualifiedName, String>> generateArtifacts() {
-      return Stream.of( applyTemplate( Aspect.class, new StructureElementJavaArtifactGenerator<>(), config ),
+      return Stream.of(
+                  applyTemplate( Aspect.class, new StructureElementJavaArtifactGenerator<>(), config ),
                   applyTemplate( ComplexType.class, new StructureElementJavaArtifactGenerator<>(
-                        elements( ComplexType.class ).filter( element -> element.getExtends().isPresent() ).collect( Collectors.toSet() ) ), config ),
+                        elements( ComplexType.class ).filter( element ->
+                              element.getExtends().isPresent() ).collect( Collectors.toSet() ) ), config ),
+                  applyTemplate( Event.class, new StructureElementJavaArtifactGenerator<>(), config ),
                   applyTemplate( Enumeration.class, new EnumerationJavaArtifactGenerator<>(), config ) )
             .flatMap( Function.identity() );
    }