diff options
author | Trygve Laugstøl <trygvis@inamo.no> | 2013-08-03 20:21:19 +0200 |
---|---|---|
committer | Trygve Laugstøl <trygvis@inamo.no> | 2013-08-03 20:21:19 +0200 |
commit | df92538ab3d83da9839f08b28fc8a67317565463 (patch) | |
tree | a958d3cce1d3a9045c79371b182adb1ea6cf5a73 /container-compiler-plugin/src/main/java/io/trygvis/container/compiler/SqlUnitModel.java | |
parent | e1cf8889628d2d31cf7067b8c002f229fc22007d (diff) | |
download | container-playground-df92538ab3d83da9839f08b28fc8a67317565463.tar.gz container-playground-df92538ab3d83da9839f08b28fc8a67317565463.tar.bz2 container-playground-df92538ab3d83da9839f08b28fc8a67317565463.tar.xz container-playground-df92538ab3d83da9839f08b28fc8a67317565463.zip |
wip
Diffstat (limited to 'container-compiler-plugin/src/main/java/io/trygvis/container/compiler/SqlUnitModel.java')
-rw-r--r-- | container-compiler-plugin/src/main/java/io/trygvis/container/compiler/SqlUnitModel.java | 33 |
1 files changed, 33 insertions, 0 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 new file mode 100644 index 0000000..20d5f3b --- /dev/null +++ b/container-compiler-plugin/src/main/java/io/trygvis/container/compiler/SqlUnitModel.java @@ -0,0 +1,33 @@ +package io.trygvis.container.compiler; + +import io.trygvis.container.compiler.model.TypeRef; +import io.trygvis.persistence.EntityMirror; + +import javax.lang.model.element.Element; +import java.util.Map; +import java.util.TreeMap; + +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 EntityMirror get(TypeRef type) { + EntityMirror entity = entities.get(type); + if (entity == null) { + throw new InternalErrorException("Could not find entity for " + type.fqName); + } + return entity; + } + + public Map<TypeRef, EntityMirror> getEntities() { + return entities; + } + + public Element elementForEntity(EntityMirror entity) { + return elements.get(entity); + } +} |