org.hibernate.sql.ast.tree.update.UpdateStatement Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of hibernate-core Show documentation

Hibernate's core ORM functionality

There is a newer version: 7.0.0.Beta1

/*
 * Hibernate, Relational Persistence for Idiomatic Java
 *
 * License: GNU Lesser General Public License (LGPL), version 2.1 or later
 * See the lgpl.txt file in the root directory or http://www.gnu.org/licenses/lgpl-2.1.html
 */
package org.hibernate.sql.ast.tree.update;

import java.util.ArrayList;
import java.util.List;

import org.hibernate.sql.ast.spi.SqlAstTreeHelper;
import org.hibernate.sql.ast.tree.MutationStatement;
import org.hibernate.sql.ast.tree.cte.CteConsumer;
import org.hibernate.sql.ast.tree.from.TableReference;
import org.hibernate.sql.ast.tree.predicate.Predicate;

/**
 * @author Steve Ebersole
 */
public class UpdateStatement implements MutationStatement, CteConsumer {
	private final TableReference targetTable;
	private final List assignments;
	private final Predicate restriction;

	public UpdateStatement(
			TableReference targetTable,
			List assignments,
			Predicate restriction) {
		this.targetTable = targetTable;
		this.assignments = assignments;
		this.restriction = restriction;
	}

	public TableReference getTargetTable() {
		return targetTable;
	}

	public List getAssignments() {
		return assignments;
	}

	public Predicate getRestriction() {
		return restriction;
	}


	public static class UpdateStatementBuilder {
		private final TableReference targetTableRef;
		private List assignments;
		private Predicate restriction;

		public UpdateStatementBuilder(TableReference targetTableRef) {
			this.targetTableRef = targetTableRef;
		}

		public UpdateStatementBuilder addAssignment(Assignment assignment) {
			if ( assignments == null ) {
				assignments = new ArrayList<>();
			}
			assignments.add( assignment );

			return this;
		}

		public UpdateStatementBuilder addRestriction(Predicate restriction) {
			this.restriction = SqlAstTreeHelper.combinePredicates( this.restriction, restriction );
			return this;
		}

		public UpdateStatementBuilder setRestriction(Predicate restriction) {
			this.restriction = restriction;
			return this;
		}

//		public SqlAstUpdateDescriptor createUpdateDescriptor() {
//			return new SqlAstUpdateDescriptorImpl(
//					createUpdateAst(),
//					Collections.singleton( targetTableRef.getTable().getTableExpression() )
//			);
//		}

		public UpdateStatement createUpdateAst() {
			if ( assignments == null || assignments.isEmpty() ) {
				return null;
			}

			return new UpdateStatement( targetTableRef, assignments, restriction );
		}
	}
}