Examples with ConstantScoreQueryBuilder org.opensearch.index.query.ConstantScoreQueryBuilder used on opensource projects

Example 1 with ConstantScoreQueryBuilder

use of org.opensearch.index.query.ConstantScoreQueryBuilder in project OpenSearch by opensearch-project.

the class NestedSortBuilderTests method testRewrite.

/**
 * Test that filters and inner nested sorts get rewritten
 */
public void testRewrite() throws IOException {
    QueryBuilder filterThatRewrites = new MatchNoneQueryBuilder() {

        @Override
        protected QueryBuilder doRewrite(org.opensearch.index.query.QueryRewriteContext queryShardContext) throws IOException {
            return new MatchAllQueryBuilder();
        }
    };
    // test that filter gets rewritten
    NestedSortBuilder original = new NestedSortBuilder("path").setFilter(filterThatRewrites);
    QueryRewriteContext mockRewriteContext = Mockito.mock(QueryRewriteContext.class);
    NestedSortBuilder rewritten = original.rewrite(mockRewriteContext);
    assertNotSame(rewritten, original);
    assertNotSame(rewritten.getFilter(), original.getFilter());
    // test that inner nested sort gets rewritten
    original = new NestedSortBuilder("path");
    original.setNestedSort(new NestedSortBuilder("otherPath").setFilter(filterThatRewrites));
    rewritten = original.rewrite(mockRewriteContext);
    assertNotSame(rewritten, original);
    assertNotSame(rewritten.getNestedSort(), original.getNestedSort());
    // test that both filter and inner nested sort get rewritten
    original = new NestedSortBuilder("path");
    original.setFilter(filterThatRewrites);
    original.setNestedSort(new NestedSortBuilder("otherPath").setFilter(filterThatRewrites));
    rewritten = original.rewrite(mockRewriteContext);
    assertNotSame(rewritten, original);
    assertNotSame(rewritten.getFilter(), original.getFilter());
    assertNotSame(rewritten.getNestedSort(), original.getNestedSort());
    // test that original stays unchanged if no element rewrites
    original = new NestedSortBuilder("path");
    original.setFilter(new MatchNoneQueryBuilder());
    original.setNestedSort(new NestedSortBuilder("otherPath").setFilter(new MatchNoneQueryBuilder()));
    rewritten = original.rewrite(mockRewriteContext);
    assertSame(rewritten, original);
    assertSame(rewritten.getFilter(), original.getFilter());
    assertSame(rewritten.getNestedSort(), original.getNestedSort());
    // test that rewrite works recursively
    original = new NestedSortBuilder("firstLevel");
    ConstantScoreQueryBuilder constantScoreQueryBuilder = new ConstantScoreQueryBuilder(filterThatRewrites);
    original.setFilter(constantScoreQueryBuilder);
    NestedSortBuilder nestedSortThatRewrites = new NestedSortBuilder("thirdLevel").setFilter(filterThatRewrites);
    original.setNestedSort(new NestedSortBuilder("secondLevel").setNestedSort(nestedSortThatRewrites));
    rewritten = original.rewrite(mockRewriteContext);
    assertNotSame(rewritten, original);
    assertNotSame(rewritten.getFilter(), constantScoreQueryBuilder);
    assertNotSame(((ConstantScoreQueryBuilder) rewritten.getFilter()).innerQuery(), constantScoreQueryBuilder.innerQuery());
    assertEquals("secondLevel", rewritten.getNestedSort().getPath());
    assertNotSame(rewritten.getNestedSort(), original.getNestedSort());
    assertEquals("thirdLevel", rewritten.getNestedSort().getNestedSort().getPath());
    assertNotSame(rewritten.getNestedSort().getNestedSort(), nestedSortThatRewrites);
}

Example 2 with ConstantScoreQueryBuilder

use of org.opensearch.index.query.ConstantScoreQueryBuilder in project OpenSearch by opensearch-project.

the class PercolatorFieldMapperTests method testUnsupportedQueries.

public void testUnsupportedQueries() {
    RangeQueryBuilder rangeQuery1 = new RangeQueryBuilder("field").from("2016-01-01||/D").to("2017-01-01||/D");
    RangeQueryBuilder rangeQuery2 = new RangeQueryBuilder("field").from("2016-01-01||/D").to("now");
    PercolatorFieldMapper.verifyQuery(rangeQuery1);
    PercolatorFieldMapper.verifyQuery(rangeQuery2);
    HasChildQueryBuilder hasChildQuery = new HasChildQueryBuilder("_type", new MatchAllQueryBuilder(), ScoreMode.None);
    expectThrows(IllegalArgumentException.class, () -> PercolatorFieldMapper.verifyQuery(new BoolQueryBuilder().must(hasChildQuery)));
    expectThrows(IllegalArgumentException.class, () -> PercolatorFieldMapper.verifyQuery(new DisMaxQueryBuilder().add(hasChildQuery)));
    PercolatorFieldMapper.verifyQuery(new ConstantScoreQueryBuilder((rangeQuery1)));
    expectThrows(IllegalArgumentException.class, () -> PercolatorFieldMapper.verifyQuery(new ConstantScoreQueryBuilder(hasChildQuery)));
    PercolatorFieldMapper.verifyQuery(new BoostingQueryBuilder(rangeQuery1, new MatchAllQueryBuilder()));
    expectThrows(IllegalArgumentException.class, () -> PercolatorFieldMapper.verifyQuery(new BoostingQueryBuilder(hasChildQuery, new MatchAllQueryBuilder())));
    PercolatorFieldMapper.verifyQuery(new FunctionScoreQueryBuilder(rangeQuery1, new RandomScoreFunctionBuilder()));
    expectThrows(IllegalArgumentException.class, () -> PercolatorFieldMapper.verifyQuery(new FunctionScoreQueryBuilder(hasChildQuery, new RandomScoreFunctionBuilder())));
    expectThrows(IllegalArgumentException.class, () -> PercolatorFieldMapper.verifyQuery(hasChildQuery));
    expectThrows(IllegalArgumentException.class, () -> PercolatorFieldMapper.verifyQuery(new BoolQueryBuilder().must(hasChildQuery)));
    HasParentQueryBuilder hasParentQuery = new HasParentQueryBuilder("_type", new MatchAllQueryBuilder(), false);
    expectThrows(IllegalArgumentException.class, () -> PercolatorFieldMapper.verifyQuery(hasParentQuery));
    expectThrows(IllegalArgumentException.class, () -> PercolatorFieldMapper.verifyQuery(new BoolQueryBuilder().must(hasParentQuery)));
}

Example 3 with ConstantScoreQueryBuilder

use of org.opensearch.index.query.ConstantScoreQueryBuilder in project OpenSearch by opensearch-project.

the class PercolatorFieldMapper method verifyQuery.

/**
 * Fails if a percolator contains an unsupported query. The following queries are not supported:
 * 1) a has_child query
 * 2) a has_parent query
 */
static void verifyQuery(QueryBuilder queryBuilder) {
    if (queryBuilder.getName().equals("has_child")) {
        throw new IllegalArgumentException("the [has_child] query is unsupported inside a percolator query");
    } else if (queryBuilder.getName().equals("has_parent")) {
        throw new IllegalArgumentException("the [has_parent] query is unsupported inside a percolator query");
    } else if (queryBuilder instanceof BoolQueryBuilder) {
        BoolQueryBuilder boolQueryBuilder = (BoolQueryBuilder) queryBuilder;
        List<QueryBuilder> clauses = new ArrayList<>();
        clauses.addAll(boolQueryBuilder.filter());
        clauses.addAll(boolQueryBuilder.must());
        clauses.addAll(boolQueryBuilder.mustNot());
        clauses.addAll(boolQueryBuilder.should());
        for (QueryBuilder clause : clauses) {
            verifyQuery(clause);
        }
    } else if (queryBuilder instanceof ConstantScoreQueryBuilder) {
        verifyQuery(((ConstantScoreQueryBuilder) queryBuilder).innerQuery());
    } else if (queryBuilder instanceof FunctionScoreQueryBuilder) {
        verifyQuery(((FunctionScoreQueryBuilder) queryBuilder).query());
    } else if (queryBuilder instanceof BoostingQueryBuilder) {
        verifyQuery(((BoostingQueryBuilder) queryBuilder).negativeQuery());
        verifyQuery(((BoostingQueryBuilder) queryBuilder).positiveQuery());
    } else if (queryBuilder instanceof DisMaxQueryBuilder) {
        DisMaxQueryBuilder disMaxQueryBuilder = (DisMaxQueryBuilder) queryBuilder;
        for (QueryBuilder innerQueryBuilder : disMaxQueryBuilder.innerQueries()) {
            verifyQuery(innerQueryBuilder);
        }
    }
}