前言
上篇我们分析了spring-tx
中的AOP部分,包括TransactionAttributeSourcePointcut
如何定位潜在的事务方法,以及TransactionInterceptor
又如何结合PlatformTransactionManager
为方法应用事务管理,相信看过上篇的同学也从中get到了使用AOP的新姿势😂
不过到目前为止,除去开篇中的概念介绍,我们对PlatformTransactionManager
还知之甚少,本篇我们就深入一点儿。DataSourceTransactionManager
作为最具代表性的 PlatformTransactionManager
实现,我们就以它为例来看看隐藏在抽象下的实现细节吧。
一个小例子
来看一个使用Spring声明式事务的小例子。
@Service
public class ServiceA {
@Autowired
private ServiceB serviceB;
// Propagation.REQUIRED 有就加入,没有就自己玩
@Transactional(propagation = Propagation.REQUIRED)
public void service() {
// do some CRUD stuff...
// then call serviceB#service()
serviceB.service();
}
}
@Service
public class ServiceB {
// Propagation.REQUIRES_NEW 自扫门前雪,各用各的事务
@Transactional(propagation = Propagation.REQUIRES_NEW)
public void service() {
// do some CRUD stuff...
}
}
@RestController
public class DemoController {
@Autowired
private ServiceA serviceA;
@PostMapping("/demo")
public void demo() {
serviceA.service();
}
}
这里我们有两个事务方法,分别属于ServiceA
和ServiceB
,处理请求的入口在DemoController
。请求到达后,
- 进入服务 A 的
service()
方法 - 开启事务 tx-a
- 执行CRUD
- 进入服务 B 的
service()
方法 - 挂起 tx-a,开启新事务 tx-b
- 执行CRUD
- 结束
serviceB#service()
调用,视执行情况提交或回滚事务 - 结束
serviceA#service()
调用,视执行情况提交或回滚事务
麻雀虽小五脏俱全,这个小例子也是后续分析源码时的参照,先给出来混个脸熟吧。
DataSourceTransactionManager
Hierarchy overview
DataSourceTransactionManager
继承自AbstractPlatformTransactionManager
,基类中实现了事务管理的一整套工作流:
- 检查是否已存在事务
- 应用适当的传播行为
- 必要时暂停和恢复事务
- 提交事务时检查
rollback-only
标记 - 回滚事务时的处理——如实回滚或仅设置
rollback-only
标记 - 触发已注册的
transaction synchronization
回调
子类只需要根据事务的状态实现特定的模板方法即可,比如事务开始、暂停、恢复、提交和回滚。
TransactionInterceptor
事务拦截器想必大家都很熟悉了,简化后的模型大致如下:
// 详细的分析请看上篇
public Object invoke(MethodInvocation invocation) throws Throwable {
Object result = null;
// 1. 解析@Transactional注解
TransactionAttribute txAttr = xxx;
// 2. 获取事物管理器
PlatformTransactionManager txManager = xxx;
// 3. 根据配置的事务属性开启事务
TransactionStatus status = txManager.getTransaction(txAttr);
try {
// 4. 执行目标方法
result = invocation.proceed();
} catch (Throwable ex) {
// 5-1. 方法异常退出,根据配置决定是提交还是回滚
if (txAttr.rollbackOn(ex)) {
txManager.rollback(status);
} else {
txManager.commit(status);
}
throw ex;
}
// 5-2. 方法正常执行完成,提交事务
txManager.commit(status);
return result;
}
核心流程大致分为5步,其中和事务管理器有关的是第3和第5步。前情提要到此结束,那我们就从第3步getTransaction(...)
开始呗~
getTransaction
@Override
public final TransactionStatus getTransaction(@Nullable TransactionDefinition definition)
throws TransactionException {
// 使用默认配置,如果没有的话
TransactionDefinition def = (definition != null ? definition : TransactionDefinition.withDefaults());
// 模板方法,子类实现
Object transaction = doGetTransaction();
boolean debugEnabled = logger.isDebugEnabled();
// 检查是否已存在事务
if (isExistingTransaction(transaction)) {
// 已存在事务,根据传播行为来决定如何处理
return handleExistingTransaction(def, transaction, debugEnabled);
}
// 检查一下超时配置,默认 -1 ,不能更低了
if (def.getTimeout() < TransactionDefinition.TIMEOUT_DEFAULT) {
throw new InvalidTimeoutException("Invalid transaction timeout", def.getTimeout());
}
// 当前不存在事务,同样要根据传播行为来决定如何处理
// PROPAGATION_MANDATORY 要求一定要运行在事务中,当前事务不存在,因此这里选择抛出异常
if (def.getPropagationBehavior() == TransactionDefinition.PROPAGATION_MANDATORY) {
throw new IllegalTransactionStateException(
"No existing transaction found for transaction marked with propagation 'mandatory'");
}
// PROPAGATION_REQUIRED 在没有事务存在时会开启新的独立事务
// PROPAGATION_REQUIRES_NEW 无论是否有事务存在都会开启新的独立事务
// PROPAGATION_NESTED 在没有事务存在时和PROPAGATION_REQUIRED的行为一致
// 换言之,走到这里,这三种传播行为都会开启新的独立事务
else if (def.getPropagationBehavior() == TransactionDefinition.PROPAGATION_REQUIRED ||
def.getPropagationBehavior() == TransactionDefinition.PROPAGATION_REQUIRES_NEW ||
def.getPropagationBehavior() == TransactionDefinition.PROPAGATION_NESTED) {
// 挂起当前事务,这里传null是因为当前没有事务
// 仍然调用suspend的意义在于触发 TransactionSynchronization#suspend() 回调
SuspendedResourcesHolder suspendedResources = suspend(null);
if (debugEnabled) {
logger.debug("Creating new transaction with name [" + def.getName() + "]: " + def);
}
try {
// 开启新的独立事务
return startTransaction(def, transaction, debugEnabled, suspendedResources);
}
catch (RuntimeException | Error ex) {
// 事务开启失败,恢复已存在的事务
// 同样为了触发 TransactionSynchronization 回调,这里是 resume()
resume(null, suspendedResources);
throw ex;
}
}
// 其它传播行为在当前没有事务存在时会继续以非事务方式运行,比如PROPAGATION_SUPPORTS
// 不用为它们开启新的事务,但需要处理可能注册的transaction synchronization
else {
// Create "empty" transaction: no actual transaction, but potentially synchronization.
if (def.getIsolationLevel() != TransactionDefinition.ISOLATION_DEFAULT && logger.isWarnEnabled()) {
logger.warn("Custom isolation level specified but no actual transaction initiated; " +
"isolation level will effectively be ignored: " + def);
}
boolean newSynchronization = (getTransactionSynchronization() == SYNCHRONIZATION_ALWAYS);
return prepareTransactionStatus(def, null, true, newSynchronization, debugEnabled, null);
}
}
getTransaction(...)
首先会调用模板方法doGetTransaction()
来获取事务对象(transaction object
)。事务对象虽然特定于具体的PlatformTransactionManeger
实现,但一般都会携带相应的事务状态,我们来分析一下DataSourceTransactionManager
对它的实现。
@Override
protected Object doGetTransaction() {
DataSourceTransactionObject txObject = new DataSourceTransactionObject();
// 是否允许嵌套事务,传播行为PROPAGATION_NESTED时有用
txObject.setSavepointAllowed(isNestedTransactionAllowed());
// 以javax.sql.DataSource为键,ConnectionHolder为值绑定到线程私有存储
// 通过TransactionSynchronizationManager统一管理(重要)
ConnectionHolder conHolder =
(ConnectionHolder) TransactionSynchronizationManager.getResource(obtainDataSource());
// 若能获取到这样一个ConnectionHolder,自然是一个已存在的,因此是false
txObject.setConnectionHolder(conHolder, false);
return txObject;
}
开篇介绍ResourceHolder
接口的时候,我们曾讨论过如何保证多个方法运行在同一个事务中的话题。若将事务窄化为数据库事务的话,自然是要保证多个方法使用同一个JDBC Connection,ConnectionHolder
作为ResouceHolder
的一种实现就是用来携带JDBC Connection的。当然了,除了作为java.sql.Connection
的容器以外,ConnectionHolder
还有一些可配置属性,比如设置rollback-only
标记、超时时长等。DataSourceTransactionObject
可以说是java.sql.Connection
的一个快照,它保存了数据库连接的一系列状态(用于后续对数据库连接的复原),比如当前的隔离级别、自动提交(auto commit
)的取值以及一个指示java.sql.Connection
是通过javax.sql.DataSource
新创建的还是从线程私有存储中获取的标记。这两个类都比较简单,大家可以自行查阅一下,这里就不细说了。
了解了ConnectionHolder
和DataSourceTransactionObject
以后,再看isExistingTransaction(...)
就很简单了。
@Override
protected boolean isExistingTransaction(Object transaction) {
DataSourceTransactionObject txObject = (DataSourceTransactionObject) transaction;
// 1. 必须持有 ConnectionHolder 实例
// 2. 事务已开启,新开的事务会设置这个标志
// 满足这两个条件就可以认为已经有一个已存在的事务
return (txObject.hasConnectionHolder() && txObject.getConnectionHolder().isTransactionActive());
}
还记得前面给出的小例子吗?对于服务 A 的service()
方法来说,很明显我们是无法从线程私有存储中获取到ConnectionHolder
实例的,因此isExistingTransaction(...)
必然返回 false。这里我们也先跳过handleExistingTransaction(...)
,晚点再来分析它。接下来,在没有任何已知事务的前提下,我们一起来看看spring-tx
是如何根据传播行为的配置情况做不同处理的。
PROPAGATION_MANDATORY
要求一定要运行在事务中,而当前不存在任何事务,因此会抛出异常。PROPAGATION_REQUIRED
、PROPAGATION_REQUIRES_NEW
和PROPAGATION_NESTED
这三种传播行为在没有事务存在时的表现是一致的,它们都会开启一个新的独立事务。新事务的开启也意味着对旧事务(或者说外层事务)的挂起,所以我们看到这第一步便是调用suspend(...)
,而这里传null
也是因为当前不存在任何已知事务。
protected final SuspendedResourcesHolder suspend(@Nullable Object transaction)
throws TransactionException {
// 首先查看 transaction synchronization 是否已激活
if (TransactionSynchronizationManager.isSynchronizationActive()) {
// 执行 TransactionSynchronization#suspend() 回调,解除 transaction synchronization 激活状态
List<TransactionSynchronization> suspendedSynchronizations = doSuspendSynchronization();
try {
Object suspendedResources = null;
// 若已有事务,调用模板方法执行挂起逻辑
if (transaction != null) {
suspendedResources = doSuspend(transaction);
}
// 将当前事务的名称、readOnly标记、隔离级别以及是否事务已激活进行重置
// 并把这些属于旧事务的信息记录到 SuspendedResourcesHolder , 用于后续的复原
String name = TransactionSynchronizationManager.getCurrentTransactionName();
TransactionSynchronizationManager.setCurrentTransactionName(null);
boolean readOnly = TransactionSynchronizationManager.isCurrentTransactionReadOnly();
TransactionSynchronizationManager.setCurrentTransactionReadOnly(false);
Integer isolationLevel = TransactionSynchronizationManager.getCurrentTransactionIsolationLevel();
TransactionSynchronizationManager.setCurrentTransactionIsolationLevel(null);
boolean wasActive = TransactionSynchronizationManager.isActualTransactionActive();
TransactionSynchronizationManager.setActualTransactionActive(false);
return new SuspendedResourcesHolder(
suspendedResources, suspendedSynchronizations, name, readOnly, isolationLevel, wasActive);
}
catch (RuntimeException | Error ex) {
// 事务挂起失败不代表失效,需要进行恢复 -> 回调 TransactionSynchronization#resume()
doResumeSynchronization(suspendedSynchronizations);
throw ex;
}
}
// 事务已激活,但 transaction synchronization 未激活
else if (transaction != null) {
// 这种情况下只需要调用模板方法挂起事务即可
Object suspendedResources = doSuspend(transaction);
return new SuspendedResourcesHolder(suspendedResources);
}
// 事务未激活、transaction synchronization 也未激活,没有什么需要挂起的
else {
return null;
}
}
suspend(...)
中牵涉到spring-tx
事务同步机制(transaction synchronization mechanism
),这部分内容我们在开篇中介绍过。如若事务同步机制已激活,就需要先执行对应的回调,这也是doSuspendSynchronization()
做的工作。接下来才是调用模板方法doSuspend(...)
,DataSourceTransactionManager
实现了这个方法。
@Override
protected Object doSuspend(Object transaction) {
DataSourceTransactionObject txObject = (DataSourceTransactionObject) transaction;
// 清除持有的 ConnectionHolder
txObject.setConnectionHolder(null);
// 并将其从ThreadLocal中解绑
return TransactionSynchronizationManager.unbindResource(obtainDataSource());
}
主要就是将ConnectionHolder
从事务对象和线程私有存储中移除,然后转移到SuspendedResourcesHolder
中用于后续的恢复。回到getTransaction(...)
,挂起旧事务后自然是要开启新事物。
private TransactionStatus startTransaction(TransactionDefinition definition, Object transaction,
boolean debugEnabled, @Nullable SuspendedResourcesHolder suspendedResources) {
// 是否需要激活 transaction synchronization
boolean newSynchronization = (getTransactionSynchronization() != SYNCHRONIZATION_NEVER);
// 创建一个代表事务状态的对象
DefaultTransactionStatus status = newTransactionStatus(
definition, transaction, true, newSynchronization, debugEnabled, suspendedResources);
// 模板方法,通知新事务已启动
doBegin(transaction, definition);
// 按需初始化 transaction synchronization
prepareSynchronization(status, definition);
return status;
}
简单明了的三步:
- 创建
TransactionStatus
实例,它包含了一系列的事务状态 - 调用模板方法
doBegin(...)
,通知新事务已启动 - 按需开启事务同步机制
DeaultTransactionStatus
就是一个上下文对象或者说信息的集合体,代码好像没什么可说的,大家自行查阅一下吧,我们重点来看一下doBegin(...)
模板方法。
@Override
protected void doBegin(Object transaction, TransactionDefinition definition) {
DataSourceTransactionObject txObject = (DataSourceTransactionObject) transaction;
Connection con = null;
try {
// 事务对象未持有 ConnectionHolder
// 或事务对象持有 ConnectionHolder,并且与事务同步
if (!txObject.hasConnectionHolder() ||
txObject.getConnectionHolder().isSynchronizedWithTransaction()) {
// 获取一个新的数据库连接
Connection newCon = obtainDataSource().getConnection();
if (logger.isDebugEnabled()) {
logger.debug("Acquired Connection [" + newCon + "] for JDBC transaction");
}
// 绑定到事务对象,注意这里 newConnectionHolder 为 true
txObject.setConnectionHolder(new ConnectionHolder(newCon), true);
}
// 标记 ConnectionHolder 与事务同步,对应上面的检查,避免过多的获取数据库连接
txObject.getConnectionHolder().setSynchronizedWithTransaction(true);
con = txObject.getConnectionHolder().getConnection();
// 应用read-only标记,并返回初始的隔离级别
Integer previousIsolationLevel = DataSourceUtils.prepareConnectionForTransaction(con, definition);
// 应用上配置的隔离级别,初始的隔离级别已经记录下来了,后续可恢复
txObject.setPreviousIsolationLevel(previousIsolationLevel);
// 打 read-only 标记
txObject.setReadOnly(definition.isReadOnly());
// 事务的开启需要设置自动提交为false
if (con.getAutoCommit()) {
// 记录下需要在后续恢复自动提交
txObject.setMustRestoreAutoCommit(true);
if (logger.isDebugEnabled()) {
logger.debug("Switching JDBC Connection [" + con + "] to manual commit");
}
// 自动提交设为false,标识着事务的开启
con.setAutoCommit(false);
}
// read-only 标记相关,执行 SET TRANSACTION READ ONLY 语句
prepareTransactionalConnection(con, definition);
// 重要:标记事务已激活,此时新的数据库连接已获取,并且关闭了自动提交
txObject.getConnectionHolder().setTransactionActive(true);
// 应用超时时长
int timeout = determineTimeout(definition);
if (timeout != TransactionDefinition.TIMEOUT_DEFAULT) {
txObject.getConnectionHolder().setTimeoutInSeconds(timeout);
}
// 绑定到线程私有存储,后续可通过 TransactionSynchronizationManager#getResource(...) 获取
if (txObject.isNewConnectionHolder()) {
TransactionSynchronizationManager.bindResource(obtainDataSource(), txObject.getConnectionHolder());
}
}
// 出错的话,释放数据库连接并抛出异常
catch (Throwable ex) {
if (txObject.isNewConnectionHolder()) {
DataSourceUtils.releaseConnection(con, obtainDataSource());
txObject.setConnectionHolder(null, false);
}
throw new CannotCreateTransactionException("Could not open JDBC Connection for transaction", ex);
}
}
doBegin(...)
设置transaction active
标记,isExistingTransaction(...)
会检查这个标记,这样就构成了一个闭环。最后一步是调用prepareSynchronization(...)
。
protected void prepareSynchronization(DefaultTransactionStatus status, TransactionDefinition definition) {
// 检查是否是全新开启的事务同步机制
// 换言之 TransactionSynchronizationManager.isSynchronizationActive() 需为false
if (status.isNewSynchronization()) {
// 记录事务是否已激活,也就是是否有事务对象
TransactionSynchronizationManager.setActualTransactionActive(status.hasTransaction());
// 记录当前的隔离级别
TransactionSynchronizationManager.setCurrentTransactionIsolationLevel(
definition.getIsolationLevel() != TransactionDefinition.ISOLATION_DEFAULT ?
definition.getIsolationLevel() : null);
// 记录是否是read-only型事务
TransactionSynchronizationManager.setCurrentTransactionReadOnly(definition.isReadOnly());
// 记录事务名称
TransactionSynchronizationManager.setCurrentTransactionName(definition.getName());
// 激活事务同步机制
// initSynchronization()调用之后isSynchronizationActive()就返回true,我们已经在suspend(...)中见过这个检查
TransactionSynchronizationManager.initSynchronization();
}
}
再次回到getTransaction(...)
,可以看到startTransaction(...)
是包裹在 try 块中的,以便在出现异常时恢复已有事务,那我们就看看具体是怎么恢复的呗。
protected final void resume(@Nullable Object transaction,
@Nullable SuspendedResourcesHolder resourcesHolder) throws TransactionException {
// 如果有挂起的事务
if (resourcesHolder != null) {
// 挂起的资源,也就是doSuspend(...)中解绑的ConnectionHolder
Object suspendedResources = resourcesHolder.suspendedResources;
// 调用模板方法,重新绑定ConnectionHolder到ThreadLocal
if (suspendedResources != null) {
doResume(transaction, suspendedResources);
}
// suspend(...)的逆操作
List<TransactionSynchronization> suspendedSynchronizations = resourcesHolder.suspendedSynchronizations;
if (suspendedSynchronizations != null) {
TransactionSynchronizationManager.setActualTransactionActive(resourcesHolder.wasActive);
TransactionSynchronizationManager.setCurrentTransactionIsolationLevel(resourcesHolder.isolationLevel);
TransactionSynchronizationManager.setCurrentTransactionReadOnly(resourcesHolder.readOnly);
TransactionSynchronizationManager.setCurrentTransactionName(resourcesHolder.name);
// 回调TransactionSynchronization#resume(),重新激活 transaction synchronization
doResumeSynchronization(suspendedSynchronizations);
}
}
}
@Override
protected void doResume(@Nullable Object transaction, Object suspendedResources) {
// 重新绑定
TransactionSynchronizationManager.bindResource(obtainDataSource(), suspendedResources);
}
基本上是suspend(...)
的逆操作,至此,创建新事务的流程就分析完了。等等,你以为这就完了?还记得我们的小例子吗?ServiceB
说这才哪到哪,好戏才刚刚开始呢。没错,对ServiceB
而言,ServiceA#service()
已经开启事务了,等到它调用service()
的时候,isExistingTransaction(...)
就返回 true 而不是 false 了,那spring-tx
是如何处理当前已存在事务的情况呢?
private TransactionStatus handleExistingTransaction(
TransactionDefinition definition, Object transaction, boolean debugEnabled)
throws TransactionException {
// PROPAGATION_NEVER不允许运行在事务中,因此抛出异常
if (definition.getPropagationBehavior() == TransactionDefinition.PROPAGATION_NEVER) {
throw new IllegalTransactionStateException(
"Existing transaction found for transaction marked with propagation 'never'");
}
// PROPAGATION_NOT_SUPPORTED不支持在事务中运行
if (definition.getPropagationBehavior() == TransactionDefinition.PROPAGATION_NOT_SUPPORTED) {
if (debugEnabled) {
logger.debug("Suspending current transaction");
}
// 挂起当前事务,以非事务方式运行
Object suspendedResources = suspend(transaction);
boolean newSynchronization = (getTransactionSynchronization() == SYNCHRONIZATION_ALWAYS);
// 创建不支持事务的 TransactionStatus,因此 newTransaction 为 false
return prepareTransactionStatus(
definition, null, false, newSynchronization, debugEnabled, suspendedResources);
}
// PROPAGATION_REQUIRES_NEW需要开启新事务
// 流程和之前分析的一致,只不过这里不再传 null 了而已
if (definition.getPropagationBehavior() == TransactionDefinition.PROPAGATION_REQUIRES_NEW) {
if (debugEnabled) {
logger.debug("Suspending current transaction, creating new transaction with name [" +
definition.getName() + "]");
}
// 挂起已有事务
SuspendedResourcesHolder suspendedResources = suspend(transaction);
try {
// 开启新事务
return startTransaction(definition, transaction, debugEnabled, suspendedResources);
}
catch (RuntimeException | Error beginEx) {
// 出现异常后恢复已有事务
resumeAfterBeginException(transaction, suspendedResources, beginEx);
throw beginEx;
}
}
// PROPAGATION_NESTED在已有事务的情况下是创建嵌套事务,也就是Savepoint
// 限于篇幅,Savepoint相关的我们就不讲了,用得也不是那么多
if (definition.getPropagationBehavior() == TransactionDefinition.PROPAGATION_NESTED) {
if (!isNestedTransactionAllowed()) {
throw new NestedTransactionNotSupportedException(
"Transaction manager does not allow nested transactions by default - " +
"specify 'nestedTransactionAllowed' property with value 'true'");
}
if (debugEnabled) {
logger.debug("Creating nested transaction with name [" + definition.getName() + "]");
}
if (useSavepointForNestedTransaction()) {
DefaultTransactionStatus status =
prepareTransactionStatus(definition, transaction, false, false, debugEnabled, null);
status.createAndHoldSavepoint();
return status;
}
else {
return startTransaction(definition, transaction, debugEnabled, null);
}
}
// 剩下来还能发挥作用的就只有PROPAGATION_SUPPORTS 和 PROPAGATION_REQUIRED 了
// 这两在已有事务的情况下行为是一致的,都是加入已有事务
if (debugEnabled) {
logger.debug("Participating in existing transaction");
}
// 检验事务配置
if (isValidateExistingTransaction()) {
// 隔离级别需保持一致
if (definition.getIsolationLevel() != TransactionDefinition.ISOLATION_DEFAULT) {
Integer currentIsolationLevel = TransactionSynchronizationManager.getCurrentTransactionIsolationLevel();
if (currentIsolationLevel == null || currentIsolationLevel != definition.getIsolationLevel()) {
Constants isoConstants = DefaultTransactionDefinition.constants;
throw new IllegalTransactionStateException("Participating transaction with definition [" +
definition + "] specifies isolation level which is incompatible with existing transaction: " +
(currentIsolationLevel != null ?
isoConstants.toCode(currentIsolationLevel, DefaultTransactionDefinition.PREFIX_ISOLATION) :
"(unknown)"));
}
}
// read-only 标记要一致
if (!definition.isReadOnly()) {
if (TransactionSynchronizationManager.isCurrentTransactionReadOnly()) {
throw new IllegalTransactionStateException("Participating transaction with definition [" +
definition + "] is not marked as read-only but existing transaction is");
}
}
}
boolean newSynchronization = (getTransactionSynchronization() != SYNCHRONIZATION_NEVER);
// 创建加入已有事务的 TransactionStatus,因此 newTransaction 为false
return prepareTransactionStatus(definition, transaction, false, newSynchronization, debugEnabled, null);
}
protected final DefaultTransactionStatus prepareTransactionStatus(
TransactionDefinition definition, @Nullable Object transaction, boolean newTransaction,
boolean newSynchronization, boolean debug, @Nullable Object suspendedResources) {
// 创建 TransactionStatus
DefaultTransactionStatus status = newTransactionStatus(
definition, transaction, newTransaction, newSynchronization, debug, suspendedResources);
// 按需初始化事务同步机制
prepareSynchronization(status, definition);
return status;
}
handleExistingTransaction(...)
也对不同的传播行为做了处理,至于其中涉及到的方法调用,前面都已经分析过了。各位同学在自己看源码的时候,建议动手调试调试,彻底梳理清楚其间的状态标志及其转换为好。
rollback
嘟嘟嘟,CRUD出异常了,这会该回滚事务了。
@Override
public final void rollback(TransactionStatus status) throws TransactionException {
// 同一个事务不能多次回滚
if (status.isCompleted()) {
throw new IllegalTransactionStateException(
"Transaction is already completed - do not call commit or rollback more than once per transaction");
}
DefaultTransactionStatus defStatus = (DefaultTransactionStatus) status;
// 处理回滚,unexpected用于标记这是不是一个预期之外的回滚
// 方法执行出现异常,且符合rollbackOn配置,那么回滚是正常的,否则就是不正常的
processRollback(defStatus, false);
}
private void processRollback(DefaultTransactionStatus status, boolean unexpected) {
try {
// 预期之外的异常?
boolean unexpectedRollback = unexpected;
try {
// 回调 TransactionSynchronization#beforeCompletion()
triggerBeforeCompletion(status);
// 有Savepoint,回滚到保存点
if (status.hasSavepoint()) {
if (status.isDebug()) {
logger.debug("Rolling back transaction to savepoint");
}
status.rollbackToHeldSavepoint();
}
// 独立事务
else if (status.isNewTransaction()) {
if (status.isDebug()) {
logger.debug("Initiating transaction rollback");
}
// 调用模板方法进行回滚
doRollback(status);
}
else {
// 非独立事务,加入的一个已存在的事务
if (status.hasTransaction()) {
// 已标记为rollback-only,出现于多个事务方法的嵌套调用,比如 A -> B -> C,且传播行为的配置不会导致开启多个事务
// 或者配置了外层事务会因为内层事务的异常而回滚,默认true
if (status.isLocalRollbackOnly() || isGlobalRollbackOnParticipationFailure()) {
if (status.isDebug()) {
logger.debug("Participating transaction failed - marking existing transaction as rollback-only");
}
// 调用模板方法设置rollback-only标记
doSetRollbackOnly(status);
}
else {
if (status.isDebug()) {
logger.debug("Participating transaction failed - letting transaction originator decide on rollback");
}
}
}
else {
logger.debug("Should roll back transaction but cannot - no transaction available");
}
// 不需要fail-fast,清楚标志,后续不抛异常
if (!isFailEarlyOnGlobalRollbackOnly()) {
unexpectedRollback = false;
}
}
}
catch (RuntimeException | Error ex) {
// 回调 TransactionSynchronization#afterCompletion(...),状态未知
triggerAfterCompletion(status, TransactionSynchronization.STATUS_UNKNOWN);
throw ex;
}
// 回调 TransactionSynchronization#afterCompletion(...),已回滚
triggerAfterCompletion(status, TransactionSynchronization.STATUS_ROLLED_BACK);
// 抛出异常
if (unexpectedRollback) {
throw new UnexpectedRollbackException(
"Transaction rolled back because it has been marked as rollback-only");
}
}
finally {
// 清理
cleanupAfterCompletion(status);
}
}
回滚分为三种情况:
- 嵌套事务(创建有
Savepoint
)回滚 - 独立事务回滚
- 非独立事务回滚
篇幅有限,Savepoint
就不说了,重点看2、3两种情况。对于独立事务,调用doRollback(...)
进行回滚,基于java.sql.Connection
的回滚机制。
@Override
protected void doRollback(DefaultTransactionStatus status) {
DataSourceTransactionObject txObject = (DataSourceTransactionObject) status.getTransaction();
Connection con = txObject.getConnectionHolder().getConnection();
if (status.isDebug()) {
logger.debug("Rolling back JDBC transaction on Connection [" + con + "]");
}
try {
// 回滚
con.rollback();
}
catch (SQLException ex) {
throw new TransactionSystemException("Could not roll back JDBC transaction", ex);
}
}
对于非独立事务,调用doSetRollbackOnly(...)
,仅仅是打上rollback-only
标记用于后续的检测,processRollback(...)
本身就对这个标记进行了检测。
@Override
protected void doSetRollbackOnly(DefaultTransactionStatus status) {
DataSourceTransactionObject txObject = (DataSourceTransactionObject) status.getTransaction();
if (status.isDebug()) {
logger.debug("Setting JDBC transaction [" + txObject.getConnectionHolder().getConnection() +
"] rollback-only");
}
// 代理给ConnectionHolder,打上rollback-only标记
txObject.setRollbackOnly();
}
能做的都做了,最后该清理现场了。
private void cleanupAfterCompletion(DefaultTransactionStatus status) {
// 标记事务已完成
status.setCompleted();
// 若开启了事务同步机制,清除相关的线程私有变量
if (status.isNewSynchronization()) {
TransactionSynchronizationManager.clear();
}
// 若是独立事务,调用模板方法进行清理
if (status.isNewTransaction()) {
doCleanupAfterCompletion(status.getTransaction());
}
// 若还存在外层事务
if (status.getSuspendedResources() != null) {
if (status.isDebug()) {
logger.debug("Resuming suspended transaction after completion of inner transaction");
}
// 恢复外层事务
Object transaction = (status.hasTransaction() ? status.getTransaction() : null);
resume(transaction, (SuspendedResourcesHolder) status.getSuspendedResources());
}
}
/**
* 基本上是 doBegin(...)的逆操作
*/
@Override
protected void doCleanupAfterCompletion(Object transaction) {
DataSourceTransactionObject txObject = (DataSourceTransactionObject) transaction;
// 解绑
if (txObject.isNewConnectionHolder()) {
TransactionSynchronizationManager.unbindResource(obtainDataSource());
}
Connection con = txObject.getConnectionHolder().getConnection();
try {
// 如有必要,恢复自动提交
if (txObject.isMustRestoreAutoCommit()) {
con.setAutoCommit(true);
}
// 恢复原有隔离级别和readOnly标记
DataSourceUtils.resetConnectionAfterTransaction(
con, txObject.getPreviousIsolationLevel(), txObject.isReadOnly());
}
catch (Throwable ex) {
logger.debug("Could not reset JDBC Connection after transaction", ex);
}
// 若是独立事务,释放连接
if (txObject.isNewConnectionHolder()) {
if (logger.isDebugEnabled()) {
logger.debug("Releasing JDBC Connection [" + con + "] after transaction");
}
DataSourceUtils.releaseConnection(con, this.dataSource);
}
// 清除ConnectionHolder的状态
txObject.getConnectionHolder().clear();
}
commit
要是没出什么幺蛾子,就可以提交事务了。
@Override
public final void commit(TransactionStatus status) throws TransactionException {
// 同一个事务不能重复提交
if (status.isCompleted()) {
throw new IllegalTransactionStateException(
"Transaction is already completed - do not call commit or rollback more than once per transaction");
}
DefaultTransactionStatus defStatus = (DefaultTransactionStatus) status;
// 当前事务被标记为rollback-only
// 何时打上rollback-only标记可参考对 rollback 的分析
if (defStatus.isLocalRollbackOnly()) {
if (defStatus.isDebug()) {
logger.debug("Transactional code has requested rollback");
}
// 执行回滚而不是提交
processRollback(defStatus, false);
return;
}
// 全局回滚,XA事务可能会有吧
if (!shouldCommitOnGlobalRollbackOnly() && defStatus.isGlobalRollbackOnly()) {
if (defStatus.isDebug()) {
logger.debug("Global transaction is marked as rollback-only but transactional code requested commit");
}
// 执行回滚,unexpected为true
// 比如一个XA事务,事务A执行正常,事务B执行异常
// 此时这个大事务是不能提交的,事务协调者会标记全局rollback-only
// 这样就可以让原本正常的事务A也回滚而不是提交
processRollback(defStatus, true);
return;
}
// 终于可以提交了,开心😄
processCommit(defStatus);
}
private void processCommit(DefaultTransactionStatus status) throws TransactionException {
try {
boolean beforeCompletionInvoked = false;
try {
boolean unexpectedRollback = false;
// 准备提交,扩展点
prepareForCommit(status);
// 回调 TransactionSynchronization#beforeCommit()
triggerBeforeCommit(status);
// 回调 TransactionSynchronization#beforeCompletion()
triggerBeforeCompletion(status);
beforeCompletionInvoked = true;
// Savepoint,忽略不说了
if (status.hasSavepoint()) {
if (status.isDebug()) {
logger.debug("Releasing transaction savepoint");
}
unexpectedRollback = status.isGlobalRollbackOnly();
status.releaseHeldSavepoint();
}
// 独立事务
else if (status.isNewTransaction()) {
if (status.isDebug()) {
logger.debug("Initiating transaction commit");
}
// 读取全局rollback-only标记
unexpectedRollback = status.isGlobalRollbackOnly();
// 调用模板方法进行提交
doCommit(status);
}
// fail-fast,读取全局rollback-only标记
else if (isFailEarlyOnGlobalRollbackOnly()) {
unexpectedRollback = status.isGlobalRollbackOnly();
}
// 抛出异常,如果全局标记有 rollback-only
if (unexpectedRollback) {
throw new UnexpectedRollbackException(
"Transaction silently rolled back because it has been marked as rollback-only");
}
}
catch (UnexpectedRollbackException ex) {
// 回调 TransactionSynchronization#afterCompletion(...)
// 对应上面抛出的异常,此时事务是回滚的而不是已提交
triggerAfterCompletion(status, TransactionSynchronization.STATUS_ROLLED_BACK);
throw ex;
}
catch (TransactionException ex) {
// 如果需要在提交失败时进行回滚的话,默认false
if (isRollbackOnCommitFailure()) {
doRollbackOnCommitException(status, ex);
}
else {
// 回调 TransactionSynchronization#afterCompletion(...),状态未知
triggerAfterCompletion(status, TransactionSynchronization.STATUS_UNKNOWN);
}
throw ex;
}
catch (RuntimeException | Error ex) {
// 确保 TransactionSynchronization#beforeCompletion()得到执行
if (!beforeCompletionInvoked) {
triggerBeforeCompletion(status);
}
// 回滚事务,如果提交失败的话,注意与前面遇到TransactionException时处理的异同
doRollbackOnCommitException(status, ex);
throw ex;
}
// 回调 TransactionSynchronization#afterCommit()
try {
triggerAfterCommit(status);
}
finally {
// 回调 TransactionSynchronization#afterCompletion(...),状态已提交
triggerAfterCompletion(status, TransactionSynchronization.STATUS_COMMITTED);
}
}
finally {
// 清理
cleanupAfterCompletion(status);
}
}
commit
相对来讲是比较复杂的,它牵涉到对rollback-only
标记的处理。这里解释一下global rollback only
,区别于local rollback only
,这个标记多用于多事务协作的情况,比如两阶段提交协议下的分布式事务。
区别于rollback
,commit
不需要处理非独立事务的情况,内层事务不需要提交,它只需要等待外层事务提交即可。对于独立事务,模板方法doCommit(...)
也只是简单的代理给java.sql.Connection
的提交机制。
@Override
protected void doCommit(DefaultTransactionStatus status) {
DataSourceTransactionObject txObject = (DataSourceTransactionObject) status.getTransaction();
Connection con = txObject.getConnectionHolder().getConnection();
if (status.isDebug()) {
logger.debug("Committing JDBC transaction on Connection [" + con + "]");
}
try {
con.commit();
}
catch (SQLException ex) {
// processCommit(...)对doCommit(...)抛出的TransactionException有特殊处理
throw new TransactionSystemException("Could not commit JDBC transaction", ex);
}
}
后记
最后回顾一下我们的小例子,现在它还有秘密吗?本篇我们一起探索了PlatformTransactionManeger
抽象下的事务管理机制,篇幅较长,相信耐心看完的你一定有所收获😊