HHH-16982 Avoid unnecessary registerReloadedEntity calls

hibernate-core/src/main/java/org/hibernate/query/internal/DelegatingDomainQueryExecutionContext.java

@@ -37,6 +37,11 @@ public class DelegatingDomainQueryExecutionContext implements DomainQueryExecuti
 		return delegate.getCallback();
 	}
 
+	@Override
+	public boolean hasCallbackActions() {
+		return delegate.hasCallbackActions();
+	}
+
 	@Override
 	public SharedSessionContractImplementor getSession() {
 		return delegate.getSession();

hibernate-core/src/main/java/org/hibernate/query/spi/AbstractSelectionQuery.java

@@ -535,6 +535,11 @@ public abstract class AbstractSelectionQuery<R>
 		return callback;
 	}
 
+	@Override
+	public boolean hasCallbackActions() {
+		return callback != null && callback.hasAfterLoadActions();
+	}
+
 	protected void resetCallback() {
 		callback = null;
 	}

hibernate-core/src/main/java/org/hibernate/query/spi/DomainQueryExecutionContext.java

@@ -29,6 +29,11 @@ public interface DomainQueryExecutionContext {
 	 */
 	Callback getCallback();
 
+	default boolean hasCallbackActions() {
+		final Callback callback = getCallback();
+		return callback != null && callback.hasAfterLoadActions();
+	}
+
 	/**
 	 * The underlying session
 	 */

hibernate-core/src/main/java/org/hibernate/query/sql/internal/NativeQueryImpl.java

@@ -436,6 +436,11 @@ public class NativeQueryImpl<R>
 		return getSession().getFactory();
 	}
 
+	@Override
+	public boolean hasCallbackActions() {
+		return callback != null && callback.hasAfterLoadActions();
+	}
+
 	@Override
 	public QueryParameterBindings getQueryParameterBindings() {
 		return parameterBindings;

hibernate-core/src/main/java/org/hibernate/query/sqm/internal/SqmJdbcExecutionContextAdapter.java

@@ -65,6 +65,11 @@ public class SqmJdbcExecutionContextAdapter extends BaseExecutionContext {
 		return sqmExecutionContext.getCallback();
 	}
 
+	@Override
+	public boolean hasCallbackActions() {
+		return sqmExecutionContext.hasCallbackActions();
+	}
+
 	@Override
 	public boolean hasQueryExecutionToBeAddedToStatistics() {
 		return true;

hibernate-core/src/main/java/org/hibernate/sql/exec/internal/CallbackImpl.java

@@ -37,4 +37,9 @@ public class CallbackImpl implements Callback {
 			afterLoadActions.get( i ).afterLoad( entity, entityMappingType, session );
 		}
 	}
+
+	@Override
+	public boolean hasAfterLoadActions() {
+		return !afterLoadActions.isEmpty();
+	}
 }

hibernate-core/src/main/java/org/hibernate/sql/exec/internal/CallbackNoOp.java

@@ -29,4 +29,9 @@ public class CallbackNoOp implements Callback {
 	public void invokeAfterLoadActions(Object entity, EntityMappingType entityMappingType, SharedSessionContractImplementor session) {
 		// don't do anything
 	}
+
+	@Override
+	public boolean hasAfterLoadActions() {
+		return false;
+	}
 }

hibernate-core/src/main/java/org/hibernate/sql/exec/internal/JdbcSelectExecutorStandardImpl.java

@@ -793,5 +793,10 @@ public class JdbcSelectExecutorStandardImpl implements JdbcSelectExecutor {
 			return context.getCallback();
 		}
 
+		@Override
+		public boolean hasCallbackActions() {
+			return context.hasCallbackActions();
+		}
+
 	}
 }

hibernate-core/src/main/java/org/hibernate/sql/exec/spi/Callback.java

@@ -36,4 +36,6 @@ public interface Callback {
 	default void invokeAfterLoadActions(SharedSessionContractImplementor session, Object entity, org.hibernate.persister.entity.Loadable persister) {
 		invokeAfterLoadActions( entity, persister, session );
 	}
+
+	boolean hasAfterLoadActions();
 }

hibernate-core/src/main/java/org/hibernate/sql/exec/spi/ExecutionContext.java

@@ -36,6 +36,11 @@ public interface ExecutionContext {
 
 	Callback getCallback();
 
+	default boolean hasCallbackActions() {
+		final Callback callback = getCallback();
+		return callback != null && callback.hasAfterLoadActions();
+	}
+
 	String getQueryIdentifier(String sql);
 
 	/**

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

@@ -104,6 +104,11 @@ public class NestedRowProcessingState extends BaseExecutionContext implements Ro
 		return processingState.getCallback();
 	}
 
+	@Override
+	public boolean hasCallbackActions() {
+		return processingState.hasCallbackActions();
+	}
+
 	@Override
 	public CollectionKey getCollectionKey() {
 		return processingState.getCollectionKey();

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

@@ -716,7 +716,7 @@ public abstract class AbstractEntityInitializer extends AbstractFetchParentAcces
 	}
 
 	protected void registerReloadedEntity(RowProcessingState rowProcessingState, Object instance) {
-		if ( rowProcessingState.getCallback() != null ) {
+		if ( rowProcessingState.hasCallbackActions() ) {
 			// This is only needed for follow-on locking, so skip registering the entity if there is no callback
 			rowProcessingState.getJdbcValuesSourceProcessingState()
 					.registerReloadedEntity(

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

@@ -153,6 +153,11 @@ public class RowProcessingStateStandardImpl extends BaseExecutionContext impleme
 		return executionContext.getCallback();
 	}
 
+	@Override
+	public boolean hasCallbackActions() {
+		return executionContext.hasCallbackActions();
+	}
+
 	@Override
 	public CollectionKey getCollectionKey() {
 		return executionContext.getCollectionKey();

hibernate-core/src/test/java/org/hibernate/orm/test/query/sqm/BaseSqmUnitTest.java

@@ -55,6 +55,11 @@ public abstract class BaseSqmUnitTest
 	public void invokeAfterLoadActions(Object entity, EntityMappingType entityMappingType, SharedSessionContractImplementor session) {
 	}
 
+	@Override
+	public boolean hasAfterLoadActions() {
+		return false;
+	}
+
 	protected SqmSelectStatement<?> interpretSelect(String hql) {
 		return interpretSelect( hql, sessionFactory() );
 	}