HHH-16997 Embedded components in HibernateProxy are not initialized when entity has reference to another entity of the same type + HHH-16901 Embedded field in entity association from composite key not correctly instantiated

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

@@ -170,7 +170,7 @@ public NavigablePath createNavigablePath(SqmPath<?> parent, String alias) {
 					"`lhs` cannot be null for a sub-navigable reference - " + getName()
 			);
 		}
-		final SqmPathSource<?> parentPathSource = parent.getReferencedPathSource();
+		final SqmPathSource<?> parentPathSource = parent.getResolvedModel();
 		NavigablePath navigablePath = parent.getNavigablePath();
 		if ( parentPathSource instanceof PluralPersistentAttribute<?, ?, ?> ) {
 			navigablePath = navigablePath.append( CollectionPart.Nature.ELEMENT.getName() );

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

@@ -7,6 +7,7 @@
 package org.hibernate.metamodel.model.domain.internal;
 
 import org.hibernate.metamodel.model.domain.EmbeddableDomainType;
+import org.hibernate.query.sqm.SqmJoinable;
 import org.hibernate.query.sqm.SqmPathSource;
 import org.hibernate.query.sqm.tree.domain.SqmEmbeddedValuedSimplePath;
 import org.hibernate.query.sqm.tree.domain.SqmPath;
@@ -47,7 +48,9 @@ public boolean isGeneric() {
 	@Override
 	public SqmPath<J> createSqmPath(SqmPath<?> lhs, SqmPathSource<?> intermediatePathSource) {
 		return new SqmEmbeddedValuedSimplePath<>(
-				PathHelper.append( lhs, this, intermediatePathSource ),
+				pathModel instanceof SqmJoinable<?, ?>
+						? ( (SqmJoinable<?, ?>) pathModel ).createNavigablePath( lhs, null )
+						: PathHelper.append( lhs, this, intermediatePathSource ),
 				pathModel,
 				lhs,
 				lhs.nodeBuilder()

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

@@ -173,7 +173,7 @@ public NavigablePath createNavigablePath(SqmPath<?> parent, String alias) {
 					"LHS cannot be null for a sub-navigable reference - " + getName()
 			);
 		}
-		final SqmPathSource<?> parentPathSource = parent.getReferencedPathSource();
+		final SqmPathSource<?> parentPathSource = parent.getResolvedModel();
 		NavigablePath navigablePath = parent.getNavigablePath();
 		if ( parentPathSource instanceof PluralPersistentAttribute<?, ?, ?> ) {
 			navigablePath = navigablePath.append( CollectionPart.Nature.ELEMENT.getName() );
@@ -224,7 +224,7 @@ public NavigablePath createNavigablePath(SqmPath<?> parent, String alias) {
 				);
 			}
 			NavigablePath navigablePath = parent.getNavigablePath();
-			if ( parent.getReferencedPathSource() instanceof PluralPersistentAttribute<?, ?, ?> ) {
+			if ( parent.getResolvedModel() instanceof PluralPersistentAttribute<?, ?, ?> ) {
 				navigablePath = navigablePath.append( CollectionPart.Nature.ELEMENT.getName() );
 			}
 			if ( getDeclaringType() instanceof IdentifiableDomainType<?> ) {

hibernate-core/src/main/java/org/hibernate/query/sqm/spi/SqmCreationHelper.java

@@ -53,7 +53,7 @@ public static NavigablePath buildSubNavigablePath(SqmPath<?> lhs, String subNavi
 			);
 		}
 		NavigablePath navigablePath = lhs.getNavigablePath();
-		if ( lhs.getReferencedPathSource() instanceof PluralPersistentAttribute<?, ?, ?>
+		if ( lhs.getResolvedModel() instanceof PluralPersistentAttribute<?, ?, ?>
 				&& CollectionPart.Nature.fromName( subNavigable ) == null ) {
 			navigablePath = navigablePath.append( CollectionPart.Nature.ELEMENT.getName() );
 		}

hibernate-core/src/main/java/org/hibernate/query/sqm/sql/BaseSqmToSqlAstConverter.java

@@ -5049,7 +5049,11 @@ private Expression withTreatRestriction(Expression expression, SqmPath<?> path)
 				return expression;
 			}
 			final BasicValuedPathInterpretation<?> basicPath = (BasicValuedPathInterpretation<?>) expression;
-			final AbstractEntityPersister persister = (AbstractEntityPersister) basicPath.getTableGroup().getModelPart().getPartMappingType();
+			final TableGroup tableGroup = basicPath.getTableGroup();
+			final TableGroup elementTableGroup = tableGroup instanceof PluralTableGroup
+					? ( (PluralTableGroup) tableGroup ).getElementTableGroup()
+					: tableGroup;
+			final AbstractEntityPersister persister = (AbstractEntityPersister) elementTableGroup.getModelPart().getPartMappingType();
 			// Only need a case expression around the basic valued path for the parent treat expression
 			// if the column of the basic valued path is shared between subclasses
 			if ( persister.isSharedColumn( basicPath.getColumnReference().getColumnExpression() ) ) {

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

@@ -103,6 +103,11 @@ public EntityDomainType<S> getReferencedPathSource() {
 		return treatTarget;
 	}
 
+	@Override
+	public SqmPathSource<?> getResolvedModel() {
+		return treatTarget;
+	}
+
 	@Override
 	public SqmAttributeJoin<O, S> makeCopy(SqmCreationProcessingState creationProcessingState) {
 		return new SqmTreatedBagJoin<>( wrappedPath, treatTarget, getAlias() );

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

@@ -91,8 +91,12 @@ public SqmPathSource<S> getNodeType() {
 
 	@Override
 	public EntityDomainType<S> getReferencedPathSource() {
-		//noinspection unchecked
+		return treatTarget;
-		return (EntityDomainType<S>) wrappedPath.getReferencedPathSource();
+	}
+
+	@Override
+	public SqmPathSource<?> getResolvedModel() {
+		return treatTarget;
 	}
 
 	@Override

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

@@ -77,6 +77,11 @@ public EntityDomainType<S> getTreatTarget() {
 		return treatTarget;
 	}
 
+	@Override
+	public EntityDomainType<S> getModel() {
+		return getTreatTarget();
+	}
+
 	@Override
 	public SqmPath<T> getWrappedPath() {
 		return wrappedPath;

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

@@ -102,6 +102,11 @@ public EntityDomainType<S> getReferencedPathSource() {
 		return treatTarget;
 	}
 
+	@Override
+	public SqmPathSource<?> getResolvedModel() {
+		return treatTarget;
+	}
+
 	@Override
 	public SqmPath<?> resolveIndexedAccess(
 			SqmExpression<?> selector,

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

@@ -98,6 +98,11 @@ public EntityDomainType<S> getReferencedPathSource() {
 		return treatTarget;
 	}
 
+	@Override
+	public SqmPathSource<?> getResolvedModel() {
+		return treatTarget;
+	}
+
 	@Override
 	public SqmMapJoin<O, K, S> makeCopy(SqmCreationProcessingState creationProcessingState) {
 		return new SqmTreatedMapJoin<>(

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

@@ -91,6 +91,16 @@ public SqmPathSource<S> getNodeType() {
 		return treatTarget;
 	}
 
+	@Override
+	public SqmPathSource<S> getReferencedPathSource() {
+		return treatTarget;
+	}
+
+	@Override
+	public SqmPathSource<?> getResolvedModel() {
+		return treatTarget;
+	}
+
 	@Override
 	public void appendHqlString(StringBuilder sb) {
 		sb.append( "treat(" );

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

@@ -100,6 +100,11 @@ public EntityDomainType<S> getReferencedPathSource() {
 		return treatTarget;
 	}
 
+	@Override
+	public SqmPathSource<?> getResolvedModel() {
+		return treatTarget;
+	}
+
 	@Override
 	public SqmAttributeJoin<O, S> makeCopy(SqmCreationProcessingState creationProcessingState) {
 		return new SqmTreatedSetJoin<>( wrappedPath, treatTarget, getAlias() );

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

@@ -109,6 +109,16 @@ public EntityDomainType<S> getNodeType() {
 		return treatTarget;
 	}
 
+	@Override
+	public SqmPathSource<S> getReferencedPathSource() {
+		return treatTarget;
+	}
+
+	@Override
+	public SqmPathSource<?> getResolvedModel() {
+		return treatTarget;
+	}
+
 	@Override
 	public <S1 extends S> SqmTreatedSimplePath<S,S1> treatAs(Class<S1> treatJavaType) throws PathException {
 		return super.treatAs( treatJavaType );

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

@@ -97,7 +97,12 @@ public SqmPathSource<S> getNodeType() {
 
 	@Override
 	public EntityDomainType<S> getReferencedPathSource() {
-		return getTreatTarget();
+		return treatTarget;
+	}
+
+	@Override
+	public SqmPathSource<?> getResolvedModel() {
+		return treatTarget;
 	}
 
 	@Override

hibernate-core/src/main/java/org/hibernate/sql/results/graph/embeddable/AbstractEmbeddableInitializer.java

@@ -8,13 +8,14 @@
 
 import org.hibernate.engine.internal.ManagedTypeHelper;
 import org.hibernate.engine.spi.SessionFactoryImplementor;
+import org.hibernate.metamodel.internal.AbstractCompositeIdentifierMapping;
 import org.hibernate.metamodel.mapping.CompositeIdentifierMapping;
 import org.hibernate.metamodel.mapping.EmbeddableMappingType;
 import org.hibernate.metamodel.mapping.EmbeddableValuedModelPart;
-import org.hibernate.metamodel.mapping.EntityIdentifierMapping;
 import org.hibernate.metamodel.mapping.ForeignKeyDescriptor;
 import org.hibernate.metamodel.mapping.VirtualModelPart;
-import org.hibernate.metamodel.model.domain.NavigableRole;
+import org.hibernate.metamodel.mapping.internal.EmbeddedAttributeMapping;
+import org.hibernate.metamodel.mapping.internal.EmbeddedIdentifierMappingImpl;
 import org.hibernate.metamodel.spi.ValueAccess;
 import org.hibernate.property.access.spi.PropertyAccess;
 import org.hibernate.proxy.HibernateProxy;
@@ -68,31 +69,20 @@ public AbstractEmbeddableInitializer(
 		final int size = embeddableTypeDescriptor.getNumberOfFetchables();
 		this.rowState = new Object[ size ];
 
-		this.isPartOfKey = isPartOfKey( embedded, navigablePath );
+		this.isPartOfKey = isPartOfKey( embedded, navigablePath, fetchParentAccess );
 		// We never want to create empty composites for the FK target or PK, otherwise collections would break
 		this.createEmptyCompositesEnabled = !isPartOfKey && embeddableTypeDescriptor.isCreateEmptyCompositesEnabled();
-
 		this.sessionFactory = creationState.getSqlAstCreationContext().getSessionFactory();
 		this.assemblers = createAssemblers( resultDescriptor, creationState, embeddableTypeDescriptor );
 	}
 
-	private static boolean isPartOfKey(EmbeddableValuedModelPart modelPart, NavigablePath navigablePath) {
+	private static boolean isPartOfKey(EmbeddableValuedModelPart modelPart, NavigablePath navigablePath, FetchParentAccess fetchParentAccess) {
 		return modelPart.isEntityIdentifierMapping()
-				|| isPartOfKey( navigablePath )
+				|| ForeignKeyDescriptor.PART_NAME.equals( navigablePath.getLocalName() )
-				|| modelPart.getNavigableRole() != null && isPartOfKey( modelPart.getNavigableRole() );
-	}
-
-	private static boolean isPartOfKey(NavigablePath navigablePath) {
-		return ForeignKeyDescriptor.PART_NAME.equals( navigablePath.getLocalName() )
 				|| ForeignKeyDescriptor.TARGET_PART_NAME.equals( navigablePath.getLocalName() )
 				|| navigablePath instanceof EntityIdentifierNavigablePath
-				|| navigablePath.getParent().getParent() != null && isPartOfKey( navigablePath.getParent() );
+				|| fetchParentAccess != null && fetchParentAccess.isEmbeddableInitializer()
-	}
+				&& ( (AbstractEmbeddableInitializer) fetchParentAccess ).isPartOfKey;
-
-	private static boolean isPartOfKey(NavigableRole navigableRole) {
-		final NavigableRole parent = navigableRole.getParent();
-		return parent != null
-				&& ( parent.getLocalName().equals( EntityIdentifierMapping.ID_ROLE_NAME ) || isPartOfKey( parent ) );
 	}
 
 	protected DomainResultAssembler<?>[] createAssemblers(

hibernate-core/src/test/java/org/hibernate/orm/test/embeddable/EmbeddableAscDescQueryTest.java

@@ -74,6 +74,7 @@ public void setUp(SessionFactoryScope scope) {
 	public void tearDown(SessionFactoryScope scope) {
 		scope.inTransaction(
 				session -> {
+					session.createMutationQuery( "update EntityA a set a.componentA.entityA = null" ).executeUpdate();
 					session.createMutationQuery( "delete from EntityB" ).executeUpdate();
 					session.createMutationQuery( "delete from EntityC" ).executeUpdate();
 					session.createMutationQuery( "delete from EntityA" ).executeUpdate();