优雅解决spring事务中异步发送MQ,Redis,RPC

优雅解决spring事务中异步发送MQ,Redis,RPC

问题

我们在使用spring的@Transaction事务时,如果在事务中进行MQ,Redis,RPC等等远程调用时,spring的事务是无法回滚的

场景举例

伪代码

@Transactional
    public Result updateShopInfo(Shop shop) {
        Long id = shop.getId();
        if (id == null) {
            return Result.fail("商户id不存在");
        }
        //1.更新数据库
        db.updateById(shop);
        //2.更新redis/发送mq/发送rpc
        //这些操作事务无法回滚
        redis.delete();
        mq.send();
        rpc.post();
        return Result.ok("更新成功");
    }

解决方案

spring事务提供的TransactionSynchronization接口,提供了完成事务后,根据事务状态,如已提交,回滚,进行回调操作的实现

因此我们完全可以在事务完成后进行回调操作,我们可以编写一个工具类,这样就可以较小的侵入改写代码了

TransactionUtils.java

/**
 * @program: TransactionDemo
 * @description: 事务回调实现
 * @author: Acerola
 * @create: 2022-11-08 09:02
 **/

public class TransactionUtils {
    public static void doAfterTransaction(DoTransactionCompletion doTransactionCompletion){
        //判断事务是否被激活
        if (TransactionSynchronizationManager.isActualTransactionActive()){
            //将我们实现的afterCompletion注册到当前事务上下文的同步器里
            TransactionSynchronizationManager.registerSynchronization(doTransactionCompletion);
        }
    }
}


class DoTransactionCompletion implements TransactionSynchronization {
    private Runnable runnable;

    public DoTransactionCompletion(Runnable runnable) {
        this.runnable = runnable;
    }

    /**
     * 事务完成后操作
     */
    @Override
    public void afterCompletion(int status) {
        //判断事务提交
        if (status == TransactionSynchronization.STATUS_COMMITTED) {
            this.runnable.run();
        }
    }
}

/**
 * 使用案例
 */
@Service
class transactionTest{
    @Transactional
    public void test(){
        //开启事务
        TransactionUtils.doAfterTransaction(new DoTransactionCompletion(()->{
            //send MQ,redis,RPC....
        }));
        //事务结束
    }
}