2021-07-06 00:30:14 +02:00
|
|
|
package it.cavallium.dbengine.lucene.searcher;
|
|
|
|
|
2021-09-18 18:34:21 +02:00
|
|
|
import io.net5.buffer.api.Send;
|
2021-09-19 19:59:37 +02:00
|
|
|
import it.cavallium.dbengine.database.disk.LLIndexContext;
|
|
|
|
import it.cavallium.dbengine.database.disk.LLIndexSearcher;
|
|
|
|
import reactor.core.publisher.Flux;
|
2021-07-06 00:30:14 +02:00
|
|
|
import reactor.core.publisher.Mono;
|
|
|
|
|
|
|
|
public class AdaptiveLuceneMultiSearcher implements LuceneMultiSearcher {
|
|
|
|
|
2021-09-19 19:59:37 +02:00
|
|
|
private static final LuceneMultiSearcher countLuceneMultiSearcher
|
|
|
|
= new SimpleUnsortedUnscoredLuceneMultiSearcher(new CountLuceneLocalSearcher());
|
2021-07-06 00:30:14 +02:00
|
|
|
|
2021-09-19 19:59:37 +02:00
|
|
|
private static final LuceneMultiSearcher scoredSimpleLuceneShardSearcher
|
|
|
|
= new ScoredSimpleLuceneShardSearcher();
|
2021-07-27 19:34:51 +02:00
|
|
|
|
2021-09-19 19:59:37 +02:00
|
|
|
private static final LuceneMultiSearcher unscoredPagedLuceneMultiSearcher
|
|
|
|
= new SimpleUnsortedUnscoredLuceneMultiSearcher(new SimpleLuceneLocalSearcher());
|
2021-07-06 00:30:14 +02:00
|
|
|
|
|
|
|
@Override
|
2021-09-19 19:59:37 +02:00
|
|
|
public Mono<Send<LuceneSearchResult>> collect(Flux<Send<LLIndexContext>> indexSearchersFlux,
|
|
|
|
LocalQueryParams queryParams,
|
|
|
|
String keyFieldName) {
|
|
|
|
if (queryParams.limit() == 0) {
|
|
|
|
return countLuceneMultiSearcher.collect(indexSearchersFlux, queryParams, keyFieldName);
|
|
|
|
} else if (queryParams.isSorted() || queryParams.isScored()) {
|
|
|
|
return scoredSimpleLuceneShardSearcher.collect(indexSearchersFlux, queryParams, keyFieldName);
|
2021-07-06 00:30:14 +02:00
|
|
|
} else {
|
2021-09-19 19:59:37 +02:00
|
|
|
return unscoredPagedLuceneMultiSearcher.collect(indexSearchersFlux, queryParams, keyFieldName);
|
2021-07-06 00:30:14 +02:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|