CavalliumDBEngine/src/main/java/it/cavallium/dbengine/client/Hits.java

package it.cavallium.dbengine.client;

import io.net5.buffer.api.Drop;
import io.net5.buffer.api.Owned;
import io.net5.buffer.api.Send;
import it.cavallium.dbengine.client.query.current.data.TotalHitsCount;
import io.net5.buffer.api.internal.ResourceSupport;
import it.cavallium.dbengine.database.collections.ValueGetter;
import it.cavallium.dbengine.database.collections.ValueTransformer;
import java.util.function.Function;
import reactor.core.publisher.Flux;
import reactor.core.publisher.Mono;
import reactor.util.function.Tuples;

public final class Hits<T> extends ResourceSupport<Hits<T>, Hits<T>> {

	private static final Drop<Hits<?>> DROP = new Drop<>() {
		@Override
		public void drop(Hits<?> obj) {
			if (obj.onClose != null) {
				obj.onClose.run();
			}
		}

		@Override
		public Drop<Hits<?>> fork() {
			return this;
		}

		@Override
		public void attach(Hits<?> obj) {

		}
	};

	private Flux<T> results;
	private TotalHitsCount totalHitsCount;
	private Runnable onClose;

	@SuppressWarnings({"unchecked", "rawtypes"})
	public Hits(Flux<T> results, TotalHitsCount totalHitsCount, Runnable onClose) {
		super((Drop<Hits<T>>) (Drop) DROP);
		this.results = results;
		this.totalHitsCount = totalHitsCount;
		this.onClose = onClose;
	}

	public static <T> Hits<T> empty() {
		return new Hits<>(Flux.empty(), TotalHitsCount.of(0, true), null);
	}

	public static <K, V> Hits<LazyHitEntry<K, V>> withValuesLazy(Hits<LazyHitKey<K>> hits,
			ValueGetter<K, V> valuesGetter) {
		var hitsEntry = hits.results().map(hitKey -> hitKey.withValue(valuesGetter::get));

		return new Hits<>(hitsEntry, hits.totalHitsCount, hits::close);
	}

	public static <K, V> Hits<HitEntry<K, V>> withValues(Hits<HitKey<K>> hits, ValueGetter<K, V> valuesGetter) {
		var hitsEntry = hits.results().flatMap(hitKey -> hitKey.withValue(valuesGetter::get));

		return new Hits<>(hitsEntry, hits.totalHitsCount, hits::close);
	}

	public static <T, U> Function<Send<Hits<HitKey<T>>>, Send<Hits<LazyHitEntry<T, U>>>> generateMapper(
			ValueGetter<T, U> valueGetter) {
		return resultToReceive -> {
			var result = resultToReceive.receive();
			var hitsToTransform = result.results()
					.map(hit -> new LazyHitEntry<>(Mono.just(hit.key()), valueGetter.get(hit.key()), hit.score()));
			return new Hits<>(hitsToTransform, result.totalHitsCount(), result::close).send();
		};
	}

	public static <T, U> Function<Send<Hits<HitKey<T>>>, Send<Hits<LazyHitEntry<T, U>>>> generateMapper(
			ValueTransformer<T, U> valueTransformer) {
		return resultToReceive -> {
			var result = resultToReceive.receive();
			var hitsToTransform = result.results().map(hit -> Tuples.of(hit.score(), hit.key()));
			var transformed = valueTransformer
					.transform(hitsToTransform)
					.filter(tuple3 -> tuple3.getT3().isPresent())
					.map(tuple3 -> new LazyHitEntry<>(Mono.just(tuple3.getT2()),
							Mono.just(tuple3.getT3().orElseThrow()),
							tuple3.getT1()
					));
			return new Hits<>(transformed, result.totalHitsCount(), result::close).send();
		};
	}

	public Flux<T> results() {
		return results;
	}

	public TotalHitsCount totalHitsCount() {
		return totalHitsCount;
	}

	@Override
	public String toString() {
		return "Hits[" + "results=" + results + ", " + "totalHitsCount=" + totalHitsCount + ']';
	}

	@Override
	protected RuntimeException createResourceClosedException() {
		return new IllegalStateException("Closed");
	}

	@Override
	protected Owned<Hits<T>> prepareSend() {
		var results = this.results;
		var totalHitsCount = this.totalHitsCount;
		var onClose = this.onClose;
		return drop -> {
			var instance = new Hits<>(results, totalHitsCount, onClose);
			drop.attach(instance);
			return instance;
		};
	}

	protected void makeInaccessible() {
		this.results = null;
		this.totalHitsCount = null;
		this.onClose = null;
	}
}