SpringBoot集成Mybatis使用切面对所有Service的事务统一管理

SpringBoot集成Mybatis使用切面对所有Service的事务统一管理

package cn.com.baidu.aspect;

import org.springframework.aop.aspectj.AspectJExpressionPointcut;
import org.springframework.aop.support.DefaultPointcutAdvisor;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.jdbc.datasource.DataSourceTransactionManager;
import org.springframework.transaction.TransactionDefinition;
import org.springframework.transaction.interceptor.*;
import javax.sql.DataSource;
import java.util.Collections;
import java.util.HashMap;
import java.util.Map;

/**
 * CBC
 * 2023年12月2日00:53:35
 * 用于对数据源事务配置的初始化
 */
@Configuration
public class MyBatisTransaction {
    @Autowired
    private DataSource dataSource;

    /**
     * 获取事务管理器-修改信息
     * @return
     */
    @Bean("myBatisTransactionManager")
    public DataSourceTransactionManager myBatisTransactionManager() {
        return new DataSourceTransactionManager(dataSource);
    }

    /**
     * 事务拦截器(设置以哪些开头的需要事务)
     */
    @Bean("myBatisTransactionAdvice")
    public TransactionInterceptor myBatisTransactionAdvice(DataSourceTransactionManager txManager) {
        NameMatchTransactionAttributeSource source = new NameMatchTransactionAttributeSource();
        /* 只读事务，不做更新操作*/
        RuleBasedTransactionAttribute readOnlyTx = new RuleBasedTransactionAttribute();
        readOnlyTx.setReadOnly(true);
        /*不支持事务，当前存在事务时，将挂起事务，运行方法*/
        readOnlyTx.setPropagationBehavior(TransactionDefinition.PROPAGATION_NOT_SUPPORTED);
        /*写事务，需要更新操作 */
        RuleBasedTransactionAttribute requiredTx = new RuleBasedTransactionAttribute(TransactionDefinition.PROPAGATION_REQUIRED, Collections.singletonList(new RollbackRuleAttribute(Exception.class)));
        requiredTx.setTimeout(5);
        Map<String, TransactionAttribute> txMap = new HashMap<>();
        txMap.put("save*", requiredTx);
        txMap.put("add*", requiredTx);
        txMap.put("update*", requiredTx);
        txMap.put("delete*", requiredTx);
        txMap.put("remove*", requiredTx);
        txMap.put("get*", readOnlyTx);
        txMap.put("list*", readOnlyTx);
        txMap.put("query*", readOnlyTx);
        source.setNameMap(txMap);
        return new TransactionInterceptor(txManager, source);
    }

    /**
     * 切面拦截规则 参数会自动从容器中注入
     */
    @Bean
    public DefaultPointcutAdvisor defaultPointcutAdvisor(TransactionInterceptor myBatisTransactionAdvice) {
        DefaultPointcutAdvisor pointcutAdvisor = new DefaultPointcutAdvisor();
        pointcutAdvisor.setAdvice(myBatisTransactionAdvice);
        AspectJExpressionPointcut pointCut = new AspectJExpressionPointcut();
        pointCut.setExpression("execution(* cn.com.baidu.service.*.*.*(..))");
        pointcutAdvisor.setPointcut(pointCut);
        return pointcutAdvisor;
    }
}