Loading

Seata源码之开篇

1.源码下载

官方地址:https://seata.io/zh-cn/blog/download.html

2.AT模式回顾

官方文档:https://seata.io/zh-cn/docs/dev/mode/at-mode.html

写隔离

  • 一阶段本地事务提交前,需要确保先拿到全局锁
  • 拿不到全局锁 ,不能提交本地事务。
  • 全局锁 的尝试被限制在一定范围内,超出范围将放弃,并回滚本地事务,释放本地锁。

image

如果 tx1 的二阶段全局回滚,则 tx1 需要重新获取该数据的本地锁,进行反向补偿的更新操作,实现分支的回滚。

此时,如果 tx2 仍在等待该数据的 全局锁,同时持有本地锁,则 tx1 的分支回滚会失败。分支的回滚会一直重试,直到 tx2 的 全局锁 等锁超时,放弃 全局锁 并回滚本地事务释放本地锁,tx1 的分支回滚最终成功。

因为整个过程 全局锁 在 tx1 结束前一直是被 tx1 持有的,所以不会发生 脏写 的问题。

读隔离

在数据库本地事务隔离级别 读已提交(Read Committed) 或以上的基础上,Seata(AT 模式)的默认全局隔离级别是 读未提交(Read Uncommitted)

如果应用在特定场景下,必需要求全局的 读已提交 ,目前 Seata 的方式是通过 SELECT FOR UPDATE 语句的代理。

image

SELECT FOR UPDATE 语句的执行会申请 全局锁 ,如果 全局锁 被其他事务持有,则释放本地锁(回滚 SELECT FOR UPDATE 语句的本地执行)并重试。这个过程中,查询是被 block 住的,直到 全局锁 拿到,即读取的相关数据是 已提交 的,才返回。

出于总体性能上的考虑,Seata 目前的方案并没有对所有 SELECT 语句都进行代理,仅针对 FOR UPDATE 的 SELECT 语句。

AT二阶段

一阶段:

  1. 解析 SQL:得到 SQL 的类型(UPDATE),表(product),条件(where name = 'TXC')等相关的信息。
  2. 查询前镜像(改变之前的数据):根据解析得到的条件信息,生成查询语句,定位数据。
  3. 执行业务 SQL:更新这条数据。
  4. 查询后镜像(改变后的数据):根据前镜像的结果,通过主键 定位数据。
  5. 插入回滚日志:把前后镜像数据以及业务 SQL 相关的信息组成一条回滚日志记录,插入到UNDO_LOG 表中。
  6. 提交前,向 TC 注册分支:申请全局锁
  7. 本地事务提交:业务数据的更新和前面步骤中生成的 UNDO LOG 一并提交。
  8. 将本地事务提交的结果上报给 TC。

二阶段-回滚:

  1. 收到 TC 的分支回滚请求,开启一个本地事务,执行如下操作。
  2. 通过 XID 和 Branch ID 查找到相应的 UNDO LOG 记录。
  3. 根据 UNDO LOG 中的前镜像和业务 SQL 的相关信息生成并执行回滚的语句:
  4. 提交本地事务。并把本地事务的执行结果(即分支事务回滚的结果)上报给 TC。

二阶段-提交:

  1. 收到 TC 的分支提交请求,把请求放入一个异步任务的队列中,马上返回提交成功的结果给 TC。
  2. 异步任务阶段的分支提交请求将异步和批量地删除相应 UNDO LOG 记录。

3.源码入口

Seata客户端启动

首先一个Seata的客户端启动一般分为几个流程:

  1. 自动加载各种Bean及配置信息
  2. 初始化TM
  3. 初始化RM(具体服务)
  4. 初始化分布式事务客户端完成,代理数据源
  5. 连接TC(Seata服务端),注册RM,注册TM
  6. 开启全局事务

在这个其中,就会涉及到几个核心的类型,首先我们需要来看配置类型GlobalTransactionAutoConfiguration

所以我们直接通过官方案例引入的Seata包,找到SpringBoot项目在启动的时候自动扫描加载类型的spring.factories,然后找到GlobalTransactionAutoConfiguration(Seata自动配置类)

全局事务扫描类源码

这个类型的核心点,就是加载配置,注入相关的Bean

/**
 * seata自动配置类
 */
@Configuration
@EnableConfigurationProperties(SeataProperties.class)
public class GlobalTransactionAutoConfiguration {

	private final ApplicationContext applicationContext;

	private final SeataProperties seataProperties;

	public GlobalTransactionAutoConfiguration(ApplicationContext applicationContext,
			SeataProperties seataProperties) {
		this.applicationContext = applicationContext;
		this.seataProperties = seataProperties;
	}
	// 注入全局事务扫描器
	@Bean
	public GlobalTransactionScanner globalTransactionScanner() {

		String applicationName = applicationContext.getEnvironment()
				.getProperty("spring.application.name");

		String txServiceGroup = seataProperties.getTxServiceGroup();

		if (StringUtils.isEmpty(txServiceGroup)) {
			txServiceGroup = applicationName + "-fescar-service-group";
			seataProperties.setTxServiceGroup(txServiceGroup);
		}
		// 构建全局扫描器,传入参数:应用名、事务分组名,失败处理器
		return new GlobalTransactionScanner(applicationName, txServiceGroup);
	}
}

GlobalTransactionScanner全局事务扫描器

在这其中我们要关心的是GlobalTransactionScanner这个类型,这个类型扫描@GlobalTransactional注解,并对代理方法进行拦截增强事务的功能。

image

这里给大家展示了当前GlobalTransactionScanner的类关系图,其中我们现在继承了Aop的AbstractAutoProxyCreator类型,在这其中有一个重点方法,其实这个方法就是判断Bean对象是否需要代理,是否需要增强

protected Object wrapIfNecessary(Object bean, String beanName, Object cacheKey) {
   if (beanName != null && this.targetSourcedBeans.contains(beanName)) {
       return bean;
   }
   if (Boolean.FALSE.equals(this.advisedBeans.get(cacheKey))) {
       return bean;
   }
   if (isInfrastructureClass(bean.getClass()) || shouldSkip(bean.getClass(), beanName)) {
       this.advisedBeans.put(cacheKey, Boolean.FALSE);
       return bean;
   }

   // Create proxy if we have advice.
   Object[] specificInterceptors = getAdvicesAndAdvisorsForBean(bean.getClass(), beanName, null);
   if (specificInterceptors != DO_NOT_PROXY) {
       this.advisedBeans.put(cacheKey, Boolean.TRUE);
       Object proxy = createProxy(
           bean.getClass(), beanName, specificInterceptors, new SingletonTargetSource(bean));
       this.proxyTypes.put(cacheKey, proxy.getClass());
       return proxy;
   }

   this.advisedBeans.put(cacheKey, Boolean.FALSE);
   return bean;
}

当然这是父类提供的方法,那子类继承之后重写此方法,完成了定制化的效果

@Override
protected Object wrapIfNecessary(Object bean, String beanName, Object cacheKey) {
   try {
       // 加锁防止并发
       synchronized (PROXYED_SET) {
           if (PROXYED_SET.contains(beanName)) {
               return bean;
           }
           interceptor = null;
           //check TCC proxy
           // 检查是否是TCC模式
           if (TCCBeanParserUtils.isTccAutoProxy(bean, beanName, applicationContext)) {
               //TCC interceptor, proxy bean of sofa:reference/dubbo:reference, and LocalTCC
               // 如果是:添加TCC拦截器
               interceptor = new TccActionInterceptor(TCCBeanParserUtils.getRemotingDesc(beanName));
               ConfigurationCache.addConfigListener(ConfigurationKeys.DISABLE_GLOBAL_TRANSACTION,
                                                    (ConfigurationChangeListener)interceptor);
           } else {
               // 不是TCC模式
               Class<?> serviceInterface = SpringProxyUtils.findTargetClass(bean);
               Class<?>[] interfacesIfJdk = SpringProxyUtils.findInterfaces(bean);

               // 判断是否有相关事务注解,如果没有就不代理
               if (!existsAnnotation(new Class[]{serviceInterface})
                   && !existsAnnotation(interfacesIfJdk)) {
                   return bean;
               }

               // 当发现存在全局事务注解标注的Bean,添加拦截器
               if (globalTransactionalInterceptor == null) {
                   // 添加拦截器
                   globalTransactionalInterceptor = new GlobalTransactionalInterceptor(failureHandlerHook);
                   ConfigurationCache.addConfigListener(
                       ConfigurationKeys.DISABLE_GLOBAL_TRANSACTION,
                       (ConfigurationChangeListener)globalTransactionalInterceptor);
               }
               interceptor = globalTransactionalInterceptor;
           }

           LOGGER.info("Bean[{}] with name [{}] would use interceptor [{}]", bean.getClass().getName(), beanName, interceptor.getClass().getName());
           // 检查是否是代理对象
           if (!AopUtils.isAopProxy(bean)) {
               // 不是调用Spring代理(父级)
               bean = super.wrapIfNecessary(bean, beanName, cacheKey);
           } else {
               // 已经是代理对象,反射获取代理类中的已经存在的拦截器组合,然后添加到该集合当中
               AdvisedSupport advised = SpringProxyUtils.getAdvisedSupport(bean);
               Advisor[] advisor = buildAdvisors(beanName, getAdvicesAndAdvisorsForBean(null, null, null));
               for (Advisor avr : advisor) {
                   advised.addAdvisor(0, avr);
               }
           }
           // 将Bean添加到Set中
           PROXYED_SET.add(beanName);
           return bean;
       }
   } catch (Exception exx) {
       throw new RuntimeException(exx);
   }
}
posted @ 2022-05-21 11:02  ZT丶  阅读(184)  评论(0编辑  收藏  举报