wip

1 files changed, 9 insertions, 2 deletions

diff --git a/container-compiler-plugin/src/main/java/io/trygvis/container/compiler/SqlUnitModel.java b/container-compiler-plugin/src/main/java/io/trygvis/container/compiler/SqlUnitModel.java
index 20d5f3b..bb6fa14 100644
--- a/container-compiler-plugin/src/main/java/io/trygvis/container/compiler/SqlUnitModel.java
+++ b/container-compiler-plugin/src/main/java/io/trygvis/container/compiler/SqlUnitModel.java
@@ -4,15 +4,22 @@ import io.trygvis.container.compiler.model.TypeRef;
 import io.trygvis.persistence.EntityMirror;
 
 import javax.lang.model.element.Element;
+import java.util.Collections;
 import java.util.Map;
 import java.util.TreeMap;
 
+import static java.util.Collections.addAll;
+
 public class SqlUnitModel {
     private Map<TypeRef, EntityMirror> entities = new TreeMap<>();
     private Map<EntityMirror, Element> elements = new TreeMap<>();
 
-    public void add(EntityMirror entity) {
-        entities.put(entity.type, entity);
+    public SqlUnitModel add(EntityMirror... entities) {
+        for (EntityMirror entity : entities) {
+            this.entities.put(entity.type, entity);
+        }
+
+        return this;
     }
 
     public EntityMirror get(TypeRef type) {