大家都知道想要 Spring 的事务功能,你就必须使用 @EnableAspectJAutoProxy 注解开启事务,并且需要再方法或者类或接口上标注 @Transactional 注解。
@Configuration
@EnableTransactionManagement
@ComponentScan("com.cn.spring.service")
public class TxMainConfig {
@Bean
public DataSource dataSource() {
System.out.println("开始拦截数据源");
DriverManagerDataSource dataSource = new DriverManagerDataSource();
dataSource.setDriverClassName("com.mysql.cj.jdbc.Driver");
dataSource.setUrl("jdbc:mysql://localhost:3306/db?useSSL=false&allowPublicKeyRetrieval=true");
dataSource.setUsername("root");
dataSource.setPassword("8888");
System.out.println("数据源链接成功!");
return dataSource;
}
@Bean
public JdbcTemplate jdbcTemplate(@Autowired DataSource dataSource) {
return new JdbcTemplate(dataSource);
}
// 事物管理器
@Bean
public JdbcTransactionManager jdbcTransactionManager() {
return new JdbcTransactionManager(dataSource());
}
@Bean
public TransactionTemplate transactionTemplate() {
TransactionTemplate template = new TransactionTemplate();
template.setTransactionManager(jdbcTransactionManager());
return template;
}
}
定义业务类如下:
@Service
public class CalculateServiceImpl implements CalculateService {
@Autowired
private OrderService orderService;
@Autowired
private StockService stockService;
@Override
@Transactional // 默认传播机制
public void add(int a, int b) {
try {
orderService.crateOrder();
stockService.addStock();
} catch (Exception e) {
e.printStackTrace();
}
}
}
那么思考下,为什么在 Spring 中标注了这两个注解(@EnableTransactionManagement、@Transactional)就能够
帮我们完成事务功能?
那么现在就看下这两个注解到底干了什么什么事情。先看 @EnableTransactionManagement 源码如下:
从源码可以清楚的看到通过 @Import 注解导入了两个核心类:InfrastructureAdvisorAutoProxyCreator、ProxyTransactionManagementConfiguration
其实猜都能猜到,第一个类肯定使用来生成代理类的、第二个肯定是导入一些事务相关切面逻辑。
进入到 InfrastructureAdvisorAutoProxyCreator 类可以发现此类是 AOP 切面功能 AbstractAutoProxyCreator 抽象类的子类,而且 AbstractAutoProxyCreator 是一个 BeanPostProcessor 后置处理器,在 Spring 启动过程中会被回调。
实现了上面接口的类,只需要关注接口中定义的规范 API 即可,在 Spring 启动过程中就会去回调这些接口的方法,源码如下:
然后去查看 InfrastructureAdvisorAutoProxyCreator 类看能否找到以上规范的 API 方法,发现找不到,但是它继承了 AbstractAdvisorAutoProxyCreator 类,所以就去它的父类中查找,发现也没有,那就在往上一个父类找,也就是 AbstractAutoProxyCreator (AOP 切面功能抽象类,AOP 中王者级别的类),发现终于找到了后置处理器中定义的规范 API 方法,源码如下:
首先看到 getEarlyBeanReference() 方法,源码如下:
发现第一行是把当前过来的原生包成品 bean 保存到了一个缓存中,为什么要保存呢?因为怕重复创建代理对象,第二行就是创建代理对象的逻辑,这个 wrapIfNecessary() 方法会在两个地方被调用,第一个就是这里的 getEarlyBeanReference() 方法,还有一个地方就是后置处理器的另一个 API 方法中——postProcessAfterInitialization() 方法中。
getEarlyBeanReference() 和 postProcessAfterInitialization() 都是后置处理器的规范 API,所以必然存在先后被回调到,所以这里不管你是哪个先被调用,你肯定需要一个缓存或者是标记说你干了什么事?这里就是使用一个 Map 类型 earlyProxyReferences 缓存表示在 getEarlyBeanReference() API 中创建过了代理对象,其他 API 中不需要再去创建了,第二个 API 调用创建代理对象的源码如下:
其他 API 不需要去关注,因为没有起到关键性作用,所以这里重点关注 getEarlyBeanReference() 和 postProcessAfterInitialization(),但是会发现这两个是做的同一件事情,所以只需要关注 getEarlyBeanReference() 这个 API 的逻辑即可。
进入到 wrapIfNecessary() 方法,源码如下:
这里有两个非常重要的步骤:
进入到 getAdvicesAndAdvisorsForBean() 源码如下:
先看第一行代码,去收集实现了 Advisor 的接口子类,事务通过注解 @EnableTransactionManagement 导入了一个 Advisor,然后这里就会找到一个 Advisor 候选者,源码如下:
接着看第二行代码,就是找到的 Advisor 是否能够用于此类,能够能够用于此类,那就是表示这个类需要被增强,就需要被代理了,就需要创建这个类的代理类了。进入 findAdvisorsThatCanApply() 方法,会从方法和类进行匹配,源码如下:
获取都 ClassFilter 对象匹配类,答案是这里匹配不到,因为我们的 @Transactional 注解是标在了方法上的,所以又会往类的所有方法上去匹配,每个方法挨个去比较看是否能够找到 @Transactional 注解。
接下来我们进入事务注解的匹配过程,TransactionAttributeSourcePointcut 负责去匹配,看下到底是怎么匹配的,源码如下:
南海中一定要一个意识: 切面的匹配功过程一定会有 Pointcut 切入点,并且每个 Pointcut 一定会有两个元素:就是 ClassFilter 对类的匹配执行器,MethodMatcher 对方法的匹配执行器
进入到 getTransactionAttributeSource() 方法,源码如下:
为什么敢断定这个 transactionAttributeSource 变量不是 null ? 如下源码:在 TransactionAttributeSourceAdvisor 类的构造中赋值了这个 transactionAttributeSource 变量
这个 ProxyTransactionManagementConfiguration 是注解 @EnableTransactionManagement 导入进来的类。
然后 getTransactionAttributeSource() 就获取到了一个事务属性对象,transactionAttributeSource 这个对象主要用来封装 @Transactional 注解中解析出来的值(传播特性,隔离级别、只读等等)。
继续往下看,看是怎么匹配这个注解的
重点都是在 computeTransactionAttribute() 方法,这里面回去判断方法、类、接口是否被 @Transactional 修饰,进入源码:
进入 findTransactionAttribute() 方法,源码如下:
从上述源码中可以看出就是去查找这个类,方法、接口、接口方法中是否存在 @Transactional 注解,如果有就要开始往下代理对象了。
至此第一个步骤:匹配过程就算是结束了,接下来开始要去创建代理对象
回到外层调用处,源码如下:
可以看到我们的目标对象被包装成了一个 SingletonTargetSource 对象,进入到 createProxy() 源码如下:
首先过来就创建了一个 ProxyFactory 的代理工厂,拷贝了一下当前对象的全部属性到工厂中,把目标对象放到了 ProxyFactory、Advisors 增强器放到了 ProxyFactory。这个对象包装了代理需要的所有东西,就是工厂,里面存放着代理相关的东西
还会推算你的目标 bean 是否有接口,如果没有就会使用 Cglib、有就使用 Proxy 代理,我们这里是有接口的,所以使用 Proxy 代理。
进入方法 getProxy() 内部,源码如下:
然后再进入 getProxy() 方法,源码如下:
这里就是 JDK 动态代理代码了,this 是上面创建的对象 JdkDynamicAopProxy,此类实现了 InvocationHandler 接口,源码如下:
至此创建代理对象也算完成了,代理对象的 InvocationHandler 是 JdkDynamicAopProxy,也就是说当你调用方法的时候,就会调用到 JdkDynamicAopProxy 类中的 invoke() 方法,这是动态代理最基本调用。
测试代码如下:
因为这里的 CalculateService 类需要被代理,所以这里会生成代理对象,calculateService 对象是个$Proxy23@222 对象,当调用 add() 方法时,会调用到 JdkDynamicAopProxy 中的 invoke() 方法,因为 JdkDynamicAopProxy 实现了 InvocationHandler 接口,源码如下:
第一行其实去判断这个方法是否需要被增强,其实你走到这里,类的匹配根本就不需要做了,因为走到这里必然是代理类,现在只需要去匹配这个方法需不需被增强,进入 getInterceptorsAndDynamicInterceptionAdvice()方法,源码如下:
无非就是一个 matches() 匹配过程,如果匹配返回 true,表示这个方法需要被增强,就会进入 registry.getInterceptors(advisor) 方法,去获取对这个方法增强逻辑 Advice,源码如下:
发现是返回的一个 TransactionInterceptor transactionInterceptor 对象,其实就是一个 Advice,TransactionInterceptor 就是 Advice 的子类而已,而且这个 TransactionInterceptor 拦截器是通过 @EnableTransactionManagement 注解导入进来的,源码如下:
所以这个事物的所有增强逻辑都是在这里做的,这些增强逻辑其实就是帮我们自动开启事务,自动回滚,自动提交。所以这就是为什么我们直接添加两个注解能够帮我们完成事物控制。其实都是这块的 Advice 拦截器完成的。
至此 getInterceptorsAndDynamicInterceptionAdvice() 方法就算完成了,而且返回了一个 TransactionInterceptor Advice,再回到源码开始执行 proceed() 方法,源码如下:
此时 chain 不为空,就会执行 else 逻辑,进入到 ReflectiveMethodInvocation 类,源码如下:
执行 invoke() 方法,调用 TransactionInterceptor 事务提供的 Advice 增强功能,源码如下:
进入到 invokeWithinTransaction() 方法,源码如下:
首先过来就获取 @Transactional 解析封装好的事务对象 TransactionAttributeSource,然后获取事务管理器,继续往下看,源码如下:
看到这三个步骤是不是就恍然大悟了,原来事务相关的功能都是在这里帮我们做好了。
进入开启事务的方法 createTransactionIfNecessary() 方法,源码如下:
然后调用我们自己的业务方法,源码如下:
然后再看事务回滚的方法 completeTransactionAfterThrowing(),源码如下:
然后再看下事务提交的方法 commitTransactionAfterReturning() 源码如下:
至此声明式事务的执行流程分析已经算完成了,还有事物的传播特性没有分析,是因为这一块内容也挺多的,篇幅太大不太好,所以放在了另一篇文章中,下面看下编程式事务流程分析。
为什么还要这个编程式事务?因为很多时候注解不够灵活,如果自己来管控事务是不是比较好,而且一个注解开启的事物,如果业务逻辑中,有一个接口耗时非常的久,那么就会阻塞住 Connection 连接,连接没有释放,如果请求很多,那就完蛋了,资源可能就不够用了,但是编程式事务就比较好的可以自己控制。
代码如下:
// 编程式事务
@Override
public void reduce(int a, int b) {
System.out.println("start reduce......");
// 第一种方式: 什么都不用自己控制
template.execute(status -> {
// 这里面还可以继续嵌套事务
//orderService.crateOrder();
// ...
// 要是里面还有要执行的耗时逻辑又可以重新套一个execute 操作
// ...
return null;
});
template.execute(status -> {
// 这里面还可以继续嵌套事务
//stockService.addStock();
// ...
// 要是里面还有要执行的耗时逻辑又可以重新套一个execute 操作
// ...
return null;
});
// 第二种方式: 自己控制 commit 和 rollback
DefaultTransactionDefinition definition = new DefaultTransactionDefinition();
TransactionStatus transaction = jdbcTransactionManager.getTransaction(definition);
try {
System.out.println("一些耗时的业务逻辑代码.....");
stockService.addStock();
orderService.crateOrder();
} catch (TransactionException e) {
// 自己控制回滚操作
jdbcTransactionManager.rollback(transaction);
}
// 自己控制提交操作
jdbcTransactionManager.commit(transaction);
}
源码很简单,就对 execute() 这个分析,源码如下:
先去获取数据源对象,然后开启事务并设置手动提交,然后调用目标方法,执行完之后,调用 commit() 提交,如果出现异常或者 Error 就进行回滚。