HHH-15776 Avoid type checking for ManagedEntity in StatefulPersistenc…

…eContext#addReferenceEntry
hibernate · Nov 28, 2022 · f78e48b · f78e48b
1 parent 7f0300d
commit f78e48b
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/hibernate-core/src/main/java/org/hibernate/engine/internal/StatefulPersistenceContext.java b/hibernate-core/src/main/java/org/hibernate/engine/internal/StatefulPersistenceContext.java
@@ -68,6 +68,7 @@
 
 import org.jboss.logging.Logger;
 
+import static org.hibernate.engine.internal.ManagedTypeHelper.asManagedEntity;
 import static org.hibernate.engine.internal.ManagedTypeHelper.asPersistentAttributeInterceptable;
 import static org.hibernate.engine.internal.ManagedTypeHelper.isPersistentAttributeInterceptable;
 
@@ -573,9 +574,9 @@ public EntityEntry addEntry(
 	public void addReferenceEntry(
 			final Object entity,
 			final Status status) {
-
-		((ManagedEntity)entity).$$_hibernate_getEntityEntry().setStatus( status );
-		entityEntryContext.addEntityEntry( entity, ((ManagedEntity)entity).$$_hibernate_getEntityEntry() );
+		final EntityEntry entityEntry = asManagedEntity( entity ).$$_hibernate_getEntityEntry();
+		entityEntry.setStatus( status );
+		entityEntryContext.addEntityEntry( entity, entityEntry );
 
 		setHasNonReadOnlyEnties( status );
 	}