Apache
/
lucene
mirror of https://github.com/apache/lucene.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Update TestTopDocsCollector to no longer rely on the deprecated search(Query, Collector) (#13600)

This commit is contained in:
Luca Cavanna 2024-07-24 13:06:31 +02:00 committed by GitHub
parent 97d89c661f
commit d491dfe131
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 31 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

38
lucene/core/src/test/org/apache/lucene/search/TestTopDocsCollector.java
View File

@ -18,6 +18,7 @@ package org.apache.lucene.search;
import java.io.IOException;
import java.util.Arrays;
import java.util.Collection;
import org.apache.lucene.document.Document;
import org.apache.lucene.document.Field;
import org.apache.lucene.document.Field.Store;
@ -41,6 +42,35 @@ import org.apache.lucene.util.BytesRef;
public class TestTopDocsCollector extends LuceneTestCase {
private static final class MyTopDocsCollectorMananger
implements CollectorManager<MyTopDocsCollector, MyTopDocsCollector> {
private final int numHits;
MyTopDocsCollectorMananger(int numHits) {
this.numHits = numHits;
}
@Override
public MyTopDocsCollector newCollector() {
return new MyTopDocsCollector(numHits);
}
@Override
public MyTopDocsCollector reduce(Collection<MyTopDocsCollector> collectors) {
int totalHits = 0;
MyTopDocsCollector myTopDocsCollector = new MyTopDocsCollector(numHits);
for (MyTopDocsCollector collector : collectors) {
totalHits += collector.totalHits;
for (ScoreDoc scoreDoc : collector.pq) {
myTopDocsCollector.pq.insertWithOverflow(scoreDoc);
}
}
myTopDocsCollector.totalHits = totalHits;
return myTopDocsCollector;
}
}
private static final class MyTopDocsCollector extends TopDocsCollector<ScoreDoc> {
private int idx = 0;
@ -125,14 +155,8 @@ public class TestTopDocsCollector extends LuceneTestCase {
private TopDocsCollector<ScoreDoc> doSearch(int numResults) throws IOException {
Query q = new MatchAllDocsQuery();
return doSearch(numResults, q);
}
private TopDocsCollector<ScoreDoc> doSearch(int numResults, Query q) throws IOException {
IndexSearcher searcher = newSearcher(reader);
TopDocsCollector<ScoreDoc> tdc = new MyTopDocsCollector(numResults);
searcher.search(q, tdc);
return tdc;
return searcher.search(q, new MyTopDocsCollectorMananger(numResults));
}
private TopDocs doSearchWithThreshold(