70 lines
2.3 KiB
Java
70 lines
2.3 KiB
Java
package it.cavallium.dbengine.lucene.searcher;
|
|
|
|
import io.netty5.buffer.api.Send;
|
|
import it.cavallium.dbengine.database.LLUtils;
|
|
import it.cavallium.dbengine.database.disk.LLIndexSearchers;
|
|
import it.cavallium.dbengine.lucene.collector.Buckets;
|
|
import it.cavallium.dbengine.lucene.collector.DecimalBucketMultiCollectorManager;
|
|
import java.util.List;
|
|
import org.apache.logging.log4j.LogManager;
|
|
import org.apache.logging.log4j.Logger;
|
|
import org.apache.lucene.search.IndexSearcher;
|
|
import org.apache.lucene.search.Query;
|
|
import org.jetbrains.annotations.NotNull;
|
|
import org.jetbrains.annotations.Nullable;
|
|
import reactor.core.publisher.Flux;
|
|
import reactor.core.publisher.Mono;
|
|
import reactor.core.scheduler.Schedulers;
|
|
|
|
public class DecimalBucketMultiSearcher {
|
|
|
|
protected static final Logger logger = LogManager.getLogger(DecimalBucketMultiSearcher.class);
|
|
|
|
public Mono<Buckets> collectMulti(Mono<LLIndexSearchers> indexSearchersMono,
|
|
BucketParams bucketParams,
|
|
@NotNull List<Query> queries,
|
|
@Nullable Query normalizationQuery) {
|
|
|
|
return Mono.usingWhen(indexSearchersMono, indexSearchers -> this
|
|
// Search results
|
|
.search(indexSearchers.shards(), bucketParams, queries, normalizationQuery)
|
|
// Ensure that one result is always returned
|
|
.single(), indexSearchers -> Mono.fromCallable(() -> {
|
|
indexSearchers.close();
|
|
return null;
|
|
}).subscribeOn(Schedulers.boundedElastic()));
|
|
}
|
|
|
|
private Mono<Buckets> search(Iterable<IndexSearcher> indexSearchers,
|
|
BucketParams bucketParams,
|
|
@NotNull List<Query> queries,
|
|
@Nullable Query normalizationQuery) {
|
|
return Mono
|
|
.fromCallable(() -> {
|
|
LLUtils.ensureBlocking();
|
|
return new DecimalBucketMultiCollectorManager(bucketParams.min(),
|
|
bucketParams.max(),
|
|
bucketParams.buckets(),
|
|
bucketParams.bucketFieldName(),
|
|
bucketParams.valueSource(),
|
|
queries,
|
|
normalizationQuery,
|
|
bucketParams.collectionRate(),
|
|
bucketParams.sampleSize()
|
|
);
|
|
})
|
|
.flatMap(cmm -> Flux
|
|
.fromIterable(indexSearchers)
|
|
.flatMap(shard -> Mono.fromCallable(() -> {
|
|
LLUtils.ensureBlocking();
|
|
return cmm.search(shard);
|
|
}))
|
|
.collectList()
|
|
.flatMap(results -> Mono.fromCallable(() -> {
|
|
LLUtils.ensureBlocking();
|
|
return cmm.reduce(results);
|
|
}))
|
|
);
|
|
}
|
|
}
|