@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的一系列骚操作

dataSource.getConnection()    由具体的dataSource实现类获得，思路都是由连接池里拿一个未使用的连接出来

核心代码来了，TransactionSynchronizationManager.bindResource 将connectionHolder绑定到ThreadLocal，这样就将connection，TransactionStatus和当前线程绑定了起来

执行代理方法

执行后处理

根据执行结果判断是执行rollback还是commit

TransactionAspectSupport.completeTransactionAfterThrowing  如果抛了异常，则判断是否满足回滚条件

AbstractPlatformTransactionManager     执行回滚操作

TransactionAspectSupport.commitTransactionAfterReturning    没异常则执行commit

AbstractPlatformTransactionManager   执行提交操作   但还需要判断是否标记了rollbackOnly，有可能A,B两个方法共用同一个事务，B抛了异常，导致外层的方法A哪怕没错也得一起回滚。

至此，整个链路就分析完成了