Skip to content

Commit

Permalink
Replaces usages of FacetsCollector with FacetsCollectorManager
Browse files Browse the repository at this point in the history
In the effort of decreasing usages of IndexSearcher#search(query, Collector) by using the corresponding method that accepts a collector manager, this commit replaces many usages of FacetsCollector with its corresponding existing collector manager.
  • Loading branch information
javanna committed Mar 25, 2022
1 parent bb7568d commit 5242e93
Show file tree
Hide file tree
Showing 16 changed files with 68 additions and 130 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -31,6 +31,7 @@
import org.apache.lucene.facet.FacetResult;
import org.apache.lucene.facet.Facets;
import org.apache.lucene.facet.FacetsCollector;
import org.apache.lucene.facet.FacetsCollectorManager;
import org.apache.lucene.facet.FacetsConfig;
import org.apache.lucene.facet.range.DoubleRange;
import org.apache.lucene.facet.range.DoubleRangeFacetCounts;
Expand Down Expand Up @@ -213,9 +214,7 @@ public static Query getBoundingBoxQuery(
/** User runs a query and counts facets. */
public FacetResult search() throws IOException {

FacetsCollector fc = new FacetsCollector();

searcher.search(new MatchAllDocsQuery(), fc);
FacetsCollector fc = searcher.search(new MatchAllDocsQuery(), new FacetsCollectorManager());

Facets facets =
new DoubleRangeFacetCounts(
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -28,6 +28,7 @@
import org.apache.lucene.facet.FacetResult;
import org.apache.lucene.facet.Facets;
import org.apache.lucene.facet.FacetsCollector;
import org.apache.lucene.facet.FacetsCollectorManager;
import org.apache.lucene.facet.FacetsConfig;
import org.apache.lucene.facet.taxonomy.FastTaxonomyFacetCounts;
import org.apache.lucene.facet.taxonomy.TaxonomyReader;
Expand Down Expand Up @@ -125,12 +126,10 @@ private List<FacetResult> facetsOnly() throws IOException {
IndexSearcher searcher = new IndexSearcher(indexReader);
TaxonomyReader taxoReader = new DirectoryTaxonomyReader(taxoDir);

FacetsCollector fc = new FacetsCollector();

// MatchAllDocsQuery is for "browsing" (counts facets
// for all non-deleted docs in the index); normally
// you'd use a "normal" query:
searcher.search(new MatchAllDocsQuery(), fc);
FacetsCollector fc = searcher.search(new MatchAllDocsQuery(), new FacetsCollectorManager());

// Retrieve results
List<FacetResult> results = new ArrayList<>();
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -73,8 +73,7 @@ public void testBasic() throws IOException {
}

IndexSearcher searcher = newSearcher(writer.getReader());
FacetsCollector fc = new FacetsCollector();
searcher.search(new MatchAllDocsQuery(), fc);
FacetsCollector fc = searcher.search(new MatchAllDocsQuery(), new FacetsCollectorManager());

int visitedDocs = 0;
DocIdSetIterator disi;
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -73,8 +73,7 @@ public void testAddSameDocTwice() throws Exception {
DirectoryReader indexReader = DirectoryReader.open(indexDir);
DirectoryTaxonomyReader taxoReader = new DirectoryTaxonomyReader(taxoDir);
IndexSearcher searcher = newSearcher(indexReader);
FacetsCollector fc = new FacetsCollector();
searcher.search(new MatchAllDocsQuery(), fc);
FacetsCollector fc = searcher.search(new MatchAllDocsQuery(), new FacetsCollectorManager());

Facets facets = getTaxonomyFacetCounts(taxoReader, facetsConfig, fc);
FacetResult res = facets.getTopChildren(10, "a");
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -59,9 +59,8 @@ public void testBasic() throws Exception {
IndexReader r = w.getReader();
w.close();

FacetsCollector fc = new FacetsCollector();
IndexSearcher s = newSearcher(r);
s.search(new MatchAllDocsQuery(), fc);
FacetsCollector fc = s.search(new MatchAllDocsQuery(), new FacetsCollectorManager());

LongValueFacetCounts facets = new LongValueFacetCounts("field", fc);

Expand Down Expand Up @@ -111,9 +110,8 @@ public void testOnlyBigLongs() throws Exception {
IndexReader r = w.getReader();
w.close();

FacetsCollector fc = new FacetsCollector();
IndexSearcher s = newSearcher(r);
s.search(new MatchAllDocsQuery(), fc);
FacetsCollector fc = s.search(new MatchAllDocsQuery(), new FacetsCollectorManager());

LongValueFacetCounts facets = new LongValueFacetCounts("field", fc);

Expand Down Expand Up @@ -143,9 +141,8 @@ public void testGetAllDims() throws Exception {
IndexReader r = w.getReader();
w.close();

FacetsCollector fc = new FacetsCollector();
IndexSearcher s = newSearcher(r);
s.search(new MatchAllDocsQuery(), fc);
FacetsCollector fc = s.search(new MatchAllDocsQuery(), new FacetsCollectorManager());

Facets facets = new LongValueFacetCounts("field", fc);

Expand Down Expand Up @@ -204,7 +201,6 @@ public void testRandomSingleValued() throws Exception {

int iters = atLeast(100);
for (int iter = 0; iter < iters; iter++) {
FacetsCollector fc = new FacetsCollector();
if (VERBOSE) {
System.out.println("\nTEST: iter=" + iter);
System.out.println(" test all docs");
Expand All @@ -231,7 +227,7 @@ public void testRandomSingleValued() throws Exception {

LongValueFacetCounts facetCounts;
if (random().nextBoolean()) {
s.search(new MatchAllDocsQuery(), fc);
FacetsCollector fc = s.search(new MatchAllDocsQuery(), new FacetsCollectorManager());
if (random().nextBoolean()) {
if (VERBOSE) {
System.out.println(" use value source");
Expand Down Expand Up @@ -331,8 +327,7 @@ public void testRandomSingleValued() throws Exception {
System.out.println(" test id range " + minId + "-" + maxId);
}

fc = new FacetsCollector();
s.search(IntPoint.newRangeQuery("id", minId, maxId), fc);
FacetsCollector fc = s.search(IntPoint.newRangeQuery("id", minId, maxId), new FacetsCollectorManager());
if (random().nextBoolean()) {
if (VERBOSE) {
System.out.println(" use doc values");
Expand Down Expand Up @@ -481,7 +476,6 @@ public void testRandomMultiValued() throws Exception {

int iters = atLeast(100);
for (int iter = 0; iter < iters; iter++) {
FacetsCollector fc = new FacetsCollector();
if (VERBOSE) {
System.out.println("\nTEST: iter=" + iter);
System.out.println(" test all docs");
Expand All @@ -505,7 +499,7 @@ public void testRandomMultiValued() throws Exception {

LongValueFacetCounts facetCounts;
if (random().nextBoolean()) {
s.search(new MatchAllDocsQuery(), fc);
FacetsCollector fc = s.search(new MatchAllDocsQuery(), new FacetsCollectorManager());
if (VERBOSE) {
System.out.println(" use doc values");
}
Expand Down Expand Up @@ -577,8 +571,7 @@ public void testRandomMultiValued() throws Exception {
System.out.println(" test id range " + minId + "-" + maxId);
}

fc = new FacetsCollector();
s.search(IntPoint.newRangeQuery("id", minId, maxId), fc);
FacetsCollector fc = s.search(IntPoint.newRangeQuery("id", minId, maxId), new FacetsCollectorManager());
if (random().nextBoolean()) {
facetCounts = new LongValueFacetCounts("field", fc);
} else {
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -260,8 +260,7 @@ public void testStaleState() throws Exception {
IndexSearcher searcher = newSearcher(writer.getReader());
writer.close();

FacetsCollector c = new FacetsCollector();
searcher.search(new MatchAllDocsQuery(), c);
FacetsCollector c = searcher.search(new MatchAllDocsQuery(), new FacetsCollectorManager());

// using a stale state
expectThrows(IllegalStateException.class, () -> new StringValueFacetCounts(state, c));
Expand Down Expand Up @@ -339,8 +338,7 @@ private void checkFacetResult(
StringDocValuesReaderState state =
new StringDocValuesReaderState(searcher.getIndexReader(), "field");

FacetsCollector c = new FacetsCollector();
searcher.search(new MatchAllDocsQuery(), c);
FacetsCollector c = searcher.search(new MatchAllDocsQuery(), new FacetsCollectorManager());

for (int topN : topNs) {

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -36,6 +36,7 @@
import org.apache.lucene.facet.FacetTestCase;
import org.apache.lucene.facet.Facets;
import org.apache.lucene.facet.FacetsCollector;
import org.apache.lucene.facet.FacetsCollectorManager;
import org.apache.lucene.facet.FacetsConfig;
import org.apache.lucene.facet.LabelAndValue;
import org.apache.lucene.facet.MultiDoubleValuesSource;
Expand Down Expand Up @@ -85,9 +86,8 @@ public void testBasicLong() throws Exception {
IndexReader r = w.getReader();
w.close();

FacetsCollector fc = new FacetsCollector();
IndexSearcher s = newSearcher(r);
s.search(new MatchAllDocsQuery(), fc);
FacetsCollector fc = s.search(new MatchAllDocsQuery(), new FacetsCollectorManager());

Facets facets =
new LongRangeFacetCounts(
Expand Down Expand Up @@ -131,9 +131,8 @@ public void testBasicLongMultiValued() throws Exception {
IndexReader r = w.getReader();
w.close();

FacetsCollector fc = new FacetsCollector();
IndexSearcher s = newSearcher(r);
s.search(new MatchAllDocsQuery(), fc);
FacetsCollector fc = s.search(new MatchAllDocsQuery(), new FacetsCollectorManager());

Facets facets =
new LongRangeFacetCounts(
Expand Down Expand Up @@ -182,9 +181,8 @@ public void testBasicLongMultiValuedMixedSegmentTypes() throws Exception {
IndexReader r = w.getReader();
w.close();

FacetsCollector fc = new FacetsCollector();
IndexSearcher s = newSearcher(r);
s.search(new MatchAllDocsQuery(), fc);
FacetsCollector fc = s.search(new MatchAllDocsQuery(), new FacetsCollectorManager());

Facets facets =
new LongRangeFacetCounts(
Expand Down Expand Up @@ -223,9 +221,8 @@ public void testLongGetAllDims() throws Exception {
IndexReader r = w.getReader();
w.close();

FacetsCollector fc = new FacetsCollector();
IndexSearcher s = newSearcher(r);
s.search(new MatchAllDocsQuery(), fc);
FacetsCollector fc = s.search(new MatchAllDocsQuery(), new FacetsCollectorManager());

Facets facets =
new LongRangeFacetCounts(
Expand Down Expand Up @@ -287,9 +284,8 @@ public void testLongMinMax() throws Exception {
IndexReader r = w.getReader();
w.close();

FacetsCollector fc = new FacetsCollector();
IndexSearcher s = newSearcher(r);
s.search(new MatchAllDocsQuery(), fc);
FacetsCollector fc = s.search(new MatchAllDocsQuery(), new FacetsCollectorManager());

Facets facets =
new LongRangeFacetCounts(
Expand Down Expand Up @@ -327,9 +323,8 @@ public void testOverlappedEndStart() throws Exception {
IndexReader r = w.getReader();
w.close();

FacetsCollector fc = new FacetsCollector();
IndexSearcher s = newSearcher(r);
s.search(new MatchAllDocsQuery(), fc);
FacetsCollector fc = s.search(new MatchAllDocsQuery(), new FacetsCollectorManager());

Facets facets =
new LongRangeFacetCounts(
Expand Down Expand Up @@ -363,9 +358,8 @@ public void testEmptyRangesSingleValued() throws Exception {
IndexReader r = w.getReader();
w.close();

FacetsCollector fc = new FacetsCollector();
IndexSearcher s = newSearcher(r);
s.search(new MatchAllDocsQuery(), fc);
FacetsCollector fc = s.search(new MatchAllDocsQuery(), new FacetsCollectorManager());

Facets facets = new LongRangeFacetCounts("field", fc);

Expand Down Expand Up @@ -393,9 +387,8 @@ public void testEmptyRangesMultiValued() throws Exception {
IndexReader r = w.getReader();
w.close();

FacetsCollector fc = new FacetsCollector();
IndexSearcher s = newSearcher(r);
s.search(new MatchAllDocsQuery(), fc);
FacetsCollector fc = s.search(new MatchAllDocsQuery(), new FacetsCollectorManager());

Facets facets = new LongRangeFacetCounts("field", fc);

Expand Down Expand Up @@ -539,10 +532,8 @@ public void testBasicDouble() throws Exception {

IndexReader r = w.getReader();

FacetsCollector fc = new FacetsCollector();

IndexSearcher s = newSearcher(r);
s.search(new MatchAllDocsQuery(), fc);
FacetsCollector fc = s.search(new MatchAllDocsQuery(), new FacetsCollectorManager());
Facets facets =
new DoubleRangeFacetCounts(
"field",
Expand Down Expand Up @@ -577,10 +568,8 @@ public void testBasicDoubleMultiValued() throws Exception {

IndexReader r = w.getReader();

FacetsCollector fc = new FacetsCollector();

IndexSearcher s = newSearcher(r);
s.search(new MatchAllDocsQuery(), fc);
FacetsCollector fc = s.search(new MatchAllDocsQuery(), new FacetsCollectorManager());
Facets facets =
new DoubleRangeFacetCounts(
"field",
Expand Down Expand Up @@ -626,9 +615,8 @@ public void testBasicDoubleMultiValuedMixedSegmentTypes() throws Exception {
IndexReader r = w.getReader();
w.close();

FacetsCollector fc = new FacetsCollector();
IndexSearcher s = newSearcher(r);
s.search(new MatchAllDocsQuery(), fc);
FacetsCollector fc = s.search(new MatchAllDocsQuery(), new FacetsCollectorManager());

Facets facets =
new DoubleRangeFacetCounts(
Expand Down Expand Up @@ -756,8 +744,7 @@ public void testRandomLongsSingleValued() throws Exception {
}
}

FacetsCollector sfc = new FacetsCollector();
s.search(new MatchAllDocsQuery(), sfc);
FacetsCollector sfc = s.search(new MatchAllDocsQuery(), new FacetsCollectorManager());
Query fastMatchQuery;
if (random().nextBoolean()) {
if (random().nextBoolean()) {
Expand Down Expand Up @@ -953,8 +940,7 @@ public void testRandomLongsMultiValued() throws Exception {
}
}

FacetsCollector sfc = new FacetsCollector();
s.search(new MatchAllDocsQuery(), sfc);
FacetsCollector sfc = s.search(new MatchAllDocsQuery(), new FacetsCollectorManager());
Query fastMatchQuery;
if (random().nextBoolean()) {
if (random().nextBoolean()) {
Expand Down Expand Up @@ -1106,8 +1092,7 @@ public void testRandomDoublesSingleValued() throws Exception {
}
}

FacetsCollector sfc = new FacetsCollector();
s.search(new MatchAllDocsQuery(), sfc);
FacetsCollector sfc = s.search(new MatchAllDocsQuery(), new FacetsCollectorManager());
Query fastMatchFilter;
if (random().nextBoolean()) {
if (random().nextBoolean()) {
Expand Down Expand Up @@ -1276,8 +1261,7 @@ public void testRandomDoublesMultiValued() throws Exception {
}
}

FacetsCollector sfc = new FacetsCollector();
s.search(new MatchAllDocsQuery(), sfc);
FacetsCollector sfc = s.search(new MatchAllDocsQuery(), new FacetsCollectorManager());
Query fastMatchFilter;
if (random().nextBoolean()) {
if (random().nextBoolean()) {
Expand Down Expand Up @@ -1352,10 +1336,8 @@ public void testMissingValues() throws Exception {

IndexReader r = w.getReader();

FacetsCollector fc = new FacetsCollector();

IndexSearcher s = newSearcher(r);
s.search(new MatchAllDocsQuery(), fc);
FacetsCollector fc = s.search(new MatchAllDocsQuery(), new FacetsCollectorManager());
Facets facets =
new LongRangeFacetCounts(
"field",
Expand Down Expand Up @@ -1396,10 +1378,8 @@ public void testMissingValuesMultiValued() throws Exception {

IndexReader r = w.getReader();

FacetsCollector fc = new FacetsCollector();

IndexSearcher s = newSearcher(r);
s.search(new MatchAllDocsQuery(), fc);
FacetsCollector fc = s.search(new MatchAllDocsQuery(), new FacetsCollectorManager());
Facets facets =
new LongRangeFacetCounts(
"field",
Expand Down Expand Up @@ -1544,12 +1524,10 @@ public void testCustomDoubleValuesSource() throws Exception {

FacetsConfig config = new FacetsConfig();

FacetsCollector fc = new FacetsCollector();

IndexReader r = writer.getReader();

IndexSearcher s = newSearcher(r, false, false);
s.search(new MatchAllDocsQuery(), fc);
FacetsCollector fc = s.search(new MatchAllDocsQuery(), new FacetsCollectorManager());

final DoubleRange[] ranges =
new DoubleRange[] {
Expand Down
Loading

0 comments on commit 5242e93

Please sign in to comment.