59 lines
1.5 KiB
Java
59 lines
1.5 KiB
Java
package it.cavallium.dbengine.lucene.collector;
|
|
|
|
import java.io.IOException;
|
|
import java.util.Collection;
|
|
import java.util.List;
|
|
import org.apache.lucene.index.LeafReaderContext;
|
|
import org.apache.lucene.search.Collector;
|
|
import org.apache.lucene.search.CollectorManager;
|
|
import org.apache.lucene.search.LeafCollector;
|
|
import org.apache.lucene.search.ScoreDoc;
|
|
import org.apache.lucene.search.ScoreMode;
|
|
import reactor.core.publisher.FluxSink;
|
|
import reactor.core.publisher.Sinks.Many;
|
|
|
|
public class ReactiveCollectorMultiManager implements CollectorMultiManager<Void, Void> {
|
|
|
|
private final FluxSink<ScoreDoc> scoreDocsSink;
|
|
|
|
public ReactiveCollectorMultiManager(FluxSink<ScoreDoc> scoreDocsSink) {
|
|
this.scoreDocsSink = scoreDocsSink;
|
|
}
|
|
|
|
public CollectorManager<Collector, Void> get(int shardIndex) {
|
|
return new CollectorManager<>() {
|
|
|
|
@Override
|
|
public Collector newCollector() {
|
|
return new Collector() {
|
|
|
|
@Override
|
|
public LeafCollector getLeafCollector(LeafReaderContext leafReaderContext) {
|
|
return new ReactiveLeafCollector(leafReaderContext, scoreDocsSink, shardIndex);
|
|
}
|
|
|
|
@Override
|
|
public ScoreMode scoreMode() {
|
|
return ReactiveCollectorMultiManager.this.scoreMode();
|
|
}
|
|
};
|
|
}
|
|
|
|
@Override
|
|
public Void reduce(Collection<Collector> collection) {
|
|
return null;
|
|
}
|
|
};
|
|
}
|
|
|
|
@Override
|
|
public ScoreMode scoreMode() {
|
|
return ScoreMode.COMPLETE_NO_SCORES;
|
|
}
|
|
|
|
@Override
|
|
public Void reduce(List<Void> results) {
|
|
return null;
|
|
}
|
|
}
|