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-10-08 11:50:28 +02:00
|
|
|
import io.net5.buffer.api.internal.ResourceSupport;
|
|
|
|
import it.cavallium.dbengine.database.LLUtils;
|
2021-09-18 18:34:21 +02:00
|
|
|
import it.cavallium.dbengine.database.disk.LLIndexSearcher;
|
2021-09-20 12:51:27 +02:00
|
|
|
import it.cavallium.dbengine.database.disk.LLIndexSearchers;
|
2021-10-08 11:50:28 +02:00
|
|
|
import it.cavallium.dbengine.database.disk.LLIndexSearchers.UnshardedIndexSearchers;
|
|
|
|
import it.cavallium.dbengine.lucene.searcher.LLSearchTransformer.TransformerInput;
|
2021-10-08 02:13:33 +02:00
|
|
|
import org.apache.lucene.search.IndexSearcher;
|
2021-07-06 00:30:14 +02:00
|
|
|
import reactor.core.publisher.Mono;
|
|
|
|
|
|
|
|
public class AdaptiveLuceneLocalSearcher implements LuceneLocalSearcher {
|
|
|
|
|
|
|
|
private static final LuceneLocalSearcher localSearcher = new SimpleLuceneLocalSearcher();
|
|
|
|
|
2021-07-06 01:30:37 +02:00
|
|
|
private static final LuceneLocalSearcher countSearcher = new CountLuceneLocalSearcher();
|
|
|
|
|
2021-07-06 00:30:14 +02:00
|
|
|
@Override
|
2021-09-20 12:51:27 +02:00
|
|
|
public Mono<Send<LuceneSearchResult>> collect(Mono<Send<LLIndexSearcher>> indexSearcher,
|
2021-07-06 01:30:37 +02:00
|
|
|
LocalQueryParams queryParams,
|
2021-09-20 12:51:27 +02:00
|
|
|
String keyFieldName,
|
|
|
|
LLSearchTransformer transformer) {
|
2021-10-08 11:50:28 +02:00
|
|
|
Mono<Send<LLIndexSearchers>> indexSearchersMono = indexSearcher
|
|
|
|
.map(LLIndexSearchers::unsharded)
|
|
|
|
.map(ResourceSupport::send);
|
|
|
|
|
|
|
|
if (transformer == LLSearchTransformer.NO_TRANSFORMATION) {
|
|
|
|
return transformedCollect(indexSearcher, queryParams, keyFieldName, transformer);
|
|
|
|
} else {
|
|
|
|
return LLUtils.usingSendResource(indexSearchersMono, indexSearchers -> transformer
|
|
|
|
.transform(Mono.fromCallable(() -> new TransformerInput(indexSearchers, queryParams)))
|
|
|
|
.flatMap(queryParams2 -> this
|
|
|
|
.transformedCollect(indexSearcher, queryParams2, keyFieldName, LLSearchTransformer.NO_TRANSFORMATION)),
|
|
|
|
true);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
public Mono<Send<LuceneSearchResult>> transformedCollect(Mono<Send<LLIndexSearcher>> indexSearcher,
|
|
|
|
LocalQueryParams queryParams,
|
|
|
|
String keyFieldName,
|
|
|
|
LLSearchTransformer transformer) {
|
2021-07-06 01:30:37 +02:00
|
|
|
if (queryParams.limit() == 0) {
|
2021-09-20 12:51:27 +02:00
|
|
|
return countSearcher.collect(indexSearcher, queryParams, keyFieldName, transformer);
|
2021-07-06 01:30:37 +02:00
|
|
|
} else {
|
2021-09-20 12:51:27 +02:00
|
|
|
return localSearcher.collect(indexSearcher, queryParams, keyFieldName, transformer);
|
2021-07-06 01:30:37 +02:00
|
|
|
}
|
2021-07-06 00:30:14 +02:00
|
|
|
}
|
|
|
|
}
|