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

Question

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

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();     

我尝试使用multiselectCriteriaQuery.multiselect(root.get("column1“）,root.get("column2”））.distinct(true).getGroupRestriction()；，但它还是选择了所有列。

如果我使用@query（“query”）来定义要选择的列，它就可以工作，但是我需要它使用criteries 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();  

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