如何在Spring Data JPA Specification中实现分组聚合得到Map<Status, Integer>

不用@Query注解也能实现需求，核心是结合Specification过滤逻辑和JPA Criteria API构建动态分组统计查询，以下是具体实现步骤：

1. 完善Repository层 #

让你的PersonRepository继承JpaRepository和JpaSpecificationExecutor，支持Specification和自定义Criteria查询：

import org.springframework.data.jpa.repository.JpaRepository;
import org.springframework.data.jpa.repository.JpaSpecificationExecutor;

public interface PersonRepository extends JpaRepository<Person, String>, JpaSpecificationExecutor<Person> {
}

2. 复用过滤逻辑（Specification） #

把你已掌握的性别、年龄过滤逻辑封装成可复用的Specification：

import org.springframework.data.jpa.domain.Specification;
import javax.persistence.criteria.CriteriaBuilder;
import javax.persistence.criteria.CriteriaQuery;
import javax.persistence.criteria.Predicate;
import javax.persistence.criteria.Root;
import org.apache.commons.lang3.StringUtils;

public class PersonSpecifications {
    // 性别过滤
    public static Specification<Person> hasSex(String sex) {
        return (root, query, cb) -> StringUtils.hasText(sex) ? cb.equal(root.get("sex"), sex) : cb.conjunction();
    }

    // 年龄过滤
    public static Specification<Person> hasAge(Integer age) {
        return (root, query, cb) -> age != null ? cb.equal(root.get("age"), age) : cb.conjunction();
    }
}

3. 实现分组统计Service方法 #

通过EntityManager和Criteria API构建分组统计查询，同时复用上面的过滤Specification：

import org.springframework.stereotype.Service;
import javax.persistence.EntityManager;
import javax.persistence.Tuple;
import javax.persistence.criteria.CriteriaBuilder;
import javax.persistence.criteria.CriteriaQuery;
import javax.persistence.criteria.Predicate;
import javax.persistence.criteria.Root;
import java.util.Arrays;
import java.util.Map;
import java.util.function.Function;
import java.util.stream.Collectors;

@Service
public class PersonService {

    private final EntityManager entityManager;

    // 构造函数注入EntityManager（推荐替代@Autowired）
    public PersonService(EntityManager entityManager) {
        this.entityManager = entityManager;
    }

    public Map<Status, Integer> countByStatusWithFilters(String sex, Integer age) {
        // 初始化结果Map，确保所有Status枚举值默认计数为0，避免缺失状态
        Map<Status, Integer> result = Arrays.stream(Status.values())
                .collect(Collectors.toMap(Function.identity(), s -> 0));

        CriteriaBuilder cb = entityManager.getCriteriaBuilder();
        CriteriaQuery<Tuple> query = cb.createTupleQuery();
        Root<Person> root = query.from(Person.class);

        // 组装过滤条件（复用已有的Specification）
        Specification<Person> spec = Specification.where(PersonSpecifications.hasSex(sex))
                .and(PersonSpecifications.hasAge(age));
        Predicate filterPredicate = spec.toPredicate(root, query, cb);

        // 构建分组统计：按status分组，统计记录数
        query.multiselect(root.get("status"), cb.count(root.get("id")))
                .where(filterPredicate)
                .groupBy(root.get("status"));

        // 执行查询并填充结果
        entityManager.createQuery(query).getResultList()
                .forEach(tuple -> {
                    Status status = (Status) tuple.get(0);
                    Long count = (Long) tuple.get(1);
                    result.put(status, count.intValue());
                });

        return result;
    }
}

4. 编写Controller接口 #

对外提供查询接口，返回指定格式的统计结果：

import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.RestController;
import java.util.Map;

@RestController
@RequestMapping("/persons")
public class PersonController {

    private final PersonService personService;

    public PersonController(PersonService personService) {
        this.personService = personService;
    }

    @GetMapping("/status-count")
    public Map<Status, Integer> getStatusCount(
            @RequestParam(required = false) String sex,
            @RequestParam(required = false) Integer age) {
        return personService.countByStatusWithFilters(sex, age);
    }
}

关键说明 #

• 初始化结果Map时遍历所有Status枚举值，确保即使某个状态没有匹配数据，也会返回0，完全符合你需要的返回格式。
• 利用Specification复用已有的过滤逻辑，避免重复代码，同时保持查询的动态性。
• 使用Criteria API构建分组统计，全程无需编写原生SQL或JPQL，完全符合不使用@Query的要求。

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