2021-10-28 23:48:25 +02:00
|
|
|
package it.cavallium.dbengine.client;
|
|
|
|
|
|
|
|
import it.cavallium.dbengine.client.query.current.data.TotalHitsCount;
|
2022-06-30 15:06:10 +02:00
|
|
|
import it.cavallium.dbengine.database.DiscardingCloseable;
|
2022-07-23 15:03:59 +02:00
|
|
|
import it.cavallium.dbengine.database.SafeCloseable;
|
2021-10-28 23:48:25 +02:00
|
|
|
import it.cavallium.dbengine.database.collections.ValueGetter;
|
2022-07-23 14:25:59 +02:00
|
|
|
import it.cavallium.dbengine.lucene.LuceneCloseable;
|
2022-06-14 18:05:26 +02:00
|
|
|
import it.cavallium.dbengine.utils.SimpleResource;
|
2021-10-28 23:48:25 +02:00
|
|
|
import java.util.function.Function;
|
2023-02-09 23:34:25 +01:00
|
|
|
import java.util.stream.Stream;
|
2022-07-23 15:03:59 +02:00
|
|
|
import org.apache.logging.log4j.LogManager;
|
|
|
|
import org.apache.logging.log4j.Logger;
|
2021-10-28 23:48:25 +02:00
|
|
|
|
2022-07-23 14:25:59 +02:00
|
|
|
public class Hits<T> extends SimpleResource implements DiscardingCloseable {
|
2022-06-20 23:31:42 +02:00
|
|
|
|
2022-07-23 15:03:59 +02:00
|
|
|
private static final Logger LOG = LogManager.getLogger(Hits.class);
|
2023-02-09 23:34:25 +01:00
|
|
|
private static final Hits<?> EMPTY_HITS = new Hits<>(Stream.empty(), TotalHitsCount.of(0, true), false);
|
|
|
|
private final Stream<T> results;
|
2022-07-23 14:25:59 +02:00
|
|
|
private final TotalHitsCount totalHitsCount;
|
2021-10-28 23:48:25 +02:00
|
|
|
|
2023-02-09 23:34:25 +01:00
|
|
|
public Hits(Stream<T> results, TotalHitsCount totalHitsCount) {
|
2022-07-23 14:25:59 +02:00
|
|
|
this(results, totalHitsCount, true);
|
2022-06-20 23:31:42 +02:00
|
|
|
}
|
|
|
|
|
2023-02-09 23:34:25 +01:00
|
|
|
private Hits(Stream<T> results, TotalHitsCount totalHitsCount, boolean canClose) {
|
2022-07-23 14:25:59 +02:00
|
|
|
super(canClose);
|
2021-10-28 23:48:25 +02:00
|
|
|
this.results = results;
|
|
|
|
this.totalHitsCount = totalHitsCount;
|
|
|
|
}
|
|
|
|
|
2022-06-20 23:31:42 +02:00
|
|
|
@SuppressWarnings("unchecked")
|
2021-10-28 23:48:25 +02:00
|
|
|
public static <T> Hits<T> empty() {
|
2022-06-20 23:31:42 +02:00
|
|
|
return (Hits<T>) EMPTY_HITS;
|
2021-10-28 23:48:25 +02:00
|
|
|
}
|
|
|
|
|
2023-02-09 23:34:25 +01:00
|
|
|
public static <T, U> Function<Hits<HitKey<T>>, Hits<HitEntry<T, U>>> generateMapper(
|
2021-10-28 23:48:25 +02:00
|
|
|
ValueGetter<T, U> valueGetter) {
|
2021-11-08 11:17:52 +01:00
|
|
|
return result -> {
|
2021-10-28 23:48:25 +02:00
|
|
|
var hitsToTransform = result.results()
|
2023-02-09 23:34:25 +01:00
|
|
|
.map(hit -> new HitEntry<>(hit.key(), valueGetter.get(hit.key()), hit.score()));
|
2022-07-23 22:13:17 +02:00
|
|
|
return Hits.withResource(hitsToTransform, result.totalHitsCount(), result);
|
2021-10-28 23:48:25 +02:00
|
|
|
};
|
|
|
|
}
|
|
|
|
|
2023-02-09 23:34:25 +01:00
|
|
|
public static <T> Hits<T> withResource(Stream<T> hits, TotalHitsCount count, SafeCloseable resource) {
|
2022-07-23 15:03:59 +02:00
|
|
|
if (resource instanceof LuceneCloseable luceneCloseable) {
|
|
|
|
return new LuceneHits<>(hits, count, luceneCloseable);
|
|
|
|
} else {
|
|
|
|
return new CloseableHits<>(hits, count, resource);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2023-02-09 23:34:25 +01:00
|
|
|
public Stream<T> results() {
|
2022-06-14 18:05:26 +02:00
|
|
|
ensureOpen();
|
2021-10-28 23:48:25 +02:00
|
|
|
return results;
|
|
|
|
}
|
|
|
|
|
|
|
|
public TotalHitsCount totalHitsCount() {
|
2022-06-14 18:05:26 +02:00
|
|
|
ensureOpen();
|
2021-10-28 23:48:25 +02:00
|
|
|
return totalHitsCount;
|
|
|
|
}
|
|
|
|
|
|
|
|
@Override
|
|
|
|
public String toString() {
|
|
|
|
return "Hits[" + "results=" + results + ", " + "totalHitsCount=" + totalHitsCount + ']';
|
|
|
|
}
|
|
|
|
|
|
|
|
@Override
|
2022-06-14 18:05:26 +02:00
|
|
|
protected void onClose() {
|
2022-07-23 14:25:59 +02:00
|
|
|
}
|
|
|
|
|
2022-07-23 15:03:59 +02:00
|
|
|
public static final class LuceneHits<U> extends Hits<U> implements LuceneCloseable {
|
|
|
|
|
|
|
|
private final LuceneCloseable resource;
|
|
|
|
|
2023-02-09 23:34:25 +01:00
|
|
|
public LuceneHits(Stream<U> hits, TotalHitsCount count, LuceneCloseable resource) {
|
2022-07-23 15:03:59 +02:00
|
|
|
super(hits, count);
|
|
|
|
this.resource = resource;
|
|
|
|
}
|
|
|
|
|
|
|
|
@Override
|
|
|
|
protected void onClose() {
|
|
|
|
try {
|
|
|
|
resource.close();
|
|
|
|
} catch (Throwable ex) {
|
|
|
|
LOG.error("Failed to close resource", ex);
|
|
|
|
}
|
|
|
|
super.onClose();
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
public static final class CloseableHits<U> extends Hits<U> {
|
|
|
|
|
|
|
|
private final SafeCloseable resource;
|
|
|
|
|
2023-02-09 23:34:25 +01:00
|
|
|
public CloseableHits(Stream<U> hits, TotalHitsCount count, SafeCloseable resource) {
|
2022-07-23 15:03:59 +02:00
|
|
|
super(hits, count);
|
|
|
|
this.resource = resource;
|
|
|
|
}
|
|
|
|
|
|
|
|
@Override
|
|
|
|
protected void onClose() {
|
|
|
|
try {
|
|
|
|
resource.close();
|
|
|
|
} catch (Throwable ex) {
|
|
|
|
LOG.error("Failed to close resource", ex);
|
|
|
|
}
|
|
|
|
super.onClose();
|
|
|
|
}
|
|
|
|
}
|
2021-10-28 23:48:25 +02:00
|
|
|
}
|