1 files changed, 9 insertions, 1 deletions
diff --git a/container-compiler-plugin/src/main/java/io/trygvis/container/compiler/MyProcessor.java b/container-compiler-plugin/src/main/java/io/trygvis/container/compiler/MyProcessor.java
index fbbc56d..64ac678 100644
--- a/container-compiler-plugin/src/main/java/io/trygvis/container/compiler/MyProcessor.java
+++ b/container-compiler-plugin/src/main/java/io/trygvis/container/compiler/MyProcessor.java
@@ -25,6 +25,10 @@ import javax.tools.Diagnostic;
 import java.util.HashSet;
 import java.util.Set;
 
+import static io.trygvis.container.compiler.Utils.writeFile;
+import static io.trygvis.persistence.generators.EntityManagerGenerator.generateEntityManager;
+import static io.trygvis.persistence.generators.EntityManagerFactoryGenerator.generateEntityManagerFactory;
+import static io.trygvis.persistence.generators.SequencesGenerator.generateSequences;
 import static java.util.Arrays.asList;
 import static java.util.Collections.emptyList;
 import static java.util.Collections.emptySet;
@@ -104,7 +108,8 @@ public class MyProcessor implements Processor {
         TypeElement log = elements.getTypeElement(Log.class.getCanonicalName());
         TypeElement entity = elements.getTypeElement(Entity.class.getCanonicalName());
 
-        EntityHandler entityHandler = new EntityHandler(processingEnv);
+        SqlUnitModel unit = new SqlUnitModel();
+        EntityHandler entityHandler = new EntityHandler(processingEnv, unit);
 
         Set<TypeElement> sqlEntities = typesIn(roundEnv.getElementsAnnotatedWith(SqlEntity.class));
         Set<PackageElement> packages = ElementFilter.packagesIn(roundEnv.getElementsAnnotatedWith(SqlEntitySet.class));
@@ -136,6 +141,9 @@ public class MyProcessor implements Processor {
         }
 
         entityHandler.phase3(hadErrors);
+        writeFile(processingEnv, generateSequences(unit), null);
+        writeFile(processingEnv, generateEntityManagerFactory(unit), null);
+        writeFile(processingEnv, generateEntityManager(unit), null);
 
         return true;
     }