angry1980.neo4j.NodeCountQuery Maven / Gradle / Ivy

Go to download
package angry1980.neo4j;

import com.google.common.collect.ImmutableMap;
import org.neo4j.graphdb.Result;

import java.util.Map;
import java.util.Objects;

public class NodeCountQuery implements Query {

    private static final String QUERY = "match (node)"
            + " where {nodeType} in labels(node)"
            + " return count(node) as result"
            ;

    private final String nodeType;
    private int result;

    public NodeCountQuery(String nodeType) {
        this.nodeType = Objects.requireNonNull(nodeType);
    }

    @Override
    public String getQuery() {
        return QUERY;
    }

    @Override
    public Map getParams() {
        return ImmutableMap.of("nodeType", nodeType);

    }

    @Override
    public NodeCountQuery handle(Result result) {
        this.result = Query.getIntResult(result, "result");
        return this;
    }

    public int getResult() {
        return result;
    }
}