JpaCriteriaUpdate.java

Go to the documentation of this file.

/*
 * Turró i Cutiller Foundation. License notice.
 * Copyright (C) 2020 Lluis Turró Cutiller <http://www.turro.org/>
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Affero General Public License for more details.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */
 
package org.turro.jpa.query;
 
import javax.persistence.criteria.CollectionJoin;
import javax.persistence.criteria.CriteriaUpdate;
import javax.persistence.criteria.Expression;
import javax.persistence.criteria.Join;
import javax.persistence.criteria.JoinType;
import javax.persistence.criteria.ListJoin;
import javax.persistence.criteria.MapJoin;
import javax.persistence.criteria.Path;
import javax.persistence.criteria.Predicate;
import javax.persistence.criteria.Root;
import javax.persistence.criteria.Selection;
import javax.persistence.criteria.SetJoin;
import javax.persistence.metamodel.SingularAttribute;
 
public class JpaCriteriaUpdate<E> {
  
  private final Class<E> persistentClass;
  private final JpaCriteria jpaCriteria;
  private CriteriaUpdate<E> criteriaUpdate;
  protected Root<E> root;
 
  public JpaCriteriaUpdate(Class<E> persistentClass, JpaCriteria jpaCriteria) {
    this.persistentClass = persistentClass;
    this.jpaCriteria = jpaCriteria;
    initCriteria();
  }
 
  public CriteriaUpdate<E> update() {
    return criteriaUpdate;
  }
 
  /* Delegate CriteriaUpdate */
  
  public <Y, X extends Y> JpaCriteriaUpdate<E> set(SingularAttribute<? super E, Y> attribute, X value) {
    criteriaUpdate.set(attribute, value);
    return this;
  }
 
  public <Y> JpaCriteriaUpdate<E> set(SingularAttribute<? super E, Y> attribute, Expression<? extends Y> value) {
    criteriaUpdate.set(attribute, value);
    return this;
  }
 
  public <Y, X extends Y> JpaCriteriaUpdate<E> set(Path<Y> attribute, X value) {
    criteriaUpdate.set(attribute, value);
    return this;
  }
 
  public <Y> JpaCriteriaUpdate<E> set(Path<Y> attribute, Expression<? extends Y> value) {
    criteriaUpdate.set(attribute, value);
    return this;
  }
 
  public JpaCriteriaUpdate<E> set(String attributeName, Object value) {
    criteriaUpdate.set(attributeName, value);
    return this;
  }
 
  public JpaCriteriaUpdate<E> where(Expression<Boolean> restriction) {
    criteriaUpdate.where(restriction);
    return this;
  }
 
  public JpaCriteriaUpdate<E> where(Predicate... restrictions) {
    criteriaUpdate.where(restrictions);
    return this;
  }
 
  public <U> JpaSubquery<U> subquery(Class<U> javaClass) {
    return new JpaSubquery<>(javaClass, criteriaUpdate.subquery(javaClass));
  }
 
  public Predicate getRestriction() {
    return criteriaUpdate.getRestriction();
  }
 
  /* Delegate Root */
  
  public Path field(String fieldName) {
    return root.get(fieldName);
  }
 
  public Selection<E> alias(String name) {
    return root.alias(name);
  }
 
  public <X, Y> Join<X, Y> join(String attributeName) {
    return root.join(attributeName);
  }
 
  public <X, Y> CollectionJoin<X, Y> joinCollection(String attributeName) {
    return root.joinCollection(attributeName);
  }
 
  public <X, Y> SetJoin<X, Y> joinSet(String attributeName) {
    return root.joinSet(attributeName);
  }
 
  public <X, Y> ListJoin<X, Y> joinList(String attributeName) {
    return root.joinList(attributeName);
  }
 
  public <X, K, V> MapJoin<X, K, V> joinMap(String attributeName) {
    return root.joinMap(attributeName);
  }
 
  public <X, Y> Join<X, Y> join(String attributeName, JoinType jt) {
    return root.join(attributeName, jt);
  }
 
  public <X, Y> CollectionJoin<X, Y> joinCollection(String attributeName, JoinType jt) {
    return root.joinCollection(attributeName, jt);
  }
 
  public <X, Y> SetJoin<X, Y> joinSet(String attributeName, JoinType jt) {
    return root.joinSet(attributeName, jt);
  }
 
  public <X, Y> ListJoin<X, Y> joinList(String attributeName, JoinType jt) {
    return root.joinList(attributeName, jt);
  }
 
  public <X, K, V> MapJoin<X, K, V> joinMap(String attributeName, JoinType jt) {
    return root.joinMap(attributeName, jt);
  }
 
  private void initCriteria() {
    criteriaUpdate = jpaCriteria.builder().createCriteriaUpdate(persistentClass);
    root = criteriaUpdate.from(persistentClass);
  }  
 
}