Rework multi-table handling and non-select query plan code to allow caching and help Hibernate Reactive

Author: dreab8

hibernate-core/src/main/java/org/hibernate/query/sqm/internal/AbstractMultiTableMutationQueryPlan.java

@@ -38,6 +38,11 @@ protected abstract MultiTableHandlerBuildResult buildHandler(
 	@Override
 	public int executeUpdate(DomainQueryExecutionContext context) {
 		BulkOperationCleanupAction.schedule( context.getSession(), statement );
+		return handler.execute( getJdbcParameterBindings( context ), context );
+	}
+
+	// For Hibernate Reactive
+	protected JdbcParameterBindings getJdbcParameterBindings(DomainQueryExecutionContext context) {
 		MultiTableHandler localCopy = handler;
 		JdbcParameterBindings jdbcParameterBindings = null;
 
@@ -100,7 +105,7 @@ public int executeUpdate(DomainQueryExecutionContext context) {
 		if ( jdbcParameterBindings == null ) {
 			jdbcParameterBindings = localCopy.createJdbcParameterBindings( context );
 		}
-		return localCopy.execute( jdbcParameterBindings, context );
+		return jdbcParameterBindings;
 	}
 
 }

hibernate-core/src/main/java/org/hibernate/query/sqm/internal/ConcreteSqmSelectQueryPlan.java

@@ -455,7 +455,8 @@ private <T, X> T withCacheableSqmInterpretation(DomainQueryExecutionContext exec
 		return interpreter.interpret( context, executionContext, localCopy, jdbcParameterBindings );
 	}
 
-	private JdbcParameterBindings createJdbcParameterBindings(CacheableSqmInterpretation<SelectStatement, JdbcOperationQuerySelect> sqmInterpretation, DomainQueryExecutionContext executionContext) {
+	// For Hibernate Reactive
+	protected JdbcParameterBindings createJdbcParameterBindings(CacheableSqmInterpretation<SelectStatement, JdbcOperationQuerySelect> sqmInterpretation, DomainQueryExecutionContext executionContext) {
 		return SqmUtil.createJdbcParameterBindings(
 				executionContext.getQueryParameterBindings(),
 				domainParameterXref,
@@ -471,7 +472,8 @@ public <T> MappingModelExpressible<T> getResolvedMappingModelType(SqmParameter<T
 		);
 	}
 
-	private static CacheableSqmInterpretation<SelectStatement, JdbcOperationQuerySelect> buildInterpretation(
+	// For Hibernate Reactive
+	protected static CacheableSqmInterpretation<SelectStatement, JdbcOperationQuerySelect> buildInterpretation(
 			SqmSelectStatement<?> sqm,
 			DomainParameterXref domainParameterXref,
 			DomainQueryExecutionContext executionContext,

hibernate-core/src/main/java/org/hibernate/query/sqm/mutation/internal/temptable/TableBasedDeleteHandler.java

@@ -747,4 +747,9 @@ protected TemporaryTableStrategy getTemporaryTableStrategy() {
 	protected Function<SharedSessionContractImplementor, String> getSessionUidAccess() {
 		return sessionUidAccess;
 	}
+
+	// For Hibernate Reactive
+	protected @Nullable JdbcParameter getSessionUidParameter() {
+		return sessionUidParameter;
+	}
 }

hibernate-core/src/main/java/org/hibernate/query/sqm/mutation/internal/temptable/TableBasedInsertHandler.java

@@ -1325,4 +1325,8 @@ protected AfterUseAction getAfterUseAction() {
 		return forceDropAfterUse ? AfterUseAction.DROP : temporaryTableStrategy.getTemporaryTableAfterUseAction();
 	}
 
+	// For Hibernate Reactive
+	protected @Nullable JdbcParameter getSessionUidParameter() {
+		return sessionUidParameter;
+	}
 }

hibernate-core/src/main/java/org/hibernate/query/sqm/mutation/internal/temptable/TableBasedSoftDeleteHandler.java

@@ -512,4 +512,9 @@ protected TemporaryTableStrategy getTemporaryTableStrategy() {
 	protected Function<SharedSessionContractImplementor, String> getSessionUidAccess() {
 		return sessionUidAccess;
 	}
+
+	// For Hibernate Reactive
+	protected @Nullable JdbcParameter getSessionUidParameter() {
+		return sessionUidParameter;
+	}
 }

hibernate-core/src/main/java/org/hibernate/query/sqm/mutation/internal/temptable/TableBasedUpdateHandler.java

@@ -736,4 +736,9 @@ protected CacheableSqmInterpretation<InsertSelectStatement, JdbcOperationQueryMu
 	protected Function<SharedSessionContractImplementor, String> getSessionUidAccess() {
 		return sessionUidAccess;
 	}
+
+	// For Hibernate reactive
+	protected @Nullable JdbcParameter getSessionUidParameter() {
+		return sessionUidParameter;
+	}
 }