org.securegraph.query.QueryBase Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of securegraph-core Show documentation
The newest version!
package org.securegraph.query;

import org.securegraph.*;
import org.securegraph.util.FilterIterable;

import java.util.ArrayList;
import java.util.EnumSet;
import java.util.List;
import java.util.Map;

public abstract class QueryBase implements Query {
    private final Graph graph;
    private final Map propertyDefinitions;
    private final Parameters parameters;

    protected QueryBase(Graph graph, String queryString, Map propertyDefinitions, Authorizations authorizations) {
        this.graph = graph;
        this.propertyDefinitions = propertyDefinitions;
        this.parameters = createParameters(queryString, authorizations);
    }

    protected Parameters createParameters(String queryString, Authorizations authorizations) {
        return new Parameters(queryString, authorizations);
    }

    @Override
    public Iterable vertices() {
        return vertices(FetchHint.ALL);
    }

    @Override
    public abstract Iterable vertices(EnumSet fetchHints);

    @Override
    public Iterable edges() {
        return edges(FetchHint.ALL);
    }

    @Override
    public abstract Iterable edges(EnumSet fetchHints);

    @Override
    public Iterable edges(final String label, EnumSet fetchHints) {
        return new FilterIterable(edges()) {
            @Override
            protected boolean isIncluded(Edge o) {
                return label.equals(o.getLabel());
            }
        };
    }

    @Override
    public Iterable edges(final String label) {
        return edges(label, FetchHint.ALL);
    }

    @Override
    public  Query range(String propertyName, T startValue, T endValue) {
        this.parameters.addHasContainer(new HasContainer(propertyName, Compare.GREATER_THAN_EQUAL, startValue, this.propertyDefinitions));
        this.parameters.addHasContainer(new HasContainer(propertyName, Compare.LESS_THAN_EQUAL, endValue, this.propertyDefinitions));
        return this;
    }

    @Override
    public  Query has(String propertyName, T value) {
        this.parameters.addHasContainer(new HasContainer(propertyName, Compare.EQUAL, value, this.propertyDefinitions));
        return this;
    }

    @Override
    public  Query has(String propertyName, Predicate predicate, T value) {
        this.parameters.addHasContainer(new HasContainer(propertyName, predicate, value, this.propertyDefinitions));
        return this;
    }

    @Override
    public Query skip(int count) {
        this.parameters.setSkip(count);
        return this;
    }

    @Override
    public Query limit(int count) {
        this.parameters.setLimit(count);
        return this;
    }

    public Graph getGraph() {
        return graph;
    }

    public Parameters getParameters() {
        return parameters;
    }

    protected Map getPropertyDefinitions() {
        return propertyDefinitions;
    }

    public static class HasContainer {
        public String key;
        public Object value;
        public Predicate predicate;
        private final Map propertyDefinitions;

        public HasContainer(final String key, final Predicate predicate, final Object value, Map propertyDefinitions) {
            this.key = key;
            this.value = value;
            this.predicate = predicate;
            this.propertyDefinitions = propertyDefinitions;
        }

        public boolean isMatch(Element elem) {
            return this.predicate.evaluate(elem.getProperties(this.key), this.value, this.propertyDefinitions);
        }
    }

    public static class Parameters {
        private final Authorizations authorizations;
        private final String queryString;
        private long limit = 100;
        private long skip = 0;
        private final List hasContainers = new ArrayList();

        public Parameters(String queryString, Authorizations authorizations) {
            this.queryString = queryString;
            this.authorizations = authorizations;
        }

        public void addHasContainer(HasContainer hasContainer) {
            this.hasContainers.add(hasContainer);
        }

        public String getQueryString() {
            return queryString;
        }

        public long getLimit() {
            return limit;
        }

        public void setLimit(long limit) {
            this.limit = limit;
        }

        public long getSkip() {
            return skip;
        }

        public void setSkip(long skip) {
            this.skip = skip;
        }

        public Authorizations getAuthorizations() {
            return authorizations;
        }

        public List getHasContainers() {
            return hasContainers;
        }

        public Parameters clone() {
            Parameters result = new Parameters(this.getQueryString(), this.getAuthorizations());
            result.setSkip(this.getSkip());
            result.setLimit(this.getLimit());
            result.hasContainers.addAll(this.getHasContainers());
            return result;
        }
    }
}