HHH-19126 Correct plural attribute path to be collection-typed

Author: mbellade

hibernate-core/src/main/java/org/hibernate/metamodel/model/domain/internal/AbstractPluralAttribute.java

@@ -143,6 +143,7 @@ public Class<E> getBindableJavaType() {
 		return getElementType().getJavaType();
 	}
 
+	@SuppressWarnings("unchecked")
 	@Override
 	public SqmPath<E> createSqmPath(SqmPath<?> lhs, SqmPathSource<?> intermediatePathSource) {
 		final NavigablePath navigablePath;
@@ -152,7 +153,9 @@ public SqmPath<E> createSqmPath(SqmPath<?> lhs, SqmPathSource<?> intermediatePat
 		else {
 			navigablePath = lhs.getNavigablePath().append( intermediatePathSource.getPathName() ).append( getPathName() );
 		}
-		return new SqmPluralValuedSimplePath<>(
+		// We need an unchecked cast here : PluralPersistentAttribute implements path source with its element type
+		//  but resolving paths from it must produce collection-typed expressions.
+		return (SqmPath<E>) new SqmPluralValuedSimplePath<>(
 				navigablePath,
 				this,
 				lhs,

hibernate-core/src/main/java/org/hibernate/query/sqm/tree/domain/SqmPluralValuedSimplePath.java

@@ -18,50 +18,55 @@
 import org.hibernate.query.hql.spi.SqmPathRegistry;
 import org.hibernate.query.sqm.NodeBuilder;
 import org.hibernate.query.sqm.SemanticQueryWalker;
+import org.hibernate.query.sqm.SqmPathSource;
 import org.hibernate.query.sqm.tree.SqmCopyContext;
 import org.hibernate.query.sqm.tree.SqmJoinType;
 import org.hibernate.query.sqm.tree.expression.SqmExpression;
 import org.hibernate.query.sqm.tree.from.SqmFrom;
+import org.hibernate.type.descriptor.java.JavaType;
 import org.hibernate.query.sqm.tree.from.SqmQualifiedJoin;
 
 /**
  * An SqmPath for plural attribute paths
  *
- * @param <E> The collection element type, which is the "bindable" type in the SQM tree
+ * @param <C> The collection type
  *
  * @author Steve Ebersole
  */
-public class SqmPluralValuedSimplePath<E> extends AbstractSqmSimplePath<E> {
+public class SqmPluralValuedSimplePath<C> extends AbstractSqmSimplePath<C> {
 	public SqmPluralValuedSimplePath(
 			NavigablePath navigablePath,
-			PluralPersistentAttribute<?, ?, E> referencedNavigable,
+			PluralPersistentAttribute<?, C, ?> referencedNavigable,
 			SqmPath<?> lhs,
 			NodeBuilder nodeBuilder) {
 		this( navigablePath, referencedNavigable, lhs, null, nodeBuilder );
 	}
 
 	public SqmPluralValuedSimplePath(
 			NavigablePath navigablePath,
-			PluralPersistentAttribute<?, ?, E> referencedNavigable,
+			PluralPersistentAttribute<?, C, ?> referencedNavigable,
 			SqmPath<?> lhs,
 			String explicitAlias,
 			NodeBuilder nodeBuilder) {
-		super( navigablePath, referencedNavigable, lhs, explicitAlias, nodeBuilder );
+		// We need to do an unchecked cast here: PluralPersistentAttribute implements path source with
+		//  the element type, but paths generated from it must be collection-typed.
+		//noinspection unchecked
+		super( navigablePath, (SqmPathSource<C>) referencedNavigable, lhs, explicitAlias, nodeBuilder );
 	}
 
 	@Override
-	public SqmPluralValuedSimplePath<E> copy(SqmCopyContext context) {
-		final SqmPluralValuedSimplePath<E> existing = context.getCopy( this );
+	public SqmPluralValuedSimplePath<C> copy(SqmCopyContext context) {
+		final SqmPluralValuedSimplePath<C> existing = context.getCopy( this );
 		if ( existing != null ) {
 			return existing;
 		}
 
 		final SqmPath<?> lhsCopy = getLhs().copy( context );
-		final SqmPluralValuedSimplePath<E> path = context.registerCopy(
+		final SqmPluralValuedSimplePath<C> path = context.registerCopy(
 				this,
 				new SqmPluralValuedSimplePath<>(
 						getNavigablePathCopy( lhsCopy ),
-						getModel(),
+						(PluralPersistentAttribute<?,C,?>) getModel(),
 						lhsCopy,
 						getExplicitAlias(),
 						nodeBuilder()
@@ -71,19 +76,13 @@ public SqmPluralValuedSimplePath<E> copy(SqmCopyContext context) {
 		return path;
 	}
 
-	@Override
-	public PluralPersistentAttribute<?, ?, E> getReferencedPathSource() {
-		return (PluralPersistentAttribute<?, ?, E>) super.getReferencedPathSource();
-	}
-
-	@Override
-	public PluralPersistentAttribute<?, ?, E> getModel() {
-		return (PluralPersistentAttribute<?, ?, E>) super.getModel();
+	public PluralPersistentAttribute<?, C, ?> getPluralAttribute() {
+		return (PluralPersistentAttribute<?, C, ?>) getModel();
 	}
 
 	@Override
-	public PluralPersistentAttribute<?,?,E> getNodeType() {
-		return getReferencedPathSource();
+	public JavaType<C> getJavaTypeDescriptor() {
+		return getPluralAttribute().getAttributeJavaType();
 	}
 
 	@Override
@@ -123,12 +122,11 @@ public SqmPath<?> resolveIndexedAccess(
 		}
 		SqmFrom<?, ?> path = pathRegistry.findFromByPath( navigablePath.getParent() );
 		if ( path == null ) {
-			final PluralPersistentAttribute<?, ?, E> referencedPathSource = getReferencedPathSource();
+			final SqmPathSource<C> referencedPathSource = getReferencedPathSource();
 			final SqmFrom<?, Object> parent = pathRegistry.resolveFrom( getLhs() );
 			final SqmQualifiedJoin<Object, ?> join;
 			final SqmExpression<?> index;
 			if ( referencedPathSource instanceof ListPersistentAttribute<?, ?> ) {
-				//noinspection unchecked
 				join = new SqmListJoin<>(
 						parent,
 						(ListPersistentAttribute<Object, ?>) referencedPathSource,
@@ -140,7 +138,6 @@ public SqmPath<?> resolveIndexedAccess(
 				index = ( (SqmListJoin<?, ?>) join ).index();
 			}
 			else if ( referencedPathSource instanceof MapPersistentAttribute<?, ?, ?> ) {
-				//noinspection unchecked
 				join = new SqmMapJoin<>(
 						parent,
 						(MapPersistentAttribute<Object, ?, ?>) referencedPathSource,
@@ -168,33 +165,12 @@ else if ( referencedPathSource instanceof MapPersistentAttribute<?, ?, ?> ) {
 	}
 
 	@Override
-	public <S extends E> SqmTreatedSimplePath<E,S> treatAs(Class<S> treatJavaType) throws PathException {
-		return (SqmTreatedSimplePath<E, S>) treatAs( nodeBuilder().getDomainModel().entity( treatJavaType ) );
+	public <S extends C> SqmTreatedSimplePath<C,S> treatAs(Class<S> treatJavaType) throws PathException {
+		return (SqmTreatedSimplePath<C, S>) treatAs( nodeBuilder().getDomainModel().entity( treatJavaType ) );
 	}
 
 	@Override
-	public <S extends E> SqmTreatedPath<E, S> treatAs(EntityDomainType<S> treatTarget) throws PathException {
+	public <S extends C> SqmTreatedPath<C, S> treatAs(EntityDomainType<S> treatTarget) throws PathException {
 		return getTreatedPath( treatTarget );
 	}
-
-//	@Override
-//	public DomainResult createDomainResult(
-//			String resultVariable,
-//			DomainResultCreationState creationState,
-//			DomainResultCreationContext creationContext) {
-//		return new CollectionResultImpl(
-//				getReferencedNavigable().getPluralAttribute().getDescribedAttribute(),
-//				getNavigablePath(),
-//				resultVariable,
-//				LockMode.NONE,
-//				getReferencedNavigable().getPluralAttribute().getCollectionKeyDescriptor().createDomainResult(
-//						getNavigablePath().append( "{id}" ),
-//						null,
-//						creationState,
-//						creationContext
-//				),
-//				initializerProducerCreator.createProducer( resultVariable, creationState, creationContext )
-//		);
-//	}
-
 }

hibernate-core/src/main/java/org/hibernate/query/sqm/tree/predicate/SqmMemberOfPredicate.java

@@ -35,7 +35,7 @@ public SqmMemberOfPredicate(
 		this.leftHandExpression = leftHandExpression;
 
 		leftHandExpression.applyInferableType(
-				( (SqmPluralValuedSimplePath<?>) pluralPath ).getReferencedPathSource().getElementType()
+				( (SqmPluralValuedSimplePath<?>) pluralPath ).getPluralAttribute().getElementType()
 		);
 	}