+++ /dev/null
-package org.apache.lucene.facet.example.simple;
-
-import java.util.Iterator;
-import java.util.List;
-
-import org.apache.lucene.index.IndexReader;
-import org.apache.lucene.index.Term;
-import org.apache.lucene.search.IndexSearcher;
-import org.apache.lucene.search.Query;
-import org.apache.lucene.search.TermQuery;
-import org.apache.lucene.search.TopScoreDocCollector;
-
-import org.apache.lucene.search.MultiCollector;
-import org.apache.lucene.facet.example.ExampleUtils;
-import org.apache.lucene.facet.index.params.DefaultFacetIndexingParams;
-import org.apache.lucene.facet.index.params.FacetIndexingParams;
-import org.apache.lucene.facet.search.DrillDown;
-import org.apache.lucene.facet.search.FacetsCollector;
-import org.apache.lucene.facet.search.params.CountFacetRequest;
-import org.apache.lucene.facet.search.params.FacetRequest;
-import org.apache.lucene.facet.search.params.FacetSearchParams;
-import org.apache.lucene.facet.search.results.FacetResult;
-import org.apache.lucene.facet.search.results.FacetResultNode;
-import org.apache.lucene.facet.taxonomy.CategoryPath;
-import org.apache.lucene.facet.taxonomy.TaxonomyReader;
-
-/**
- * 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.
- */
-
-/**
- * SampleSearcer searches index with facets.
- *
- * @lucene.experimental
- */
-public class SimpleSearcher {
-
- /**
- * Search an index with facets.
- * @param indexReader index reader.
- * @param taxoReader taxonomy reader.
- * @throws Exception on error (no detailed exception handling here for sample simplicity
- * @return facet results
- */
- public static List<FacetResult> searchWithFacets (IndexReader indexReader,
- TaxonomyReader taxoReader) throws Exception {
- CountFacetRequest facetRequest = new CountFacetRequest(new CategoryPath("root","a"), 10);
- return searchWithRequest(indexReader, taxoReader, null, facetRequest);
- }
-
- /**
- * Search an index with facets for given facet requests.
- * @param indexReader index reader.
- * @param taxoReader taxonomy reader.
- * @param indexingParams the facet indexing params
- * @param facetRequests facet requests of interest
- * @throws Exception on error (no detailed exception handling here for sample simplicity
- * @return facet results
- */
- public static List<FacetResult> searchWithRequest(IndexReader indexReader,
- TaxonomyReader taxoReader, FacetIndexingParams indexingParams,
- FacetRequest... facetRequests) throws Exception {
- Query q = new TermQuery(new Term(SimpleUtils.TEXT, "white"));
- return searchWithRequestAndQuery(q, indexReader, taxoReader,
- indexingParams, facetRequests);
- }
-
- /**
- * Search an index with facets for given query and facet requests.
- * @param q query of interest
- * @param indexReader index reader.
- * @param taxoReader taxonomy reader.
- * @param indexingParams the facet indexing params
- * @param facetRequests facet requests of interest
- * @throws Exception on error (no detailed exception handling here for sample simplicity
- * @return facet results
- */
- public static List<FacetResult> searchWithRequestAndQuery(Query q,
- IndexReader indexReader, TaxonomyReader taxoReader,
- FacetIndexingParams indexingParams, FacetRequest... facetRequests)
- throws Exception {
-
- ExampleUtils.log("Query: " + q);
- // prepare searcher to search against
- IndexSearcher searcher = new IndexSearcher(indexReader);
-
- // collect matching documents into a collector
- TopScoreDocCollector topDocsCollector = TopScoreDocCollector.create(10, true);
-
- if (indexingParams == null) {
- indexingParams = new DefaultFacetIndexingParams();
- }
-
- // Faceted search parameters indicate which facets are we interested in
- FacetSearchParams facetSearchParams = new FacetSearchParams(indexingParams);
-
- // Add the facet requests of interest to the search params
- for (FacetRequest frq : facetRequests) {
- facetSearchParams.addFacetRequest(frq);
- }
-
- FacetsCollector facetsCollector = new FacetsCollector(facetSearchParams, indexReader, taxoReader);
-
- // perform documents search and facets accumulation
- searcher.search(q, MultiCollector.wrap(topDocsCollector, facetsCollector));
-
- // Obtain facets results and print them
- List<FacetResult> res = facetsCollector.getFacetResults();
-
- int i = 0;
- for (FacetResult facetResult : res) {
- ExampleUtils.log("Res " + (i++) + ": " + facetResult);
- }
-
- return res;
- }
-
- /**
- * Search an index with facets drill-down.
- * @param indexReader index reader.
- * @param taxoReader taxonomy reader.
- * @throws Exception on error (no detailed exception handling here for sample simplicity
- * @return facet results
- */
- public static List<FacetResult> searchWithDrillDown(IndexReader indexReader,
- TaxonomyReader taxoReader) throws Exception {
-
- // base query the user is interested in
- Query baseQuery = new TermQuery(new Term(SimpleUtils.TEXT, "white"));
-
- // facet of interest
- CountFacetRequest facetRequest = new CountFacetRequest(new CategoryPath("root","a"), 10);
-
- // initial search - all docs matching the base query will contribute to the accumulation
- List<FacetResult> res1 = searchWithRequest(indexReader, taxoReader, null, facetRequest);
-
- // a single result (because there was a single request)
- FacetResult fres = res1.get(0);
-
- // assume the user is interested in the second sub-result
- // (just take the second sub-result returned by the iterator - we know there are 3 results!)
- Iterator<? extends FacetResultNode> resIterator = fres.getFacetResultNode().getSubResults().iterator();
- resIterator.next(); // skip first result
- CategoryPath categoryOfInterest = resIterator.next().getLabel();
-
- // drill-down preparation: turn the base query into a drill-down query for the category of interest
- Query q2 = DrillDown.query(baseQuery, categoryOfInterest);
-
- // that's it - search with the new query and we're done!
- // only documents both matching the base query AND containing the
- // category of interest will contribute to the new accumulation
- return searchWithRequestAndQuery(q2, indexReader, taxoReader, null, facetRequest);
- }
-
-}
fnp / pylucene.git / blobdiff