HSEARCH-3319 Address super-type references

Author: marko-bekhta

documentation/src/test/java/org/hibernate/search/documentation/search/predicate/FieldReferenceIT.java

@@ -67,6 +67,13 @@ void smoke() {
 							.fetchHits( 20 )
 			).containsOnly( "c" );
 
+			assertThat(
+					searchSession.search( EntityC_.scope )
+							.select( f -> f.field( EntityC_.stringA ) )
+							.where( f -> f.match().field( EntityC_.stringA ).matching( "c" ) )
+							.fetchHits( 20 )
+			).containsOnly( "c" );
+
 			SearchScope<EntityB_union_EntityC_, EntityB> scope = EntityB_union_EntityC_.scope.create( searchSession );
 
 			SearchPredicate searchPredicate =
@@ -200,30 +207,24 @@ public static class EntityA_ {
 		}
 	}
 
-	public static class EntityB_ {
-		public static ValueFieldReference1<EntityB_, String, String, String> stringA;
+	public static class EntityB_ extends EntityA_ {
 		public static ValueFieldReference1<EntityB_, String, String, String> stringB;
 
 		public static RootReferenceScope<EntityB_, EntityB> scope;
 
 		static {
-			stringA = ValueFieldReference1.of( "stringA", EntityB_.class, String.class, String.class, String.class );
 			stringB = ValueFieldReference1.of( "stringB", EntityB_.class, String.class, String.class, String.class );
 
 			scope = RootReferenceScopeImpl.of( EntityB_.class, EntityB.class );
 		}
 	}
 
-	public static class EntityC_ {
-		public static ValueFieldReference1<EntityC_, String, String, String> stringA;
-		public static ValueFieldReference1<EntityC_, String, String, String> stringB;
+	public static class EntityC_ extends EntityB_ {
 		public static ValueFieldReference1<EntityC_, String, String, String> stringC;
 
 		public static RootReferenceScope<EntityC_, EntityC> scope;
 
 		static {
-			stringA = ValueFieldReference1.of( "stringA", EntityC_.class, String.class, String.class, String.class );
-			stringB = ValueFieldReference1.of( "stringB", EntityC_.class, String.class, String.class, String.class );
 			stringC = ValueFieldReference1.of( "stringC", EntityC_.class, String.class, String.class, String.class );
 
 			scope = RootReferenceScopeImpl.of( EntityC_.class, EntityC.class );

engine/src/main/java/org/hibernate/search/engine/search/aggregation/dsl/RangeAggregationFieldStep.java

@@ -10,6 +10,7 @@
 
 import org.hibernate.search.engine.search.common.ValueConvert;
 import org.hibernate.search.engine.search.predicate.dsl.SearchPredicateFactory;
+import org.hibernate.search.engine.search.reference.aggregation.RangeAggregationFieldReference;
 
 /**
  * The initial step in a "range" aggregation definition, where the target field can be set.
@@ -43,4 +44,15 @@ public interface RangeAggregationFieldStep<SR, PDF extends SearchPredicateFactor
 	 */
 	<F> RangeAggregationRangeStep<SR, ?, PDF, F> field(String fieldPath, Class<F> type, ValueConvert convert);
 
+	/**
+	 * Target the given field in the range aggregation.
+	 *
+	 * @param fieldReference The field reference representing a <a href="SearchAggregationFactory.html#field-paths">path</a> to the index field to aggregate.
+	 * @param <F> The type of field values.
+	 * @return The next step.
+	 */
+	default <F> RangeAggregationRangeStep<SR, ?, PDF, F> field(RangeAggregationFieldReference<? super SR, F> fieldReference) {
+		return field( fieldReference.absolutePath(), fieldReference.aggregationType(), fieldReference.valueConvert() );
+	}
+
 }

engine/src/main/java/org/hibernate/search/engine/search/aggregation/dsl/SearchAggregationFactory.java

@@ -9,6 +9,7 @@
 import java.util.function.Function;
 
 import org.hibernate.search.engine.search.common.NamedValues;
+import org.hibernate.search.engine.search.reference.object.ObjectFieldReference;
 import org.hibernate.search.util.common.SearchException;
 import org.hibernate.search.util.common.annotation.Incubating;
 
@@ -101,6 +102,20 @@ <T> AggregationFinalStep<T> withParameters(
 	@Incubating
 	SearchAggregationFactory<SR> withRoot(String objectFieldPath);
 
+	/**
+	 * Create a new aggregation factory whose root for all paths passed to the DSL
+	 * will be the given object field.
+	 * <p>
+	 * See <a href="#field-paths">here</a> for more information.
+	 *
+	 * @param objectFieldReference The reference representing the path from the current root to an object field that will become the new root.
+	 * @return A new aggregation factory using the given object field as root.
+	 */
+	@Incubating
+	default SearchAggregationFactory<SR> withRoot(ObjectFieldReference<? super SR> objectFieldReference) {
+		return withRoot( objectFieldReference.absolutePath() );
+	}
+
 	/**
 	 * @param relativeFieldPath The path to a field, relative to the {@link #withRoot(String) root} of this factory.
 	 * @return The absolute path of the field, for use in native aggregations for example.

engine/src/main/java/org/hibernate/search/engine/search/aggregation/dsl/TermsAggregationFieldStep.java

@@ -11,6 +11,7 @@
 
 import org.hibernate.search.engine.search.common.ValueConvert;
 import org.hibernate.search.engine.search.predicate.dsl.SearchPredicateFactory;
+import org.hibernate.search.engine.search.reference.aggregation.TermsAggregationFieldReference;
 
 /**
  * The initial step in a "terms" aggregation definition, where the target field can be set.
@@ -45,4 +46,16 @@ public interface TermsAggregationFieldStep<SR, PDF extends SearchPredicateFactor
 	<F> TermsAggregationOptionsStep<SR, ?, PDF, F, Map<F, Long>> field(String fieldPath, Class<F> type,
 			ValueConvert convert);
 
+	/**
+	 * Target the given field in the terms aggregation.
+	 *
+	 * @param fieldReference The field reference representing a <a href="SearchAggregationFactory.html#field-paths">path</a> to the index field to aggregate.
+	 * @param <F> The type of field values.
+	 * @return The next step.
+	 */
+	default <F> TermsAggregationOptionsStep<SR, ?, PDF, F, Map<F, Long>> field(
+			TermsAggregationFieldReference<? super SR, F> fieldReference) {
+		return field( fieldReference.absolutePath(), fieldReference.aggregationType(), fieldReference.valueConvert() );
+	}
+
 }

engine/src/main/java/org/hibernate/search/engine/search/predicate/dsl/ExistsPredicateFieldStep.java

@@ -33,7 +33,7 @@ public interface ExistsPredicateFieldStep<SR, N extends ExistsPredicateOptionsSt
 	 * @return The next step.
 	 */
 	@Incubating
-	default N field(ExistsPredicateFieldReference<SR> field) {
+	default N field(ExistsPredicateFieldReference<? super SR> field) {
 		return field( field.absolutePath() );
 	}
 

engine/src/main/java/org/hibernate/search/engine/search/predicate/dsl/KnnPredicateFieldStep.java

@@ -22,5 +22,5 @@ public interface KnnPredicateFieldStep<SR> {
 	 */
 	KnnPredicateVectorStep<SR> field(String fieldPath);
 
-	<T> KnnPredicateVectorGenericStep<SR, T> field(KnnPredicateFieldReference<SR, T> field);
+	<T> KnnPredicateVectorGenericStep<SR, T> field(KnnPredicateFieldReference<? super SR, T> field);
 }

engine/src/main/java/org/hibernate/search/engine/search/predicate/dsl/MatchPredicateFieldStep.java

@@ -61,8 +61,8 @@ default N field(String fieldPath) {
 	 * @return The next step.
 	 */
 	@SuppressWarnings("unchecked")
-	default <T> MatchPredicateFieldMoreGenericStep<?, ?, T, MatchPredicateFieldReference<SR, T>> field(
-			MatchPredicateFieldReference<SR, T> fieldReference) {
+	default <T> MatchPredicateFieldMoreGenericStep<?, ?, T, MatchPredicateFieldReference<? super SR, T>> field(
+			MatchPredicateFieldReference<? super SR, T> fieldReference) {
 		return fields( fieldReference );
 	}
 
@@ -82,6 +82,6 @@ default N field(String fieldPath) {
 	 * @see #field(MatchPredicateFieldReference)
 	 */
 	@SuppressWarnings("unchecked")
-	<T> MatchPredicateFieldMoreGenericStep<?, ?, T, MatchPredicateFieldReference<SR, T>> fields(
-			MatchPredicateFieldReference<SR, T>... fieldReferences);
+	<T> MatchPredicateFieldMoreGenericStep<?, ?, T, MatchPredicateFieldReference<? super SR, T>> fields(
+			MatchPredicateFieldReference<? super SR, T>... fieldReferences);
 }

engine/src/main/java/org/hibernate/search/engine/search/predicate/dsl/NestedPredicateFieldStep.java

@@ -38,7 +38,7 @@ public interface NestedPredicateFieldStep<SR, N extends NestedPredicateNestStep<
 	 * to apply the predicate on.
 	 * @return The next step.
 	 */
-	default N objectField(NestedPredicateFieldReference<SR> field) {
+	default N objectField(NestedPredicateFieldReference<? super SR> field) {
 		return objectField( field.absolutePath() );
 	}
 

engine/src/main/java/org/hibernate/search/engine/search/predicate/dsl/PhrasePredicateFieldMoreStep.java

@@ -75,7 +75,7 @@ default S field(String fieldPath) {
 	 */
 	@Incubating
 	@SuppressWarnings("unchecked")
-	default S field(PhrasePredicateFieldReference<SR, ?> field) {
+	default S field(PhrasePredicateFieldReference<? super SR, ?> field) {
 		return fields( field );
 	}
 
@@ -95,7 +95,7 @@ default S field(PhrasePredicateFieldReference<SR, ?> field) {
 	 */
 	@Incubating
 	@SuppressWarnings("unchecked")
-	default S fields(PhrasePredicateFieldReference<SR, ?>... fields) {
+	default S fields(PhrasePredicateFieldReference<? super SR, ?>... fields) {
 		String[] paths = new String[fields.length];
 		for ( int i = 0; i < fields.length; i++ ) {
 			paths[i] = fields[i].absolutePath();

engine/src/main/java/org/hibernate/search/engine/search/predicate/dsl/PhrasePredicateFieldStep.java

@@ -74,7 +74,7 @@ default N field(String fieldPath) {
 	 */
 	@Incubating
 	@SuppressWarnings("unchecked")
-	default N field(PhrasePredicateFieldReference<SR, ?> field) {
+	default N field(PhrasePredicateFieldReference<? super SR, ?> field) {
 		return fields( field );
 	}
 
@@ -97,7 +97,7 @@ default N field(PhrasePredicateFieldReference<SR, ?> field) {
 	 */
 	@Incubating
 	@SuppressWarnings("unchecked")
-	default N fields(PhrasePredicateFieldReference<SR, ?>... fields) {
+	default N fields(PhrasePredicateFieldReference<? super SR, ?>... fields) {
 		String[] paths = new String[fields.length];
 		for ( int i = 0; i < fields.length; i++ ) {
 			paths[i] = fields[i].absolutePath();