spring动态切换数据库支持事务

http://blog.csdn.net/y_wave/article/details/52469795

在项目中有mysql的多个库，在代码中同一个方法可能会操作不同的表。在网上学习了各种方法。大概总结了一下。

1.mycat、cobar等分布式数据库中间件。

可以很好的支持，但是太重量级了，对我们项目有点大材小用。

2.spring的AbstractRoutingDataSource实现数据库连接切换。

可以动态的切换数据源，但是对事务有影响，可以用JTA实现事务一致，但是效率较低。而且我们项目事务可以单库一致就满足需求。所以采用了这种方式。

下面是具体的实现过程：

1）spring的配置文件中配置多个数据源。

[html] view plain copy

1. <bean id="dataSource" class="org.apache.commons.dbcp.BasicDataSource"  
2. destroy-method="close">  
3. <property name="driverClassName" value="com.mysql.jdbc.Driver" />  
4. <property name="url" value="${mysql.url}" />  
5. <property name="username" value="${mysql.username}" />  
6. <property name="password" value="${mysql.password}" />  
7. <!-- 连接池启动时的初始值 -->  
8. <property name="initialSize" value="1" />  
9. <!-- 最大空闲值.当经过一个高峰时间后，连接池可以慢慢将已经用不到的连接慢慢释放一部分，一直减少到maxIdle为止 -->  
10. <property name="maxIdle" value="2" />  
11. <!-- 最小空闲值.当空闲的连接数少于阀值时，连接池就会预申请一些连接，以避免洪峰来时再申请而造成的性能开销 -->  
12. <property name="minIdle" value="1" />  
13. </bean>  
14.   
15. <bean id="xiaomi" class="org.apache.commons.dbcp.BasicDataSource"  
16. destroy-method="close">  
17. <property name="driverClassName" value="com.mysql.jdbc.Driver" />  
18. <property name="url" value="${mysql.url.xiaomi}" />  
19. <property name="username" value="${mysql.username}" />  
20. <property name="password" value="${mysql.password}" />  
21. <!-- 连接池启动时的初始值 -->  
22. <property name="initialSize" value="1" />  
23. <!-- 最大空闲值.当经过一个高峰时间后，连接池可以慢慢将已经用不到的连接慢慢释放一部分，一直减少到maxIdle为止 -->  
24. <property name="maxIdle" value="2" />  
25. <!-- 最小空闲值.当空闲的连接数少于阀值时，连接池就会预申请一些连接，以避免洪峰来时再申请而造成的性能开销 -->  
26. <property name="minIdle" value="1" />  
27. </bean>  

2）定义动态的数据源

[html] view plain copy

1. <bean class="com.futuren.wzk.common.datasource.DynamicDataSource"  
2. id="dynamicDataSource">  
3. <property name="targetDataSources">  
4. <map key-type="java.lang.String">  
5. <entry value-ref="dataSource" key="wzk"></entry>  
6. <entry value-ref="xiaomi" key="xiaomi"></entry>  
7. </map>  
8. </property>  
9. <property name="defaultTargetDataSource" ref="dataSource" />  
10. </bean>  

3）定义动态数据源和辅助类

[java] view plain copy

1. public class DynamicDataSource extends AbstractRoutingDataSource {  
2. @Override  
3. protected Object determineCurrentLookupKey() {  
4. String type = DataSourceContextHolder.getDataSourceType();  
5. return type;  
6. }  
7. }  
8.   
9. public class DataSourceContextHolder {  
10. private static final ThreadLocal<String> contextHolder = new ThreadLocal<String>();  
11.   
12. /** 
13. * @Description: 设置数据源类型 
14. * @param dataSourceType 
15. * 数据库类型 
16. * @return void 
17. * @throws 
18. */  
19. public static void setDataSourceType(String dataSourceType) {  
20. contextHolder.set(dataSourceType);  
21. }  
22.   
23. /** 
24. * @Description: 获取数据源类型 
25. * @param 
26. * @return String 
27. * @throws 
28. */  
29. public static String getDataSourceType() {  
30. return contextHolder.get();  
31. }  
32.   
33. /** 
34. * @Description: 清除数据源类型 
35. * @param 
36. * @return void 
37. * @throws 
38. */  
39. public static void clearDataSourceType() {  
40. contextHolder.remove();  
41. }  
42. }  

4）修改事务管理器的数据源为动态数据源，指定事务注解的排序为2，我们会指定切换数据源的注解为1，这样在事务之前切换数据源，否则在事务之后切换的的话，无效。

[html] view plain copy

1. <!-- 注解事务处理 -->  
2. <bean id="transactionManager"  
3. class="org.springframework.jdbc.datasource.DataSourceTransactionManager">  
4. <property name="dataSource" ref="dynamicDataSource" />  
5. <!-- 启用注解 -->  
6. <tx:annotation-driven transaction-manager="transactionManager" order="2"/>  

5）定义切换数据库的注解和aop切面，指定排序为1，这里有个疑问，通过切点获取代理方法的注解数据，我用的是反射，但是网上有说可以直接作为参数传入的，我一直没有试验成功，不知道哪里有错，后续哪位大神指导的，可以分享一下。

[java] view plain copy

1. @Target({ElementType.METHOD, ElementType.TYPE})  
2. @Retention(RetentionPolicy.RUNTIME)  
3. @Inherited  
4. @Documented  
5. public @interface DataSource {  
6. String name();  
7.   
8. }  
9.   
10. @Component  
11. @Aspect  
12. @Order(1)  
13. public class DataSourceProxy {  
14.   
15. @Before(value="@annotation(com.futuren.wzk.common.datasource.DataSource)")  
16. public void before(JoinPoint jp) {  
17. String methodName = jp.getSignature().getName();  
18. Method[] methods = jp.getTarget().getClass().getMethods();  
19. for(Method method : methods) {  
20. if(method.getName().equals(methodName)) {  
21. DataSource ds = method.getAnnotation(DataSource.class);  
22. DataSourceContextHolder.setDataSourceType(ds.name());  
23. }  
24. }  
25. }  
26. }  

6）在项目中使用

[java] view plain copy

1. @Override  
2. @Transactional  
3. @DataSource(name="ucenter")  
4. public int addUser(User user) {  
5. userMapper.insert(user);  
6. return user.getUid();  
7. }  

这种方法，只支持单库事务，如果要多库事务，可能要引入JTA，或者是其他自定义实现。或者其他我不知道的技术。欢迎讨论！