如何在spring事务提交成功后再进行异步操作

业务场景:

我们经常会有些方法会调用一些方法异步执行,比如入库操作后要异步发送mq消息,发送短信或者发布事件等等
但是往往我们的业务方法总是包含事务的,要么全部成功,要么全部失败。
有的时候事务执行失败了或者还没有提交,异步方法就执行完成了

我们希望:

  1. 当我们事务失败回滚时,我们的异步操作也不执行
  2. 我们的异步操作需要等待事务完成后才执行

比如:

@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() 方法,这个是不管事务成功还是失败,当事务完成后就会执行。


版权声明:本文为qq_32519415原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。