如何拿到数据库事务的控制权

一。编程事务

自己在pjp.proceed()前后控制事务 TransactionUtil

1.TransactionUtil

@Component
public class TransactionUtil {

    //获取数据源
    private DataSourceTransactionManager dataSourceTransactionManager;

    public TransactionUtil(DataSourceTransactionManager dataSourceTransactionManager) {
        this.dataSourceTransactionManager = dataSourceTransactionManager;
    }

    //开启事务
    public TransactionStatus begin() {
        return dataSourceTransactionManager.getTransaction(new DefaultTransactionAttribute());
    }

    /*提交事务*/
    public void commit(TransactionStatus transactionStatus) {
        dataSourceTransactionManager.commit(transactionStatus);
    }

    /*回滚事务*/
    public void rollBack(TransactionStatus transactionStatus) {
        dataSourceTransactionManager.commit(transactionStatus);
    }
}

2.使用

@Autowired
TransactionUtil transactionUtil;

@Around("com.aop.DIY.AOPTransaction.pointCut()")
    public Object around(ProceedingJoinPoint pjp) throws Throwable {
        //1.获取到目标对象的方法
        //2.获取该方法上是否加上注解
        //3.如果存在事务，开启事务
        //4.调用目标对象方法
        //5.判断该方法是否存在注解
        //6.如果存在注解提交事务
        System.out.println("AOP类进入拦截");
        Object proceed = null;
        TransactionStatus transactionStatus = null;

        //1.百度SpringAOP获取目标对象的方法
        Signature signature = pjp.getSignature();
        MethodSignature methodSignature = (MethodSignature)signature;
        Method targetMethod = methodSignature.getMethod();

        //2.判断方法是否存在注解
        ExtTransaction extTransaction = targetMethod.getDeclaredAnnotation(ExtTransaction.class);
        if(extTransaction != null){
            //3.如果存在注解，开启事务
            transactionStatus = transactionUtils.begin();
        }
        //4.调用目标对象方法
        proceed = pjp.proceed();

        //5.判断该方法是否存在注解
        if(extTransaction != null){
            //6.如果存在注解，提交事务
            transactionUtils.commit(transactionStatus);
        }
        return proceed;
    }

二。切面到获取连接源，返回自定义的连接器

1.写一个切面类，切到javax.sql.DataSource.getConnection方法，返回自己定义的连接器

@Aspect
@Component
public class DataSourceAspect {
    @Around("execution(* javax.sql.DataSource.getConnection(..))")
    public Connection around(ProceedingJoinPoint pjp) {
        System.out.println("进入DataSourceAspect切面，返回自己定义的连接器");
        try {
            Connection connection = (Connection) pjp.proceed();
            return new MyConnection(connection);
        }catch (Throwable throwable){
            throwable.printStackTrace();
        }
        return null;
    }
}

2.自定义连接器

public class MyConnection implements Connection {
    private Connection connection;

    public MyConnection(Connection connection) {
        this.connection = connection;
    }

    @Override
    public void commit() throws SQLException {
        System.out.println("自定义提交事务方法");
    }

    @Override
    public void rollback() throws SQLException {
        System.out.println("自定义回滚事务方法");
    }

    @Override
    public void close() throws SQLException {
        System.out.println("自定义关闭事务方法");
    }