--- a/flys-backend/src/main/java/de/intevation/flys/importer/ImporterSession.java	Sun May 08 22:41:07 2011 +0000
+++ b/flys-backend/src/main/java/de/intevation/flys/importer/ImporterSession.java	Mon May 09 00:15:45 2011 +0000
@@ -3,6 +3,7 @@
 import java.util.Iterator;
 import java.util.Map;
 import java.util.HashMap;
+import java.util.LinkedHashMap;
 
 import java.math.BigDecimal;
 
@@ -26,6 +27,9 @@
 {
     private static Logger log = Logger.getLogger(ImporterSession.class);
 
+    public static final int MAX_WST_CACHE_SIZE = 20;
+    public static final int MAX_AT_CACHE_SIZE  = 20;
+
     private static final ThreadLocal<ImporterSession> SESSION =
         new ThreadLocal<ImporterSession>() {
             @Override
@@ -36,9 +40,10 @@
 
     protected Session databaseSession;
 
-    protected Map<IdValueKey, WstColumnValue> wstColumnValues;
+    protected Map<Integer, Map<ValueKey, WstColumnValue>> wstColumnValues;
 
-    protected Map<IdValueKey, DischargeTableValue> dischargeTableValues;
+    protected Map<Integer, Map<ValueKey, DischargeTableValue>>
+        dischargeTableValues;
 
     protected Map<IdValueKey, Range> ranges;
 
@@ -51,6 +56,26 @@
             SessionFactoryProvider.createSessionFactory();
         databaseSession = sessionFactory.openSession();
         databaseSession.setFlushMode(FlushMode.MANUAL);
+
+        wstColumnValues =
+            new LinkedHashMap<Integer, Map<ValueKey, WstColumnValue>>() {
+                @Override
+                protected boolean removeEldestEntry(
+                    Map.Entry<Integer, Map<ValueKey, WstColumnValue>> eldest
+                ) {
+                    return size() > MAX_WST_CACHE_SIZE;
+                }
+            };
+
+        dischargeTableValues =
+            new LinkedHashMap<Integer, Map<ValueKey, DischargeTableValue>>() {
+                @Override
+                protected boolean removeEldestEntry(
+                    Map.Entry<Integer, Map<ValueKey, DischargeTableValue>> eldest
+                ) {
+                    return size() > MAX_AT_CACHE_SIZE;
+                }
+            };
     }
 
     public Session getDatabaseSession() {
@@ -62,13 +87,25 @@
         BigDecimal position,
         BigDecimal w
     ) {
-        if (wstColumnValues == null) {
-            loadWstColumnValues();
+        Integer c = column.getId();
+
+        Map<ValueKey, WstColumnValue> map = wstColumnValues.get(c);
+
+        if (map == null) {
+            map = new HashMap<ValueKey, WstColumnValue>();
+            wstColumnValues.put(c, map);
+            Query query = databaseSession.createQuery(
+                "from WstColumnValue where wstColumn.id=:cid");
+            query.setParameter("cid", c);
+            for (Iterator iter = query.iterate(); iter.hasNext();) {
+                WstColumnValue wcv = (WstColumnValue)iter.next();
+                map.put(new ValueKey(wcv.getPosition(), wcv.getW()), wcv);
+            }
         }
 
-        IdValueKey key = new IdValueKey(column.getId(), position, w);
+        ValueKey key = new ValueKey(position, w);
 
-        WstColumnValue wcv = wstColumnValues.get(key);
+        WstColumnValue wcv = map.get(key);
 
         if (wcv != null) {
             return wcv;
@@ -78,62 +115,48 @@
 
         databaseSession.save(wcv);
 
-        wstColumnValues.put(key, wcv);
+        map.put(key, wcv);
 
         return wcv;
     }
 
-    protected void loadWstColumnValues() {
-        log.info("load wst column values");
-        wstColumnValues = new HashMap<IdValueKey, WstColumnValue>();
-
-        Query query = databaseSession.createQuery("from WstColumnValue");
-
-        for (Iterator iter = query.iterate(); iter.hasNext();) {
-            WstColumnValue wcv = (WstColumnValue)iter.next();
-            wstColumnValues.put(new IdValueKey(wcv), wcv);
-        }
-        log.info(wstColumnValues.size() + " values loaded");
-    }
-
     public DischargeTableValue getDischargeTableValue(
         DischargeTable table,
         BigDecimal     q,
         BigDecimal     w
     ) {
-        if (dischargeTableValues == null) {
-            loadDischargeTableValues();
-        }
+        Integer t = table.getId();
 
-        IdValueKey key = new IdValueKey(table.getId(), q, w);
+        Map<ValueKey, DischargeTableValue> map =
+            dischargeTableValues.get(t);
 
-        DischargeTableValue dtv = dischargeTableValues.get(key);
-
-        if (dtv != null) {
-            return dtv;
+        if (map == null) {
+            map = new HashMap<ValueKey, DischargeTableValue>();
+            dischargeTableValues.put(t, map);
+            Query query = databaseSession.createQuery(
+                "from DischargeTableValue where dischargeTable.id=:tid");
+            query.setParameter("tid", t);
+            for (Iterator iter = query.iterate(); iter.hasNext();) {
+                DischargeTableValue dctv = (DischargeTableValue)iter.next();
+                map.put(new ValueKey(q, w), dctv);
+            }
         }
 
-        dtv = new DischargeTableValue(table, q, w);
-
-        databaseSession.save(dtv);
-
-        dischargeTableValues.put(key, dtv);
-
-        return dtv;
-    }
+        ValueKey key = new ValueKey(q, w);
 
-    protected void loadDischargeTableValues() {
-        log.info("load discharge table values");
-
-        dischargeTableValues = new HashMap<IdValueKey, DischargeTableValue>();
+        DischargeTableValue dctv = map.get(key);
 
-        Query query = databaseSession.createQuery("from DischargeTableValue");
+        if (dctv != null) {
+            return dctv;
+        }
 
-        for (Iterator iter = query.iterate(); iter.hasNext();) {
-            DischargeTableValue dtv = (DischargeTableValue)iter.next();
-            dischargeTableValues.put(new IdValueKey(dtv), dtv);
-        }
-        log.info(dischargeTableValues.size() + " values loaded");
+        dctv = new DischargeTableValue(table, q, w);
+
+        databaseSession.save(dctv);
+
+        map.put(key, dctv);
+
+        return dctv;
     }
 
     public Range getRange(River river, BigDecimal a, BigDecimal b) {