HSEARCH-5133 Introduce metric aggregation values

To test metric aggregations on different types
hibernate · Aug 1, 2024 · cc7c324 · cc7c324
1 parent 197e230
commit cc7c324
Show file tree

Hide file tree

Showing 26 changed files with 281 additions and 2 deletions.
diff --git a/...arch/integrationtest/backend/tck/testsupport/types/AnalyzedStringFieldTypeDescriptor.java b/...arch/integrationtest/backend/tck/testsupport/types/AnalyzedStringFieldTypeDescriptor.java
@@ -14,6 +14,7 @@
 import org.hibernate.search.integrationtest.backend.tck.testsupport.types.expectations.IndexNullAsMatchPredicateExpectactions;
 import org.hibernate.search.integrationtest.backend.tck.testsupport.types.values.AscendingUniqueTermValues;
 import org.hibernate.search.integrationtest.backend.tck.testsupport.types.values.IndexableValues;
+import org.hibernate.search.integrationtest.backend.tck.testsupport.types.values.MetricAggregationsValues;
 
 public class AnalyzedStringFieldTypeDescriptor extends StandardFieldTypeDescriptor<String> {
 
@@ -62,6 +63,11 @@ protected List<List<String>> createMultiResultingInSingleAfterMedian() {
 		};
 	}
 
+	@Override
+	public MetricAggregationsValues<String> metricAggregationsValues() {
+		return null;
+	}
+
 	@Override
 	protected IndexableValues<String> createIndexableValues() {
 		return new IndexableValues<String>() {

diff --git a/...e/search/integrationtest/backend/tck/testsupport/types/BigDecimalFieldTypeDescriptor.java b/...e/search/integrationtest/backend/tck/testsupport/types/BigDecimalFieldTypeDescriptor.java
@@ -19,6 +19,7 @@
 import org.hibernate.search.integrationtest.backend.tck.testsupport.types.expectations.IndexNullAsMatchPredicateExpectactions;
 import org.hibernate.search.integrationtest.backend.tck.testsupport.types.values.AscendingUniqueTermValues;
 import org.hibernate.search.integrationtest.backend.tck.testsupport.types.values.IndexableValues;
+import org.hibernate.search.integrationtest.backend.tck.testsupport.types.values.MetricAggregationsValues;
 import org.hibernate.search.integrationtest.backend.tck.testsupport.util.TckConfiguration;
 
 public class BigDecimalFieldTypeDescriptor extends StandardFieldTypeDescriptor<BigDecimal> {
@@ -66,6 +67,16 @@ protected BigDecimal applyDelta(BigDecimal value, int multiplierForDelta) {
 		};
 	}
 
+	@Override
+	public MetricAggregationsValues<BigDecimal> metricAggregationsValues() {
+		return new MetricAggregationsValues<BigDecimal>() {
+			@Override
+			protected BigDecimal valueOf(int value) {
+				return BigDecimal.valueOf( value ).setScale( 2, RoundingMode.FLOOR );
+			}
+		};
+	}
+
 	@Override
 	protected IndexableValues<BigDecimal> createIndexableValues() {
 		return new IndexableValues<BigDecimal>() {

diff --git a/...e/search/integrationtest/backend/tck/testsupport/types/BigIntegerFieldTypeDescriptor.java b/...e/search/integrationtest/backend/tck/testsupport/types/BigIntegerFieldTypeDescriptor.java
@@ -14,6 +14,7 @@
 import org.hibernate.search.integrationtest.backend.tck.testsupport.types.expectations.IndexNullAsMatchPredicateExpectactions;
 import org.hibernate.search.integrationtest.backend.tck.testsupport.types.values.AscendingUniqueTermValues;
 import org.hibernate.search.integrationtest.backend.tck.testsupport.types.values.IndexableValues;
+import org.hibernate.search.integrationtest.backend.tck.testsupport.types.values.MetricAggregationsValues;
 
 public class BigIntegerFieldTypeDescriptor extends StandardFieldTypeDescriptor<BigInteger> {
 
@@ -60,6 +61,16 @@ protected BigInteger applyDelta(BigInteger value, int multiplierForDelta) {
 		};
 	}
 
+	@Override
+	public MetricAggregationsValues<BigInteger> metricAggregationsValues() {
+		return new MetricAggregationsValues<BigInteger>() {
+			@Override
+			protected BigInteger valueOf(int value) {
+				return BigInteger.valueOf( value * 100L );
+			}
+		};
+	}
+
 	@Override
 	protected IndexableValues<BigInteger> createIndexableValues() {
 		return new IndexableValues<BigInteger>() {

diff --git a/...nate/search/integrationtest/backend/tck/testsupport/types/BooleanFieldTypeDescriptor.java b/...nate/search/integrationtest/backend/tck/testsupport/types/BooleanFieldTypeDescriptor.java
@@ -12,6 +12,7 @@
 import org.hibernate.search.integrationtest.backend.tck.testsupport.types.expectations.IndexNullAsMatchPredicateExpectactions;
 import org.hibernate.search.integrationtest.backend.tck.testsupport.types.values.AscendingUniqueTermValues;
 import org.hibernate.search.integrationtest.backend.tck.testsupport.types.values.IndexableValues;
+import org.hibernate.search.integrationtest.backend.tck.testsupport.types.values.MetricAggregationsValues;
 
 public class BooleanFieldTypeDescriptor extends StandardFieldTypeDescriptor<Boolean> {
 
@@ -46,6 +47,11 @@ protected List<List<Boolean>> createMultiResultingInSingleAfterMedian() {
 		};
 	}
 
+	@Override
+	public MetricAggregationsValues<Boolean> metricAggregationsValues() {
+		return null;
+	}
+
 	@Override
 	protected IndexableValues<Boolean> createIndexableValues() {
 		return new IndexableValues<Boolean>() {

diff --git a/...bernate/search/integrationtest/backend/tck/testsupport/types/ByteFieldTypeDescriptor.java b/...bernate/search/integrationtest/backend/tck/testsupport/types/ByteFieldTypeDescriptor.java
@@ -8,9 +8,11 @@
 import java.util.List;
 import java.util.Optional;
 
+import org.hibernate.search.engine.cfg.spi.ConvertUtils;
 import org.hibernate.search.integrationtest.backend.tck.testsupport.types.expectations.IndexNullAsMatchPredicateExpectactions;
 import org.hibernate.search.integrationtest.backend.tck.testsupport.types.values.AscendingUniqueTermValues;
 import org.hibernate.search.integrationtest.backend.tck.testsupport.types.values.IndexableValues;
+import org.hibernate.search.integrationtest.backend.tck.testsupport.types.values.MetricAggregationsValues;
 
 public class ByteFieldTypeDescriptor extends StandardFieldTypeDescriptor<Byte> {
 
@@ -56,6 +58,16 @@ private byte toByteExact(int value) {
 		};
 	}
 
+	@Override
+	public MetricAggregationsValues<Byte> metricAggregationsValues() {
+		return new MetricAggregationsValues<Byte>() {
+			@Override
+			protected Byte valueOf(int value) {
+				return ConvertUtils.convertByte( value );
+			}
+		};
+	}
+
 	@Override
 	protected IndexableValues<Byte> createIndexableValues() {
 		return new IndexableValues<Byte>() {

diff --git a/...rnate/search/integrationtest/backend/tck/testsupport/types/DoubleFieldTypeDescriptor.java b/...rnate/search/integrationtest/backend/tck/testsupport/types/DoubleFieldTypeDescriptor.java
@@ -8,9 +8,11 @@
 import java.util.List;
 import java.util.Optional;
 
+import org.hibernate.search.engine.cfg.spi.ConvertUtils;
 import org.hibernate.search.integrationtest.backend.tck.testsupport.types.expectations.IndexNullAsMatchPredicateExpectactions;
 import org.hibernate.search.integrationtest.backend.tck.testsupport.types.values.AscendingUniqueTermValues;
 import org.hibernate.search.integrationtest.backend.tck.testsupport.types.values.IndexableValues;
+import org.hibernate.search.integrationtest.backend.tck.testsupport.types.values.MetricAggregationsValues;
 
 public class DoubleFieldTypeDescriptor extends StandardFieldTypeDescriptor<Double> {
 
@@ -49,6 +51,16 @@ protected Double applyDelta(Double value, int multiplierForDelta) {
 		};
 	}
 
+	@Override
+	public MetricAggregationsValues<Double> metricAggregationsValues() {
+		return new MetricAggregationsValues<Double>() {
+			@Override
+			protected Double valueOf(int value) {
+				return ConvertUtils.convertDouble( value );
+			}
+		};
+	}
+
 	@Override
 	protected IndexableValues<Double> createIndexableValues() {
 		return new IndexableValues<Double>() {

diff --git a/...g/hibernate/search/integrationtest/backend/tck/testsupport/types/FieldTypeDescriptor.java b/...g/hibernate/search/integrationtest/backend/tck/testsupport/types/FieldTypeDescriptor.java
@@ -15,6 +15,7 @@
 import org.hibernate.search.integrationtest.backend.tck.testsupport.types.expectations.IndexNullAsMatchPredicateExpectactions;
 import org.hibernate.search.integrationtest.backend.tck.testsupport.types.values.AscendingUniqueTermValues;
 import org.hibernate.search.integrationtest.backend.tck.testsupport.types.values.IndexableValues;
+import org.hibernate.search.integrationtest.backend.tck.testsupport.types.values.MetricAggregationsValues;
 import org.hibernate.search.integrationtest.backend.tck.testsupport.util.TckConfiguration;
 
 public abstract class FieldTypeDescriptor<F, S extends SearchableProjectableIndexFieldTypeOptionsStep<?, F>> {
@@ -181,6 +182,8 @@ public final AscendingUniqueTermValues<F> getAscendingUniqueTermValues() {
 
 	protected abstract AscendingUniqueTermValues<F> createAscendingUniqueTermValues();
 
+	public abstract MetricAggregationsValues<F> metricAggregationsValues();
+
 	/**
 	 * @return A set of indexables values, not necessarily unique.
 	 */

diff --git a/...ernate/search/integrationtest/backend/tck/testsupport/types/FloatFieldTypeDescriptor.java b/...ernate/search/integrationtest/backend/tck/testsupport/types/FloatFieldTypeDescriptor.java
@@ -8,9 +8,11 @@
 import java.util.List;
 import java.util.Optional;
 
+import org.hibernate.search.engine.cfg.spi.ConvertUtils;
 import org.hibernate.search.integrationtest.backend.tck.testsupport.types.expectations.IndexNullAsMatchPredicateExpectactions;
 import org.hibernate.search.integrationtest.backend.tck.testsupport.types.values.AscendingUniqueTermValues;
 import org.hibernate.search.integrationtest.backend.tck.testsupport.types.values.IndexableValues;
+import org.hibernate.search.integrationtest.backend.tck.testsupport.types.values.MetricAggregationsValues;
 
 public class FloatFieldTypeDescriptor extends StandardFieldTypeDescriptor<Float> {
 
@@ -49,6 +51,16 @@ protected Float applyDelta(Float value, int multiplierForDelta) {
 		};
 	}
 
+	@Override
+	public MetricAggregationsValues<Float> metricAggregationsValues() {
+		return new MetricAggregationsValues<Float>() {
+			@Override
+			protected Float valueOf(int value) {
+				return ConvertUtils.convertFloat( value );
+			}
+		};
+	}
+
 	@Override
 	protected IndexableValues<Float> createIndexableValues() {
 		return new IndexableValues<Float>() {

diff --git a/...ate/search/integrationtest/backend/tck/testsupport/types/GeoPointFieldTypeDescriptor.java b/...ate/search/integrationtest/backend/tck/testsupport/types/GeoPointFieldTypeDescriptor.java
@@ -12,6 +12,7 @@
 import org.hibernate.search.integrationtest.backend.tck.testsupport.types.expectations.IndexNullAsMatchPredicateExpectactions;
 import org.hibernate.search.integrationtest.backend.tck.testsupport.types.values.AscendingUniqueTermValues;
 import org.hibernate.search.integrationtest.backend.tck.testsupport.types.values.IndexableValues;
+import org.hibernate.search.integrationtest.backend.tck.testsupport.types.values.MetricAggregationsValues;
 
 public class GeoPointFieldTypeDescriptor extends StandardFieldTypeDescriptor<GeoPoint> {
 
@@ -26,6 +27,11 @@ protected AscendingUniqueTermValues<GeoPoint> createAscendingUniqueTermValues()
 		return null; // Value lookup is not supported
 	}
 
+	@Override
+	public MetricAggregationsValues<GeoPoint> metricAggregationsValues() {
+		return null;
+	}
+
 	@Override
 	protected IndexableValues<GeoPoint> createIndexableValues() {
 		return new IndexableValues<GeoPoint>() {

diff --git a/...nate/search/integrationtest/backend/tck/testsupport/types/InstantFieldTypeDescriptor.java b/...nate/search/integrationtest/backend/tck/testsupport/types/InstantFieldTypeDescriptor.java
@@ -18,6 +18,7 @@
 import org.hibernate.search.integrationtest.backend.tck.testsupport.types.expectations.IndexNullAsMatchPredicateExpectactions;
 import org.hibernate.search.integrationtest.backend.tck.testsupport.types.values.AscendingUniqueTermValues;
 import org.hibernate.search.integrationtest.backend.tck.testsupport.types.values.IndexableValues;
+import org.hibernate.search.integrationtest.backend.tck.testsupport.types.values.MetricAggregationsValues;
 
 public class InstantFieldTypeDescriptor extends StandardFieldTypeDescriptor<Instant> {
 
@@ -57,6 +58,16 @@ protected Instant applyDelta(Instant value, int multiplierForDelta) {
 		};
 	}
 
+	@Override
+	public MetricAggregationsValues<Instant> metricAggregationsValues() {
+		return new MetricAggregationsValues<Instant>() {
+			@Override
+			protected Instant valueOf(int value) {
+				return Instant.ofEpochSecond( value * 100_000L );
+			}
+		};
+	}
+
 	@Override
 	protected IndexableValues<Instant> createIndexableValues() {
 		return new IndexableValues<Instant>() {

diff --git a/...nate/search/integrationtest/backend/tck/testsupport/types/IntegerFieldTypeDescriptor.java b/...nate/search/integrationtest/backend/tck/testsupport/types/IntegerFieldTypeDescriptor.java
@@ -11,6 +11,7 @@
 import org.hibernate.search.integrationtest.backend.tck.testsupport.types.expectations.IndexNullAsMatchPredicateExpectactions;
 import org.hibernate.search.integrationtest.backend.tck.testsupport.types.values.AscendingUniqueTermValues;
 import org.hibernate.search.integrationtest.backend.tck.testsupport.types.values.IndexableValues;
+import org.hibernate.search.integrationtest.backend.tck.testsupport.types.values.MetricAggregationsValues;
 
 public class IntegerFieldTypeDescriptor extends StandardFieldTypeDescriptor<Integer> {
 
@@ -49,6 +50,16 @@ protected Integer applyDelta(Integer value, int multiplierForDelta) {
 		};
 	}
 
+	@Override
+	public MetricAggregationsValues<Integer> metricAggregationsValues() {
+		return new MetricAggregationsValues<Integer>() {
+			@Override
+			protected Integer valueOf(int value) {
+				return value;
+			}
+		};
+	}
+
 	@Override
 	protected IndexableValues<Integer> createIndexableValues() {
 		return new IndexableValues<Integer>() {

diff --git a/...earch/integrationtest/backend/tck/testsupport/types/KeywordStringFieldTypeDescriptor.java b/...earch/integrationtest/backend/tck/testsupport/types/KeywordStringFieldTypeDescriptor.java
@@ -11,6 +11,7 @@
 import org.hibernate.search.integrationtest.backend.tck.testsupport.types.expectations.IndexNullAsMatchPredicateExpectactions;
 import org.hibernate.search.integrationtest.backend.tck.testsupport.types.values.AscendingUniqueTermValues;
 import org.hibernate.search.integrationtest.backend.tck.testsupport.types.values.IndexableValues;
+import org.hibernate.search.integrationtest.backend.tck.testsupport.types.values.MetricAggregationsValues;
 
 public class KeywordStringFieldTypeDescriptor extends StandardFieldTypeDescriptor<String> {
 
@@ -54,6 +55,11 @@ protected List<List<String>> createMultiResultingInSingleAfterMedian() {
 		};
 	}
 
+	@Override
+	public MetricAggregationsValues<String> metricAggregationsValues() {
+		return null;
+	}
+
 	@Override
 	protected IndexableValues<String> createIndexableValues() {
 		return new IndexableValues<String>() {

diff --git a/...te/search/integrationtest/backend/tck/testsupport/types/LocalDateFieldTypeDescriptor.java b/...te/search/integrationtest/backend/tck/testsupport/types/LocalDateFieldTypeDescriptor.java
@@ -13,6 +13,7 @@
 import org.hibernate.search.integrationtest.backend.tck.testsupport.types.expectations.IndexNullAsMatchPredicateExpectactions;
 import org.hibernate.search.integrationtest.backend.tck.testsupport.types.values.AscendingUniqueTermValues;
 import org.hibernate.search.integrationtest.backend.tck.testsupport.types.values.IndexableValues;
+import org.hibernate.search.integrationtest.backend.tck.testsupport.types.values.MetricAggregationsValues;
 
 public class LocalDateFieldTypeDescriptor extends StandardFieldTypeDescriptor<LocalDate> {
 
@@ -51,6 +52,16 @@ protected LocalDate applyDelta(LocalDate value, int multiplierForDelta) {
 		};
 	}
 
+	@Override
+	public MetricAggregationsValues<LocalDate> metricAggregationsValues() {
+		return new MetricAggregationsValues<LocalDate>() {
+			@Override
+			protected LocalDate valueOf(int value) {
+				return LocalDate.ofEpochDay( value );
+			}
+		};
+	}
+
 	@Override
 	protected IndexableValues<LocalDate> createIndexableValues() {
 		return new IndexableValues<LocalDate>() {

diff --git a/...earch/integrationtest/backend/tck/testsupport/types/LocalDateTimeFieldTypeDescriptor.java b/...earch/integrationtest/backend/tck/testsupport/types/LocalDateTimeFieldTypeDescriptor.java
@@ -5,6 +5,7 @@
 package org.hibernate.search.integrationtest.backend.tck.testsupport.types;
 
 import java.time.LocalDateTime;
+import java.time.ZoneOffset;
 import java.time.temporal.ChronoUnit;
 import java.util.Arrays;
 import java.util.List;
@@ -14,6 +15,7 @@
 import org.hibernate.search.integrationtest.backend.tck.testsupport.types.expectations.IndexNullAsMatchPredicateExpectactions;
 import org.hibernate.search.integrationtest.backend.tck.testsupport.types.values.AscendingUniqueTermValues;
 import org.hibernate.search.integrationtest.backend.tck.testsupport.types.values.IndexableValues;
+import org.hibernate.search.integrationtest.backend.tck.testsupport.types.values.MetricAggregationsValues;
 
 public class LocalDateTimeFieldTypeDescriptor extends StandardFieldTypeDescriptor<LocalDateTime> {
 
@@ -53,6 +55,16 @@ protected LocalDateTime applyDelta(LocalDateTime value, int multiplierForDelta)
 		};
 	}
 
+	@Override
+	public MetricAggregationsValues<LocalDateTime> metricAggregationsValues() {
+		return new MetricAggregationsValues<LocalDateTime>() {
+			@Override
+			protected LocalDateTime valueOf(int value) {
+				return LocalDateTime.ofEpochSecond( value, 0, ZoneOffset.UTC );
+			}
+		};
+	}
+
 	@Override
 	protected IndexableValues<LocalDateTime> createIndexableValues() {
 		return new IndexableValues<LocalDateTime>() {

diff --git a/...te/search/integrationtest/backend/tck/testsupport/types/LocalTimeFieldTypeDescriptor.java b/...te/search/integrationtest/backend/tck/testsupport/types/LocalTimeFieldTypeDescriptor.java
@@ -14,6 +14,7 @@
 import org.hibernate.search.integrationtest.backend.tck.testsupport.types.expectations.IndexNullAsMatchPredicateExpectactions;
 import org.hibernate.search.integrationtest.backend.tck.testsupport.types.values.AscendingUniqueTermValues;
 import org.hibernate.search.integrationtest.backend.tck.testsupport.types.values.IndexableValues;
+import org.hibernate.search.integrationtest.backend.tck.testsupport.types.values.MetricAggregationsValues;
 
 public class LocalTimeFieldTypeDescriptor extends StandardFieldTypeDescriptor<LocalTime> {
 
@@ -53,6 +54,11 @@ protected LocalTime applyDelta(LocalTime value, int multiplierForDelta) {
 		};
 	}
 
+	@Override
+	public MetricAggregationsValues<LocalTime> metricAggregationsValues() {
+		return null;
+	}
+
 	@Override
 	protected IndexableValues<LocalTime> createIndexableValues() {
 		return new IndexableValues<LocalTime>() {

diff --git a/...bernate/search/integrationtest/backend/tck/testsupport/types/LongFieldTypeDescriptor.java b/...bernate/search/integrationtest/backend/tck/testsupport/types/LongFieldTypeDescriptor.java
@@ -10,9 +10,11 @@
 import java.util.List;
 import java.util.Optional;
 
+import org.hibernate.search.engine.cfg.spi.ConvertUtils;
 import org.hibernate.search.integrationtest.backend.tck.testsupport.types.expectations.IndexNullAsMatchPredicateExpectactions;
 import org.hibernate.search.integrationtest.backend.tck.testsupport.types.values.AscendingUniqueTermValues;
 import org.hibernate.search.integrationtest.backend.tck.testsupport.types.values.IndexableValues;
+import org.hibernate.search.integrationtest.backend.tck.testsupport.types.values.MetricAggregationsValues;
 import org.hibernate.search.integrationtest.backend.tck.testsupport.util.TckConfiguration;
 
 public class LongFieldTypeDescriptor extends StandardFieldTypeDescriptor<Long> {
@@ -61,6 +63,16 @@ protected Long applyDelta(Long value, int multiplierForDelta) {
 		};
 	}
 
+	@Override
+	public MetricAggregationsValues<Long> metricAggregationsValues() {
+		return new MetricAggregationsValues<Long>() {
+			@Override
+			protected Long valueOf(int value) {
+				return ConvertUtils.convertLong( value );
+			}
+		};
+	}
+
 	@Override
 	protected IndexableValues<Long> createIndexableValues() {
 		return new IndexableValues<Long>() {