是否可以将 count(*) 和 groupBy 添加到 Spring Data Jpa 规范？

Question

我正在使用 Spring Boot 和 Spring Data Jpa 编写应用程序。

我有一个使用 RSQL（我使用 this 库）和规范的方法，通过 where 子句对 db 进行过滤器调用。

我的仓库：

public interface BaseRepository<M extends BaseModel> extends JpaRepository<M, Long>, JpaSpecificationExecutor<M>{

}

在服务中我调用了存储库：

String filter = "createdDate > 2019-09-04T11:52:59.449";
return repository.findAll(toSpecification(filter), pageable).getContent();  

所以我使用上面的库来生成规范并将其传递给存储库。

现在我需要按某些字段对结果进行分组。

如何在规范中添加 count 和 groupBy 的问题？或者我可能需要创建一个谓词，在其中定义这个 count 和 groupBy （不知何故）并以此制定规范？有什么更好的方法？

更新 #0

我已经设法将一个组按部分添加到结果 sql 查询中。我创建了一个新的规范对象，在其中按参数定义所有组，然后将此规范与原始规范结合起来。

String filter = "createdDate > 2019-09-04T11:52:59.449";

Specification<M> groupBySpec = new Specification<M>() {

            @Override
            public Predicate toPredicate(Root<M> root, CriteriaQuery<?> query, CriteriaBuilder criteriaBuilder) {

                List<Expression<?>> groupByParams = new ArrayList<>();

                groupByParams.add(root.get("param1"));
                groupByParams.add(root.get("param2"));

                query.groupBy(groupByParams);

                return query.getGroupRestriction();
            }
        };


return repository.findAll(groupBySpec.and(toSpecification(filter)), pageable).getContent();     

问题是它总是选择所有列，但我只需要选择那些按部分分组的列。

结果我得到一个 sql 异常：ERROR: column "columnName" must appear in the GROUP BY clause or be used in an aggregate function。

我尝试过像 criteriaQuery.multiselect(root.get("column1"), root.get("column2")).distinct(true).getGroupRestriction(); 这样使用多选，但它还是选择了所有列。

如果我使用 @Query("query") 来定义要选择的列，但我需要它使用 Criteria Api/Specification 才能工作。

Answer 1

您可以创建自己的 GroupBySpecification 类，将 groupBy 表达式添加到其他规范：

public class GroupBySpecification implements Specification {
  Specification original;
  Parameters parameters; // whatever data needed for calculating groupBy expression

  public GroupBySpecification(Specification original, Parameters parameters){
      this.original = original;
      this.parameters = parameters;
  }
  public Predicate toPredicate(Root<T> root, CriteriaQuery<?> query, CriteriaBuilder criteriaBuilder) {

      Expression<?> expression = ...; // whatever logic needed to be used on parameters...
      criteriaBuilder.groupBy(expression);
      return original.toPredicate(root, query, criteriabuilder);
  }

}

那么你可以这样使用它：

String filter = "createdDate > 2019-09-04T11:52:59.449";
return repository.findAll(new GroupBySpecification(toSpecification(filter),parameters),
   pageable).getContent();  

如果您需要 count，则调用相应的存储库方法：

repository.count(new GroupBySpecification(toSpecification(filter),parameters));