Как сохранить древовидную структуру, используя neo4j и gremlin - PullRequest
4 голосов
/ 01 августа 2011

Я хочу сохранить следующую структуру дерева каталогов, используя neo4j локальную базу данных и Gremlin в Java.

           (ROOT)
        /          \
    Dir2           Dir3
    /    \             \
  Dir4   Dir5        Dir6
  /
Dir7 

Я определил метод StorePath (String path).
Что я хочу: когда я вызываю StorePath (путь) с path = "Root \ Dir2 \ Dir4 \ Dir7", тогда данные должны храниться следующим образом

         Root
         /
       Dir2
       /
     Dir4
     /
   Dir7 

где Root и Dir * - вершины с пустыми краями. Пожалуйста, помогите мне с кодом Java.

1 Ответ

6 голосов
/ 04 августа 2011
private static final RelationshipType SUB_DIR = DynamicRelationshipType.withName("SUB_DIR");

public void storePath(String path) {
    Node dir = graphDb.getReferenceNode();
    for (String name : path.split(File.separator)) {
        dir = obtainSubDir(dir, name);
    }
}

private Node obtainSubDir(Node dir, String name) {
    Node subDir = getSubDir(dir,name);
    if (subDir!=null) return subDir;
    return createSubDir(dir, name);
}

private Node getSubDir(Node dir, String name) {
    for (Relationship rel : dir.getRelationships(SUB_DIR, Direction.OUTGOING)) {
        final Node subDir = rel.getEndNode();
        if (subDir.getProperty("name", "").equals(name)) return subDir;
    }
    return null;
}

private Node createSubDir(Node dir, String name) {
    Node subDir = dir.getGraphDatabase().createNode();
    subDir.setProperty("name", name);
    dir.createRelationshipTo(subDir, SUB_DIR);
    return subDir;
}
...