场景:一个更新报表的接口,调用一次更新一次。大致逻辑是先查询,存在即把旧数据逻辑删除,新增一条新数据。偶然一次调用方短时间内发起了多次调用,导致查询的时候报错。
报错信息:org.springframework.dao.IncorrectResultSizeDataAccessException: query did not return a unique result: 2; nested exception is javax.persistence.NonUniqueResultException: query did not return a unique result: 2
源代码:
public Boolean saveFileFromTemplate(SaveReportFileDTO dto) {
// 使用分布式锁防止多个容器并发导致数据问题
RLock redissonClientLock = redissonClient.getLock("saveTemplate:" + dto.getId() + ":" + dto.getReviewType());
try {
boolean tryLock = redissonClientLock.tryLock(5, 10, TimeUnit.SECONDS);
if (!tryLock) {
return false;
}
log.info("******获取到锁");
//2.查询是否存在文件信息
Short fileType = FileTypeEnums.report_count.getCode();
FileEntity fileEntity = fileRepository.findByIdAndFileTypeAndStage(dto.getTenantId(), dto.getId(),
fileType, dto.getStage());
if (fileEntity != null) {
//存在,删除重新生成
fileEntity.setDeleted(true);
fileEntity.setModifyTime(System.currentTimeMillis());
fileRepository.save(fileEntity);
}
FileEntity fileEntity = new FileEntity();
// 省略无用代码
fileEntity.setCreateTime(System.currentTimeMillis());
log.info("【saveTemplate-阶段{}】==>fileEntityNew:{}", stage, fileEntityNew);
FileEntity save = fileRepository.save(fileEntity);
log.info("===return.....");
return save != null;
} catch (Exception e) {
e.printStackTrace();
} finally {
if (redissonClientLock.isLocked()) {
if (redissonClientLock.isHeldByCurrentThread()) {
redissonClientLock.unlock();
log.info("******释放锁");
}
}
}
return false;
}
加日志后JMeter测试也是有问题
通过SQL日志看到逻辑删除在线程9持有分布式锁时间之外,这就导致线程2获取到锁查询数据库有两条数据(线程9未更新的和新增的)。找到问题就好解决了,把save更新方法改成手动的update就解决问题。
if (fileEntity != null) {
fileRepository.deleteByTenantIdAndIdAndFileTypeAndKbStage(dto.getTenantId(), dto.getId(), fileType, stage);
}