Spring?JPA使用CriteriaBuilder動態(tài)構(gòu)造查詢方式

更新時間：2021年12月07日 09:25:58 作者：一路斜陽

這篇文章主要介紹了Spring?JPA使用CriteriaBuilder動態(tài)構(gòu)造查詢方式，具有很好的參考價值，希望對大家有所幫助。如有錯誤或未考慮完全的地方，望不吝賜教

Spring JPA使用CriteriaBuilder動態(tài)構(gòu)造查詢

在使用Spring JPA提供的方法只能進行簡單的CRUD，如果遇到復雜的情況就需要我們動態(tài)來構(gòu)建查詢條件了。這里我們來看使用CriteriaBuilder如何來構(gòu)造查詢。

核心代碼：

 CriteriaBuilder criteriaBuilder = entityManager.getCriteriaBuilder();
 CriteriaQuery<Long> query = criteriaBuilder.createQuery(Long.class);
 Root<Order> root = query.from(Order.class);
 query.select(criteriaBuilder.count(root.get("id")));
 Predicate predicate = criteriaBuilder.equal(root.get("id"), 1);
 query.where(predicate);
 Long singleResult = entityManager.createQuery(query).getSingleResult();

調(diào)用entityManager.getCriteriaBuilder()來獲取CriteriaBuilder。CriteriaBuilder可以用于創(chuàng)建CriteriaQuery、CriteriaUpdate和CriteriaDelete。除此之外類似count、max等函數(shù)也是由CriteriaBuilder來創(chuàng)建的。其中Entitymanager可以使用@PersistenceContext注解來進行注入。

調(diào)用criteriaBuilder.createQuery來創(chuàng)建CriteriaQuery。其中createQuery的參數(shù)是Query返回值類型。

調(diào)用query.from(Order.class)。參數(shù)是對應(yīng)于order表的實體類，query.from類似于sql中的from語句，該方法的執(zhí)行等價于sql中的from order。

調(diào)用 query.select創(chuàng)建映射。 query.select(criteriaBuilder.count(root.get(“id”)))等價于select count(id)。如果執(zhí)行query.select(root)則等價于select *。

使用CriteriaBuilder構(gòu)造查詢條件Predicate，該predicate也就是在where后面的條件子句。

將Predicate放在 query.where中。

最后執(zhí)行查詢獲取數(shù)據(jù)。

JPA CriteriaBuilder中一些運算的使用

最近使用jpa時，需要使用訂單中的金額除以單價算出每個訂單的數(shù)量，然后求和。找了好多資料才解決，在此整理一下。

首先了解一下CriteriaBuilder的一些運算

// Create path and parameter expressions:
  Expression<Integer> path = country.get("population");
  Expression<Integer> param = cb.parameter(Integer.class);
  // Addition (+)
  Expression<Integer> sum1 = cb.sum(path, param); // expression + expression
  Expression<Integer> sum2 = cb.sum(path, 1000); // expression + number
  Expression<Integer> sum3 = cb.sum(1000, path); // number + expression
  // Subtraction (-)
  Expression<Integer> diff1 = cb.diff(path, param); // expression - expression
  Expression<Integer> diff2 = cb.diff(path, 1000); // expression - number
  Expression<Integer> diff3 = cb.diff(1000, path); // number - expression
  // Multiplication (*)
  Expression<Integer> prod1 = cb.prod(path, param); // expression * expression
  Expression<Integer> prod2 = cb.prod(path, 1000); // expression * number
  Expression<Integer> prod3 = cb.prod(1000, path); // number * expression
  // Division (/)
  Expression<Integer> quot1 = cb.quot(path, param); // expression / expression
  Expression<Integer> quot2 = cb.quot(path, 1000); // expression / number
  Expression<Integer> quot3 = cb.quot(1000, path); // number / expression
  // Modulo (%)
  Expression<Integer> mod1 = cb.mod(path, param); // expression % expression
  Expression<Integer> mod2 = cb.mod(path, 1000); // expression % number
  Expression<Integer> mod3 = cb.mod(1000, path); // number % expression
  // Math(abs, exp, sqrt)
  Expression<Integer> abs = cb.abs(param); // 求絕對值A(chǔ)BS(expression)
  Expression<Integer> neg = cb.neg(path); // 求相反數(shù) -expression
  Expression<Integer> sqrt = cb.sqrt(cb.literal(100)); //求平方根 SQRT(expression)

由于CriteriaBuilder提供的加減乘除方法的名字和平常使用的不太一樣，所以用了好久才找出來。

單字段求和可以直接使用

CriteriaBuilder cb = em.getCriteriaBuilder();
Expression<Number> sum = cb.sum(root.get(字段名)).alias(別名)

前邊也說了需求是用金額（amount）除以單價（unitPrice），然后求和，所以這時需要先用amount除以unitPrice

Expression<Number> quot = cb.quot(root.get("amount"), root.get("unitPrice"));

算出數(shù)量后就可以使用sum求和了。

cb.sum(quot)

完事之后，還需要對數(shù)據(jù)四舍五入，需要用到mysql的round方法

在CriteriaBuilder沒有找到round方法，那怎么辦呢？沒關(guān)系，CriteriaBuilder還提供了function方法，在function方法里可以直接傳方法名進去

/**
name: 方法名
returnType： 返回類型
arguments：表達式
**/
public <T> Expression<T> function(String name, Class<T> returnType, Expression... arguments) {
        return new ParameterizedFunctionExpression(this, returnType, name, arguments);
    }

調(diào)用代碼如下

Expression<BigDecimal> round = cb.function("round", BigDecimal.class, quot);

到這一步發(fā)現(xiàn)，還需要保留兩位小數(shù)。可是，Expression里沒有相關(guān)的方法。天無絕人之路，在看了cb.function()后發(fā)現(xiàn)，這個方法的最后一個參數(shù)是可變參數(shù)，所以上邊的代碼改成了如下

List<Integer> list = new ArrayList<>();
list.add(1);
list.add(2);
Expression<Integer> size = cb.size(list);
Expression<BigDecimal> round = cb.function("round", BigDecimal.class, quot,size);

至此，完成了先求兩列的商，然后對商求和的功能。

以上為個人經(jīng)驗，希望能給大家一個參考，也希望大家多多支持腳本之家。

您可能感興趣的文章: