From df92538ab3d83da9839f08b28fc8a67317565463 Mon Sep 17 00:00:00 2001
From: Trygve Laugstøl <trygvis@inamo.no>
Date: Sat, 3 Aug 2013 20:21:19 +0200
Subject: wip

---
 .../trygvis/container/compiler/SqlUnitModel.java   | 33 ++++++++++++++++++++++
 1 file changed, 33 insertions(+)
 create mode 100644 container-compiler-plugin/src/main/java/io/trygvis/container/compiler/SqlUnitModel.java

(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
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);
+    }
+}
-- 
cgit v1.2.3