org.vertexium.elasticsearch5.ElasticsearchSearchExtendedDataQuery Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of vertexium-elasticsearch5 Show documentation

There is a newer version: 4.10.0

Show newest version

package org.vertexium.elasticsearch5;

import org.elasticsearch.client.Client;
import org.elasticsearch.index.query.BoolQueryBuilder;
import org.elasticsearch.index.query.QueryBuilder;
import org.elasticsearch.index.query.QueryBuilders;
import org.vertexium.Authorizations;
import org.vertexium.FetchHints;
import org.vertexium.Graph;
import org.vertexium.query.GraphQuery;

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

public class ElasticsearchSearchExtendedDataQuery extends ElasticsearchSearchQueryBase implements GraphQuery {
    private final String elementId;
    private final String tableName;

    public ElasticsearchSearchExtendedDataQuery(
        Client client,
        Graph graph,
        String elementId,
        String tableName,
        String queryString,
        Options options,
        Authorizations authorizations
    ) {
        super(client, graph, queryString, options, authorizations);
        this.elementId = elementId;
        this.tableName = tableName;
    }

    @Override
    protected List getFilters(EnumSet elementTypes, FetchHints fetchHints) {
        List filters = super.getFilters(elementTypes, fetchHints);
        BoolQueryBuilder filter = QueryBuilders.boolQuery()
            .must(QueryBuilders.termsQuery(
                Elasticsearch5SearchIndex.ELEMENT_TYPE_FIELD_NAME,
                ElasticsearchDocumentType.VERTEX_EXTENDED_DATA.getKey(),
                ElasticsearchDocumentType.EDGE_EXTENDED_DATA.getKey()
            ))
            .must(QueryBuilders.termQuery(Elasticsearch5SearchIndex.ELEMENT_ID_FIELD_NAME, elementId));
        if (tableName != null) {
            filter = filter.must(QueryBuilders.termQuery(Elasticsearch5SearchIndex.EXTENDED_DATA_TABLE_NAME_FIELD_NAME, tableName));
        }
        filters.add(filter);
        return filters;
    }

    @Override
    public String toString() {
        return super.toString() +
            ", elementId=" + elementId +
            ", tableName=" + tableName;
    }
}