Examples with BooleanQuery org.apache.lucene.search.BooleanQuery used on opensource projects

Example 41 with BooleanQuery

use of org.apache.lucene.search.BooleanQuery in project zm-mailbox by Zimbra.

the class LuceneQueryOperation method combineOps.

@Override
protected QueryOperation combineOps(QueryOperation other, boolean union) {
    assert (!haveRunSearch);
    if (union) {
        if (other.hasNoResults()) {
            queryInfo.addAll(other.getResultInfo());
            // a query for (other OR nothing) == other
            return this;
        }
    } else {
        if (other.hasAllResults()) {
            if (other.hasSpamTrashSetting()) {
                forceHasSpamTrashSetting();
            }
            queryInfo.addAll(other.getResultInfo());
            // we match all results.  (other AND anything) == other
            return this;
        }
    }
    if (other instanceof LuceneQueryOperation) {
        LuceneQueryOperation otherLucene = (LuceneQueryOperation) other;
        if (union) {
            queryString = '(' + queryString + ") OR (" + otherLucene.queryString + ')';
        } else {
            queryString = '(' + queryString + ") AND (" + otherLucene.queryString + ')';
        }
        BooleanQuery top = new BooleanQuery();
        if (union) {
            if (luceneQuery instanceof BooleanQuery) {
                orCopy((BooleanQuery) luceneQuery, top);
            } else {
                top.add(new BooleanClause(luceneQuery, Occur.SHOULD));
            }
            if (otherLucene.luceneQuery instanceof BooleanQuery) {
                orCopy((BooleanQuery) otherLucene.luceneQuery, top);
            } else {
                top.add(new BooleanClause(otherLucene.luceneQuery, Occur.SHOULD));
            }
        } else {
            if (luceneQuery instanceof BooleanQuery) {
                andCopy((BooleanQuery) luceneQuery, top);
            } else {
                top.add(new BooleanClause(luceneQuery, Occur.MUST));
            }
            if (otherLucene.luceneQuery instanceof BooleanQuery) {
                andCopy((BooleanQuery) otherLucene.luceneQuery, top);
            } else {
                top.add(new BooleanClause(otherLucene.luceneQuery, Occur.MUST));
            }
        }
        luceneQuery = top;
        queryInfo.addAll(other.getResultInfo());
        if (other.hasSpamTrashSetting()) {
            forceHasSpamTrashSetting();
        }
        return this;
    }
    return null;
}

Example 42 with BooleanQuery

use of org.apache.lucene.search.BooleanQuery in project lucene-solr by apache.

the class SynonymTokenizer method testOverlapAnalyzer2.

public void testOverlapAnalyzer2() throws Exception {
    TestHighlightRunner helper = new TestHighlightRunner() {

        @Override
        public void run() throws Exception {
            String s = "Hi-Speed10 foo";
            Query query;
            Highlighter highlighter;
            String result;
            query = new TermQuery(new Term("text", "foo"));
            highlighter = getHighlighter(query, "text", HighlighterTest.this);
            result = highlighter.getBestFragments(getTS2(), s, 3, "...");
            assertEquals("Hi-Speed10 <B>foo</B>", result);
            query = new TermQuery(new Term("text", "10"));
            highlighter = getHighlighter(query, "text", HighlighterTest.this);
            result = highlighter.getBestFragments(getTS2(), s, 3, "...");
            assertEquals("Hi-Speed<B>10</B> foo", result);
            query = new TermQuery(new Term("text", "hi"));
            highlighter = getHighlighter(query, "text", HighlighterTest.this);
            result = highlighter.getBestFragments(getTS2(), s, 3, "...");
            assertEquals("<B>Hi</B>-Speed10 foo", result);
            query = new TermQuery(new Term("text", "speed"));
            highlighter = getHighlighter(query, "text", HighlighterTest.this);
            result = highlighter.getBestFragments(getTS2(), s, 3, "...");
            assertEquals("Hi-<B>Speed</B>10 foo", result);
            query = new TermQuery(new Term("text", "hispeed"));
            highlighter = getHighlighter(query, "text", HighlighterTest.this);
            result = highlighter.getBestFragments(getTS2(), s, 3, "...");
            assertEquals("<B>Hi-Speed</B>10 foo", result);
            BooleanQuery.Builder booleanQuery = new BooleanQuery.Builder();
            booleanQuery.add(new TermQuery(new Term("text", "hi")), Occur.SHOULD);
            booleanQuery.add(new TermQuery(new Term("text", "speed")), Occur.SHOULD);
            query = booleanQuery.build();
            highlighter = getHighlighter(query, "text", HighlighterTest.this);
            result = highlighter.getBestFragments(getTS2(), s, 3, "...");
            assertEquals("<B>Hi-Speed</B>10 foo", result);
            // ///////////////// same tests, just put the bigger overlapping token
            // first
            query = new TermQuery(new Term("text", "foo"));
            highlighter = getHighlighter(query, "text", HighlighterTest.this);
            result = highlighter.getBestFragments(getTS2a(), s, 3, "...");
            assertEquals("Hi-Speed10 <B>foo</B>", result);
            query = new TermQuery(new Term("text", "10"));
            highlighter = getHighlighter(query, "text", HighlighterTest.this);
            result = highlighter.getBestFragments(getTS2a(), s, 3, "...");
            assertEquals("Hi-Speed<B>10</B> foo", result);
            query = new TermQuery(new Term("text", "hi"));
            highlighter = getHighlighter(query, "text", HighlighterTest.this);
            result = highlighter.getBestFragments(getTS2a(), s, 3, "...");
            assertEquals("<B>Hi</B>-Speed10 foo", result);
            query = new TermQuery(new Term("text", "speed"));
            highlighter = getHighlighter(query, "text", HighlighterTest.this);
            result = highlighter.getBestFragments(getTS2a(), s, 3, "...");
            assertEquals("Hi-<B>Speed</B>10 foo", result);
            query = new TermQuery(new Term("text", "hispeed"));
            highlighter = getHighlighter(query, "text", HighlighterTest.this);
            result = highlighter.getBestFragments(getTS2a(), s, 3, "...");
            assertEquals("<B>Hi-Speed</B>10 foo", result);
            query = booleanQuery.build();
            highlighter = getHighlighter(query, "text", HighlighterTest.this);
            result = highlighter.getBestFragments(getTS2a(), s, 3, "...");
            assertEquals("<B>Hi-Speed</B>10 foo", result);
        }
    };
    helper.start();
}

Example 43 with BooleanQuery

use of org.apache.lucene.search.BooleanQuery in project lucene-solr by apache.

the class TestUnifiedHighlighter method testMultipleTerms.

public void testMultipleTerms() throws Exception {
    RandomIndexWriter iw = new RandomIndexWriter(random(), dir, indexAnalyzer);
    Field body = new Field("body", "", fieldType);
    Document doc = new Document();
    doc.add(body);
    body.setStringValue("This is a test. Just a test highlighting from postings. Feel free to ignore.");
    iw.addDocument(doc);
    body.setStringValue("Highlighting the first term. Hope it works.");
    iw.addDocument(doc);
    IndexReader ir = iw.getReader();
    iw.close();
    IndexSearcher searcher = newSearcher(ir);
    UnifiedHighlighter highlighter = new UnifiedHighlighter(searcher, indexAnalyzer);
    BooleanQuery query = new BooleanQuery.Builder().add(new TermQuery(new Term("body", "highlighting")), BooleanClause.Occur.SHOULD).add(new TermQuery(new Term("body", "just")), BooleanClause.Occur.SHOULD).add(new TermQuery(new Term("body", "first")), BooleanClause.Occur.SHOULD).build();
    TopDocs topDocs = searcher.search(query, 10, Sort.INDEXORDER);
    assertEquals(2, topDocs.totalHits);
    String[] snippets = highlighter.highlight("body", query, topDocs);
    assertEquals(2, snippets.length);
    assertEquals("<b>Just</b> a test <b>highlighting</b> from postings. ", snippets[0]);
    assertEquals("<b>Highlighting</b> the <b>first</b> term. ", snippets[1]);
    ir.close();
}

Example 44 with BooleanQuery

use of org.apache.lucene.search.BooleanQuery in project lucene-solr by apache.

the class TestUnifiedHighlighter method testBooleanMustNot.

public void testBooleanMustNot() throws Exception {
    RandomIndexWriter iw = new RandomIndexWriter(random(), dir, indexAnalyzer);
    Field body = new Field("body", "This sentence has both terms.  This sentence has only terms.", fieldType);
    Document document = new Document();
    document.add(body);
    iw.addDocument(document);
    IndexReader ir = iw.getReader();
    iw.close();
    IndexSearcher searcher = newSearcher(ir);
    BooleanQuery query2 = new BooleanQuery.Builder().add(new TermQuery(new Term("body", "both")), BooleanClause.Occur.MUST_NOT).build();
    BooleanQuery query = new BooleanQuery.Builder().add(new TermQuery(new Term("body", "terms")), BooleanClause.Occur.SHOULD).add(query2, BooleanClause.Occur.SHOULD).build();
    TopDocs topDocs = searcher.search(query, 10);
    assertEquals(1, topDocs.totalHits);
    UnifiedHighlighter highlighter = new UnifiedHighlighter(searcher, indexAnalyzer);
    highlighter.setMaxLength(Integer.MAX_VALUE - 1);
    String[] snippets = highlighter.highlight("body", query, topDocs, 2);
    assertEquals(1, snippets.length);
    assertFalse(snippets[0].contains("<b>both</b>"));
    ir.close();
}

Example 45 with BooleanQuery

use of org.apache.lucene.search.BooleanQuery in project lucene-solr by apache.

the class TestUnifiedHighlighter method testCambridgeMA.

public void testCambridgeMA() throws Exception {
    BufferedReader r = new BufferedReader(new InputStreamReader(this.getClass().getResourceAsStream("CambridgeMA.utf8"), StandardCharsets.UTF_8));
    String text = r.readLine();
    r.close();
    RandomIndexWriter iw = new RandomIndexWriter(random(), dir, indexAnalyzer);
    Field body = new Field("body", text, fieldType);
    Document document = new Document();
    document.add(body);
    iw.addDocument(document);
    IndexReader ir = iw.getReader();
    iw.close();
    IndexSearcher searcher = newSearcher(ir);
    BooleanQuery query = new BooleanQuery.Builder().add(new TermQuery(new Term("body", "porter")), BooleanClause.Occur.SHOULD).add(new TermQuery(new Term("body", "square")), BooleanClause.Occur.SHOULD).add(new TermQuery(new Term("body", "massachusetts")), BooleanClause.Occur.SHOULD).build();
    TopDocs topDocs = searcher.search(query, 10);
    assertEquals(1, topDocs.totalHits);
    UnifiedHighlighter highlighter = new UnifiedHighlighter(searcher, indexAnalyzer);
    highlighter.setMaxLength(Integer.MAX_VALUE - 1);
    String[] snippets = highlighter.highlight("body", query, topDocs, 2);
    assertEquals(1, snippets.length);
    assertTrue(snippets[0].contains("<b>Square</b>"));
    assertTrue(snippets[0].contains("<b>Porter</b>"));
    ir.close();
}