Changeset 5:bbf0cc5bee29 in AnnotationManagerN4J for src/main/java/de/mpiwg/itgroup/annotations/neo4j/AnnotationStore.java

Timestamp:

Jul 3, 2012, 7:23:17 PM (12 years ago)

Author:

casties

Branch:

default

Message:

version 0.2 really works now

File:

• src/main/java/de/mpiwg/itgroup/annotations/neo4j/AnnotationStore.java (modified) (14 diffs)

src/main/java/de/mpiwg/itgroup/annotations/neo4j/AnnotationStore.java

@@ -9 +9 @@
 
 import org.apache.log4j.Logger;
+import org.neo4j.graphdb.Direction;
 import org.neo4j.graphdb.GraphDatabaseService;
 import org.neo4j.graphdb.Node;
@@ -59 +60 @@
         annot.setBodyText((String) annotNode.getProperty("bodyText", null));
         annot.setBodyUri((String) annotNode.getProperty("bodyUri", null));
-        Iterable<Relationship> targets = annotNode.getRelationships(RelationTypes.ANNOTATES);
-        for (Relationship target : targets) {
+        // get annotation target from relation
+        Iterable<Relationship> targetRels = annotNode.getRelationships(RelationTypes.ANNOTATES);
+        for (Relationship targetRel : targetRels) {
+            Node target = targetRel.getEndNode();
             annot.setTargetBaseUri((String) target.getProperty("uri", null));
+            // just the first one
             break;
         }
@@ -69 +73 @@
             annot.setFragmentType(FragmentTypes.valueOf(ft));
         }
-        Iterable<Relationship> creators = annotNode.getRelationships(RelationTypes.CREATED);
-        for (Relationship creator : creators) {
+        // get creator form relation
+        Iterable<Relationship> creatorRels = annotNode.getRelationships(RelationTypes.CREATED);
+        for (Relationship creatorRel : creatorRels) {
+            Node creator = creatorRel.getStartNode();
             annot.setCreatorUri((String) creator.getProperty("uri", null));
+            annot.setCreatorName((String) creator.getProperty("name", null));
+            // just the first one
             break;
         }
@@ -86 +94 @@
      */
     public Annotation storeAnnotation(Annotation annot) {
+        Node annotNode = null;
         Transaction tx = graphDb.beginTx();
-        Node annotNode = null;
-        String id = annot.getUri();
-        if (id == null) {
-            id = createRessourceURI("annot:");
-        }
-        annotNode = createAnnotationNode(id);
-
         try {
-            // Mutating operations go here
+            /*
+             * create or get the annotation
+             */
+            String id = annot.getUri();
+            if (id == null) {
+                id = createRessourceURI("annot:");
+            }
+            annotNode = getOrCreateAnnotationNode(id);
+
             /*
              * the annotation body
@@ -113 +123 @@
             String targetBaseUri = annot.getTargetBaseUri();
             if (targetBaseUri != null) {
-                Node target = createTargetNode(targetBaseUri);
-                annotNode.createRelationshipTo(target, RelationTypes.ANNOTATES);
+                Node target = getOrCreateTargetNode(targetBaseUri);
+                getOrCreateRelation(annotNode, RelationTypes.ANNOTATES, target);
             }
 
@@ -128 +138 @@
 
             /*
+             * The name of the creator of this annotation.
+             */
+            String creatorName = annot.getCreatorName();
+
+            /*
              * The URI of the creator of this annotation.
              */
             String creatorUri = annot.getCreatorUri();
             if (creatorUri != null) {
-                Node creator = createPersonNode(creatorUri, null);
-                creator.createRelationshipTo(annotNode, RelationTypes.CREATED);
+                Node creator = getOrCreatePersonNode(creatorUri, creatorName);
+                getOrCreateRelation(creator, RelationTypes.CREATED, annotNode);
             }
 
@@ -181 +196 @@
             if (target != null) {
                 Iterable<Relationship> relations = target.getRelationships(RelationTypes.ANNOTATES);
-                for (Relationship relation: relations) {
+                for (Relationship relation : relations) {
                     Node ann = relation.getStartNode();
                     if (ann.getProperty("TYPE", "").equals("ANNOTATION")) {
@@ -187 +202 @@
                         annotations.add(annot);
                     } else {
-                        logger.error("ANNOTATES relation does not start with ANNOTATION: "+ann);
+                        logger.error("ANNOTATES relation does not start with ANNOTATION: " + ann);
                     }
                 }
@@ -198 +213 @@
             if (person != null) {
                 Iterable<Relationship> relations = person.getRelationships(RelationTypes.CREATED);
-                for (Relationship relation: relations) {
+                for (Relationship relation : relations) {
                     Node ann = relation.getEndNode();
                     if (ann.getProperty("TYPE", "").equals("ANNOTATION")) {
@@ -204 +219 @@
                         annotations.add(annot);
                     } else {
-                        logger.error("CREATED relation does not end with ANNOTATION: "+ann);
+                        logger.error("CREATED relation does not end with ANNOTATION: " + ann);
                     }
                 }
@@ -212 +227 @@
     }
 
-    protected Node createAnnotationNode(String id) {
+    protected Relationship getOrCreateRelation(Node start, RelationshipType type, Node end) {
+        if (start.hasRelationship()) {
+            // there are relations
+            Iterable<Relationship> rels = start.getRelationships(type, Direction.OUTGOING);
+            for (Relationship rel : rels) {
+                if (rel.getEndNode().equals(end)) {
+                    // relation exists
+                    return rel;
+                }
+            }
+        }
+        // create new one
+        Relationship rel;
+        Transaction tx = graphDb.beginTx();
+        try {
+            rel = start.createRelationshipTo(end, type);
+            tx.success();
+        } finally {
+            tx.finish();
+        }
+        return rel;
+    }
+
+    protected Node getOrCreateAnnotationNode(String id) {
         Index<Node> idx = graphDb.index().forNodes("annotations");
         IndexHits<Node> annotations = idx.get("id", id);
@@ -232 +270 @@
     }
 
-    protected Node createTargetNode(String uri) {
+    protected Node getOrCreateTargetNode(String uri) {
         Index<Node> idx = graphDb.index().forNodes("targets");
         IndexHits<Node> targets = idx.get("uri", uri);
@@ -252 +290 @@
     }
 
-    protected Node createPersonNode(String uri, String name) {
+    protected Node getOrCreatePersonNode(String uri, String name) {
         Index<Node> idx = graphDb.index().forNodes("persons");
         // Person is identified by URI
@@ -291 +329 @@
     }
 
-
 }