[#HHH-17204] visibility changes for Reactive upsert() support

hibernate · Sep 13, 2023 · 3dbb9c6 · 3dbb9c6
1 parent 2561a50
commit 3dbb9c6
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 2 deletions.
diff --git a/hibernate-core/src/main/java/org/hibernate/persister/entity/AbstractEntityPersister.java b/hibernate-core/src/main/java/org/hibernate/persister/entity/AbstractEntityPersister.java
@@ -1051,6 +1051,11 @@ public DeleteCoordinator getDeleteCoordinator() {
 		return deleteCoordinator;
 	}
 
+	@Internal
+	public UpdateCoordinator getMergeCoordinator() {
+		return mergeCoordinator;
+	}
+
 	public String getVersionSelectString() {
 		return sqlVersionSelectString;
 	}

diff --git a/hibernate-core/src/main/java/org/hibernate/persister/entity/mutation/MergeCoordinator.java b/hibernate-core/src/main/java/org/hibernate/persister/entity/mutation/MergeCoordinator.java
@@ -24,7 +24,7 @@ public MergeCoordinator(AbstractEntityPersister entityPersister, SessionFactoryI
 	}
 
 	@Override
-	<O extends MutationOperation> AbstractTableUpdateBuilder<O> newTableUpdateBuilder(EntityTableMapping tableMapping) {
+	protected <O extends MutationOperation> AbstractTableUpdateBuilder<O> newTableUpdateBuilder(EntityTableMapping tableMapping) {
 		return new TableMergeBuilder<>( entityPersister(), tableMapping, factory() );
 	}
 

diff --git a/...core/src/main/java/org/hibernate/persister/entity/mutation/UpdateCoordinatorStandard.java b/...core/src/main/java/org/hibernate/persister/entity/mutation/UpdateCoordinatorStandard.java
@@ -1088,7 +1088,7 @@ protected MutationOperationGroup generateDynamicUpdateGroup(
 		return createOperationGroup( valuesAnalysis, updateGroupBuilder.buildMutationGroup() );
 	}
 
-	<O extends MutationOperation> AbstractTableUpdateBuilder<O> newTableUpdateBuilder(EntityTableMapping tableMapping) {
+	protected <O extends MutationOperation> AbstractTableUpdateBuilder<O> newTableUpdateBuilder(EntityTableMapping tableMapping) {
 		return new TableUpdateBuilderStandard<>( entityPersister(), tableMapping, factory() );
 	}