HHH-19364 add Mutators

Author: gavinking

hibernate-core/src/main/java/org/hibernate/query/programmatic/MutationSpecification.java

@@ -4,8 +4,11 @@
  */
 package org.hibernate.query.programmatic;
 
+import jakarta.persistence.criteria.CommonAbstractCriteria;
 import jakarta.persistence.criteria.CriteriaUpdate;
 import jakarta.persistence.criteria.CriteriaDelete;
+import jakarta.persistence.criteria.CriteriaBuilder;
+import jakarta.persistence.criteria.Root;
 import org.hibernate.Incubating;
 import org.hibernate.SharedSessionContract;
 import org.hibernate.query.IllegalMutationQueryException;
@@ -42,6 +45,13 @@ public interface MutationSpecification<T> extends QuerySpecification<T> {
 	@Override
 	MutationSpecification<T> addRestriction(Restriction<T> restriction);
 
+	@FunctionalInterface
+	interface Mutator<T> {
+		void mutate(CriteriaBuilder builder, CommonAbstractCriteria query, Root<T> mutationTarget);
+	}
+
+	MutationSpecification<T> mutate(Mutator<T> mutation);
+
 	/**
 	 * Finalize the building and create the {@linkplain SelectionQuery} instance.
 	 */

hibernate-core/src/main/java/org/hibernate/query/programmatic/SelectionSpecification.java

@@ -5,6 +5,8 @@
 package org.hibernate.query.programmatic;
 
 import jakarta.persistence.criteria.CriteriaQuery;
+import jakarta.persistence.criteria.CriteriaBuilder;
+import jakarta.persistence.criteria.Root;
 import org.hibernate.Incubating;
 import org.hibernate.SharedSessionContract;
 import org.hibernate.query.IllegalSelectQueryException;
@@ -89,6 +91,13 @@ public interface SelectionSpecification<T> extends QuerySpecification<T> {
 	@Override
 	SelectionSpecification<T> addRestriction(Restriction<T> restriction);
 
+	@FunctionalInterface
+	interface Mutator<T> {
+		void mutate(CriteriaBuilder builder, CriteriaQuery<T> query, Root<T> root);
+	}
+
+	SelectionSpecification<T> mutate(Mutator<T> mutation);
+
 	/**
 	 * Covariant override.
 	 */

hibernate-core/src/main/java/org/hibernate/query/programmatic/internal/MutationSpecificationImpl.java

@@ -73,6 +73,13 @@ public MutationSpecification<T> addRestriction(Restriction<T> restriction) {
 		return this;
 	}
 
+	@Override
+	public MutationSpecification<T> mutate(Mutator<T> mutation) {
+		specifications.add( (sqmStatement, mutationTargetRoot) ->
+				mutation.mutate( sqmStatement.nodeBuilder(), sqmStatement, mutationTargetRoot ) );
+		return this;
+	}
+
 	@Override
 	public MutationQuery createQuery(SharedSessionContract session) {
 		final var sessionImpl = (SharedSessionContractImplementor) session;

hibernate-core/src/main/java/org/hibernate/query/programmatic/internal/SelectionSpecificationImpl.java

@@ -71,6 +71,13 @@ public SelectionSpecification<T> addRestriction(Restriction<T> restriction) {
 		return this;
 	}
 
+	@Override
+	public SelectionSpecification<T> mutate(Mutator<T> mutation) {
+		specifications.add( (sqmStatement, root) ->
+				mutation.mutate( sqmStatement.nodeBuilder(), sqmStatement, root ) );
+		return this;
+	}
+
 	@Override
 	public SelectionSpecification<T> addOrdering(Order<T> order) {
 		specifications.add( (sqmStatement, root) -> {