1 files changed, 2 insertions, 7 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 f63c4b2..85523dc 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
@@ -1,23 +1,17 @@
 package io.trygvis.persistence;
 
-import io.trygvis.container.compiler.NotImplementedException;
-import io.trygvis.container.compiler.SqlUnitModel;
 import io.trygvis.container.compiler.model.TypeRef;
 
 import java.util.ArrayList;
 import java.util.List;
 
-import static io.trygvis.persistence.FieldMirror.FieldType.PRIMITIVE;
-import static io.trygvis.persistence.FieldMirror.FieldType.REFERENCE;
-import static java.lang.String.format;
-import static org.apache.commons.lang.StringUtils.join;
-
 public class EntityMirror implements Comparable<EntityMirror> {
     public final GeneratorConfiguration generatorConfiguration;
     public final List<FieldMirror> fields = new ArrayList<>();
     public final List<FieldMirror> idFields = new ArrayList<>();
     public final TypeRef type;
     public final String tableName;
+    public final TypeRef rowType;
     public final TypeRef daoType;
     public final TypeRef utilsType;
     public TypeRef idType;
@@ -27,6 +21,7 @@ public class EntityMirror implements Comparable<EntityMirror> {
         this.type = type;
         this.tableName = tableName;
 
+        this.rowType = new TypeRef(type.plainName + "Row");
         this.daoType = new TypeRef(type.plainName + "Dao").args(type.args);
         this.utilsType = new TypeRef(type.plainName + "Dao.Utils").args(type.args);
     }