angry1980.audio.dao.Neo4j Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of audio-neo4j Show documentation

There is a newer version: 0.0.10

package angry1980.audio.dao;

import angry1980.audio.model.Neo4jNodeType;
import angry1980.audio.model.Neo4jRelationType;
import angry1980.neo4j.Query;
import angry1980.neo4j.Template;
import org.neo4j.graphdb.*;
import org.neo4j.helpers.collection.IteratorUtil;

import java.util.Objects;
import java.util.Optional;
import java.util.function.Consumer;
import java.util.function.Function;
import java.util.stream.Stream;
import java.util.stream.StreamSupport;

public abstract class Neo4j {

    public static final String ID_PROPERTY_NAME = "id";

    private Template template;

    public Neo4j(GraphDatabaseService graphDB) {
        this.template = new Template(graphDB);
    }

    protected Template getTemplate() {
        return template;
    }

    protected long getId(Node node){
        return (long) node.getProperty(ID_PROPERTY_NAME);
    }

    protected Optional getNode(GraphDatabaseService graphDB, Neo4jNodeType type, long id){
        return Optional.ofNullable(graphDB.findNode(type, ID_PROPERTY_NAME, id));
    }

    protected Node getOrCreateNode(GraphDatabaseService graphDB, Neo4jNodeType type, long id, Function create){
        return getNode(graphDB, type, id).orElseGet(() -> create.apply(createNode(graphDB, type, id)));
    }

    protected Node createNode(GraphDatabaseService graphDB, Neo4jNodeType type, long id){
        Node node = graphDB.createNode(type);
        node.setProperty(ID_PROPERTY_NAME, id);
        return node;
    }

    protected Stream getNodesAsStream(GraphDatabaseService graphDB, Neo4jNodeType type) {
        return StreamSupport.stream(IteratorUtil.asIterable(graphDB.findNodes(type)).spliterator(), false);
    }

    protected Stream getNodeConnectionsAsStream(Node node, Neo4jRelationType type){
        return StreamSupport.stream(node.getRelationships(Direction.OUTGOING, type).spliterator(), false);
    }

}