快捷導(dǎo)航

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

更新時(shí)間：2021年12月07日 09:25:58 作者：一路斜陽(yáng)

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

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

在使用Spring JPA提供的方法只能進(jìn)行簡(jiǎn)單的CRUD，如果遇到復(fù)雜的情況就需要我們動(dòng)態(tài)來(lái)構(gòu)建查詢條件了。這里我們來(lái)看使用CriteriaBuilder如何來(lái)構(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()來(lái)獲取CriteriaBuilder。CriteriaBuilder可以用于創(chuàng)建CriteriaQuery、CriteriaUpdate和CriteriaDelete。除此之外類似count、max等函數(shù)也是由CriteriaBuilder來(lái)創(chuàng)建的。其中Entitymanager可以使用@PersistenceContext注解來(lái)進(jìn)行注入。

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

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

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

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

將Predicate放在 query.where中。

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

JPA CriteriaBuilder中一些運(yùn)算的使用

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

首先了解一下CriteriaBuilder的一些運(yùn)算

// 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); // 求絕對(duì)值A(chǔ)BS(expression)
  Expression<Integer> neg = cb.neg(path); // 求相反數(shù) -expression
  Expression<Integer> sqrt = cb.sqrt(cb.literal(100)); //求平方根 SQRT(expression)

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

單字段求和可以直接使用

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

前邊也說(shuō)了需求是用金額（amount）除以單價(jià)（unitPrice），然后求和，所以這時(shí)需要先用amount除以u(píng)nitPrice

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

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

cb.sum(quot)

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

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

/**
name: 方法名
returnType： 返回類型
arguments：表達(dá)式
**/
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)的方法。天無(wú)絕人之路，在看了cb.function()后發(fā)現(xiàn)，這個(gè)方法的最后一個(gè)參數(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);

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

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

您可能感興趣的文章: