如何用Criteria API实现PostgreSQL日期算术运算？是否需自定义Expression？

优先方案：使用内置函数实现 #

不需要自定义Expression，JPA Criteria API（以Hibernate为例）已经支持通过CriteriaBuilder.function()调用PostgreSQL原生日期函数，完全能实现你要的逻辑。

实现示例1：调用PostgreSQL原生date函数 #

直接利用PostgreSQL的date函数处理日期偏移，代码如下：

CriteriaBuilder cb = entityManager.getCriteriaBuilder();
CriteriaQuery<Appointment> query = cb.createQuery(Appointment.class);
Root<Appointment> root = query.from(Appointment.class);

// 目标对比日期
LocalDate targetDate = LocalDate.of(2022, 10, 21);

// 构建「日期+2天」的表达式
Expression<LocalDate> datePlusTwoDays = cb.function(
    "date", LocalDate.class,
    root.get("date"),
    cb.literal("2 days")
);

// 添加查询条件
query.where(cb.greaterThan(datePlusTwoDays, targetDate));

// 执行查询
List<Appointment> result = entityManager.createQuery(query).getResultList();

实现示例2：使用JPA标准date_add函数 #

如果你的JPA实现（如Hibernate 5.3+）支持JPA 2.2标准日期函数，也可以用更通用的写法：

Expression<LocalDate> datePlusTwoDays = cb.function(
    "date_add", LocalDate.class,
    root.get("date"),
    cb.literal(2),
    cb.literal("DAY")
);

query.where(cb.greaterThan(datePlusTwoDays, targetDate));

自定义Expression的问题修复 #

如果坚持要自己实现自定义Expression，你写的代码有几个语法错误：

1. SQL渲染时多套了不必要的双引号，导致生成的SQL格式错误
2. ChronoUnit的toString()返回大写值（如DAYS），但PostgreSQL要求interval单位为小写（days）
3. 未正确注册日期表达式的参数

修复后的代码如下：

private class DateArithmeticExpression<T extends Date> extends ExpressionImpl<T> {

    private final int value;
    private final ChronoUnit unit;
    private final BinaryArithmeticOperation.Operation operator;
    private final Expression<T> date;

    public DateArithmeticExpression(CriteriaBuilderImpl cb, Class<T> clazz, Expression<T> date, BinaryArithmeticOperation.Operation operator, int value, ChronoUnit unit) {
        super(cb, clazz);
        this.value = value;
        this.unit = unit;
        this.operator = operator;
        this.date = date;
    }

    @Override
    public void registerParameters(ParameterRegistry registry) {
        // 注册日期表达式的参数
        Helper.possibleParameter(date, registry);
    }

    @Override
    public String render(RenderingContext renderingContext) {
        String op = switch (operator) {
            case ADD -> "+";
            case SUBTRACT -> "-";
            default -> throw new UnsupportedOperationException("不支持的操作符: " + operator);
        };
        String renderedDateExpr = ((Renderable) this.date).render(renderingContext);
        // 把ChronoUnit转为小写单位字符串
        String unitStr = unit.toString().toLowerCase();
        return "%s %s interval '%d %s'".formatted(renderedDateExpr, op, value, unitStr);
    }
}

使用修复后的类构建条件：

CriteriaBuilderImpl cb = (CriteriaBuilderImpl) entityManager.getCriteriaBuilder();
Root<Appointment> root = query.from(Appointment.class);

Expression<Date> datePlusTwoDays = new DateArithmeticExpression<>(
    cb, Date.class,
    root.get("date"),
    BinaryArithmeticOperation.Operation.ADD,
    2,
    ChronoUnit.DAYS
);
query.where(cb.greaterThan(datePlusTwoDays, Date.valueOf("2022-10-21")));

总结 #

优先使用内置的function方法调用数据库原生或标准JPA日期函数，代码更简洁、维护性更强。只有当内置方法无法满足特殊需求时，再考虑自定义Expression，同时注意SQL渲染的语法正确性。

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