如何拆分JPA Specification并结合分页实现联合查询？

针对你需要分别过滤教师、护理员并返回合并分页结果的需求，提供两种可行方案：

方案一：组合Specification一次性查询（推荐） #

如果两类员工的过滤逻辑可以通过OR组合，直接用Spring Data JPA的原生分页能力即可，无需手动合并。

步骤1：定义两类员工的精确过滤Specification #

// 教师精确过滤规则（示例：匹配类型为TEACHER且name为"张三"）
Specification<EmployeeEntity> teacherSpec = (root, query, cb) -> cb.and(
    cb.equal(root.get("type"), EmployeeType.TEACHER),
    cb.equal(root.get("name"), "张三")
);

// 护理员精确过滤规则（示例：匹配类型为CARER且name为"李四"）
Specification<EmployeeEntity> carerSpec = (root, query, cb) -> cb.and(
    cb.equal(root.get("type"), EmployeeType.CARER),
    cb.equal(root.get("name"), "李四")
);

步骤2：组合Specification并执行分页查询 #

将两个Specification用OR连接，直接调用Spring Data JPA的分页查询方法：

// 组合规则：满足教师过滤条件 或 护理员过滤条件
Specification<EmployeeEntity> combinedSpec = Specification.where(teacherSpec).or(carerSpec);

// 构造分页参数（第0页，每页10条，按id升序）
Pageable pageable = PageRequest.of(0, 10, Sort.by("id").ascending());

// 执行分页查询，直接得到合并后的分页结果
Page<EmployeeEntity> resultPage = employeeRepository.findAll(combinedSpec, pageable);

这种方案性能最优，所有分页逻辑由JPA底层处理，无需手动计算偏移量或合并数据。

方案二：分开查询后手动合并分页（适用于无法合并的复杂逻辑） #

如果两类员工的过滤逻辑完全无法通过OR组合，需要分别查询后手动处理分页：

步骤1：统计两类员工的总符合条数 #

long teacherTotal = employeeRepository.count(teacherSpec);
long carerTotal = employeeRepository.count(carerSpec);
long totalCount = teacherTotal + carerTotal;

步骤2：计算分页偏移量，分别获取对应数据 #

根据当前页的偏移量，计算从两类结果中分别取多少数据：

Pageable pageable = PageRequest.of(pageNum, pageSize);
long offset = pageable.getOffset();
int remainSize = pageable.getPageSize();

List<EmployeeEntity> teacherData = Collections.emptyList();
List<EmployeeEntity> carerData = Collections.emptyList();

// 先从教师结果中取数据
if (offset < teacherTotal) {
    int takeFromTeacher = (int) Math.min(teacherTotal - offset, remainSize);
    Pageable teacherPage = PageRequest.of((int)(offset / pageSize), takeFromTeacher);
    teacherData = employeeRepository.findAll(teacherSpec, teacherPage).getContent();
    remainSize -= takeFromTeacher;
    offset -= teacherTotal;
}

// 剩余数量从护理员结果中取
if (remainSize > 0 && offset < carerTotal) {
    int carerPageNum = (int)(offset / pageable.getPageSize());
    Pageable carerPage = PageRequest.of(carerPageNum, remainSize);
    carerData = employeeRepository.findAll(carerSpec, carerPage).getContent();
}

步骤3：合并数据并构造分页对象 #

// 合并两类数据
List<EmployeeEntity> combinedContent = new ArrayList<>();
combinedContent.addAll(teacherData);
combinedContent.addAll(carerData);

// 构造符合Spring Data JPA规范的Page对象
Page<EmployeeEntity> resultPage = new PageImpl<>(combinedContent, pageable, totalCount);

注意：如果需要全局排序，这种方案需要在合并后额外排序，数据量大时会影响性能，优先推荐方案一。

内容的提问来源于stack exchange，提问作者2dor