From c0c9c358e8703c1af917d7270adbb04160ad34b3 Mon Sep 17 00:00:00 2001 From: Trygve Laugstøl Date: Sun, 4 Aug 2013 10:24:43 +0200 Subject: wip --- .../main/java/io/trygvis/container/compiler/SqlUnitModel.java | 11 +++++++++-- 1 file changed, 9 insertions(+), 2 deletions(-) (limited to 'container-compiler-plugin/src/main/java/io/trygvis/container/compiler/SqlUnitModel.java') 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 entities = new TreeMap<>(); private Map 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) { -- cgit v1.2.3