CavalliumDBEngine/src/main/java/it/cavallium/dbengine/lucene/searcher/LocalQueryParams.java

package it.cavallium.dbengine.lucene.searcher;

import static it.cavallium.dbengine.lucene.LuceneUtils.safeLongToInt;

import it.cavallium.dbengine.lucene.LuceneUtils;
import it.cavallium.dbengine.lucene.PageLimits;
import java.time.Duration;
import java.util.Objects;
import org.apache.lucene.search.Query;
import org.apache.lucene.search.Sort;
import org.jetbrains.annotations.NotNull;
import org.jetbrains.annotations.Nullable;

public record LocalQueryParams(@NotNull Query query, int offsetInt, long offsetLong, int limitInt, long limitLong,
															 @NotNull PageLimits pageLimits, @Nullable Sort sort,
															 @Nullable Boolean computePreciseHitsCount, Duration timeout) {

	/**
	 * Sorted params with long offsets
	 */
	public LocalQueryParams(@NotNull Query query,
			long offsetLong,
			long limitLong,
			@NotNull PageLimits pageLimits,
			@Nullable Sort sort,
			@Nullable Boolean computePreciseHitsCount,
			Duration timeout) {
		this(query,
				safeLongToInt(offsetLong),
				offsetLong,
				safeLongToInt(limitLong),
				limitLong,
				pageLimits,
				sort,
				computePreciseHitsCount,
				timeout
		);
	}

	/**
	 * Sorted params with int offsets
	 */
	public LocalQueryParams(@NotNull Query query,
			int offsetInt,
			int limitInt,
			@NotNull PageLimits pageLimits,
			@Nullable Sort sort,
			boolean computePreciseHitsCount,
			Duration timeout) {
		this(query, offsetInt, offsetInt, limitInt, limitInt, pageLimits, sort, computePreciseHitsCount, timeout);
	}

	/**
	 * Unsorted params with int offsets
	 */
	public LocalQueryParams(@NotNull Query query,
			int offsetInt,
			int limitInt,
			@NotNull PageLimits pageLimits,
			Duration timeout) {
		this(query, offsetInt, offsetInt, limitInt, limitInt, pageLimits, null, null, timeout);
	}


	/**
	 * Unsorted params with long offsets
	 */
	public LocalQueryParams(@NotNull Query query,
			long offsetLong,
			long limitLong,
			@NotNull PageLimits pageLimits,
			Duration timeout) {
		this(query,
				safeLongToInt(offsetLong),
				offsetLong,
				safeLongToInt(limitLong),
				limitLong,
				pageLimits,
				null,
				null,
				timeout
		);
	}

	public boolean isSorted() {
		return sort != null;
	}

	public boolean isSortedByScore() {
		return Objects.equals(sort, Sort.RELEVANCE);
	}

	public boolean needsScores() {
		return sort != null && sort.needsScores();
	}

	public int getTotalHitsThresholdInt() {
		return LuceneUtils.totalHitsThreshold(this.computePreciseHitsCount);
	}

	public long getTotalHitsThresholdLong() {
		return LuceneUtils.totalHitsThresholdLong(this.computePreciseHitsCount);
	}
}