@Transactional和@DS在事務(wù)中切換數(shù)據(jù)源

在一次需求中，需要對(duì)兩個(gè)數(shù)據(jù)庫(kù)進(jìn)行讀寫(xiě)操作，并且要保證對(duì)這兩個(gè)庫(kù)的操作的原子性。

所以就在一個(gè)service方法中加入了@Transaction,方法中包含了對(duì)兩個(gè)庫(kù)的操作。

spring:
  datasource:
    dynamic:
      primary: A 
      datasource:
        A:
          url:..
          driver-class-name: com.mysql.jdbc.Driver
          username: root
          password:
        B:
          url: ..
          driver-class-name: com.mysql.jdbc.Driver
          username: root
          password:
        C:
          url: ..
          driver-class-name: com.mysql.jdbc.Driver
          username: root
          password:

@Mapper
@DS("A")
public interface AMapper{
      @Insert("insert into a ...")
      void save();
}

@Mapper
@DS("B")
public interface BMapper{
      @Insert("insert into b ...")
      void save();
}

public class aService{
@Autowired
private aMapper、bMapper、cMapper
    @Transactional
    public void save(){
	    aMapper.save();
	    bMapper.save(); #報(bào)錯(cuò)
    }    
}

在執(zhí)行測(cè)試時(shí)bMapper.save()發(fā)生了錯(cuò)誤，報(bào)錯(cuò)信息：找不到b表，很明顯sql是從A庫(kù)中找b表，當(dāng)然會(huì)報(bào)錯(cuò)找不到b表。

如果把@Transactional注解去掉就可以正常運(yùn)行，數(shù)據(jù)也成功分別寫(xiě)入兩個(gè)庫(kù)中。

但是我們命名已經(jīng)在2個(gè)Mapper上加了@Ds()注解來(lái)切換要是用的數(shù)據(jù)源，那為什么加入了事務(wù)就報(bào)錯(cuò)了呢？

@Transactional執(zhí)行流程

1、save方法添加了 @Transactional 注解，Spring 事務(wù)就會(huì)生效。此時(shí)，Spring TransactionInterceptor 會(huì)通過(guò) AOP 攔截該方法，創(chuàng)建事務(wù)。而創(chuàng)建事務(wù)，勢(shì)必就會(huì)獲得數(shù)據(jù)源。

• 那么，TransactionInterceptor 會(huì)使用 Spring DataSourceTransactionManager 創(chuàng)建事務(wù)，并將事務(wù)信息通過(guò) ThreadLocal 綁定在當(dāng)前線程。
• 而事務(wù)信息，就包括事務(wù)對(duì)應(yīng)的 Connection 連接。那也就意味著，還沒(méi)走到 OrderMapper 的查詢操作，Connection 就已經(jīng)被創(chuàng)建出來(lái)了。
• 并且，因?yàn)槭聞?wù)信息會(huì)和當(dāng)前線程綁定在一起，在 OrderMapper 在查詢操作需要獲得 Connection 時(shí)，就直接拿到當(dāng)前線程綁定的 Connection ，而不是 OrderMapper 添加 @DS 注解所對(duì)應(yīng)的 DataSource 所對(duì)應(yīng)的 Connection 。

2、OK ，那么我們現(xiàn)在可以把問(wèn)題聚焦到 DataSourceTransactionManager 是怎么獲取 DataSource 從而獲得 Connection 的了。

• 對(duì)于每個(gè) DataSourceTransactionManager 數(shù)據(jù)庫(kù)事務(wù)管理器，創(chuàng)建時(shí)都會(huì)傳入其需要管理的 DataSource 數(shù)據(jù)源。
• 在使用 dynamic-datasource-spring-boot-starter 時(shí)，它創(chuàng)建了一個(gè) DynamicRoutingDataSource ，傳入到 DataSourceTransactionManager 中。
• 而 DynamicRoutingDataSource 負(fù)責(zé)管理我們配置的多個(gè)數(shù)據(jù)源。例如說(shuō)，本示例中就管理了 a、b兩個(gè)數(shù)據(jù)源，并且默認(rèn)使用 a 數(shù)據(jù)源。
• 那么在當(dāng)前場(chǎng)景下，DynamicRoutingDataSource 需要基于 @DS 獲得數(shù)據(jù)源名，從而獲得對(duì)應(yīng)的 DataSource ，結(jié)果因?yàn)槲覀冊(cè)?Service 方法上，并沒(méi)有添加 @DS 注解，所以它只好返回默認(rèn)數(shù)據(jù)源，也就是 a 。故此，就發(fā)生了 找不到表 的異常。

我們?cè)谏厦媪私獾?，因?yàn)锧Transactional會(huì)創(chuàng)建事務(wù)然后獲得數(shù)據(jù)源，因?yàn)槲覀僺ervice方法上沒(méi)有@DS注解，就拿了默認(rèn)數(shù)據(jù)源，并且在這之后，這個(gè)事務(wù)信息會(huì)通過(guò)threadLocal跟當(dāng)前線程綁定，事務(wù)信息包括了connection連接，也就意味著，在進(jìn)入這個(gè)service方法的時(shí)候，當(dāng)前事務(wù)就綁定了數(shù)據(jù)源a，在運(yùn)行到bMapper.save()時(shí)，因?yàn)閏onnection已經(jīng)存在，所以拿到的數(shù)據(jù)源還是a，這時(shí)候就找不到b庫(kù)里的表了。

解決方法

把對(duì)B庫(kù)操作的方法都加上@DS和 @Transactional注解（在service上加）。

ps:在完成了aMapper.save()之后去調(diào)用bMapper.save()時(shí)，一定要把@Transactional設(shè)置為Propagation.REQUIRES_NEW，這樣在調(diào)用另一個(gè)事務(wù)方法時(shí)，TransactionInterceptor 會(huì)將原事務(wù)掛起，暫時(shí)性的將原事務(wù)信息和當(dāng)前線程解綁，然后創(chuàng)建一個(gè)新的事務(wù)，并且從數(shù)據(jù)源中取出一個(gè)connection鏈接，而此時(shí)的數(shù)據(jù)源已經(jīng)被切換成我們需要的數(shù)據(jù)源。如果B操作發(fā)生了異常，B事務(wù)將回滾并將異常進(jìn)行拋出到A，A事務(wù)自然也會(huì)回滾。

在對(duì)B庫(kù)的操作如果不需要事務(wù)的話，可以在對(duì)B庫(kù)操作的方法上指定事務(wù)的隔離級(jí)別為NOT_SUPPORTED，這樣執(zhí)行到該方法時(shí)會(huì)暫停掛起當(dāng)前事務(wù)，等待對(duì)B庫(kù)的方法執(zhí)行完畢再恢復(fù)事務(wù)。

pps：當(dāng)前事務(wù)方法里用this來(lái)調(diào)用另一個(gè)事務(wù)方法時(shí)，當(dāng)前這個(gè)事務(wù)方法的@DS也會(huì)起作用，原因是@Ds是基于AOP切面在方法執(zhí)行前切換數(shù)據(jù)源，而this調(diào)用的不是通過(guò)代理生成的事務(wù)對(duì)象，而是自己本身的原對(duì)象，不會(huì)開(kāi)啟事務(wù)。

總結(jié)

以上為個(gè)人經(jīng)驗(yàn)，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。

最新評(píng)論