SpringBoot整合Transaction——源码解析

前言

@Transactional的底层实现是通过aop代理来完成的，当调用被@Transactional声明的方法时，实际上调用的是aop的一个代理类，事务失败会自动捕获异常进行rollback。下面就一起通过源码来了解一下吧。

在哪里实现代理

TransactionAttributeSourcePointcut.java

    public boolean matches(Method method, Class targetClass) {
        TransactionAttributeSource tas = getTransactionAttributeSource();
        return (tas == null || tas.getTransactionAttribute(method, targetClass) != null);
    }

可以看到，TransactionAttributeSourcePointcut的matches方法实现了对@Transactional注解的匹配。aop会对每一个方法调用一次matches。再往下走，进入getTransactionAttribute方法里。

AbstractFallbackTransactionAttributeSource.java

   public TransactionAttribute getTransactionAttribute(Method method, @Nullable Class targetClass) {
        if (method.getDeclaringClass() == Object.class) {
            return null;
        }

        // 先通过缓存来获取 TransactionAttribute
        Object cacheKey = getCacheKey(method, targetClass);
        TransactionAttribute cached = this.attributeCache.get(cacheKey);
        if (cached != null) {
            if (cached == NULL_TRANSACTION_ATTRIBUTE) {
                return null;
            } else {
                // 如果TransactionAttribute不为空且是一个事务属性，则直接返回
                return cached;
            }
        } else {
            /**
             * 通过以下情况来寻找解析TransactionAttribute
             * 1.@Transactional 在方法上
             * 2.@Transactional 在类上
             * 3.@Transactional 在接口方法上
             * 4.@Transactional 在接口类上
             * 只要一上任意一种情况满足，则匹配成功
             */
            TransactionAttribute txAttr = computeTransactionAttribute(method, targetClass);
            if (txAttr == null) {
                //未找到，进行缓存
                this.attributeCache.put(cacheKey, NULL_TRANSACTION_ATTRIBUTE);
            } else {
                String methodIdentification = ClassUtils.getQualifiedMethodName(method, targetClass);
                if (txAttr instanceof DefaultTransactionAttribute) {
                    ((DefaultTransactionAttribute) txAttr).setDescriptor(methodIdentification);
                }
                if (logger.isTraceEnabled()) {
                    logger.trace("Adding transactional method '" + methodIdentification + "' with attribute: " + txAttr);
                }
                this.attributeCache.put(cacheKey, txAttr);
            }
            return txAttr;
        }
    }

以上就是一个完整的匹配过程。

invoke方法在哪里调用

TransactionInterceptor.java

    public Object invoke(MethodInvocation invocation) throws Throwable {
        Class targetClass = (invocation.getThis() != null ? AopUtils.getTargetClass(invocation.getThis()) : null);

        // 真正的开启事务方法
        return invokeWithinTransaction(invocation.getMethod(), targetClass, invocation::proceed);
    }

走进invokeWithinTransaction方法看看

    protected Object invokeWithinTransaction(Method method, @Nullable Class targetClass,
                                             final InvocationCallback invocation) throws Throwable {
        TransactionAttributeSource tas = getTransactionAttributeSource();
        //获取注解属性
        final TransactionAttribute txAttr = (tas != null ? tas.getTransactionAttribute(method, targetClass) : null);
        //得到事务管理器的顶级接口，一般是DataSourceTransactiuonManager，实现了commit 和 rollback方法
        final PlatformTransactionManager tm = determineTransactionManager(txAttr);
        final String joinpointIdentification = methodIdentification(method, targetClass, txAttr);
        if (txAttr == null || !(tm instanceof CallbackPreferringPlatformTransactionManager)) {
            //看是否有必要创建一个事务，根据事务传播行为，做出相应的判断
            TransactionInfo txInfo = createTransactionIfNecessary(tm, txAttr, joinpointIdentification);
            Object retVal = null;
            try {
                //执行目标方法
                retVal = invocation.proceedWithInvocation();
            }
            catch (Throwable ex) {
                //对异常的捕获处理
                completeTransactionAfterThrowing(txInfo, ex);
                throw ex;
            }
            finally {
                //清楚事务信息
                cleanupTransactionInfo(txInfo);
            }
            commitTransactionAfterReturning(txInfo);
            return retVal;
        }
        else {
            //编程试事务处理，略过...
        }
    }