Java tutorial
/** * Licensed to the Apache Software Foundation (ASF) under one * or more contributor license agreements. See the NOTICE file * distributed with this work for additional information * regarding copyright ownership. The ASF licenses this file * to you under the Apache License, Version 2.0 (the * "License"); you may not use this file except in compliance * with the License. You may obtain a copy of the License at * * http://www.apache.org/licenses/LICENSE-2.0 * * Unless required by applicable law or agreed to in writing, * software distributed under the License is distributed on an * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY * KIND, either express or implied. See the License for the * specific language governing permissions and limitations * under the License. */ package org.apache.cxf.jaxrs.ext.search.lucene; import java.util.Collections; import org.apache.cxf.jaxrs.ext.search.SearchBean; import org.apache.cxf.jaxrs.ext.search.SearchCondition; import org.apache.cxf.jaxrs.ext.search.SearchConditionParser; import org.apache.cxf.jaxrs.ext.search.SearchConditionVisitor; import org.apache.lucene.analysis.Analyzer; import org.apache.lucene.analysis.standard.StandardAnalyzer; import org.apache.lucene.document.Document; import org.apache.lucene.document.Field; import org.apache.lucene.document.IntField; import org.apache.lucene.document.TextField; import org.apache.lucene.index.DirectoryReader; import org.apache.lucene.index.IndexWriter; import org.apache.lucene.index.IndexWriterConfig; import org.apache.lucene.index.IndexableField; import org.apache.lucene.search.IndexSearcher; import org.apache.lucene.search.Query; import org.apache.lucene.search.ScoreDoc; import org.apache.lucene.store.Directory; import org.apache.lucene.store.RAMDirectory; import org.apache.lucene.util.Version; import org.junit.After; import org.junit.Assert; import org.junit.Before; public abstract class AbstractLuceneQueryVisitorTest extends Assert { private DirectoryReader ireader; private IndexSearcher isearcher; private Directory directory; private Analyzer analyzer; @Before public void setUp() throws Exception { analyzer = new StandardAnalyzer(Version.LUCENE_4_9); directory = new RAMDirectory(); IndexWriterConfig config = new IndexWriterConfig(Version.LUCENE_4_9, analyzer); IndexWriter iwriter = new IndexWriter(directory, config); Document doc = new Document(); doc.add(new Field("contents", "name=text", TextField.TYPE_STORED)); IntField intField = new IntField("intfield", 4, Field.Store.YES); doc.add(intField); iwriter.addDocument(doc); iwriter.close(); ireader = DirectoryReader.open(directory); isearcher = new IndexSearcher(ireader); } @After public void tearDown() throws Exception { ireader.close(); directory.close(); } protected abstract SearchConditionParser<SearchBean> getParser(); protected void doTestTextContentMatch(String expression) throws Exception { doTestTextContentMatch(expression, false); } protected void doTestTextContentMatchWithAnalyzer(String expression) throws Exception { doTestTextContentMatch(expression, true); } protected void doTestTextContentMatch(String expression, boolean useAnalyzer) throws Exception { Query query = createTermQuery("contents", expression, useAnalyzer); doTestTextContentMatchWithQuery(query); } protected void doTestNoMatch(Query query) throws Exception { ScoreDoc[] hits = isearcher.search(query, null, 1000).scoreDocs; assertEquals(0, hits.length); } protected void doTestTextContentMatchWithQuery(Query query) throws Exception { ScoreDoc[] hits = isearcher.search(query, null, 1000).scoreDocs; assertEquals(1, hits.length); // Iterate through the results: for (int i = 0; i < hits.length; i++) { Document hitDoc = isearcher.doc(hits[i].doc); assertEquals("name=text", hitDoc.get("contents")); } } protected void doTestIntContentMatch(String expression) throws Exception { Query query = createTermQuery("intfield", expression); doTestIntContentMatchWithQuery(query); } protected void doTestIntContentMatchWithQuery(Query query) throws Exception { ScoreDoc[] hits = isearcher.search(query, null, 1000).scoreDocs; assertEquals(1, hits.length); // Iterate through the results: for (int i = 0; i < hits.length; i++) { Document hitDoc = isearcher.doc(hits[i].doc); IndexableField field = hitDoc.getField("intfield"); assertEquals(4, field.numericValue().intValue()); } } protected Query createTermQuery(String expression) throws Exception { return createTermQuery(expression, false); } protected Query createTermQueryWithAnalyzer(String expression) throws Exception { return createTermQuery(expression, true); } protected Query createTermQuery(String expression, boolean useAnalyzer) throws Exception { SearchCondition<SearchBean> filter = getParser().parse(expression); SearchConditionVisitor<SearchBean, Query> lucene = new LuceneQueryVisitor<SearchBean>( useAnalyzer ? analyzer : null); lucene.visit(filter); return lucene.getQuery(); } protected Query createTermQueryWithFieldClassWithAnalyzer(String expression, Class<?> cls) throws Exception { return createTermQueryWithFieldClass(expression, cls, true); } protected Query createTermQueryWithFieldClass(String expression, Class<?> cls) throws Exception { return createTermQueryWithFieldClass(expression, cls, false); } protected Query createTermQueryWithFieldClass(String expression, Class<?> cls, boolean useAnalyzer) throws Exception { SearchCondition<SearchBean> filter = getParser().parse(expression); LuceneQueryVisitor<SearchBean> lucene = new LuceneQueryVisitor<SearchBean>(useAnalyzer ? analyzer : null); lucene.setPrimitiveFieldTypeMap(Collections.<String, Class<?>>singletonMap("intfield", cls)); lucene.visit(filter); return lucene.getQuery(); } protected Query createTermQuery(String fieldName, String expression) throws Exception { return createTermQuery(fieldName, expression, false); } protected Query createTermQuery(String fieldName, String expression, boolean useAnalyzer) throws Exception { SearchCondition<SearchBean> filter = getParser().parse(expression); LuceneQueryVisitor<SearchBean> lucene = new LuceneQueryVisitor<SearchBean>("ct", fieldName, useAnalyzer ? analyzer : null); lucene.visit(filter); return lucene.getQuery(); } protected Query createTermQueryWithFieldClass(String fieldName, String expression, Class<?> cls) throws Exception { SearchCondition<SearchBean> filter = getParser().parse(expression); LuceneQueryVisitor<SearchBean> lucene = new LuceneQueryVisitor<SearchBean>("ct", fieldName); lucene.setPrimitiveFieldTypeMap(Collections.<String, Class<?>>singletonMap(fieldName, cls)); lucene.visit(filter); return lucene.getQuery(); } protected Query createPhraseQuery(String fieldName, String expression) throws Exception { return createPhraseQuery(fieldName, expression, false); } protected Query createPhraseQueryWithAnalyzer(String fieldName, String expression) throws Exception { return createPhraseQuery(fieldName, expression, true); } protected Query createPhraseQuery(String fieldName, String expression, boolean useAnalyzer) throws Exception { SearchCondition<SearchBean> filter = getParser().parse(expression); LuceneQueryVisitor<SearchBean> lucene = new LuceneQueryVisitor<SearchBean>(fieldName, useAnalyzer ? analyzer : null); lucene.visit(filter); return lucene.getQuery(); } }