Close

JPA Criteria API - Update Operations

[Last Updated: Aug 11, 2020]

In Criteria API, the javax.persistence.criteria.CriteriaUpdate interface defines functionality for performing bulk update operations.

Following method of CriteriaBuilder can be used to obtain CriteriaUpdate instance.

CriteriaUpdate<T> createCriteriaUpdate(Class<T> targetEntity)

Following is the CriteriaUpdate snippet:

public interface CriteriaUpdate<T> extends CommonAbstractCriteria {
   Root<T> from(Class<T> entityClass);
   Root<T> from(EntityType<T> entity);
   Root<T> getRoot();
   //Update the value of the specified attribute.   
   <Y, X extends Y> CriteriaUpdate<T> set( SingularAttribute<? super T, Y> attribute, X value);
   <Y> CriteriaUpdate<T> set( SingularAttribute<? super T, Y> attribute, Expression<? extends Y> value);
   <Y, X extends Y> CriteriaUpdate<T> set(Path<Y> attribute, X value);
   <Y> CriteriaUpdate<T> set(Path<Y> attribute, Expression<? extends Y> value);
   CriteriaUpdate<T> set(String attributeName, Object value);
   CriteriaUpdate<T> where(Expression<Boolean> restriction);
   CriteriaUpdate<T> where(Predicate... restrictions);
}

Quick Example

    .................. 
    CriteriaUpdate<Employee> criteriaUpdate = criteriaBuilder.createCriteriaUpdate(Employee.class);
    Root<Employee> employeeRoot = criteriaUpdate.from(Employee.class);
    criteriaUpdate.set(employeeRoot.get(Employee_.salary),
            criteriaBuilder.sum(employeeRoot.get(Employee_.salary), 500d))
                  .where(criteriaBuilder.equal(employeeRoot.get(Employee_.DEPT), "IT"));
    //the equivalent JPQL
    //UPDATE Employee e SET e.salary = e.salary + 500 WHERE e.dept = 'IT'
    entityManager.getTransaction().begin();
    int i = entityManager.createQuery(criteriaUpdate).executeUpdate();
    System.out.println("Entities updated: " + i);
    entityManager.getTransaction().commit();
    ..................

Complete Example

JPA Entity

@Entity
public class Employee {
  @Id
  @GeneratedValue
  private long id;
  private String name;
  private double salary;
  private String dept;
    .............
}

Using CriteriaUpdate

public class ExampleMain {
  private static EntityManagerFactory entityManagerFactory =
          Persistence.createEntityManagerFactory("example-unit");

  public static void main(String[] args) {
      try {
          persistEmployees();
          findAllEmployees();
          updateITEmployeesSalaries();
          findAllEmployees();
          updateAllEmployeesSalaries();
          findAllEmployees();
      } finally {
          entityManagerFactory.close();
      }
  }

  private static void updateITEmployeesSalaries() {
      System.out.println("-- increasing IT employees salaries by 500 --");
      EntityManager entityManager = entityManagerFactory.createEntityManager();
      CriteriaBuilder criteriaBuilder = entityManager.getCriteriaBuilder();
      CriteriaUpdate<Employee> criteriaUpdate = criteriaBuilder.createCriteriaUpdate(Employee.class);
      Root<Employee> employeeRoot = criteriaUpdate.from(Employee.class);
      criteriaUpdate.set(employeeRoot.get(Employee_.salary),
              criteriaBuilder.sum(employeeRoot.get(Employee_.salary), 500d))
                    .where(criteriaBuilder.equal(employeeRoot.get(Employee_.DEPT), "IT"));
      //the equivalent JPQL
      //UPDATE Employee e SET e.salary = e.salary + 500 WHERE e.dept = 'IT'
      entityManager.getTransaction().begin();
      int i = entityManager.createQuery(criteriaUpdate).executeUpdate();
      System.out.println("Entities updated: " + i);
      entityManager.getTransaction().commit();
      entityManager.close();
  }

  private static void updateAllEmployeesSalaries() {
      System.out.println("-- increasing all employees salaries by different percentage --");
      EntityManager entityManager = entityManagerFactory.createEntityManager();
      CriteriaBuilder criteriaBuilder = entityManager.getCriteriaBuilder();
      CriteriaUpdate<Employee> criteriaUpdate = criteriaBuilder.createCriteriaUpdate(Employee.class);
      Root<Employee> root = criteriaUpdate.from(Employee.class);

      CriteriaBuilder.SimpleCase<String, Double> deptCase =
              criteriaBuilder.selectCase(root.get(Employee_.DEPT));
      Expression<Double> caseExpression =
              deptCase.when("IT", criteriaBuilder.prod(root.get(Employee_.SALARY), 1.20))
                      .when("Admin", criteriaBuilder.prod(root.get(Employee_.SALARY), 1.15))
                      .otherwise(criteriaBuilder
                              .prod(root.get(Employee_.SALARY), 1.12));
      criteriaUpdate.set(root.get(Employee_.salary), caseExpression);
      //the equivalent JPQL
      /*UPDATE Employee e SET e.salary =
               CASE
                WHEN e.dept = 'IT' THEN (e.salary * 1.20)
                WHEN e.dept = 'Admin' THEN (e.salary * 1.15)
                ELSE (e.salary * 1.12)
                END*/

      entityManager.getTransaction().begin();
      int rowsUpdated = entityManager.createQuery(criteriaUpdate).executeUpdate();
      System.out.println("entities Updated: " + rowsUpdated);
      entityManager.getTransaction().commit();
      entityManager.close();
  }

  public static void persistEmployees() {
      Employee employee1 = Employee.create("Diana", 2000, "IT");
      Employee employee2 = Employee.create("Rose", 3500, "Admin");
      Employee employee3 = Employee.create("Denise", 2500, "Admin");
      Employee employee4 = Employee.create("Mike", 4000, "IT");
      Employee employee5 = Employee.create("Linda", 4500, "Sales");
      EntityManager em = entityManagerFactory.createEntityManager();
      em.getTransaction().begin();
      em.persist(employee1);
      em.persist(employee2);
      em.persist(employee3);
      em.persist(employee4);
      em.persist(employee5);
      em.getTransaction().commit();
      em.close();
  }

  private static void findAllEmployees() {
      EntityManager em = entityManagerFactory.createEntityManager();
      System.out.println("-- All employees --");
      Query query = em.createQuery(
              "SELECT e FROM Employee e");
      List<Employee> resultList = query.getResultList();
      resultList.forEach(System.out::println);
      em.close();
  }
}
-- All employees --
Employee{id=1, name='Diana', salary=2000.0, dept='IT'}
Employee{id=2, name='Rose', salary=3500.0, dept='Admin'}
Employee{id=3, name='Denise', salary=2500.0, dept='Admin'}
Employee{id=4, name='Mike', salary=4000.0, dept='IT'}
Employee{id=5, name='Linda', salary=4500.0, dept='Sales'}
-- increasing IT employees salaries by 500 --
Entities updated: 2
-- All employees --
Employee{id=1, name='Diana', salary=2500.0, dept='IT'}
Employee{id=2, name='Rose', salary=3500.0, dept='Admin'}
Employee{id=3, name='Denise', salary=2500.0, dept='Admin'}
Employee{id=4, name='Mike', salary=4500.0, dept='IT'}
Employee{id=5, name='Linda', salary=4500.0, dept='Sales'}
-- increasing all employees salaries by different percentage --
entities Updated: 5
-- All employees --
Employee{id=1, name='Diana', salary=3000.0, dept='IT'}
Employee{id=2, name='Rose', salary=4024.9999999999995, dept='Admin'}
Employee{id=3, name='Denise', salary=2875.0, dept='Admin'}
Employee{id=4, name='Mike', salary=5400.0, dept='IT'}
Employee{id=5, name='Linda', salary=5040.000000000001, dept='Sales'}

Example Project

Dependencies and Technologies Used:

  • h2 1.4.199: H2 Database Engine.
  • hibernate-core 5.4.2.Final: Hibernate's core ORM functionality.
    Implements javax.persistence:javax.persistence-api version 2.2
  • hibernate-jpamodelgen 5.4.2.Final: Annotation Processor to generate JPA 2 static metamodel classes.
  • JDK 1.8
  • Maven 3.5.4

JPA - Update Operation in Criteria API Select All Download
  • criteria-api-update-statement
    • src
      • main
        • java
          • com
            • logicbig
              • example
                • ExampleMain.java
          • resources
            • META-INF

    See Also