Spring Data JPA使用Criteria实现指定部门分组分页查询员工列表的方案咨询

嗨，我来帮你理清这个问题的解决思路，你遇到的核心问题其实有两个：一是原SQL本身的语法不符合标准规范，二是如何用Criteria API实现分组+分页的需求，同时避免在内存中处理大量数据。

先说说你原SQL的问题 #

你提到的查询语句 SELECT e.dept, e FROM Employee e GROUP BY e.dept HAVING e.dept IN ('IT', 'Admin') 其实不符合标准SQL规则。因为在GROUP BY指定按dept分组后，SELECT列表中的列要么是分组列（也就是e.dept），要么是聚合函数（比如COUNT(e.id)），直接选择整个Employee实体是不被允许的——除非你的数据库开启了非标准的宽松模式（比如MySQL关闭ONLY_FULL_GROUP_BY），但这绝对不是推荐的做法，也是你没法直接在Spring Data JPA仓库中使用这个查询的核心原因。

针对你的需求，分两种常见场景给出解决方案 #

你的核心需求应该是：获取属于IT、Admin部门的员工数据，按部门分组展示，同时支持分页，且尽量不在Java内存中处理全量数据。我分两种业务场景给你具体的实现方案：

场景1：分页获取员工列表，结果按部门自然聚合（分页对象是员工）

如果你的需求是分页展示员工（比如每页返回20条员工数据），同时希望同一部门的员工在分页结果中集中在一起，那最优的方案是先通过Criteria API完成部门过滤、排序和分页，再对分页后的少量数据按部门分组——因为分页后的数据量很小，用Stream分组完全不会有性能问题：

首先，实现自定义的仓库逻辑：

@Repository
public class EmployeeCustomRepositoryImpl implements EmployeeCustomRepository {

    @PersistenceContext
    private EntityManager entityManager;

    @Override
    public Page<Employee> getPaginatedEmployeesByDept(Pageable pageable) {
        CriteriaBuilder cb = entityManager.getCriteriaBuilder();
        CriteriaQuery<Employee> cq = cb.createQuery(Employee.class);
        Root<Employee> employeeRoot = cq.from(Employee.class);

        // 1. 过滤出IT和Admin部门的员工
        Predicate deptFilter = employeeRoot.get("dept").in("IT", "Admin");
        cq.where(deptFilter);

        // 2. 按部门排序，确保同一部门的员工在分页结果里集中展示
        cq.orderBy(cb.asc(employeeRoot.get("dept")), cb.asc(employeeRoot.get("name")));

        // 3. 执行分页查询
        TypedQuery<Employee> query = entityManager.createQuery(cq);
        query.setFirstResult((int) pageable.getOffset());
        query.setMaxResults(pageable.getPageSize());
        List<Employee> paginatedEmployees = query.getResultList();

        // 4. 统计符合条件的总员工数
        CriteriaQuery<Long> countQuery = cb.createQuery(Long.class);
        countQuery.select(cb.count(countQuery.from(Employee.class)));
        countQuery.where(deptFilter);
        long totalCount = entityManager.createQuery(countQuery).getSingleResult();

        // 如果需要把分页后的员工按部门分组成Map，直接用下面的代码即可
        // Map<String, List<Employee>> groupedEmployees = paginatedEmployees.stream()
        //         .collect(Collectors.groupingBy(Employee::getDept));

        // 返回Spring Data的Page对象
        return new PageImpl<>(paginatedEmployees, pageable, totalCount);
    }
}

然后定义对应的仓库接口：

// 自定义仓库接口
public interface EmployeeCustomRepository {
    Page<Employee> getPaginatedEmployeesByDept(Pageable pageable);
}

// 主仓库接口，继承JpaRepository和自定义接口
public interface EmployeeRepository extends JpaRepository<Employee, Long>, EmployeeCustomRepository {
}

场景2：分页获取部门（仅IT/Admin），每个部门携带对应员工列表（分页对象是部门）

如果你的需求是分页展示部门（比如每页返回5个部门），每个部门下显示该部门的所有员工，那可以通过“先分页部门，再批量查询对应员工”的方式实现，确保所有数据处理都在数据库层面完成：

首先，定义一个DTO来接收分组后的结果：

public class DeptEmployeeGroupDTO {
    private String dept;
    private List<Employee> employees;

    // 注意：构造方法的参数顺序必须和Criteria查询中SELECT的列顺序一致
    public DeptEmployeeGroupDTO(String dept, List<Employee> employees) {
        this.dept = dept;
        this.employees = employees;
    }

    // getter和setter方法
    public String getDept() { return dept; }
    public void setDept(String dept) { this.dept = dept; }
    public List<Employee> getEmployees() { return employees; }
    public void setEmployees(List<Employee> employees) { this.employees = employees; }
}

然后实现自定义仓库逻辑：

@Repository
public class EmployeeCustomRepositoryImpl implements EmployeeCustomRepository {

    @PersistenceContext
    private EntityManager entityManager;

    @Override
    public Page<DeptEmployeeGroupDTO> getPaginatedDeptEmployeeGroups(Pageable pageable) {
        CriteriaBuilder cb = entityManager.getCriteriaBuilder();

        // 1. 先分页查询符合条件的部门（仅IT和Admin）
        CriteriaQuery<String> deptQuery = cb.createQuery(String.class);
        Root<Employee> deptRoot = deptQuery.from(Employee.class);
        deptQuery.select(deptRoot.get("dept")).distinct(true);
        deptQuery.where(deptRoot.get("dept").in("IT", "Admin"));
        deptQuery.orderBy(cb.asc(deptRoot.get("dept")));

        TypedQuery<String> deptTypedQuery = entityManager.createQuery(deptQuery);
        deptTypedQuery.setFirstResult((int) pageable.getOffset());
        deptTypedQuery.setMaxResults(pageable.getPageSize());
        List<String> paginatedDepts = deptTypedQuery.getResultList();

        // 2. 获取符合条件的部门总数
        CriteriaQuery<Long> deptCountQuery = cb.createQuery(Long.class);
        deptCountQuery.select(cb.countDistinct(deptCountQuery.from(Employee.class).get("dept")));
        deptCountQuery.where(deptCountQuery.from(Employee.class).get("dept").in("IT", "Admin"));
        long totalDepts = entityManager.createQuery(deptCountQuery).getSingleResult();

        // 3. 批量查询这些分页部门对应的所有员工
        CriteriaQuery<Employee> empQuery = cb.createQuery(Employee.class);
        Root<Employee> empRoot = empQuery.from(Employee.class);
        empQuery.where(empRoot.get("dept").in(paginatedDepts));
        List<Employee> allMatchedEmployees = entityManager.createQuery(empQuery).getResultList();

        // 4. 组装成DTO列表
        Map<String, List<Employee>> empGroupMap = allMatchedEmployees.stream()
                .collect(Collectors.groupingBy(Employee::getDept));
        List<DeptEmployeeGroupDTO> result = paginatedDepts.stream()
                .map(dept -> new DeptEmployeeGroupDTO(dept, empGroupMap.getOrDefault(dept, Collections.emptyList())))
                .collect(Collectors.toList());

        // 返回分页结果
        return new PageImpl<>(result, pageable, totalDepts);
    }
}

几个关键注意点 #

• 不要用非标准SQL：永远不要依赖数据库的宽松模式来执行不符合标准的GROUP BY查询，这会导致你的代码在切换数据库时出现兼容性问题。
• 性能优化的核心：避免加载全量数据到内存，通过分页把数据量控制在合理范围内，即使后续用Stream分组也完全没问题。
• 分页对象要明确：一定要搞清楚你是对员工分页还是对部门分页，这会直接决定你的实现方案。

内容来源于stack exchange