diff flys-backend/src/main/java/de/intevation/flys/importer/ImportRiver.java @ 188:003ac16812dd

Store annotations in backend. flys-backend/trunk@1518 c6561f87-3c4e-4783-a992-168aeb5c3f6f
author Sascha L. Teichmann <sascha.teichmann@intevation.de>
date Fri, 18 Mar 2011 12:10:33 +0000
parents cf8cbcb6a10d
children bc3747a371cc
line wrap: on
line diff
--- a/flys-backend/src/main/java/de/intevation/flys/importer/ImportRiver.java	Thu Mar 17 18:07:13 2011 +0000
+++ b/flys-backend/src/main/java/de/intevation/flys/importer/ImportRiver.java	Fri Mar 18 12:10:33 2011 +0000
@@ -7,8 +7,13 @@
 
 import org.apache.log4j.Logger;
 
+import de.intevation.flys.model.River;
+
 import de.intevation.flys.utils.FileTools;
 
+import org.hibernate.Session;
+import org.hibernate.Query;
+
 public class ImportRiver
 {
     private static Logger log = Logger.getLogger(ImportRiver.class);
@@ -25,6 +30,8 @@
 
     protected List<ImportAnnotation> annotations;
 
+    protected River peer;
+
     public ImportRiver() {
     }
 
@@ -85,5 +92,29 @@
 
         annotations = aparser.getAnnotations();
     }
+
+    public void storeAnnotations() {
+        River river = getPeer();
+        for (ImportAnnotation annotation: annotations) {
+            annotation.getPeer(river);
+        }
+    }
+
+    public River getPeer() {
+        if (peer == null) {
+            Session session = Importer.sessionHolder.get();
+            Query query = session.createQuery("from River where name=:name");
+            query.setString("name", name);
+            List<River> rivers = query.list();
+            if (rivers.isEmpty()) {
+                peer = new River(name);
+                session.save(peer);
+            }
+            else {
+                peer = rivers.get(0);
+            }
+        }
+        return peer;
+    }
 }
 // vim:set ts=4 sw=4 si et sta sts=4 fenc=utf8 :

http://dive4elements.wald.intevation.org