Call reactive autoflushIfRequired

Author: DavideD

hibernate-reactive-core/src/main/java/org/hibernate/reactive/query/sqm/mutation/internal/cte/ReactiveAbstractCteMutationHandler.java

@@ -33,6 +33,7 @@
 import org.hibernate.query.sqm.tree.expression.SqmParameter;
 import org.hibernate.query.sqm.tree.expression.SqmStar;
 import org.hibernate.reactive.query.sqm.mutation.spi.ReactiveAbstractMutationHandler;
+import org.hibernate.reactive.session.ReactiveSession;
 import org.hibernate.reactive.sql.exec.internal.StandardReactiveSelectExecutor;
 import org.hibernate.reactive.sql.results.spi.ReactiveListResultsConsumer;
 import org.hibernate.sql.ast.SqlAstTranslator;
@@ -183,16 +184,18 @@ default CompletionStage<Integer> reactiveExecute(DomainQueryExecutionContext exe
 				executionContext.getQueryOptions()
 		);
 		lockOptions.setAliasSpecificLockMode( explicitDmlTargetAlias, lockMode );
-		executionContext.getSession().autoFlushIfRequired( select.getAffectedTableNames() );
-
-		return StandardReactiveSelectExecutor.INSTANCE.list(
-						select,
-						jdbcParameterBindings,
-						SqmJdbcExecutionContextAdapter.omittingLockingAndPaging( executionContext ),
-						row -> row[0],
-						ReactiveListResultsConsumer.UniqueSemantic.NONE
-				)
-				.thenApply( list -> ( (Number) list.get( 0 ) ).intValue() );
+
+		return ( (ReactiveSession) executionContext.getSession() )
+				.reactiveAutoFlushIfRequired( select.getAffectedTableNames() )
+				.thenCompose( v -> StandardReactiveSelectExecutor.INSTANCE.list(
+											  select,
+											  jdbcParameterBindings,
+											  SqmJdbcExecutionContextAdapter.omittingLockingAndPaging( executionContext ),
+											  row -> row[0],
+											  ReactiveListResultsConsumer.UniqueSemantic.NONE
+									  )
+									  .thenApply( list -> ( (Number) list.get( 0 ) ).intValue() )
+				);
 	}
 
 	/**

hibernate-reactive-core/src/main/java/org/hibernate/reactive/query/sqm/mutation/internal/cte/ReactiveCteInsertHandler.java

@@ -47,6 +47,7 @@
 import org.hibernate.reactive.logging.impl.Log;
 import org.hibernate.reactive.logging.impl.LoggerFactory;
 import org.hibernate.reactive.query.sqm.mutation.internal.ReactiveHandler;
+import org.hibernate.reactive.session.ReactiveSession;
 import org.hibernate.reactive.sql.exec.internal.StandardReactiveSelectExecutor;
 import org.hibernate.reactive.sql.results.spi.ReactiveListResultsConsumer;
 import org.hibernate.spi.NavigablePath;
@@ -589,15 +590,15 @@ public <T> MappingModelExpressible<T> getResolvedMappingModelType(SqmParameter<T
 				executionContext.getSession()
 		);
 		final JdbcOperationQuerySelect select = translator.translate( jdbcParameterBindings, executionContext.getQueryOptions() );
-
-		executionContext.getSession().autoFlushIfRequired( select.getAffectedTableNames() );
-		return StandardReactiveSelectExecutor.INSTANCE.list(
-						select,
-						jdbcParameterBindings,
-						SqmJdbcExecutionContextAdapter.omittingLockingAndPaging( executionContext ),
-						row -> row[0],
-						ReactiveListResultsConsumer.UniqueSemantic.NONE
-				)
-				.thenApply( list -> ( (Number) list.get( 0 ) ).intValue() );
+		return ( (ReactiveSession) executionContext.getSession() )
+				.reactiveAutoFlushIfRequired( select.getAffectedTableNames() )
+				.thenCompose( v -> StandardReactiveSelectExecutor.INSTANCE.list(
+								select,
+								jdbcParameterBindings,
+								SqmJdbcExecutionContextAdapter.omittingLockingAndPaging( executionContext ),
+								row -> row[0],
+								ReactiveListResultsConsumer.UniqueSemantic.NONE
+						)
+						.thenApply( list -> ( (Number) list.get( 0 ) ).intValue() ) );
 	}
 }