diff options
author | Trygve Laugstøl <trygvis@inamo.no> | 2013-08-11 22:35:02 +0200 |
---|---|---|
committer | Trygve Laugstøl <trygvis@inamo.no> | 2013-08-11 22:35:02 +0200 |
commit | e78c0a1e4a4ebc71502dceccc9ae640862b7ce9e (patch) | |
tree | 38517a79011ee280d08da06ce20dba1ac7aebd62 /container-compiler-plugin/src/main/java/io/trygvis/container/compiler/SqlUnitModel.java | |
parent | 41704c6a57d53b90b4e764ea4fb9c8ff81ced500 (diff) | |
download | container-playground-e78c0a1e4a4ebc71502dceccc9ae640862b7ce9e.tar.gz container-playground-e78c0a1e4a4ebc71502dceccc9ae640862b7ce9e.tar.bz2 container-playground-e78c0a1e4a4ebc71502dceccc9ae640862b7ce9e.tar.xz container-playground-e78c0a1e4a4ebc71502dceccc9ae640862b7ce9e.zip |
o Overhauling inheritance, adding a decent start of support for @MappedSuperclass.
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 | 13 |
1 files changed, 10 insertions, 3 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 717715e..1c0f538 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 @@ -10,7 +10,8 @@ import java.util.TreeMap; public class SqlUnitModel { private Map<TypeRef, EntityMirror> entities = new TreeMap<>(); - private Map<EntityMirror, Element> entityElements = new TreeMap<>(); + private Map<Element, EntityMirror> elementToEntity = new TreeMap<>(new ElementComparator()); + private Map<EntityMirror, Element> entityToElement = new TreeMap<>(); private Map<String, SequenceMirror> sequences = new TreeMap<>(); private Map<SequenceMirror, Element> sequenceElements = new TreeMap<>(); private String packageName; @@ -44,7 +45,8 @@ public class SqlUnitModel { public SqlUnitModel add(EntityMirror entity, Element element) { entities.put(entity.type, entity); - entityElements.put(entity, element); + entityToElement.put(entity, element); + elementToEntity.put(element, entity); return this; } @@ -70,7 +72,11 @@ public class SqlUnitModel { } public Element element(EntityMirror entity) { - return entityElements.get(entity); + return entityToElement.get(entity); + } + + public EntityMirror entity(Element element) { + return elementToEntity.get(element); } // ----------------------------------------------------------------------- @@ -120,4 +126,5 @@ public class SqlUnitModel { public Element element(SequenceMirror sequence) { return sequenceElements.get(sequence); } + } |