1 files changed, 17 insertions, 2 deletions
diff --git a/container-compiler-plugin/src/main/java/io/trygvis/persistence/EntityMirror.java b/container-compiler-plugin/src/main/java/io/trygvis/persistence/EntityMirror.java
index 3f6816d..e3d0ec8 100644
--- a/container-compiler-plugin/src/main/java/io/trygvis/persistence/EntityMirror.java
+++ b/container-compiler-plugin/src/main/java/io/trygvis/persistence/EntityMirror.java
@@ -3,10 +3,12 @@ package io.trygvis.persistence;
 import io.trygvis.container.compiler.CompilerException;
 import io.trygvis.container.compiler.model.TypeRef;
 
+import javax.lang.model.element.ExecutableElement;
 import java.util.ArrayList;
 import java.util.List;
 
 import static java.util.Collections.addAll;
+import static java.util.Collections.unmodifiableList;
 
 public class EntityMirror implements Comparable<EntityMirror> {
     public final GeneratorConfiguration generatorConfiguration;
@@ -21,14 +23,19 @@ public class EntityMirror implements Comparable<EntityMirror> {
     /**
      * The nearest @MappedSuperclass
      */
-    public EntityMirror superEntity;
+    public final EntityMirror superEntity;
+    private final List<String> constructorParameters;
+    public ExecutableElement constructor;
 
     public EntityMirror(GeneratorConfiguration generatorConfiguration, TypeRef type, boolean concrete,
-                        EntityMirror superEntity, String tableName) {
+                        EntityMirror superEntity, ExecutableElement constructor, List<String> constructorParameters,
+                        String tableName) {
         this.generatorConfiguration = generatorConfiguration;
         this.type = type;
         this.concrete = concrete;
         this.superEntity = superEntity;
+        this.constructor = constructor;
+        this.constructorParameters = unmodifiableList(constructorParameters);
 
         if (concrete) {
             this.tableName = tableName;
@@ -67,6 +74,14 @@ public class EntityMirror implements Comparable<EntityMirror> {
         return fields;
     }
 
+    public List<String> getConstructorParameters() {
+        return constructorParameters;
+    }
+
+    public boolean hasConstructorParameter(String parameter) {
+        return constructorParameters.contains(parameter);
+    }
+
     // -----------------------------------------------------------------------
     //
     // -----------------------------------------------------------------------