Examples with QueryBuilder org.graylog.shaded.elasticsearch7.org.elasticsearch.index.query.QueryBuilder used on opensource projects

Example 56 with QueryBuilder

use of org.graylog.shaded.elasticsearch7.org.elasticsearch.index.query.QueryBuilder in project unipop by unipop-graph.

the class DocVertexSchema method getSearch.

@Override
public QueryBuilder getSearch(DeferredVertexQuery query) {
    PredicatesHolder predicatesHolder = this.toPredicates(query.getVertices());
    QueryBuilder queryBuilder = createQueryBuilder(predicatesHolder);
    return queryBuilder;
}

Example 57 with QueryBuilder

use of org.graylog.shaded.elasticsearch7.org.elasticsearch.index.query.QueryBuilder in project unipop by unipop-graph.

the class NestedEdgeSchema method createQueryBuilder.

public QueryBuilder createQueryBuilder(SearchVertexQuery query) {
    PredicatesHolder edgePredicates = this.toPredicates(query.getPredicates());
    if (edgePredicates.isAborted())
        return null;
    PredicatesHolder childPredicates = childVertexSchema.toPredicates(query.getVertices());
    childPredicates = PredicatesHolderFactory.and(edgePredicates, childPredicates);
    QueryBuilder childQuery = createNestedQueryBuilder(childPredicates);
    if (query.getDirection().equals(parentDirection.opposite())) {
        if (childPredicates.isAborted())
            return null;
        return childQuery;
    } else if (!query.getDirection().equals(Direction.BOTH))
        childQuery = null;
    PredicatesHolder parentPredicates = parentVertexSchema.toPredicates(query.getVertices());
    QueryBuilder parentQuery = createQueryBuilder(parentPredicates);
    if (parentQuery != null) {
        // if (parentPredicates.isAborted()) return null;
        QueryBuilder edgeQuery = createNestedQueryBuilder(edgePredicates);
        if (edgeQuery != null) {
            parentQuery = QueryBuilders.boolQuery().must(parentQuery).must(edgeQuery);
        }
    }
    if (query.getDirection().equals(parentDirection) && parentPredicates.notAborted())
        return parentQuery;
    else if (childQuery == null && parentPredicates.notAborted())
        return parentQuery;
    else if (parentQuery == null && childPredicates.notAborted())
        return childQuery;
    else if (parentPredicates.isAborted() && childPredicates.isAborted())
        return null;
    else
        return QueryBuilders.boolQuery().should(parentQuery).should(childQuery);
}

Example 58 with QueryBuilder

use of org.graylog.shaded.elasticsearch7.org.elasticsearch.index.query.QueryBuilder in project unipop by unipop-graph.

the class NestedVertexSchema method getSearch.

@Override
public QueryBuilder getSearch(DeferredVertexQuery query) {
    PredicatesHolder predicatesHolder = this.toPredicates(query.getVertices());
    QueryBuilder queryBuilder = createQueryBuilder(predicatesHolder);
    return queryBuilder;
}

Example 59 with QueryBuilder

use of org.graylog.shaded.elasticsearch7.org.elasticsearch.index.query.QueryBuilder in project unipop by unipop-graph.

the class DocEdgeSchema method getSearch.

@Override
public QueryBuilder getSearch(SearchVertexQuery query) {
    PredicatesHolder edgePredicates = this.toPredicates(query.getPredicates());
    PredicatesHolder vertexPredicates = this.getVertexPredicates(query.getVertices(), query.getDirection());
    PredicatesHolder predicatesHolder = PredicatesHolderFactory.and(edgePredicates, vertexPredicates);
    if (predicatesHolder.isAborted())
        return null;
    QueryBuilder queryBuilder = createQueryBuilder(predicatesHolder);
    return queryBuilder;
// return createSearch(query, queryBuilder);
}

Example 60 with QueryBuilder

use of org.graylog.shaded.elasticsearch7.org.elasticsearch.index.query.QueryBuilder in project unipop by unipop-graph.

the class DocumentController method fetchProperties.

@Override
public void fetchProperties(DeferredVertexQuery uniQuery) {
    Map<DocumentVertexSchema, QueryBuilder> schemas = vertexSchemas.stream().filter(schema -> this.traversalFilter.filter(schema, uniQuery.getTraversal())).collect(new SearchCollector<>((schema) -> schema.getSearch(uniQuery)));
    Iterator<Vertex> search = search(uniQuery, schemas);
    Map<Object, DeferredVertex> vertexMap = uniQuery.getVertices().stream().collect(Collectors.toMap(UniElement::id, Function.identity(), (a, b) -> a));
    search.forEachRemaining(newVertex -> {
        DeferredVertex deferredVertex = vertexMap.get(newVertex.id());
        if (deferredVertex != null)
            deferredVertex.loadProperties(newVertex);
    });
}