JPA Criteria 中 DATEDIFF 函數的使用

項目中有個查詢使用了很多 left join 語句，查詢出來的總記錄數有二十幾萬條，而每次查詢時就只要查詢某一天的數據，及時這樣分頁查詢時依舊很慢。
優化時把條件一個一個刪掉，發現日期條件比較那里消耗了很多時間。sql 里面的日期比較是使用大小比較的，改成 DATEDIFF 函數后速度明顯提升了。

項目查詢時使用的是 Spring-Data-JPA Criteria，其中 DATEDIFF 查詢代碼如下：

Specification<Xxx> specification = (root, query, cb) -> {
    List<Predicate> predicates = new ArrayList<>();
    ...
    Expression<String> day = new MyFunctionExpression(null, String.class, "DAY");
    Expression<Integer> diff = cb.function("DATEDIFF", Integer.class, day, cb.literal(paramDTO.getReportDate()), root.get("createdDate"));
    predicates.add(cb.equal(diff, 0));
    ...
    if (predicates.size() > 0) {
        Predicate[] pre = new Predicate[predicates.size()];
        return query.where(predicates.toArray(pre)).getRestriction();
    }
    return null;
};

import org.hibernate.query.criteria.internal.CriteriaBuilderImpl;
import org.hibernate.query.criteria.internal.compile.RenderingContext;
import org.hibernate.query.criteria.internal.expression.function.BasicFunctionExpression;

public class MyFunctionExpression extends BasicFunctionExpression {
    public MyFunctionExpression(CriteriaBuilderImpl criteriaBuilder, Class javaType, String functionName) {
        super(criteriaBuilder, javaType, functionName);
    }

    @Override
    public String render(RenderingContext renderingContext) {
        return getFunctionName();
    }

}

剛開始參考網上寫成：

Expression<String> day = cb.literal("day");
Expression<Integer> diff = cb.function("DATEDIFF", Integer.class, day, cb.literal(paramDTO.getReportDate()), root.get("createdDate"));
predicates.add(cb.equal(diff, 0));

運行時會報：

為 datediff 指定的參數 1 無效。

原因是上面的寫法，sql 解析時會把 "day" 字符串放到 DATEDIFF 函數的第一個參數位置上，即 DATEDIFF('day', x, y)，而不是 DATEDIFF(day, x, y) 這種形式。

參考：

http://jpwh.org/examples/jpwh2/jpwh-2e-examples-20151103/examples/src/test/java/org/jpwh/test/querying/criteria/Restriction.java

How to call SQL native function, requiring constant argument, with JPA?