org.vertexium.query.ExtendedDataQueryableIterable 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 java.util.Iterator;

public class ExtendedDataQueryableIterable implements QueryableIterable {
    private final Graph graph;
    private final Element element;
    private final String tableName;
    private final Iterable rows;

    public ExtendedDataQueryableIterable(Graph graph, Element element, String tableName, Iterable rows) {
        this.graph = graph;
        this.element = element;
        this.tableName = tableName;
        this.rows = rows;
    }

    public Graph getGraph() {
        return graph;
    }

    public Element getElement() {
        return element;
    }

    public String getTableName() {
        return tableName;
    }

    @Override
    public Query query(Authorizations authorizations) {
        return query(null, authorizations);
    }

    @Override
    public Query query(String queryString, Authorizations authorizations) {
        if (getGraph() instanceof GraphWithSearchIndex) {
            GraphWithSearchIndex graphWithSearchIndex = (GraphWithSearchIndex) getGraph();
            return graphWithSearchIndex.getSearchIndex().queryExtendedData(
                getGraph(),
                getElement(),
                getTableName(),
                queryString,
                authorizations
            );
        }
        return new DefaultExtendedDataQuery(getGraph(), getElement(), getTableName(), queryString, authorizations);
    }

    @SuppressWarnings("unchecked")
    @Override
    public Iterator iterator() {
        return (Iterator) rows.iterator();
    }
}