JPA如何将两个分页查询结果合并为一个Page对象

JPA本身没有原生方法直接合并来自不同数据源的分页查询结果，需要手动处理数据列表和分页元信息，以下是两种常见场景的实现方式：

场景一：合并两个已分页的结果集 #

如果已经分别从两个数据源拿到了对应页码的分页数据，直接合并内容并重新构建Page对象即可：

// 1. 从两个数据源获取各自的分页结果（确保排序规则一致）
Page<User> pageDb1 = userRepo1.findByStatus(1, PageRequest.of(0, 10, Sort.by("id")));
Page<User> pageDb2 = userRepo2.findByStatus(1, PageRequest.of(0, 10, Sort.by("id")));

// 2. 合并查询结果列表
List<User> mergedContent = new ArrayList<>(pageDb1.getContent());
mergedContent.addAll(pageDb2.getContent());

// 可选：如果存在重复数据，按业务主键去重
mergedContent = new ArrayList<>(new LinkedHashSet<>(mergedContent));

// 3. 构建合并后的Page对象
long totalElements = pageDb1.getTotalElements() + pageDb2.getTotalElements();
Page<User> mergedPage = new PageImpl<>(
    mergedContent,
    PageRequest.of(0, 10),
    totalElements
);

场景二：模拟统一分页查询 #

如果需要像单数据源分页那样，直接返回合并后的指定页码结果（比如用户要第2页，每页10条），需要先获取足够数据再做分页：

int targetPage = 1; // JPA页码从0开始，第2页对应index=1
int pageSize = 10;
Sort sort = Sort.by("createTime").descending();

// 1. 从两个数据源查询按排序后的数据集（数据量大时，建议查询(targetPage+2)*pageSize条，避免合并后数据不足）
List<User> listDb1 = userRepo1.findByCondition(condition1, sort);
List<User> listDb2 = userRepo2.findByCondition(condition2, sort);

// 2. 合并并统一排序（如果两个数据源的查询排序规则不一致，需重新排序）
List<User> mergedList = new ArrayList<>(listDb1);
mergedList.addAll(listDb2);
mergedList.sort(sort.comparator());

// 3. 裁剪出目标页的数据
int startIdx = targetPage * pageSize;
int endIdx = Math.min(startIdx + pageSize, mergedList.size());
if (startIdx >= mergedList.size()) {
    mergedList = Collections.emptyList();
} else {
    mergedList = mergedList.subList(startIdx, endIdx);
}

// 4. 计算总条数并构建Page对象
long totalCount = userRepo1.countByCondition(condition1) + userRepo2.countByCondition(condition2);
Page<User> mergedPage = new PageImpl<>(
    mergedList,
    PageRequest.of(targetPage, pageSize, sort),
    totalCount
);

关键注意事项 #

• 数据源隔离：确保两个数据源配置了独立的EntityManager和Repository，避免查询串库。
• 性能优化：数据量大时，场景二中不要查询全量数据，按需查询足够条数；如果是跨库场景，JPA无法直接做联合查询，可考虑使用JDBC或跨库查询工具。
• 去重逻辑：根据业务规则实现实体的equals()和hashCode()，或手动过滤重复数据。
• 排序一致性：合并前后的排序规则必须统一，否则结果顺序不符合预期。

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