SpringData Jpa 之 Specification 构建子查询

一些简单的查询条件使用 SpringData Jpa 提供的查询方法和按命名规则匹配的方法就能完成，当面对复杂查询时，需要使用 Specification 来动态构建查询条件。

使用 Specification 构建子查询 ：

private Specification<Post> buildSpec(@NonNull ParentParam parentParam) {
    
    return (parentRoot, criteriaQuery, criteriaBuilder) -> {
        List<Predicate> predicates = new LinkedList<>();

        if (parentParam.getStatus() != null) {
            predicates.add(criteriaBuilder.equal(parentRoot.get("status"), parentParam.getStatus()));
        }
        
        if (parentParam.getId() != null) {
            Subquery<Parent> subQuery = criteriaQuery.subquery(Parent.class);
            Root<Subsidiary> subsidiaryRoot = subQuery.from(Subsidiary.class);
            subQuery.select(subsidiaryRoot.get("subId"));
            subQuery.where(
                criteriaBuilder.equal(parentRoot.get("id"), subsidiaryRoot.get("subId")),
                criteriaBuilder.equal(subsidiaryRoot.get("categoryId"), parentParam.getCategoryId()));
            predicates.add(criteriaBuilder.exists(subQuery));
        }
        
         return criteriaQuery.where(predicates.toArray(new Predicate[0])).getRestriction();
    }
}