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;
|
|
|
|
import it.cavallium.dbengine.database.LLUtils;
|
2021-07-06 00:30:14 +02:00
|
|
|
import reactor.core.publisher.Mono;
|
|
|
|
|
|
|
|
public class AdaptiveLuceneMultiSearcher implements LuceneMultiSearcher {
|
|
|
|
|
2021-07-08 17:01:56 +02:00
|
|
|
private static final LuceneMultiSearcher scoredLuceneMultiSearcher = new ScoredLuceneMultiSearcher();
|
2021-07-06 00:30:14 +02:00
|
|
|
|
2021-07-27 19:34:51 +02:00
|
|
|
private static final LuceneMultiSearcher unscoredPagedLuceneMultiSearcher = new UnscoredPagedLuceneMultiSearcher();
|
|
|
|
|
|
|
|
private static final LuceneMultiSearcher unscoredIterableLuceneMultiSearcher = new UnscoredUnsortedContinuousLuceneMultiSearcher();
|
2021-07-06 00:30:14 +02:00
|
|
|
|
|
|
|
private static final LuceneMultiSearcher countLuceneMultiSearcher = new CountLuceneMultiSearcher();
|
|
|
|
|
|
|
|
@Override
|
2021-09-18 18:34:21 +02:00
|
|
|
public Mono<Send<LuceneMultiSearcher>> createShardSearcher(LocalQueryParams queryParams) {
|
|
|
|
Mono<Send<LuceneMultiSearcher>> shardSearcherCreationMono;
|
2021-07-06 00:30:14 +02:00
|
|
|
if (queryParams.limit() <= 0) {
|
2021-09-18 18:34:21 +02:00
|
|
|
shardSearcherCreationMono = countLuceneMultiSearcher.createShardSearcher(queryParams);
|
2021-07-08 17:01:56 +02:00
|
|
|
} else if (queryParams.isScored()) {
|
2021-09-18 18:34:21 +02:00
|
|
|
shardSearcherCreationMono = scoredLuceneMultiSearcher.createShardSearcher(queryParams);
|
2021-07-27 19:34:51 +02:00
|
|
|
} else if (queryParams.offset() == 0 && queryParams.limit() >= 2147483630 && !queryParams.isSorted()) {
|
2021-09-18 18:34:21 +02:00
|
|
|
shardSearcherCreationMono = unscoredIterableLuceneMultiSearcher.createShardSearcher(queryParams);
|
2021-07-06 00:30:14 +02:00
|
|
|
} else {
|
2021-09-18 18:34:21 +02:00
|
|
|
shardSearcherCreationMono = unscoredPagedLuceneMultiSearcher.createShardSearcher(queryParams);
|
2021-07-06 00:30:14 +02:00
|
|
|
}
|
2021-09-18 18:34:21 +02:00
|
|
|
return Mono.fromRunnable(LLUtils::ensureBlocking).then(shardSearcherCreationMono);
|
2021-07-06 00:30:14 +02:00
|
|
|
}
|
|
|
|
}
|