Changeset 42:aa2bb7ac04d9 in AnnotationManagerN4J for src

Timestamp:

Sep 26, 2012, 2:12:46 PM (13 years ago)

Author:

casties

Branch:

default

Message:

more work on resources and targets.

Location:

src/main/java/de/mpiwg/itgroup/annotations

Files:

• Resource.java (added)
• Target.java (added)
• Uri.java (added)
• neo4j/AnnotationStore.java (modified) (6 diffs)

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

@@ -26 +26 @@
 import de.mpiwg.itgroup.annotations.Group;
 import de.mpiwg.itgroup.annotations.Person;
+import de.mpiwg.itgroup.annotations.Resource;
 import de.mpiwg.itgroup.annotations.Tag;
+import de.mpiwg.itgroup.annotations.Target;
+import de.mpiwg.itgroup.annotations.Uri;
 
 /**
@@ -92 +95 @@
     }
 
-    public List<Actor> getActors(String key, String query, NodeTypes type) {
-        ArrayList<Actor> actors = new ArrayList<Actor>();
+    /**
+     * Returns the Node with the given key and value.
+     * Key has to be indexed.
+     * 
+     * @param key
+     * @param value
+     * @param type
+     * @return
+     */
+    public Node getNodeFromIndex(String key, String value, NodeTypes type) {
+        if (key == null || value == null) return null;
+        Node person = getNodeIndex(type).get(key, value).getSingle();
+        return person;
+    }
+
+    /**
+     * Returns list of Actors of given type (Group or Person).
+     * Key has to be indexed.
+     * 
+     * @param key
+     * @param query
+     * @param type
+     * @return
+     */
+    @SuppressWarnings("unchecked")
+    public <T extends Actor> List<T> getActors(String key, String query, NodeTypes type) {
+        ArrayList<T> actors = new ArrayList<T>();
         Index<Node> idx = getNodeIndex(type);
         if (key == null) {
@@ -102 +130 @@
         for (Node actorNode : actorNodes) {
             Actor actor = createActorFromNode(actorNode);
-            actors.add(actor);
+            actors.add((T) actor);
         }
         return actors;
     }
-
-    /**
-     * Returns List of Groups.
-     * Key has to be indexed.
-     * 
+    
+    /**
+     * Returns list of groups. Key has to be indexed.
      * @param key
      * @param query
@@ -116 +142 @@
      */
     public List<Group> getGroups(String key, String query) {
-        ArrayList<Group> groups = new ArrayList<Group>();
-        Index<Node> idx = getNodeIndex(NodeTypes.GROUP);
+        List<Group> groups = getActors(key, query, NodeTypes.GROUP);
+        return groups;
+    }
+    
+    /**
+     * Returns list of Persons. Key has to be indexed.
+     * @param key
+     * @param query
+     * @return
+     */
+    public List<Person> getPersons(String key, String query) {
+        List<Person> persons = getActors(key, query, NodeTypes.PERSON);
+        return persons;
+    }
+    
+    /**
+     * Returns list of uri-like objects of given type (Target or Resource).
+     * Key has to be indexed.
+     * 
+     * @param key
+     * @param query
+     * @param type
+     * @return
+     */
+    @SuppressWarnings("unchecked")
+    public <T extends Uri> List<T> getUris(String key, String query, NodeTypes type) {
+        ArrayList<T> uris = new ArrayList<T>();
+        Index<Node> idx = getNodeIndex(type);
         if (key == null) {
             key = "uri";
             query = "*";
         }
-        IndexHits<Node> groupNodes = idx.query(key, query);
-        for (Node groupNode : groupNodes) {
-            Actor group = createActorFromNode(groupNode);
-            groups.add((Group) group);
-        }
-        return groups;
-    }
-
+        IndexHits<Node> actorNodes = idx.query(key, query);
+        for (Node actorNode : actorNodes) {
+            Uri uri = createUriFromNode(actorNode);
+            uris.add((T) uri);
+        }
+        return uris;
+    }
+
+    
+    /**
+     * Returns list of Targets. Key has to be indexed.
+     * @param key
+     * @param query
+     * @return
+     */
+    public List<Target> getTargets(String key, String query) {
+        List<Target> targets = getUris(key, query, NodeTypes.TARGET);
+        return targets;
+    }
+    
+    /**
+     * Returns list of Resources. Key has to be indexed.
+     * @param key
+     * @param query
+     * @return
+     */
+    public List<Resource> getResources(String key, String query) {
+        List<Resource> targets = getUris(key, query, NodeTypes.RESOURCE);
+        return targets;
+    }
+    
     /**
      * Returns List of Annotations.
@@ -490 +565 @@
         return new Tag(id, uri, name);
 
+    }
+
+    protected Uri createUriFromNode(Node uriNode) {
+        if (uriNode == null) return null;
+        String uri = (String) uriNode.getProperty("uri", null);
+        String type = (String) uriNode.getProperty("TYPE", null);
+        if (type != null && type.equals("TARGET")) {
+            return new Target(uri);
+        } else if (type != null && type.equals("RESOURCE")) {
+            return new Resource(uri);
+        }
+        return null;
     }
 
@@ -981 +1068 @@
     }
 
+    public List<Annotation> getAnnotationsByResource(String resourceUri) {
+        ArrayList<Annotation> ret = new ArrayList<Annotation>();
+        Node tag = getNodeFromIndex("uri", resourceUri, NodeTypes.RESOURCE);
+        Iterable<Relationship> rels = tag.getRelationships(Direction.INCOMING, RelationTypes.ANNOTATES);
+        for (Relationship rel : rels) {
+            Node node = rel.getStartNode();
+            ret.add(createAnnotationFromNode(node));
+        }
+        return ret;
+    }
+
 }