2021-11-18 17:13:53 +01:00
|
|
|
package it.cavallium.dbengine.lucene.searcher;
|
|
|
|
|
2023-02-24 17:19:25 +01:00
|
|
|
import static it.cavallium.dbengine.utils.StreamUtils.LUCENE_SCHEDULER;
|
|
|
|
import static it.cavallium.dbengine.utils.StreamUtils.collectOn;
|
|
|
|
import static it.cavallium.dbengine.utils.StreamUtils.fastListing;
|
|
|
|
|
2023-02-09 23:34:25 +01:00
|
|
|
import com.google.common.collect.Streams;
|
2021-11-18 17:13:53 +01:00
|
|
|
import it.cavallium.dbengine.database.disk.LLIndexSearchers;
|
2021-11-19 19:03:31 +01:00
|
|
|
import it.cavallium.dbengine.lucene.collector.Buckets;
|
|
|
|
import it.cavallium.dbengine.lucene.collector.DecimalBucketMultiCollectorManager;
|
2023-02-22 16:59:35 +01:00
|
|
|
import it.cavallium.dbengine.utils.DBException;
|
2023-02-09 23:34:25 +01:00
|
|
|
import java.io.IOException;
|
2023-02-24 17:19:25 +01:00
|
|
|
import java.util.Collection;
|
2021-11-19 19:03:31 +01:00
|
|
|
import java.util.List;
|
2021-12-17 01:48:49 +01:00
|
|
|
import org.apache.logging.log4j.LogManager;
|
|
|
|
import org.apache.logging.log4j.Logger;
|
2021-11-18 17:13:53 +01:00
|
|
|
import org.apache.lucene.search.IndexSearcher;
|
2021-11-19 19:03:31 +01:00
|
|
|
import org.apache.lucene.search.Query;
|
|
|
|
import org.jetbrains.annotations.NotNull;
|
|
|
|
import org.jetbrains.annotations.Nullable;
|
2021-11-18 17:13:53 +01:00
|
|
|
|
|
|
|
public class DecimalBucketMultiSearcher {
|
|
|
|
|
2021-12-17 01:48:49 +01:00
|
|
|
protected static final Logger logger = LogManager.getLogger(DecimalBucketMultiSearcher.class);
|
2021-11-18 17:13:53 +01:00
|
|
|
|
2023-02-09 23:34:25 +01:00
|
|
|
public Buckets collectMulti(LLIndexSearchers indexSearchers,
|
2021-11-18 17:13:53 +01:00
|
|
|
BucketParams bucketParams,
|
2021-11-19 19:03:31 +01:00
|
|
|
@NotNull List<Query> queries,
|
|
|
|
@Nullable Query normalizationQuery) {
|
2023-02-09 23:34:25 +01:00
|
|
|
try {
|
|
|
|
// Search results
|
|
|
|
return this.search(indexSearchers.shards(), bucketParams, queries, normalizationQuery);
|
|
|
|
} finally {
|
2022-06-14 13:10:38 +02:00
|
|
|
indexSearchers.close();
|
2023-02-09 23:34:25 +01:00
|
|
|
}
|
2021-11-18 17:13:53 +01:00
|
|
|
}
|
|
|
|
|
2023-02-24 17:19:25 +01:00
|
|
|
private Buckets search(Collection<IndexSearcher> indexSearchers,
|
2021-11-18 17:13:53 +01:00
|
|
|
BucketParams bucketParams,
|
2021-11-19 19:03:31 +01:00
|
|
|
@NotNull List<Query> queries,
|
|
|
|
@Nullable Query normalizationQuery) {
|
2023-02-09 23:34:25 +01:00
|
|
|
var cmm = new DecimalBucketMultiCollectorManager(bucketParams.min(),
|
|
|
|
bucketParams.max(),
|
|
|
|
bucketParams.buckets(),
|
|
|
|
bucketParams.bucketFieldName(),
|
|
|
|
bucketParams.valueSource(),
|
|
|
|
queries,
|
|
|
|
normalizationQuery,
|
|
|
|
bucketParams.collectionRate(),
|
|
|
|
bucketParams.sampleSize()
|
|
|
|
);
|
2023-02-24 17:19:25 +01:00
|
|
|
return cmm.reduce(collectOn(LUCENE_SCHEDULER, indexSearchers.stream().map(indexSearcher -> {
|
2023-02-09 23:34:25 +01:00
|
|
|
try {
|
2023-02-22 16:59:35 +01:00
|
|
|
return cmm.search(indexSearcher);
|
2023-02-09 23:34:25 +01:00
|
|
|
} catch (IOException e) {
|
2023-02-22 16:59:35 +01:00
|
|
|
throw new DBException(e);
|
2023-02-09 23:34:25 +01:00
|
|
|
}
|
2023-02-24 17:19:25 +01:00
|
|
|
}), fastListing()));
|
2021-11-18 17:13:53 +01:00
|
|
|
}
|
|
|
|
}
|