如何用JPA CriteriaBuilder实现多字段相加运算？

当CriteriaBuilder.sum()仅支持2个参数无法满足多字段求和时，你可以通过嵌套累加的方式实现类似SQL中a + b + c的运算逻辑，以下是两种可行方案：

方案一：嵌套使用sum() #

利用sum()返回的Expression作为下一次sum()的参数，逐步累加多个字段：

// 先累加前两个字段
Expression<Number> sumTwo = cb.sum(functionListToAdd.get(0), functionListToAdd.get(1));
// 再将结果与第三个字段累加
Expression<Number> totalSum = cb.sum(sumTwo, functionListToAdd.get(2));

// 应用min和max聚合函数
select.add(cb.min(totalSum));
select.add(cb.max(totalSum));
// 添加cast('units' as text)对应的逻辑
select.add(cb.literal("units").as(String.class));

cq.multiselect(select);
em.createQuery(cq).getResultList();

方案二：使用plus()方法（更贴近SQL"+"语义） #

CriteriaBuilder.plus()同样支持两个参数，嵌套使用后生成的SQL与直接写+运算完全一致：

// 链式嵌套实现多字段相加
Expression<Number> total = cb.plus(
    cb.plus(functionListToAdd.get(0), functionListToAdd.get(1)),
    functionListToAdd.get(2)
);

// 绑定聚合函数和常量字段
select.add(cb.min(total));
select.add(cb.max(total));
select.add(cb.literal("units").as(String.class));

cq.multiselect(select);
em.createQuery(cq).getResultList();

处理更多字段的通用写法 #

如果需要累加的字段数量不确定（比如3个以上），可以用循环批量构建累加表达式：

// 初始化第一个字段为初始值
Expression<Number> total = functionListToAdd.get(0);

// 循环累加剩余字段
for (int i = 1; i < functionListToAdd.size(); i++) {
    total = cb.plus(total, functionListToAdd.get(i));
}

// 后续聚合函数逻辑不变
select.add(cb.min(total));
select.add(cb.max(total));
select.add(cb.literal("units").as(String.class));

以上方案最终生成的SQL会和你提供的示例完全一致，实现多字段相加后取min、max的需求。

内容的提问来源于stack exchange，提问作者Alexandre DONAZZAN