HHH-18854 Allow HR to change DeleteOrUpsertOperation

Author: DavideD

hibernate-core/src/main/java/org/hibernate/sql/model/jdbc/DeleteOrUpsertOperation.java

@@ -44,7 +44,6 @@ public class DeleteOrUpsertOperation implements SelfExecutingUpdateOperation {
 
 	private final OptionalTableUpdate optionalTableUpdate;
 
-
 	public DeleteOrUpsertOperation(
 			EntityMutationTarget mutationTarget,
 			EntityTableMapping tableMapping,
@@ -56,6 +55,16 @@ public DeleteOrUpsertOperation(
 		this.optionalTableUpdate = optionalTableUpdate;
 	}
 
+	/*
+	 * Used by Hibernate Reactive
+	 */
+	protected DeleteOrUpsertOperation(DeleteOrUpsertOperation original) {
+		this.mutationTarget = original.mutationTarget;
+		this.tableMapping = original.tableMapping;
+		this.upsertOperation = original.upsertOperation;
+		this.optionalTableUpdate = original.optionalTableUpdate;
+	}
+
 	@Override
 	public MutationType getMutationType() {
 		return MutationType.UPDATE;
@@ -197,4 +206,18 @@ private void performUpsert(JdbcValueBindings jdbcValueBindings, SharedSessionCon
 
 		MODEL_MUTATION_LOGGER.tracef( "`%s` rows upserted into `%s`", rowCount, tableMapping.getTableName() );
 	}
+
+	/*
+	 * Used by Hibernate Reactive
+	 */
+	public UpsertOperation getUpsertOperation() {
+		return upsertOperation;
+	}
+
+	/*
+	 * Used by Hibernate Reactive
+	 */
+	public OptionalTableUpdate getOptionalTableUpdate() {
+		return optionalTableUpdate;
+	}
 }