Spring JPA使用CriteriaBuilder动态构造查询方式
作者：一路斜阳
Spring JPA使用CriteriaBuilder动态构造查询
在使用Spring JPA提供的方法只能进行简单的CRUD，如果遇到复杂的情况就需要我们动态来构建查询条件了。这里我们来看使用CriteriaBuilder如何来构造查询。
核心代码：
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();
调用entityManager.getCriteriaBuilder()来获取CriteriaBuilder。CriteriaBuilder可以用于创建CriteriaQuery、CriteriaUpdate和CriteriaDelete。除此之外类似count、max等函数也是由CriteriaBuilder来创建的。其中Entitymanager可以使用@PersistenceContext注解来进行注入。
调用criteriaBuilder.createQuery来创建CriteriaQuery。其中createQuery的参数是Query返回值类型。
调用query.from(Order.class)。参数是对应于order表的实体类，query.from类似于sql中的from语句，该方法的执行等价于sql中的from order。
调用 query.select创建映射。 query.select(criteriaBuilder.count(root.get(“id”)))等价于select count(id)。如果执行query.select(root)则等价于select *。
使用CriteriaBuilder构造查询条件Predicate，该predicate也就是在where后面的条件子句。
将Predicate放在 query.where中。
最后执行查询获取数据。
JPA CriteriaBuilder中一些运算的使用
最近使用jpa时，需要使用订单中的金额除以单价算出每个订单的数量，然后求和。找了好多资料才解决，在此整理一下。
首先了解一下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); // 求绝对值ABS(expression)
Expression<Integer> neg = cb.neg(path); // 求相反数 -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"));
算出数量后就可以使用sum求和了。
cb.sum(quot)
完事之后，还需要对数据四舍五入，需要用到mysql的round方法
在CriteriaBuilder没有找到round方法，那怎么办呢？没关系，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);
}
调用代码如下
Expression<BigDecimal> round = cb.function("round", BigDecimal.class, quot);
到这一步发现，还需要保留两位小数。可是，Expression里 没有相关的方法。天无绝人之路，在看了cb.function()后发现，这个方法的最后一个参数是可变参数，所以上边的代码改成了如下
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);
至此，完成了先求两列的商，然后对商求和的功能。
以上为个人经验，希望能给大家一个参考，也希望大家多多支持脚本之家。