Improve light DAOs

[lgpl/argeo-commons.git] / server / runtime / org.argeo.server.core / src / main / java / org / argeo / server / dao / AbstractTabularDaoSupport.java

diff --git a/server/runtime/org.argeo.server.core/src/main/java/org/argeo/server/dao/AbstractTabularDaoSupport.java b/server/runtime/org.argeo.server.core/src/main/java/org/argeo/server/dao/AbstractTabularDaoSupport.java
new file mode 100644 (file)
index 0000000..a2195b4
--- /dev/null
+++ b/server/runtime/org.argeo.server.core/src/main/java/org/argeo/server/dao/AbstractTabularDaoSupport.java
@@ -0,0 +1,49 @@
+package org.argeo.server.dao;
+
+import java.util.ArrayList;
+import java.util.HashMap;
+import java.util.List;
+import java.util.Map;
+
+public abstract class AbstractTabularDaoSupport extends
+               AbstractMemoryDaoSupport {
+       private Map<String, List<Object>> tabularView = new HashMap<String, List<Object>>();
+
+       @Override
+       protected Object findInternalRef(Reference reference) {
+               TabularInternalReference tabReference = (TabularInternalReference) reference;
+               return getFromTabularView(tabReference.getTargetTableName(),
+                               tabReference.getTargetRow());
+       }
+
+       protected Object getFromTabularView(String tableName, Integer row) {
+               return tabularView.get(tableName).get(row - 2);
+       }
+
+       protected void registerInTabularView(String tableName, Object object) {
+               if (!tabularView.containsKey(tableName))
+                       tabularView.put(tableName, new ArrayList<Object>());
+               tabularView.get(tableName).add(object);
+       }
+
+       protected static class TabularInternalReference extends Reference {
+               private String targetTableName;
+               private Integer targetRow;
+
+               public TabularInternalReference(Object object, String property,
+                               String targetSheet, Integer targetRow) {
+                       super(object, property, null);
+                       this.targetTableName = targetSheet;
+                       this.targetRow = targetRow;
+               }
+
+               public String getTargetTableName() {
+                       return targetTableName;
+               }
+
+               public Integer getTargetRow() {
+                       return targetRow;
+               }
+
+       }
+}