andreacavalli/CavalliumDBEngine
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Fix unscored-unsorted collectors

Browse Source
This commit is contained in:
Andrea Cavalli 2021-07-31 23:12:13 +02:00
parent 586be8751c
commit 328a45f40d
1 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
src/main/java/it/cavallium/dbengine/lucene/searcher/UnscoredUnsortedContinuousLuceneMultiSearcher.java
View File

@ -10,6 +10,7 @@ import java.util.List;
import java.util.concurrent.atomic.AtomicBoolean;
import java.util.concurrent.atomic.AtomicInteger;
import java.util.concurrent.locks.LockSupport;
import org.apache.lucene.index.LeafReaderContext;
import org.apache.lucene.search.Collector;
import org.apache.lucene.search.CollectorManager;
import org.apache.lucene.search.IndexSearcher;
@ -44,10 +45,11 @@ public class UnscoredUnsortedContinuousLuceneMultiSearcher implements LuceneMult
class IterableCollector extends SimpleCollector {
private int shardIndex;
private LeafReaderContext context;
@Override
public void collect(int i) {
var scoreDoc = new ScoreDoc(i, 0, shardIndex);
var scoreDoc = new ScoreDoc(context.docBase + i, 0, shardIndex);
synchronized (scoreDocsSink) {
while (scoreDocsSink.tryEmitNext(scoreDoc) == EmitResult.FAIL_OVERFLOW) {
LockSupport.parkNanos(10);
@ -55,6 +57,11 @@ public class UnscoredUnsortedContinuousLuceneMultiSearcher implements LuceneMult
}
}
@Override
protected void doSetNextReader(LeafReaderContext context) throws IOException {
this.context = context;
}
@Override
public ScoreMode scoreMode() {
return ScoreMode.COMPLETE_NO_SCORES;