HHH-7304 bugfix proposal

hibernate-core/src/main/java/org/hibernate/action/internal/AbstractEntityInsertAction.java

@@ -72,8 +72,10 @@ public abstract class AbstractEntityInsertAction extends EntityAction {
 		this.isExecuted = false;
 		this.areTransientReferencesNullified = false;
 
+		if (id != null) {
 			handleNaturalIdPreSaveNotifications();
 		}
+	}
 
 	/**
 	 * Returns the entity state.
@@ -193,7 +195,17 @@ public abstract class AbstractEntityInsertAction extends EntityAction {
 	/**
 	 * Handle sending notifications needed for natural-id after saving
 	 */
-	protected void handleNaturalIdPostSaveNotifications() {
+	public void handleNaturalIdPostSaveNotifications(Serializable id) {
+		if (isEarlyInsert()) {
+			// with early insert, we still need to add a local (transactional) natural id cross-reference
+			getSession().getPersistenceContext().getNaturalIdHelper().manageLocalNaturalIdCrossReference(
+					getPersister(),
+					id,
+					state,
+					null,
+					CachedNaturalIdValueSource.INSERT
+			);
+		}
 		// after save, we need to manage the shared cache entries
 		getSession().getPersistenceContext().getNaturalIdHelper().manageSharedNaturalIdCrossReference(
 				getPersister(),

hibernate-core/src/main/java/org/hibernate/action/internal/EntityInsertAction.java

@@ -123,7 +123,7 @@ public final class EntityInsertAction extends AbstractEntityInsertAction {
 			}
 		}
 
-		handleNaturalIdPostSaveNotifications();
+		handleNaturalIdPostSaveNotifications(id);
 
 		postInsert();
 

hibernate-core/src/main/java/org/hibernate/event/internal/AbstractSaveEventListener.java

@@ -298,6 +298,8 @@ public abstract class AbstractSaveEventListener extends AbstractReassociateEvent
 								insert.getClass().getName() );
 			}
 			id = ( ( EntityIdentityInsertAction ) insert ).getGeneratedId();
+			
+			insert.handleNaturalIdPostSaveNotifications(id);
 		}
 
 		markInterceptorDirty( entity, persister, source );