如何在QueryDSL中使用LEAST/GREATEST函数计算多表达式最值

核心差异说明 #

• 此处使用的LEAST/GREATEST为行内横向比较函数，作用是对同一行内的多个入参表达式比较取最值，和用于多行聚合的MIN/MAX聚合运算符语义完全不同
• JPA 原生CriteriaBuilder提供的least()方法仅支持单个入参，无法实现多表达式横向比较的需求

规范实现方式 #

基于QueryDSL官方提供的模板表达式能力封装即可，完全符合QueryDSL类型安全规范，支持任意数量的Double类型表达式入参：

import com.querydsl.core.types.dsl.Expressions;
import com.querydsl.core.types.dsl.NumberExpression;

public class QueryDslPostgresExtensions {
    @SafeVarargs
    public static NumberExpression<Double> least(NumberExpression<Double>... exprs) {
        if (exprs == null || exprs.length < 2) {
            throw new IllegalArgumentException("LEAST函数至少需要传入2个待比较表达式");
        }
        StringBuilder tpl = new StringBuilder("least(");
        for (int i = 0; i < exprs.length; i++) {
            tpl.append(i > 0 ? ", {" : "{").append(i).append("}");
        }
        tpl.append(")");
        return Expressions.numberTemplate(Double.class, tpl.toString(), (Object[]) exprs);
    }

    @SafeVarargs
    public static NumberExpression<Double> greatest(NumberExpression<Double>... exprs) {
        if (exprs == null || exprs.length < 2) {
            throw new IllegalArgumentException("GREATEST函数至少需要传入2个待比较表达式");
        }
        StringBuilder tpl = new StringBuilder("greatest(");
        for (int i = 0; i < exprs.length; i++) {
            tpl.append(i > 0 ? ", {" : "{").append(i).append("}");
        }
        tpl.append(")");
        return Expressions.numberTemplate(Double.class, tpl.toString(), (Object[]) exprs);
    }
}

使用示例 #

通用多值取最小场景 #

和预期写法完全一致：

NumberExpression<Double> minimal = least(exp1, exp2, expN);
query.select(minimal);

空间点最小距离计算场景 #

直接传入两个距离计算表达式即可：

JPAQuery<Tuple> query = new JPAQuery<>(this.entityManager);
NumberExpression<Double> minDistance = least(
        qEntity.point.distance(pointA),
        qEntity.point.distance(pointB)
);
query.select(minDistance);

补充说明 #

• 该实现直接生成标准PostgreSQL函数调用语句，不会和聚合函数产生语义混淆
• 模板表达式是QueryDSL官方支持的自定义函数扩展方式，天然做了参数转义，无SQL注入风险
• 若需要支持Integer、BigDecimal等其他数值类型，只需调整方法泛型定义和对应类型参数即可

内容的提问来源于stack exchange，提问作者Paul C