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

Update TestTopDocsMerge to not rely on search(Query, Collector) (#13601) 

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

19
lucene/core/src/test/org/apache/lucene/search/TestTopDocsMerge.java
View File

@ -292,26 +292,25 @@ public class TestTopDocsMerge extends LuceneTestCase {
topHits = searcher.search(query, numHits);
}
} else {
final TopFieldCollector c = TopFieldCollector.create(sort, numHits, Integer.MAX_VALUE);
searcher.search(query, c);
TopFieldDocs topFieldDocs =
searcher.search(query, new TopFieldCollectorManager(sort, numHits, Integer.MAX_VALUE));
if (useFrom) {
from = TestUtil.nextInt(random(), 0, numHits - 1);
size = numHits - from;
TopDocs tempTopHits = c.topDocs();
if (from < tempTopHits.scoreDocs.length) {
if (from < topFieldDocs.scoreDocs.length) {
// Can't use TopDocs#topDocs(start, howMany), since it has different behaviour when
// start >= hitCount
// than TopDocs#merge currently has
ScoreDoc[] newScoreDocs =
new ScoreDoc[Math.min(size, tempTopHits.scoreDocs.length - from)];
System.arraycopy(tempTopHits.scoreDocs, from, newScoreDocs, 0, newScoreDocs.length);
tempTopHits.scoreDocs = newScoreDocs;
topHits = tempTopHits;
new ScoreDoc[Math.min(size, topFieldDocs.scoreDocs.length - from)];
System.arraycopy(topFieldDocs.scoreDocs, from, newScoreDocs, 0, newScoreDocs.length);
topFieldDocs.scoreDocs = newScoreDocs;
topHits = topFieldDocs;
} else {
topHits = new TopDocs(tempTopHits.totalHits, new ScoreDoc[0]);
topHits = new TopDocs(topFieldDocs.totalHits, new ScoreDoc[0]);
}
} else {
topHits = c.topDocs(0, numHits);
topHits = topFieldDocs;
}
}