[#1598] Fix ReactiveJoinedSubclassEntityPersister

Author: DavideD

hibernate-reactive-core/src/main/java/org/hibernate/reactive/persister/entity/impl/ReactiveJoinedSubclassEntityPersister.java

@@ -41,6 +41,7 @@
 import org.hibernate.reactive.persister.entity.mutation.ReactiveDeleteCoordinator;
 import org.hibernate.reactive.persister.entity.mutation.ReactiveInsertCoordinator;
 import org.hibernate.reactive.persister.entity.mutation.ReactiveUpdateCoordinator;
+import org.hibernate.reactive.sql.results.internal.ReactiveEntityResultJoinedSubclassImpl;
 import org.hibernate.reactive.util.impl.CompletionStages;
 import org.hibernate.spi.NavigablePath;
 import org.hibernate.sql.ast.tree.from.TableGroup;
@@ -163,7 +164,7 @@ public <T> DomainResult<T> createDomainResult(
 			String resultVariable,
 			DomainResultCreationState creationState) {
 		if ( hasSubclasses() ) {
-			final EntityResultJoinedSubclassImpl entityResultJoinedSubclass = new EntityResultJoinedSubclassImpl(
+			final EntityResultJoinedSubclassImpl entityResultJoinedSubclass = new ReactiveEntityResultJoinedSubclassImpl(
 					navigablePath,
 					this,
 					tableGroup,

hibernate-reactive-core/src/main/java/org/hibernate/reactive/sql/results/internal/ReactiveEntityResultJoinedSubclassImpl.java

@@ -0,0 +1,52 @@
+/* Hibernate, Relational Persistence for Idiomatic Java
+ *
+ * SPDX-License-Identifier: Apache-2.0
+ * Copyright: Red Hat Inc. and Hibernate Authors
+ */
+package org.hibernate.reactive.sql.results.internal;
+
+import org.hibernate.metamodel.mapping.EntityValuedModelPart;
+import org.hibernate.reactive.sql.results.graph.entity.internal.ReactiveEntityAssembler;
+import org.hibernate.reactive.sql.results.graph.entity.internal.ReactiveEntityResultInitializer;
+import org.hibernate.spi.NavigablePath;
+import org.hibernate.sql.ast.tree.from.TableGroup;
+import org.hibernate.sql.results.graph.AssemblerCreationState;
+import org.hibernate.sql.results.graph.DomainResultAssembler;
+import org.hibernate.sql.results.graph.FetchParentAccess;
+import org.hibernate.sql.results.graph.Initializer;
+import org.hibernate.sql.results.graph.entity.internal.EntityResultJoinedSubclassImpl;
+
+/**
+ * @see org.hibernate.sql.results.graph.entity.internal.EntityResultJoinedSubclassImpl
+ */
+public class ReactiveEntityResultJoinedSubclassImpl extends EntityResultJoinedSubclassImpl {
+
+	public ReactiveEntityResultJoinedSubclassImpl(
+			NavigablePath navigablePath,
+			EntityValuedModelPart entityValuedModelPart,
+			TableGroup tableGroup,
+			String resultVariable) {
+		super( navigablePath, entityValuedModelPart, tableGroup, resultVariable );
+	}
+
+	@Override
+	public DomainResultAssembler createResultAssembler(
+			FetchParentAccess parentAccess,
+			AssemblerCreationState creationState) {
+		final Initializer initializer = creationState.resolveInitializer(
+				getNavigablePath(),
+				getReferencedModePart(),
+				() -> new ReactiveEntityResultInitializer(
+						this,
+						getNavigablePath(),
+						getLockMode( creationState ),
+						getIdentifierFetch(),
+						getDiscriminatorFetch(),
+						getRowIdResult(),
+						creationState
+				)
+		);
+
+		return new ReactiveEntityAssembler( this.getResultJavaType(), initializer.asEntityInitializer() );
+	}
+}