前言
在Seata官网中,我们可以知道AT模式一阶段的处理流程如下:
1.解析 SQL:得到 SQL 的类型(UPDATE),表(product),条件(where name = 'TXC')等相关的信息。
2.查询前镜像:根据解析得到的条件信息,生成查询语句,定位数据。
3.执行业务 SQL。
4.查询后镜像:根据前镜像的结果,通过 主键 定位数据。
5.插入回滚日志:把前后镜像数据以及业务 SQL 相关的信息组成一条回滚日志记录,插入到
UNDO_LOG
表中......
前镜像的作用是保证在分布式事务失败时能够成功回滚的重要依据,后镜像是在回滚前校验是否脏写的数据依据,那么我们一阶段的前后镜像在真实的代码实现中是如何生成的呢?
前后镜像的生成
为了能够探寻到前后镜像的生成原理,我们需要看一看seata源码。最终我们把入口定位到AbstractDMLBaseExecutor.executeAutoCommitFalse()
方法:
protected T executeAutoCommitFalse(Object[] args) throws Exception { // 获取前镜像 TableRecords beforeImage = beforeImage(); // 执行业务SQL T result = statementCallback.execute(statementProxy.getTargetStatement(), args); // 获取后镜像 TableRecords afterImage = afterImage(beforeImage); // 存储前后镜像 prepareUndoLog(beforeImage, afterImage); return result; } 复制代码
- 前镜像
继续深入探究一下到底是怎么获取beforeImage
的,根据源码来看,我们发现beforeImage()
方法有很多实现:
也就是说,Seata会根据不同的业务SQL来生成beforeImage
,有点经验的小伙伴能够看出,这里其实使用到了模版模式加上策略模式,我们挑一个DeleteExecutor
来看一下:
@Override protected TableRecords beforeImage() throws SQLException { SQLDeleteRecognizer visitor = (SQLDeleteRecognizer) sqlRecognizer; // 根据表名解析出元数据 TableMeta tmeta = getTableMeta(visitor.getTableName()); ArrayList<List<Object>> paramAppenderList = new ArrayList<>(); // 生成查询beforeImage的SQL语句 // SELECT [列名,] FROM [表名] (别名) (WHERE) (ORDER BY) (LIMIT) FOR UPDATE String selectSQL = buildBeforeImageSQL(visitor, tmeta, paramAppenderList); // 执行SQL查询beforeImage return buildTableRecords(tmeta, selectSQL, paramAppenderList); } 复制代码
1.关键原理就是根据业务SQL反向查询出被影响的数据;
2.为了保证查询到的数据不是快照数据,一定要记得加上
FOR UPDATE
;
另外的话,我们发现其实UpdateExecutor
的前镜像生成方式和DeleteExecutor
也差不多,像普通的insert
这种SQL的前镜像就更简单了:
@Override protected TableRecords beforeImage() throws SQLException { return TableRecords.empty(getTableMeta()); } 复制代码
因为普通insert
语句不存在任何前镜像,所以直接返回空记录;
- 后镜像
我们再来看一下后镜像是如何生成的,这次我们看一下UpdateExecutor
的后镜像生成方法afterImage()
:
@Override protected TableRecords afterImage(TableRecords beforeImage) throws SQLException { // 获取元数据 TableMeta tmeta = getTableMeta(); // 没有前镜像,也不存在后镜像,说明没有数据被修改 if (beforeImage == null || beforeImage.size() == 0) { return TableRecords.empty(getTableMeta()); } // 生成查询后镜像SQL // SELECT [列名,] FROM [表名] (别名) WHERE 主键 in (前镜像的主键值) // 这里面有一个配置项[client.undo.onlyCareUpdateColumns],是否只关心被修改的列名,默认是true String selectSQL = buildAfterImageSQL(tmeta, beforeImage); ResultSet rs = null; try (PreparedStatement pst = statementProxy.getConnection().prepareStatement(selectSQL)) { SqlGenerateUtils.setParamForPk(beforeImage.pkRows(), getTableMeta().getPrimaryKeyOnlyName(), pst); // 执行查询后镜像 rs = pst.executeQuery(); // 包装查询结果 return TableRecords.buildRecords(tmeta, rs); } finally { IOUtil.close(rs); } } 复制代码
后镜像的生成原理与前镜像的生成原理差不多,不过还是有一些小小的区别的:
1.后镜像的查询条件使用的是前镜像对应的主键值,就没有用业务SQL的查询条件;不同的Executor处理方式不同,需要根据具体的业务SQL来区分;
2.查询后镜像的SQL没有使用
FOR UPDATE
加锁,直接拿的快照数据;
小结
通过对seata源码的分析,我们现在已经了解了前后镜像的生成原理了:
1.通过业务SQL来判断SQL语句的类型,从而选择不同的Executor来获取前后镜像;
2.前镜像是通过业务SQL的查询条件,并加上
FOR UPDATE
来查询业务SQL执行前的数据;(不同的Executor实现不同)3.后镜像是在业务SQL执行完毕后,根据前镜像内的主键数据来获取的数据;(不同的Executor实现不同)
4.通过前后镜像的多种实现可以判断出seata AT模式所支持的SQL语句的所有类型;
作者:梦想实现家_Z
链接:https://juejin.cn/post/7164311701595095053
来源:稀土掘金
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。