org.vertexium.query.DefaultExtendedDataQuery 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.EnumSet;

public class DefaultExtendedDataQuery extends QueryBase {
    private final Element element;
    private final String tableName;

    public DefaultExtendedDataQuery(
            Graph graph,
            Element element,
            String tableName,
            String queryString,
            Authorizations authorizations
    ) {
        super(graph, queryString, authorizations);
        this.element = element;
        this.tableName = tableName;
    }

    public Element getElement() {
        return element;
    }

    public String getTableName() {
        return tableName;
    }

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

    @Override
    protected QueryResultsIterable extendedData(FetchHints fetchHints) {
        return new DefaultGraphQueryIterableWithAggregations<>(
                getParameters(),
                getElement().getExtendedData(getTableName()),
                true,
                true,
                true,
                getAggregations()
        );
    }
}