Examples with NestedHelper org.elasticsearch.index.search.NestedHelper used on opensource projects

Example 1 with NestedHelper

use of org.elasticsearch.index.search.NestedHelper in project elasticsearch by elastic.

the class NestedQueryBuilder method doToQuery.

@Override
protected Query doToQuery(QueryShardContext context) throws IOException {
    ObjectMapper nestedObjectMapper = context.getObjectMapper(path);
    if (nestedObjectMapper == null) {
        if (ignoreUnmapped) {
            return new MatchNoDocsQuery();
        } else {
            throw new IllegalStateException("[" + NAME + "] failed to find nested object under path [" + path + "]");
        }
    }
    if (!nestedObjectMapper.nested().isNested()) {
        throw new IllegalStateException("[" + NAME + "] nested object under path [" + path + "] is not of nested type");
    }
    final BitSetProducer parentFilter;
    Query innerQuery;
    ObjectMapper objectMapper = context.nestedScope().getObjectMapper();
    if (objectMapper == null) {
        parentFilter = context.bitsetFilter(Queries.newNonNestedFilter());
    } else {
        parentFilter = context.bitsetFilter(objectMapper.nestedTypeFilter());
    }
    try {
        context.nestedScope().nextLevel(nestedObjectMapper);
        innerQuery = this.query.toQuery(context);
    } finally {
        context.nestedScope().previousLevel();
    }
    // in its child space
    if (new NestedHelper(context.getMapperService()).mightMatchNonNestedDocs(innerQuery, path)) {
        innerQuery = Queries.filtered(innerQuery, nestedObjectMapper.nestedTypeFilter());
    }
    return new ESToParentBlockJoinQuery(innerQuery, parentFilter, scoreMode, objectMapper == null ? null : objectMapper.fullPath());
}

Example 2 with NestedHelper

use of org.elasticsearch.index.search.NestedHelper in project elasticsearch by elastic.

the class DefaultSearchContext method buildFilteredQuery.

@Override
public Query buildFilteredQuery(Query query) {
    List<Query> filters = new ArrayList<>();
    Query typeFilter = createTypeFilter(queryShardContext.getTypes());
    if (typeFilter != null) {
        filters.add(typeFilter);
    }
    if (mapperService().hasNested() && // when a _type filter is set, it will automatically exclude nested docs
    typeFilter == null && new NestedHelper(mapperService()).mightMatchNestedDocs(query) && (aliasFilter == null || new NestedHelper(mapperService()).mightMatchNestedDocs(aliasFilter))) {
        filters.add(Queries.newNonNestedFilter());
    }
    if (aliasFilter != null) {
        filters.add(aliasFilter);
    }
    if (sliceBuilder != null) {
        filters.add(sliceBuilder.toFilter(queryShardContext, shardTarget().getShardId().getId(), queryShardContext.getIndexSettings().getNumberOfShards()));
    }
    if (filters.isEmpty()) {
        return query;
    } else {
        BooleanQuery.Builder builder = new BooleanQuery.Builder();
        builder.add(query, Occur.MUST);
        for (Query filter : filters) {
            builder.add(filter, Occur.FILTER);
        }
        return builder.build();
    }
}