Examples with Result org.opensearch.percolator.QueryAnalyzer.Result used on opensource projects

Example 26 with Result

use of org.opensearch.percolator.QueryAnalyzer.Result in project OpenSearch by opensearch-project.

the class QueryAnalyzerTests method testExtractQueryMetadata_spanFirstQuery.

public void testExtractQueryMetadata_spanFirstQuery() {
    SpanTermQuery spanTermQuery1 = new SpanTermQuery(new Term("_field", "_short_term"));
    SpanFirstQuery spanFirstQuery = new SpanFirstQuery(spanTermQuery1, 20);
    Result result = analyze(spanFirstQuery, Version.CURRENT);
    assertThat(result.verified, is(false));
    assertThat(result.minimumShouldMatch, equalTo(1));
    assertTermsEqual(result.extractions, spanTermQuery1.getTerm());
}

Example 27 with Result

use of org.opensearch.percolator.QueryAnalyzer.Result in project OpenSearch by opensearch-project.

the class QueryAnalyzerTests method testToParentBlockJoinQuery.

public void testToParentBlockJoinQuery() {
    TermQuery termQuery = new TermQuery(new Term("field", "value"));
    QueryBitSetProducer queryBitSetProducer = new QueryBitSetProducer(new TermQuery(new Term("_type", "child")));
    OpenSearchToParentBlockJoinQuery query = new OpenSearchToParentBlockJoinQuery(termQuery, queryBitSetProducer, ScoreMode.None, "child");
    Result result = analyze(query, Version.CURRENT);
    assertFalse(result.verified);
    assertThat(result.minimumShouldMatch, equalTo(1));
    assertEquals(1, result.extractions.size());
    assertNull(result.extractions.toArray(new QueryExtraction[0])[0].range);
    assertEquals(new Term("field", "value"), result.extractions.toArray(new QueryExtraction[0])[0].term);
}

Example 28 with Result

use of org.opensearch.percolator.QueryAnalyzer.Result in project OpenSearch by opensearch-project.

the class QueryAnalyzerTests method testSelectBestResult.

public void testSelectBestResult() {
    Set<QueryExtraction> queryTerms1 = terms(new int[0], "12", "1234", "12345");
    Result result1 = new Result(true, queryTerms1, 1);
    Set<QueryAnalyzer.QueryExtraction> queryTerms2 = terms(new int[0], "123", "1234", "12345");
    Result result2 = new Result(true, queryTerms2, 1);
    Result result = selectBestResult(result1, result2);
    assertSame(queryTerms2, result.extractions);
    assertFalse(result.verified);
    queryTerms1 = terms(new int[] { 1, 2, 3 });
    result1 = new Result(true, queryTerms1, 1);
    queryTerms2 = terms(new int[] { 2, 3, 4 });
    result2 = new Result(true, queryTerms2, 1);
    result = selectBestResult(result1, result2);
    assertSame(queryTerms1, result.extractions);
    assertFalse(result.verified);
    queryTerms1 = terms(new int[] { 4, 5, 6 });
    result1 = new Result(true, queryTerms1, 1);
    queryTerms2 = terms(new int[] { 1, 2, 3 });
    result2 = new Result(true, queryTerms2, 1);
    result = selectBestResult(result1, result2);
    assertSame(queryTerms2, result.extractions);
    assertFalse(result.verified);
    queryTerms1 = terms(new int[] { 1, 2, 3 }, "123", "456");
    result1 = new Result(true, queryTerms1, 1);
    queryTerms2 = terms(new int[] { 2, 3, 4 }, "123", "456");
    result2 = new Result(true, queryTerms2, 1);
    result = selectBestResult(result1, result2);
    assertSame(queryTerms1, result.extractions);
    assertFalse(result.verified);
    queryTerms1 = terms(new int[] { 10 });
    result1 = new Result(true, queryTerms1, 1);
    queryTerms2 = terms(new int[] { 1 });
    result2 = new Result(true, queryTerms2, 1);
    result = selectBestResult(result1, result2);
    assertSame(queryTerms2, result.extractions);
    queryTerms1 = terms(new int[] { 10 }, "123");
    result1 = new Result(true, queryTerms1, 1);
    queryTerms2 = terms(new int[] { 1 });
    result2 = new Result(true, queryTerms2, 1);
    result = selectBestResult(result1, result2);
    assertSame(queryTerms1, result.extractions);
    assertFalse(result.verified);
    queryTerms1 = terms(new int[] { 10 }, "1", "123");
    result1 = new Result(true, queryTerms1, 1);
    queryTerms2 = terms(new int[] { 1 }, "1", "2");
    result2 = new Result(true, queryTerms2, 1);
    result = selectBestResult(result1, result2);
    assertSame(queryTerms1, result.extractions);
    assertFalse(result.verified);
    queryTerms1 = terms(new int[] { 1, 2, 3 }, "123", "456");
    result1 = new Result(true, queryTerms1, 1);
    queryTerms2 = terms(new int[] { 2, 3, 4 }, "1", "456");
    result2 = new Result(true, queryTerms2, 1);
    result = selectBestResult(result1, result2);
    assertSame("Ignoring ranges, so then prefer queryTerms1, because it has the longest shortest term", queryTerms1, result.extractions);
    assertFalse(result.verified);
    queryTerms1 = terms(new int[] {});
    result1 = new Result(false, queryTerms1, 0);
    queryTerms2 = terms(new int[] {});
    result2 = new Result(false, queryTerms2, 0);
    result = selectBestResult(result1, result2);
    assertSame("In case query extractions are empty", queryTerms2, result.extractions);
    assertFalse(result.verified);
    queryTerms1 = terms(new int[] { 1 });
    result1 = new Result(true, queryTerms1, 1);
    queryTerms2 = terms(new int[] {});
    result2 = new Result(false, queryTerms2, 0);
    result = selectBestResult(result1, result2);
    assertSame("In case query a single extraction is empty", queryTerms1, result.extractions);
    assertFalse(result.verified);
    queryTerms1 = terms(new int[] {});
    result1 = new Result(false, queryTerms1, 0);
    queryTerms2 = terms(new int[] { 1 });
    result2 = new Result(true, queryTerms2, 1);
    result = selectBestResult(result1, result2);
    assertSame("In case query a single extraction is empty", queryTerms2, result.extractions);
    assertFalse(result.verified);
    result1 = new Result(true, true);
    queryTerms2 = terms(new int[] { 1 });
    result2 = new Result(true, queryTerms2, 1);
    result = selectBestResult(result1, result2);
    assertSame("Conjunction with a match_all", result2, result);
    assertTrue(result.verified);
    queryTerms1 = terms(new int[] { 1 });
    result1 = new Result(true, queryTerms2, 1);
    result2 = new Result(true, true);
    result = selectBestResult(result1, result2);
    assertSame("Conjunction with a match_all", result1, result);
    assertTrue(result.verified);
}

Example 29 with Result

use of org.opensearch.percolator.QueryAnalyzer.Result in project OpenSearch by opensearch-project.

the class QueryAnalyzerTests method testExtractQueryMetadata_disjunctionMaxQuery.

public void testExtractQueryMetadata_disjunctionMaxQuery() {
    TermQuery termQuery1 = new TermQuery(new Term("_field", "_term1"));
    TermQuery termQuery2 = new TermQuery(new Term("_field", "_term2"));
    TermQuery termQuery3 = new TermQuery(new Term("_field", "_term3"));
    TermQuery termQuery4 = new TermQuery(new Term("_field", "_term4"));
    DisjunctionMaxQuery disjunctionMaxQuery = new DisjunctionMaxQuery(Arrays.asList(termQuery1, termQuery2, termQuery3, termQuery4), 0.1f);
    Result result = analyze(disjunctionMaxQuery, Version.CURRENT);
    assertThat(result.verified, is(true));
    assertThat(result.minimumShouldMatch, equalTo(1));
    List<QueryAnalyzer.QueryExtraction> terms = new ArrayList<>(result.extractions);
    terms.sort(Comparator.comparing(qt -> qt.term));
    assertThat(terms.size(), equalTo(4));
    assertThat(terms.get(0).field(), equalTo(termQuery1.getTerm().field()));
    assertThat(terms.get(0).bytes(), equalTo(termQuery1.getTerm().bytes()));
    assertThat(terms.get(1).field(), equalTo(termQuery2.getTerm().field()));
    assertThat(terms.get(1).bytes(), equalTo(termQuery2.getTerm().bytes()));
    assertThat(terms.get(2).field(), equalTo(termQuery3.getTerm().field()));
    assertThat(terms.get(2).bytes(), equalTo(termQuery3.getTerm().bytes()));
    assertThat(terms.get(3).field(), equalTo(termQuery4.getTerm().field()));
    assertThat(terms.get(3).bytes(), equalTo(termQuery4.getTerm().bytes()));
    disjunctionMaxQuery = new DisjunctionMaxQuery(Arrays.asList(termQuery1, termQuery2, termQuery3, new PhraseQuery("_field", "_term4")), 0.1f);
    result = analyze(disjunctionMaxQuery, Version.CURRENT);
    assertThat(result.verified, is(false));
    assertThat(result.minimumShouldMatch, equalTo(1));
    terms = new ArrayList<>(result.extractions);
    terms.sort(Comparator.comparing(qt -> qt.term));
    assertThat(terms.size(), equalTo(4));
    assertThat(terms.get(0).field(), equalTo(termQuery1.getTerm().field()));
    assertThat(terms.get(0).bytes(), equalTo(termQuery1.getTerm().bytes()));
    assertThat(terms.get(1).field(), equalTo(termQuery2.getTerm().field()));
    assertThat(terms.get(1).bytes(), equalTo(termQuery2.getTerm().bytes()));
    assertThat(terms.get(2).field(), equalTo(termQuery3.getTerm().field()));
    assertThat(terms.get(2).bytes(), equalTo(termQuery3.getTerm().bytes()));
    assertThat(terms.get(3).field(), equalTo(termQuery4.getTerm().field()));
    assertThat(terms.get(3).bytes(), equalTo(termQuery4.getTerm().bytes()));
}

Example 30 with Result

use of org.opensearch.percolator.QueryAnalyzer.Result in project OpenSearch by opensearch-project.

the class QueryAnalyzerTests method testExtractQueryMetadata_boostQuery.

public void testExtractQueryMetadata_boostQuery() {
    TermQuery termQuery1 = new TermQuery(new Term("_field", "_term"));
    BoostQuery constantScoreQuery = new BoostQuery(termQuery1, 1f);
    Result result = analyze(constantScoreQuery, Version.CURRENT);
    assertThat(result.verified, is(true));
    assertThat(result.minimumShouldMatch, equalTo(1));
    List<QueryExtraction> terms = new ArrayList<>(result.extractions);
    assertThat(terms.size(), equalTo(1));
    assertThat(terms.get(0).field(), equalTo(termQuery1.getTerm().field()));
    assertThat(terms.get(0).bytes(), equalTo(termQuery1.getTerm().bytes()));
}