HHH-19487 move dirty-checking related code out of AbstractEntityPersister

Author: gavinking

hibernate-core/src/main/java/org/hibernate/engine/internal/EntityEntryImpl.java

@@ -306,6 +306,12 @@ public void postUpdate(Object entity, Object[] updatedState, Object nextVersion)
 				.resetDirty( entity, persister, (SessionImplementor) session );
 	}
 
+	@Override
+	public void postLoad(Object entity) {
+		processIfSelfDirtinessTracker( entity, EntityEntryImpl::clearDirtyAttributes );
+		processIfManagedEntity( entity, EntityEntryImpl::useTracker );
+	}
+
 	private static void clearDirtyAttributes(final SelfDirtinessTracker entity) {
 		entity.$$_hibernate_clearDirtyAttributes();
 	}

hibernate-core/src/main/java/org/hibernate/engine/spi/EntityEntry.java

@@ -70,6 +70,8 @@ public interface EntityEntry {
 
 	Object getRowId();
 
+	void postLoad(Object entity);
+
 	/**
 	 * Handle updating the internal state of the entry after actually performing
 	 * the database update. Specifically, we update the snapshot information and

hibernate-core/src/main/java/org/hibernate/loader/internal/CacheLoadHelper.java

@@ -334,21 +334,23 @@ private static Object convertCacheEntryToEntity(
 			);
 		}
 		final Object version = getVersion( values, subclassPersister );
+		final EntityEntry entityEntry = persistenceContext.addEntry(
+				entity,
+				isReadOnly ? Status.READ_ONLY : Status.MANAGED,
+				values,
+				null,
+				entityId,
+				version,
+				LockMode.NONE,
+				true,
+				subclassPersister,
+				false
+		);
 		holder.setEntityEntry(
-				persistenceContext.addEntry(
-						entity,
-						isReadOnly ? Status.READ_ONLY : Status.MANAGED,
-						values,
-						null,
-						entityId,
-						version,
-						LockMode.NONE,
-						true,
-						subclassPersister,
-						false
-				)
+				entityEntry
 		);
 		subclassPersister.afterInitialize( entity, source );
+		entityEntry.postLoad( entity );
 		persistenceContext.initializeNonLazyCollections();
 
 		return entity;

hibernate-core/src/main/java/org/hibernate/persister/entity/AbstractEntityPersister.java

@@ -57,12 +57,10 @@
 import org.hibernate.engine.spi.EntityHolder;
 import org.hibernate.engine.spi.EntityKey;
 import org.hibernate.engine.spi.LoadQueryInfluencers;
-import org.hibernate.engine.spi.ManagedEntity;
 import org.hibernate.engine.spi.NaturalIdResolutions;
 import org.hibernate.engine.spi.PersistenceContext;
 import org.hibernate.engine.spi.PersistentAttributeInterceptable;
 import org.hibernate.engine.spi.PersistentAttributeInterceptor;
-import org.hibernate.engine.spi.SelfDirtinessTracker;
 import org.hibernate.engine.spi.SessionFactoryImplementor;
 import org.hibernate.engine.spi.SessionImplementor;
 import org.hibernate.engine.spi.SharedSessionContractImplementor;
@@ -283,9 +281,7 @@
 import static org.hibernate.boot.model.internal.SoftDeleteHelper.resolveSoftDeleteMapping;
 import static org.hibernate.engine.internal.ManagedTypeHelper.asPersistentAttributeInterceptable;
 import static org.hibernate.engine.internal.ManagedTypeHelper.isPersistentAttributeInterceptable;
-import static org.hibernate.engine.internal.ManagedTypeHelper.processIfManagedEntity;
 import static org.hibernate.engine.internal.ManagedTypeHelper.processIfPersistentAttributeInterceptable;
-import static org.hibernate.engine.internal.ManagedTypeHelper.processIfSelfDirtinessTracker;
 import static org.hibernate.generator.EventType.INSERT;
 import static org.hibernate.generator.EventType.UPDATE;
 import static org.hibernate.internal.util.ReflectHelper.isAbstractClass;
@@ -4042,38 +4038,36 @@ public boolean isVersionPropertyGenerated() {
 			&& ( isVersionGeneratedOnExecution() || isVersionGeneratedBeforeExecution() );
 	}
 
+	private Generator versionPropertyGenerator() {
+		return getEntityMetamodel().getGenerators()[ getVersionProperty() ];
+	}
+
 	public boolean isVersionGeneratedOnExecution() {
-		final Generator strategy = getEntityMetamodel().getGenerators()[ getVersionProperty() ];
-		return strategy != null && strategy.generatesSometimes() && strategy.generatedOnExecution();
+		final Generator strategy = versionPropertyGenerator();
+		return strategy != null
+			&& strategy.generatesSometimes()
+			&& strategy.generatedOnExecution();
 	}
 
 	public boolean isVersionGeneratedBeforeExecution() {
-		final Generator strategy = getEntityMetamodel().getGenerators()[ getVersionProperty() ];
-		return strategy != null && strategy.generatesSometimes() && !strategy.generatedOnExecution();
+		final Generator strategy = versionPropertyGenerator();
+		return strategy != null
+			&& strategy.generatesSometimes()
+			&& !strategy.generatedOnExecution();
 	}
 
 	@Override
 	public void afterInitialize(Object entity, SharedSessionContractImplementor session) {
-		if ( isPersistentAttributeInterceptable( entity ) && getRepresentationStrategy().getMode() == POJO ) {
-			final BytecodeLazyAttributeInterceptor interceptor = getEntityMetamodel().getBytecodeEnhancementMetadata()
-					.extractLazyInterceptor( entity );
+		if ( isPersistentAttributeInterceptable( entity )
+				&& getRepresentationStrategy().getMode() == POJO ) {
+			final BytecodeLazyAttributeInterceptor interceptor =
+					getEntityMetamodel().getBytecodeEnhancementMetadata()
+							.extractLazyInterceptor( entity );
 			assert interceptor != null;
 			if ( interceptor.getLinkedSession() == null ) {
 				interceptor.setSession( session );
 			}
 		}
-
-		// clear the fields that are marked as dirty in the dirtiness tracker
-		processIfSelfDirtinessTracker( entity, AbstractEntityPersister::clearDirtyAttributes );
-		processIfManagedEntity( entity, AbstractEntityPersister::useTracker );
-	}
-
-	private static void clearDirtyAttributes(final SelfDirtinessTracker entity) {
-		entity.$$_hibernate_clearDirtyAttributes();
-	}
-
-	private static void useTracker(final ManagedEntity entity) {
-		entity.$$_hibernate_setUseTracker( true );
 	}
 
 	@Override

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

@@ -1403,6 +1403,7 @@ protected void initializeEntityInstance(EntityInitializerData data) {
 		takeSnapshot( data, session, persistenceContext, entityEntry, resolvedEntityState );
 
 		data.concreteDescriptor.afterInitialize( entityInstanceForNotify, session );
+		entityEntry.postLoad( entityInstanceForNotify );
 
 		assert data.concreteDescriptor.getIdentifier( entityInstanceForNotify, session ) != null;