Changes for Hibernate Reactive

Author: DavideD

hibernate-core/src/main/java/org/hibernate/sql/results/graph/entity/internal/EntityInitializerImpl.java

@@ -1264,7 +1264,8 @@ protected void upgradeLockMode(EntityInitializerData data) {
 		}
 	}
 
-	private boolean isProxyInstance(Object proxy) {
+	// Used by Hibernate Reactive
+	protected boolean isProxyInstance(Object proxy) {
 		return proxy != null
 			&& ( proxy instanceof MapProxy || entityDescriptor.getJavaType().getJavaTypeClass().isInstance( proxy ) );
 	}
@@ -1356,7 +1357,8 @@ protected Object instantiateEntity(EntityInitializerData data) {
 				.instantiate( data.concreteDescriptor, data.entityKey.getIdentifier() );
 	}
 
-	private Object resolveToOptionalInstance(EntityInitializerData data) {
+	// Used by Hibernate Reactive
+	protected Object resolveToOptionalInstance(EntityInitializerData data) {
 		if ( isResultInitializer() ) {
 			// this isEntityReturn bit is just for entity loaders, not hql/criteria
 			final var processingOptions =
@@ -1372,7 +1374,8 @@ private Object resolveToOptionalInstance(EntityInitializerData data) {
 		}
 	}
 
-	private boolean matchesOptionalInstance(
+	// Used by Hibernate Reactive
+	protected boolean matchesOptionalInstance(
 			EntityInitializerData data,
 			JdbcValuesSourceProcessingOptions processingOptions) {
 		final Object optionalEntityInstance = processingOptions.getEffectiveOptionalObject();
@@ -1382,7 +1385,8 @@ private boolean matchesOptionalInstance(
 			&& areKeysEqual( requestedEntityId, data.entityKey.getIdentifier() );
 	}
 
-	private Object resolveInstanceFromCache(EntityInitializerData data) {
+	// Used by Hibernate Reactive
+	protected Object resolveInstanceFromCache(EntityInitializerData data) {
 		return loadFromSecondLevelCache(
 				data.getRowProcessingState().getSession().asEventSource(),
 				null,
@@ -1970,6 +1974,25 @@ public String toString() {
 	//#########################
 	// For Hibernate Reactive
 	//#########################
+	protected EntityPersister getRootEntityDescriptor() {
+		return rootEntityDescriptor;
+	}
+
+	protected @Nullable Initializer<?>[][] getEagerSubInitializers() {
+		return eagerSubInitializers;
+	}
+
+	protected @Nullable Initializer<?>[][] getCollectionContainingSubInitializers() {
+		return collectionContainingSubInitializers;
+	}
+
+	protected ImmutableBitSet[] getLazySets() {
+		return lazySets;
+	}
+
+	public boolean isHasLazyInitializingSubAssemblers() {
+		return hasLazyInitializingSubAssemblers;
+	}
 
 	protected @Nullable DomainResultAssembler<?> getVersionAssembler() {
 		return versionAssembler;