基于spring,Atomikos,mybatis的分布式动态数据源JTA实现

     本文的几个关键词，分布式数据源，数据源的动态寻找，分布式事务JTA实现。
     对于一些较大规模的应用，单个数据源是无法支撑起庞大的用户量，需要引入多数据源，水平层面进行分库分表，降低单个DB的负载。接下来，我们程序里里面需 要管理不同数据源之前的程序调用，保证功能是WORK的。另外，跨库就意味着之前单DB的事务就失效了，所以J2EE提出了JTA，分布式的事务管理，往 简单了说，就是2步提交（two phase）,比单步提交更苛刻。实际上他有两个容器来管理，一个是资源管理器，一个是事务管理。小伙伴们可以发现，这是一个环环相扣的过程。想解决一个 问题，你就得解决这几个相关的问题。以下代码，我也是参考了前辈们的思想，进行了改造。

    第一步：XA数据源定义
     选定义一个抽象的父类源，这样子类可以直接继承

  

 1  <!-- 两个数据源的功用配置，方便下面直接引用 -->
 2      <bean id="abstractXADataSource" class="com.atomikos.jdbc.AtomikosDataSourceBean" init-method="init"
 3              destroy-method="close" abstract="true">
 4         <property name="xaDataSourceClassName" value="com.mysql.jdbc.jdbc2.optional.MysqlXADataSource"/>
 5         <property name="poolSize" value="10" />
 6         <property name="minPoolSize" value="10"/>
 7         <property name="maxPoolSize" value="30"/>
 8         <property name="borrowConnectionTimeout" value="60"/>
 9         <property name="reapTimeout" value="20"/>
10         <!-- 最大空闲时间 -->
11         <property name="maxIdleTime" value="60"/>
12         <property name="maintenanceInterval" value="60" />
13         <property name="loginTimeout" value="60"/>
14         <property name="logWriter" value="60"/>
15         <property name="testQuery">
16             <value>select 1</value>
17         </property>
18         
19     </bean>
20 

   A源
     

 1 <!-- 配置第一个数据源 -->
 2     <bean id="dataSource_a" parent="abstractXADataSource">
 3     <!-- value只要两个数据源不同就行，随便取名 -->
 4         <property name="uniqueResourceName" value="mysql/sitestone" />
 5         <property name="xaDataSourceClassName"
 6             value="com.mysql.jdbc.jdbc2.optional.MysqlXADataSource" />
 7         <property name="xaProperties">
 8             <props>
 9                 <prop key="URL">${jdbc.url.spider}</prop>
10                 <prop key="user">${jdbc.username}</prop>
11                 <prop key="password">${jdbc.password}</prop>
12             </props>
13         </property>
14     </bean>

   

B   源

   

 1 <!-- 配置第二个数据源-->
 2     <bean id="dataSource_b" parent="abstractXADataSource">
 3 <!-- value只要两个数据源不同就行，随便取名 -->
 4         <property name="uniqueResourceName" value="mysql/sitesttwo" />
 5         <property name="xaDataSourceClassName"
 6             value="com.mysql.jdbc.jdbc2.optional.MysqlXADataSource" />
 7         <property name="xaProperties">
 8             <props>
 9                <prop key="URL">${jdbc_tb.url.spider}</prop>
10                 <prop key="user">${jdbc_tb.username}</prop>
11                 <prop key="password">${jdbc_tb.password}</prop>
12             </props>
13         </property>
14     </bean>

基于SPRING的AbstractRoutingDataSource动态数据路由定义

1   <bean name="dynamicDatasource" class="com.***.spring.datasource.CustomerDatasource">
2         <property name="targetDataSources">
3             <map>
4                 <entry key="ds_1" value-ref="dataSource_a"/>
5                 <entry key="ds_2" value-ref="dataSource_b"/>
6             </map>
7         </property>
8         <property name="defaultTargetDataSource" ref="dataSource_a"    />
9     </bean>

我这里是使用MYBATIS来进行ORM映射，配置如下

 1  <bean id="sqlSessionFactorya" class="org.mybatis.spring.SqlSessionFactoryBean">
 2         <property name="dataSource" ref="dataSource_a"/>
 3          <property name="typeAliasesPackage" value="com.****.spring.dschange.bean" />
 4          <!-- mapper和resultmap配置路径 --> 
 5         <property name="mapperLocations">
 6             <list>
 7                 <!-- 表示在com.**目录下的任意包下的resultmap包目录中，以-resultmap.xml或-mapper.xml结尾所有文件 --> 
 8                 <value>classpath:com/***/spring/dschange/mapper/ShopMapper.xml</value>
 9             </list>
10         </property>
11     </bean>
12     <bean id="sqlSessionFactoryb" class="org.mybatis.spring.SqlSessionFactoryBean">
13         <property name="dataSource" ref="dataSource_b"/>
14          <property name="typeAliasesPackage" value="com.****.spring.dschange.bean" />
15           <!-- mapper和resultmap配置路径 --> 
16         <property name="mapperLocations">
17             <list>
18                 <!-- 表示在com.***目录下的任意包下的resultmap包目录中，以-resultmap.xml或-mapper.xml结尾所有文件 --> 
19                 <value>classpath:com/***/spring/dschange/mapper/ShopMapper.xml</value>
20             </list>
21         </property>
22     </bean>

接下来，一个比较关键的地方是对MYBATIS的CustomSqlSessionTemplate的重写，主要是引入动态数据源sqlSessionFactory。针对不同的数据库，调用其对应的会话工厂，这对JTA是否启用，比较重要。

 1  @Override
 2     public SqlSessionFactory getSqlSessionFactory() {
 3  
 4         SqlSessionFactory targetSqlSessionFactory = targetSqlSessionFactorys.get(DataSourceKeyHolder.getDataSourceKey());
 5         if (targetSqlSessionFactory != null) {
 6             return targetSqlSessionFactory;
 7         } else if (defaultTargetSqlSessionFactory != null) {
 8             return defaultTargetSqlSessionFactory;
 9         } else {
10             Assert.notNull(targetSqlSessionFactorys, "Property 'targetSqlSessionFactorys' or 'defaultTargetSqlSessionFactory' are required");
11             Assert.notNull(defaultTargetSqlSessionFactory, "Property 'defaultTargetSqlSessionFactory' or 'targetSqlSessionFactorys' are required");
12         }
13         return this.sqlSessionFactory;
14     }
XML配置

 1  <!-- 配置自定义的SqlSessionTemplate模板，注入相关配置 -->
 2     <bean id="sqlSessionTemplate" class="com.amos.spring.mybatis.CustomSqlSessionTemplate" scope="prototype">
 3         <constructor-arg ref="sqlSessionFactorya" />
 4         <property name="targetSqlSessionFactorys">
 5             <map>     
 6                 <entry value-ref="sqlSessionFactorya" key="ds_1"/>
 7                 <entry value-ref="sqlSessionFactoryb" key="ds_2"/>
 8             </map> 
 9         </property>
10     </bean>

扫描配置

1  <bean id="mapperScannerConfigurer" class="org.mybatis.spring.mapper.MapperScannerConfigurer">

 1     <!-- jta -->
 2     <bean id="atomikosTransactionManager" class="com.atomikos.icatch.jta.UserTransactionManager"
 3         init-method="init" destroy-method="close">
 4         <property name="forceShutdown">
 5             <value>true</value>
 6         </property>
 7     </bean>
 8  
 9     <bean id="atomikosUserTransaction" class="com.atomikos.icatch.jta.UserTransactionImp">
10         <property name="transactionTimeout" value="300" />
11     </bean>
12  
13     <bean id="springTransactionManager"
14         class="org.springframework.transaction.jta.JtaTransactionManager">
15         <property name="transactionManager">
16             <ref bean="atomikosTransactionManager" />
17         </property>
18         <property name="userTransaction">
19             <ref bean="atomikosUserTransaction" />
20         </property>
21     </bean>
22  <tx:annotation-driven transaction-manager="springTransactionManager" proxy-target-class="true" />

2         <property name="basePackage" value="com.****.spring.dschange.mapper" />
3         <property name="sqlSessionTemplateBeanName" value="sqlSessionTemplate"/>
4         <property name="markerInterface" value="com.*****.spring.dschange.mapper.SqlMapper"/>
5     </bean>

最后就是JTA的实现配置

 1     <!-- jta -->
 2     <bean id="atomikosTransactionManager" class="com.atomikos.icatch.jta.UserTransactionManager"
 3         init-method="init" destroy-method="close">
 4         <property name="forceShutdown">
 5             <value>true</value>
 6         </property>
 7     </bean>
 8  
 9     <bean id="atomikosUserTransaction" class="com.atomikos.icatch.jta.UserTransactionImp">
10         <property name="transactionTimeout" value="300" />
11     </bean>
12  
13     <bean id="springTransactionManager"
14         class="org.springframework.transaction.jta.JtaTransactionManager">
15         <property name="transactionManager">
16             <ref bean="atomikosTransactionManager" />
17         </property>
18         <property name="userTransaction">
19             <ref bean="atomikosUserTransaction" />
20         </property>
21     </bean>
22  <tx:annotation-driven transaction-manager="springTransactionManager" proxy-target-class="true" />

以上,XML配置相关的东西已经完成。
具体的代码，请点击GITHUB查看https://github.com/igool/spring-jta-mybatis