org.hibernate.sql.ast.tree.delete.DeleteStatement Maven / Gradle / Ivy

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

Hibernate O/RM implementation of the Jakarta Persistence specification

The newest version!

/*
 * 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.delete;

import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;

import org.hibernate.sql.ast.SqlAstWalker;
import org.hibernate.sql.ast.spi.SqlAstHelper;
import org.hibernate.sql.ast.tree.AbstractMutationStatement;
import org.hibernate.sql.ast.tree.cte.CteStatement;
import org.hibernate.sql.ast.tree.expression.ColumnReference;
import org.hibernate.sql.ast.tree.from.TableReference;
import org.hibernate.sql.ast.tree.predicate.Junction;
import org.hibernate.sql.ast.tree.predicate.Predicate;

/**
 * @author Steve Ebersole
 */
public class DeleteStatement extends AbstractMutationStatement {

	private final Predicate restriction;

	public DeleteStatement(TableReference targetTable, Predicate restriction) {
		super( targetTable );
		this.restriction = restriction;
	}

	public DeleteStatement(
			TableReference targetTable,
			Predicate restriction,
			List returningColumns) {
		super( new LinkedHashMap<>(), targetTable, returningColumns );
		this.restriction = restriction;
	}

	public DeleteStatement(
			boolean withRecursive,
			Map cteStatements,
			TableReference targetTable,
			Predicate restriction,
			List returningColumns) {
		super( cteStatements, targetTable, returningColumns );
		this.restriction = restriction;
		setWithRecursive( withRecursive );
	}

	public Predicate getRestriction() {
		return restriction;
	}

	public static class DeleteStatementBuilder {
		private final TableReference targetTable;
		private Predicate restriction;

		public DeleteStatementBuilder(TableReference targetTable) {
			this.targetTable = targetTable;
		}

		public DeleteStatementBuilder addRestriction(Predicate restriction) {
			this.restriction = SqlAstHelper.combinePredicates( this.restriction, restriction );
			return this;
		}

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

		public DeleteStatement createDeleteStatement() {
			return new DeleteStatement(
					targetTable,
					restriction != null ? restriction : new Junction( Junction.Nature.CONJUNCTION )
			);
		}
	}

	@Override
	public void accept(SqlAstWalker walker) {
		walker.visitDeleteStatement( this );
	}
}