Java tutorial
/** * Copyright (c) 2014, Sindice Limited. All Rights Reserved. * * This file is part of the SIREn project. * * SIREn is a free software: you can redistribute it and/or modify * it under the terms of the GNU Affero General Public License as * published by the Free Software Foundation, either version 3 of * the License, or (at your option) any later version. * * SIREn is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the * GNU Affero General Public License for more details. * * You should have received a copy of the GNU Affero General Public * License along with this program. If not, see <http://www.gnu.org/licenses/>. */ package com.sindicetech.siren.search.node; import static com.sindicetech.siren.search.AbstractTestSirenScorer.BooleanClauseBuilder.must; import static com.sindicetech.siren.search.AbstractTestSirenScorer.BooleanClauseBuilder.should; import static com.sindicetech.siren.search.AbstractTestSirenScorer.NodeBooleanQueryBuilder.nbq; import static com.sindicetech.siren.search.AbstractTestSirenScorer.TupleQueryBuilder.tuple; import java.io.IOException; import org.apache.lucene.index.Term; import org.apache.lucene.search.BooleanClause; import org.apache.lucene.search.BooleanClause.Occur; import org.apache.lucene.search.BooleanQuery; import org.apache.lucene.search.Query; import org.apache.lucene.search.TermQuery; import org.apache.lucene.search.TopDocs; import org.junit.Test; import com.sindicetech.siren.analysis.AnyURIAnalyzer; import com.sindicetech.siren.analysis.TupleAnalyzer; import com.sindicetech.siren.index.codecs.RandomSirenCodec.PostingsFormatType; import com.sindicetech.siren.search.AbstractTestSirenScorer; import com.sindicetech.siren.search.node.NodeQuery; import com.sindicetech.siren.util.XSDDatatype; public class TestTupleQuery extends AbstractTestSirenScorer { @Override protected void configure() throws IOException { this.setAnalyzer(AnalyzerType.TUPLE); final AnyURIAnalyzer uriAnalyzer = new AnyURIAnalyzer(TEST_VERSION_CURRENT); ((TupleAnalyzer) analyzer).registerDatatype(XSDDatatype.XSD_ANY_URI.toCharArray(), uriAnalyzer); this.setPostingsFormat(PostingsFormatType.RANDOM); } @Test public void testEquality() throws Exception { final NodeQuery query1 = tuple().optional(nbq(should("value1"), should("value2"))).getQuery(); final NodeQuery query2 = tuple().optional(nbq(should("value1"), should("value2"))).getQuery(); assertEquals(query1, query2); } @Test public void testUnaryClause() throws IOException { this.addDocument("\"aaa ccc\" \"bbb ccc\" . \"aaa bbb\" \"ccc eee\" . "); // {[aaa]} Query query = tuple().optional(nbq(should("aaa"))).getLuceneProxyQuery(); TopDocs hits = searcher.search(query, 100); assertEquals(1, hits.totalHits); // {[ccc]} query = tuple().optional(nbq(should("ccc"))).getLuceneProxyQuery(); hits = searcher.search(query, 100); assertEquals(1, hits.totalHits); // {[+ccc +aaa]} query = tuple().optional(nbq(must("aaa"), must("ccc"))).getLuceneProxyQuery(); hits = searcher.search(query, 100); assertEquals(1, hits.totalHits); // {[ddd]} query = tuple().optional(nbq(should("ddd"))).getLuceneProxyQuery(); hits = searcher.search(query, 100); assertEquals(0, hits.totalHits); // {[+bbb +eee]} query = tuple().optional(nbq(must("bbb"), must("eee"))).getLuceneProxyQuery(); hits = searcher.search(query, 100); assertEquals(0, hits.totalHits); } /** * <code>aaa ({[+ddd] [+eee]})</code> * Here, the keywords are mandatory in the cell, but each cell is optional in * the tuple. So, even the first document (having only one matching cell) * should match. */ @Test public void testParenthesisMust() throws IOException { this.addDocument("\"bbb\" . \"ccc\" \"ddd bbb\" . "); this.addDocument("\"bbb\" . \"ccc eee\" \"ddd bbb\" . "); this.addDocument("\"bbb\" . "); final Query query = tuple().optional(nbq(must("ddd")), nbq(must("eee"))).getLuceneProxyQuery(); final BooleanQuery q = new BooleanQuery(); q.add(new TermQuery(new Term(DEFAULT_TEST_FIELD, "aaa")), BooleanClause.Occur.SHOULD); q.add(query, BooleanClause.Occur.SHOULD); assertEquals(2, searcher.search(q, 100).totalHits); } /** * <code>bbb ({[+ddd] [+eee]})</code> */ @Test public void testParenthesisMust2() throws IOException { this.addDocument("\"bbb\" . "); final Query query = tuple().optional(nbq(must("ddd")), nbq(must("eee"))).getLuceneProxyQuery(); final BooleanQuery q = new BooleanQuery(); q.add(new TermQuery(new Term(DEFAULT_TEST_FIELD, "bbb")), BooleanClause.Occur.SHOULD); q.add(query, BooleanClause.Occur.SHOULD); assertEquals(1, searcher.search(q, 100).totalHits); } /** * <code>bbb +({[ddd] [eee]})</code> * Here, the keywords are mandatory in the cell, but each cell is optional in * the tuple. So, even the first document (having only one cell) should match. */ @Test public void testParenthesisMust3() throws IOException { this.addDocument("\"aaa\" . \"eee\" \"bbb\" . "); this.addDocument("\"bbb\" . \"ccc\" \"ddd bbb\" . "); this.addDocument("\"bbb\" . \"ccc eee\" \"ddd bbb\" . "); final Query query = tuple().optional(nbq(should("ddd")), nbq(should("eee"))).getLuceneProxyQuery(); final BooleanQuery q = new BooleanQuery(); q.add(new TermQuery(new Term(DEFAULT_TEST_FIELD, "aaa")), BooleanClause.Occur.SHOULD); q.add(query, BooleanClause.Occur.MUST); assertEquals(3, searcher.search(q, 100).totalHits); } /** * <code>{[bbb eee]} {[ccc eee]}</code> */ @Test public void testParenthesisShould() throws IOException { this.addDocument("\"bbb eee\" . \"ccc eee\" . "); this.addDocument("\"bbb\" . \"aaa\" . "); this.addDocument("\"eee\" . \"aaa\" . "); this.addDocument("\"aaa\" . \"ccc\" . "); this.addDocument("\"aaa\" . \"eee\" . "); final Query tq1 = tuple().optional(nbq(should("bbb"), should("eee"))).getLuceneProxyQuery(); final Query tq2 = tuple().optional(nbq(should("ccc"), should("eee"))).getLuceneProxyQuery(); final BooleanQuery q = new BooleanQuery(); q.add(tq1, BooleanClause.Occur.SHOULD); q.add(tq2, BooleanClause.Occur.SHOULD); assertEquals(5, searcher.search(q, 100).totalHits); } /** * <code>+{+[+actor]} +{+[+actor]}</code> */ @Test public void testLuceneBooleanMust() throws IOException { this.addDocument( "<actor> \"actor 1\" <birthdate> \"456321\" . " + "<actor> \"actor 2\" <birthdate> \"456321\" . "); this.addDocument("<actor> \"actor 1\" . <actor> \"actor 2\" . "); final Query tq1 = tuple().with(nbq(must("actor")).bound(0, 0)).getLuceneProxyQuery(); final Query tq2 = tuple().with(nbq(must("actor")).bound(1, 1)).getLuceneProxyQuery(); final BooleanQuery bq = new BooleanQuery(); bq.add(tq1, Occur.MUST); bq.add(tq2, Occur.MUST); assertEquals(2, searcher.search(bq, 100).totalHits); } }