org.vertexium.query.DefaultVertexQuery Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of vertexium-core Show documentation
There is a newer version: 4.10.0
package org.vertexium.query;

import org.vertexium.*;
import org.vertexium.util.FilterIterable;
import org.vertexium.util.JoinIterable;

import java.util.EnumSet;
import java.util.List;

public class DefaultVertexQuery extends VertexQueryBase implements VertexQuery {
    public DefaultVertexQuery(Graph graph, Vertex sourceVertex, String queryString, Authorizations authorizations) {
        super(graph, sourceVertex, queryString, authorizations);
    }

    @Override
    public QueryResultsIterable vertices(EnumSet fetchHints) {
        Iterable vertices = allVertices(fetchHints);
        return new DefaultGraphQueryIterableWithAggregations<>(getParameters(), vertices, true, true, true, getAggregations());
    }

    private Iterable allVertices(EnumSet fetchHints) {
        List edgeLabels = getParameters().getEdgeLabels();
        String[] edgeLabelsArray = edgeLabels == null || edgeLabels.size() == 0
                ? null
                : edgeLabels.toArray(new String[edgeLabels.size()]);
        Iterable results = getSourceVertex().getVertices(
                getDirection(),
                edgeLabelsArray,
                fetchHints,
                getParameters().getAuthorizations()
        );
        if (getOtherVertexId() != null) {
            results = new FilterIterable(results) {
                @Override
                protected boolean isIncluded(Vertex otherVertex) {
                    return otherVertex.getId().equals(getOtherVertexId());
                }
            };
        }
        if (getParameters().getIds().size() > 0) {
            results = new FilterIterable(results) {
                @Override
                protected boolean isIncluded(Vertex otherVertex) {
                    return getParameters().getIds().contains(otherVertex.getId());
                }
            };
        }
        return results;
    }

    @Override
    public QueryResultsIterable edges(EnumSet fetchHints) {
        Iterable edges = allEdges(fetchHints);
        return new DefaultGraphQueryIterableWithAggregations<>(getParameters(), edges, true, true, true, getAggregations());
    }

    private Iterable allEdges(EnumSet fetchHints) {
        Iterable results = getSourceVertex().getEdges(getDirection(), fetchHints, getParameters().getAuthorizations());
        if (getOtherVertexId() != null) {
            results = new FilterIterable(results) {
                @Override
                protected boolean isIncluded(Edge edge) {
                    return edge.getOtherVertexId(getSourceVertex().getId()).equals(getOtherVertexId());
                }
            };
        }
        return results;
    }

    @Override
    protected QueryResultsIterable extendedData(EnumSet extendedDataFetchHints) {
        EnumSet fetchHints = EnumSet.of(FetchHint.EXTENDED_DATA_TABLE_NAMES);
        return extendedData(extendedDataFetchHints, new JoinIterable<>(
                allVertices(fetchHints),
                allEdges(fetchHints)
        ));
    }

    @Override
    public boolean isAggregationSupported(Aggregation aggregation) {
        if (DefaultGraphQueryIterableWithAggregations.isAggregationSupported(aggregation)) {
            return true;
        }
        return super.isAggregationSupported(aggregation);
    }
}