@Transactional TransactionManager 源码分析

先贴一篇好文章，https://www.cnblogs.com/thisiswhy/p/13948055.html

部分代码得修改下 rollback方法必须传入参数 transactionStatus ，如果用SelfTransactionMananger单实例中的属性transactionStatus，在多线程中会出现线程不安全的情况。

ArrayList 线程不安全，elementData[size++] = e是先赋值再size+1,多线程运行到赋值还没+1时,size位置上被覆盖了多次,然后多次+1,size+1,+2等位置没赋值过,下次就直接从size+n开始赋值,看起来就add了null值一样,此时不会报错,因为add时没有null所以取出时没做考虑就可能报NullPointerException了

工作上有一个需求跟文章里的很像，要求任何一个线程发生错误后，所有主线程和子线程里的数据库操作全部回滚。文章中很聪明的用到了”二次确认”的思想，利用CountDownLatch确保所有线程都执行成功后，统一执行commit，如果有一个失败，则全部rollback。

那么数据库事务是如何触发执行commit和rollback的呢？

事务是线程独立的么？

数据库连接池是什么时候借给程序又是怎么回收的呢？

带着这些疑问我们去研读下代码。

话不多说，直接先上流程图，大家先看看，能看懂多少是多少，这里我们先讲解些基本概念，后面会详细梳理流程。

基本概念

DataSourceConfiguration 创建DataSource

DataSourceTransactionManagerAutoConfiguration 创建TransactionManager manager = new DataSourceTransactionManager(dataSource);

TransactionManager 核心方法是 getTransaction，commit, rollback 都由AbstractPlatformTransactionManager的抽象类实现

TransactionInfo 包含属性TransactionManager TransactionAttribute TransactionStatus

protected TransactionInfo createTransactionIfNecessary(@Nullable PlatformTransactionManager tm,@Nullable TransactionAttribute txAttr, final String joinpointIdentification)

TransactionDefinition TransactionAttribute 主要属性 isolationLevel propagationBehavior

TransactionStatus getTransaction(@Nullable TransactionDefinition definition) throws TransactionException;

TransactionStaus 主要属性newTransaction localRollbackOnly globalRollbackOnly transaction(一般为DataSourceTransactionObject，可获取connectionHolder -> 获取connection)

TransactionStatus getTransaction(@Nullable TransactionDefinition definition)  throws TransactionException;

void commit(TransactionStatus status) throws TransactionException;

void rollback(TransactionStatus status) throws TransactionException;

TransactionInterceptor

@Transactional spring会将该方法生成cglib代理MethodInvocation，由TransactionInterceptor（MethodInterceptor)执行interceptor.invoke(methodInvocation) 从而实现方法增强

流程梳理

理解了些基础概念后，我们将参照之前的流程图跟着代码走一遍

TransactionAspectSupport.invokeWithinTransaction

核心中的核心，就是这个方法实现了对代理cglib方法的增强

createTransactionIfNecessary 创建txInfo

invocation.proceedWithInvocation 执行interceptor链路，最终执行代理方法

completeTransactionAfterThrowing 如果有异常且该异常满足回滚条件，则执行回滚操作

commitTransactionAfterReturning 如果没异常，但是rollbackOnly=true，也回滚；反之，则执行提交操作

    @Nullable
 protected Object invokeWithinTransaction(Method method, @Nullable Class<?> targetClass,
   final InvocationCallback invocation) throws Throwable {
        //...
        //创建transaction   实际调用 transaction.getTransaction(definition)
   TransactionInfo txInfo = createTransactionIfNecessary(ptm, txAttr, joinpointIdentification);

   Object retVal;
   try {
    // This is an around advice: Invoke the next interceptor in the chain.
    // This will normally result in a target object being invoked.
    retVal = invocation.proceedWithInvocation();
   }
   catch (Throwable ex) {
    // 执行rollback
    completeTransactionAfterThrowing(txInfo, ex);
    throw ex;
   }
   finally {
    cleanupTransactionInfo(txInfo);
   }

   if (retVal != null && vavrPresent && VavrDelegate.isVavrTry(retVal)) {
    // Set rollback-only in case of Vavr failure matching our rollback rules...
    TransactionStatus status = txInfo.getTransactionStatus();
    if (status != null && txAttr != null) {
     retVal = VavrDelegate.evaluateTryFailure(retVal, txAttr, status);
    }
   }
            // 执行commit
   commitTransactionAfterReturning(txInfo);
   return retVal;
        //...
}

创建txInfo

TransactionInfo txInfo = createTransactionIfNecessary() 创建transaction，实际调用transactionManager.getTransaction(definition);definition 是TransactionDefinition 类，默认事务级别为 PROPAGATION_REQUIRED

再经过DataTransactionalManager的一系列骚操作