业务场景:
我们经常会有些方法会调用一些方法异步执行,比如入库操作后要异步发送mq消息,发送短信或者发布事件等等
但是往往我们的业务方法总是包含事务的,要么全部成功,要么全部失败。
有的时候事务执行失败了或者还没有提交,异步方法就执行完成了
我们希望:
- 当我们事务失败回滚时,我们的异步操作也不执行
- 我们的异步操作需要等待事务完成后才执行
比如:
@Transactional(rollbackFor = Exception.class)
public boolean testTransactional() {
Warehouse warehouse = warehouseService.getById(1);
warehouse.setUpdateTime(null);
warehouseService.updateById(warehouse);
sendMsg();
warehouseService.save(warehouse);
return Boolean.TRUE;
}
public void sendMsg(){
ThreadUtils.execute(new Runnable() {
@Override
public void run() {
log.info("发送消息到mq");
}
});
}
这里我们编辑一个仓库, 然后再新增一个数据一样的仓库。 仓库表中code字段是有唯一索引的,所以新增肯定会失败,事务会回滚
这里,我们发现事务提交失败回滚了,但是mq消息还是发送成功了,不满足我们的预期
解决方案:
/**
* 事务测试
*
* @return
*/
@Transactional(rollbackFor = Exception.class)
public boolean testTransactional1() {
Warehouse warehouse = warehouseService.getById(1);
warehouse.setUpdateTime(LocalDateTime.now());
warehouseService.updateById(warehouse);
TransactionSynchronizationManager.registerSynchronization(new TransactionSynchronization() {
@Override
public void afterCommit() {
sendMsg();
}
});
warehouseService.save(warehouse);
return Boolean.TRUE;
}
使用TransactionSynchronizationManager, 重写afterCommit方法,把需要事务提交之后再执行的代码放到afterCommit中去,然后执行
发现并没有执行发送mq的方法。成功了
最后我们再去掉唯一索引试一下
发现发送mq永远都是在事务提交之后执行的,问题解决
我们看TransactionSynchronization 里还有 afterCompletion() 方法,这个是不管事务成功还是失败,当事务完成后就会执行。