add UpdateSpecification.reassign()

Author: gavinking

hibernate-core/src/main/java/org/hibernate/query/specification/UpdateSpecification.java

@@ -11,6 +11,8 @@
 import org.hibernate.query.restriction.Restriction;
 import org.hibernate.query.specification.internal.UpdateSpecificationImpl;
 
+import java.util.List;
+
 /**
  * Specialization of {@link MutationSpecification} for programmatic customization
  * of update queries.
@@ -36,18 +38,25 @@
 @Incubating
 public interface UpdateSpecification<T> extends MutationSpecification<T> {
 	/**
-	 * If this {@code MutationSpecification} represents an {@code update}
-	 * statement, add an assigment to a field or property of the target
-	 * entity.
+	 * Add an assigment to a field or property of the target entity.
 	 *
 	 * @param assignment The assignment to add
 	 *
 	 * @return {@code this} for method chaining.
-	 *
-	 * @since 7.2
 	 */
 	UpdateSpecification<T> assign(Assignment<? super T> assignment);
 
+	/**
+	 * Sets the assignments to fields or properties of the target entity.
+	 * If assignments were already specified, this method drops the previous
+	 * assignments in favor of the passed {@code assignments}.
+	 *
+	 * @param assignments The new assignments
+	 *
+	 * @return {@code this} for method chaining.
+	 */
+	UpdateSpecification<T> reassign(List<? extends Assignment<? super T>> assignments);
+
 	@Override
 	UpdateSpecification<T> restrict(Restriction<? super T> restriction);
 

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

@@ -55,6 +55,7 @@ public enum MutationType {
 	}
 
 	final List<BiConsumer<SqmDeleteOrUpdateStatement<T>, SqmRoot<T>>> specifications = new ArrayList<>();
+
 	private final String hql;
 	private final Class<T> mutationTarget;
 	private final SqmDeleteOrUpdateStatement<T> deleteOrUpdateStatement;

hibernate-core/src/main/java/org/hibernate/query/specification/internal/UpdateSpecificationImpl.java

@@ -11,6 +11,8 @@
 import org.hibernate.query.specification.UpdateSpecification;
 import org.hibernate.query.sqm.tree.update.SqmUpdateStatement;
 
+import java.util.List;
+
 /**
  * @author Gavin King
  */
@@ -39,6 +41,23 @@ public UpdateSpecification<T> assign(Assignment<? super T> assignment) {
 		return this;
 	}
 
+	@Override
+	public UpdateSpecification<T> reassign(List<? extends Assignment<? super T>> assignments) {
+		specifications.add( (sqmStatement, mutationTargetRoot) -> {
+			if ( sqmStatement instanceof SqmUpdateStatement<T> sqmUpdateStatement ) {
+				final var setClause = sqmUpdateStatement.getSetClause();
+				if ( setClause != null ) {
+					setClause.clearAssignments();
+				}
+				assignments.forEach( assignment -> assignment.apply( sqmUpdateStatement ) );
+			}
+			else {
+				throw new IllegalStateException( "Delete query cannot perform assignment" );
+			}
+		} );
+		return this;
+	}
+
 	@Override
 	public UpdateSpecification<T> restrict(Restriction<? super T> restriction) {
 		super.restrict( restriction );

hibernate-core/src/main/java/org/hibernate/query/sqm/tree/update/SqmSetClause.java

@@ -15,6 +15,8 @@
 import org.hibernate.query.sqm.tree.domain.SqmPath;
 import org.hibernate.query.sqm.tree.expression.SqmExpression;
 
+import static java.util.Collections.unmodifiableList;
+
 /**
  * @author Steve Ebersole
  */
@@ -38,7 +40,11 @@ public SqmSetClause copy(SqmCopyContext context) {
 	}
 
 	public List<SqmAssignment<?>> getAssignments() {
-		return Collections.unmodifiableList( assignments );
+		return unmodifiableList( assignments );
+	}
+
+	public void clearAssignments() {
+		assignments.clear();
 	}
 
 	public void addAssignment(SqmAssignment<?> assignment) {