[hibernate#1885] Add createMutationQuery method that accept criterias

Author: dreab8

hibernate-reactive-core/src/main/java/org/hibernate/reactive/mutiny/Mutiny.java

@@ -957,6 +957,28 @@ default Uni<Void> lock(Object entity, LockModeType lockModeType) {
 		 */
 		MutationQuery createMutationQuery(String queryString);
 
+		/**
+		 * Create an instance of {@link MutationQuery} for the given update tree.
+		 *
+		 * @param updateQuery the update criteria query
+		 *
+		 * @return The {@link MutationQuery} instance for manipulation and execution
+		 *
+		 * @see org.hibernate.query.QueryProducer#createMutationQuery(CriteriaUpdate)
+		 */
+		<R> MutationQuery createMutationQuery(CriteriaUpdate<R> updateQuery);
+
+		/**
+		 * Create an instance of {@link MutationQuery} for the given delete tree.
+		 *
+		 * @param deleteQuery the delete criteria query
+		 *
+		 * @return The {@link MutationQuery} instance for manipulation and execution
+		 *
+		 * @see org.hibernate.query.QueryProducer#createMutationQuery(CriteriaDelete)
+		 */
+		<R> MutationQuery createMutationQuery(CriteriaDelete<R> deleteQuery);
+
 		/**
 		 * Create an instance of {@link Query} for the given HQL/JPQL query
 		 * string or HQL/JPQL update or delete statement. In the case of an

hibernate-reactive-core/src/main/java/org/hibernate/reactive/mutiny/impl/MutinySessionImpl.java

@@ -124,6 +124,16 @@ public MutationQuery createMutationQuery(String queryString) {
 		return new MutinyMutationQueryImpl<>( delegate.createReactiveQuery( queryString ), factory );
 	}
 
+	@Override
+	public <R> MutationQuery createMutationQuery(CriteriaUpdate<R> updateQuery) {
+		return new MutinyMutationQueryImpl<>( delegate.createReactiveMutationQuery( updateQuery ), factory );
+	}
+
+	@Override
+	public <R> MutationQuery createMutationQuery(CriteriaDelete<R> deleteQuery) {
+		return new MutinyMutationQueryImpl<>( delegate.createReactiveMutationQuery( deleteQuery ), factory );
+	}
+
 	@Override @Deprecated
 	public <R> Query<R> createQuery(String queryString) {
 		return new MutinyQueryImpl<>( delegate.createReactiveQuery( queryString ), factory );

hibernate-reactive-core/src/main/java/org/hibernate/reactive/stage/Stage.java

@@ -994,6 +994,28 @@ default CompletionStage<Void> lock(Object entity, LockModeType lockModeType) {
 		 */
 		MutationQuery createMutationQuery(String queryString);
 
+		/**
+		 * Create an instance of {@link MutationQuery} for the given update tree.
+		 *
+		 * @param updateQuery the update criteria query
+		 *
+		 * @return The {@link MutationQuery} instance for manipulation and execution
+		 *
+		 * @see org.hibernate.query.QueryProducer#createMutationQuery(CriteriaUpdate)
+		 */
+		<R> MutationQuery createMutationQuery(CriteriaUpdate<R> updateQuery);
+
+		/**
+		 * Create an instance of {@link MutationQuery} for the given delete tree.
+		 *
+		 * @param deleteQuery the delete criteria query
+		 *
+		 * @return The {@link MutationQuery} instance for manipulation and execution
+		 *
+		 * @see org.hibernate.query.QueryProducer#createMutationQuery(CriteriaDelete)
+		 */
+		<R> MutationQuery createMutationQuery(CriteriaDelete<R> deleteQuery);
+
 		/**
 		 * Create an instance of {@link Query} for the given HQL/JPQL query
 		 * string or HQL/JPQL update or delete statement. In the case of an
@@ -1663,6 +1685,28 @@ default <T> CompletionStage<T> get(Class<T> entityClass, Object id, LockModeType
 		 */
 		MutationQuery createMutationQuery(String queryString);
 
+		/**
+		 * Create an instance of {@link MutationQuery} for the given update tree.
+		 *
+		 * @param updateQuery the update criteria query
+		 *
+		 * @return The {@link MutationQuery} instance for manipulation and execution
+		 *
+		 * @see org.hibernate.query.QueryProducer#createMutationQuery(CriteriaUpdate)
+		 */
+		<R> MutationQuery createMutationQuery(CriteriaUpdate<R> updateQuery);
+
+		/**
+		 * Create an instance of {@link MutationQuery} for the given delete tree.
+		 *
+		 * @param deleteQuery the delete criteria query
+		 *
+		 * @return The {@link MutationQuery} instance for manipulation and execution
+		 *
+		 * @see org.hibernate.query.QueryProducer#createMutationQuery(CriteriaDelete)
+		 */
+		<R> MutationQuery createMutationQuery(CriteriaDelete<R> deleteQuery);
+
 		/**
 		 * Create an instance of {@link Query} for the given SQL query string,
 		 * or SQL update, insert, or delete statement. In the case of an update,

hibernate-reactive-core/src/main/java/org/hibernate/reactive/stage/impl/StageSessionImpl.java

@@ -117,6 +117,16 @@ public MutationQuery createMutationQuery(String queryString) {
 		return new StageMutationQueryImpl<>( delegate.createReactiveMutationQuery( queryString ) );
 	}
 
+	@Override
+	public <R> MutationQuery createMutationQuery(CriteriaUpdate<R> updateQuery) {
+		return new StageMutationQueryImpl<>( delegate.createReactiveMutationQuery( updateQuery ) );
+	}
+
+	@Override
+	public <R> MutationQuery createMutationQuery(CriteriaDelete<R> deleteQuery) {
+		return new StageMutationQueryImpl<>( delegate.createReactiveMutationQuery( deleteQuery ) );
+	}
+
 	@Override
 	public <T> CompletionStage<T> find(Class<T> entityClass, Object primaryKey) {
 		return delegate.reactiveFind( entityClass, primaryKey, null, null );

hibernate-reactive-core/src/main/java/org/hibernate/reactive/stage/impl/StageStatelessSessionImpl.java

@@ -276,6 +276,16 @@ public MutationQuery createMutationQuery(String queryString) {
 		return new StageMutationQueryImpl<>( delegate.createReactiveMutationQuery( queryString ) );
 	}
 
+	@Override
+	public <R> MutationQuery createMutationQuery(CriteriaUpdate<R> updateQuery) {
+		return new StageMutationQueryImpl<>( delegate.createReactiveMutationQuery( updateQuery ) );
+	}
+
+	@Override
+	public <R> MutationQuery createMutationQuery(CriteriaDelete<R> deleteQuery) {
+		return new StageMutationQueryImpl<>( delegate.createReactiveMutationQuery( deleteQuery )  );
+	}
+
 	@Override
 	public <R> SelectionQuery<R> createQuery(String queryString, Class<R> resultType) {
 		return new StageSelectionQueryImpl<>( delegate.createReactiveQuery( queryString, resultType ) );