- 业务场景
遇到一个比较奇葩的需求,前端调用接口,传入一个id,想查询每个id对应的任务status(状态),查得到的要返回,查不到的也要返回传入的id(是不是很奇葩,正常来讲我只需要把查得到的给他就行),这样就没办法直接使用mybatis功能对sql进行分页查询,只能自己手动分页,对数据进行切割,然后每次批量查询。 - 代码处理
int pageSize = 100;
int size = req.getId().size(); // 这里是入参的id数组长度
int totalPage = PageUtil.totalPage(size, pageSize);
ArrayList<Result> result = new ArrayList<>();// 结果
for (int page = 0; page < totalPage; page++) {
// 手动分页,截取需要查询的数组
int startIndex = page * pageSize;
int endIndex = Math.min(startIndex + pageSize, size);
ArrayList<String> wantSearchIds = new ArrayList<>(req.getIds().subList(startIndex, endIndex));
// 执行查询,我这里写伪代码
List<Respxxx> hasInSql= saerach(wantSearchIds)
// 定义一个存在数据库的数组
ArrayList<String> existIds = new ArrayList<>();
if (ObjectUtil.isNotEmpty(hasInSql)) {
hasInSql.forEach(item->{
// 写入存在的
result.add(item);
existIds.add(item.getId());
});
}
// 获取不存在的Id
ArrayList<String> noExistIds = new ArrayList<>(wantSearchIds );
noExistIds .removeAll(existIds);
noExistIds .forEach(item->{
result.add(item);
});
}