小码农的代码（三）----------SpringJDBC多数据源应用

前面项目中基础的SpringJdbc的配置说明已经简单总结，接下去就是项目中使用到多数据源这个问题了。 
对于多数据源的实现，我的想法是首先配置多个数据源，然后在数据操作时通过切换数据源的方式来达到目的，那么如何在保持jdbctmplate只有一个的时候进行数据源切换呢，我考虑到可以起一个线程来实现在操作中的自由切换。 
首先配置3个数据源，为方便测试就直接在本机新建了三个mysql数据库（实际过程中是使用mysql与多个sql server）来写测试用例

Java代码  

1. <bean id="defaultDateSource" class="com.mchange.v2.c3p0.ComboPooledDataSource"  
2.         destroy-method="close">  
3.         <property name="driverClass" value="${jdbc.driverClass}" />  
4.         <property name="jdbcUrl" value="${jdbc.jdbcUrl}" />  
5.         <property name="user" value="${jdbc.user}" />  
6.         <property name="password" value="${jdbc.password}" />  
7.     </bean>  

Java代码  

1. <bean id="dataSource01" class="com.mchange.v2.c3p0.ComboPooledDataSource"  
2.         destroy-method="close">  
3.         <property name="driverClass" value="${jdbc.driverClass01}" />  
4.         <property name="jdbcUrl" value="${jdbc.jdbcUrl01}" />  
5.         <property name="user" value="${jdbc.user01}" />  
6.         <property name="password" value="${jdbc.password01}" />  
7.     </bean>  

Java代码  

1. <bean id="dataSource02" class="com.mchange.v2.c3p0.ComboPooledDataSource"  
2.         destroy-method="close">  
3.         <property name="driverClass" value="${jdbc.driverClass02}" />  
4.         <property name="jdbcUrl" value="${jdbc.jdbcUrl02}" />  
5.         <property name="user" value="${jdbc.user02}" />  
6.         <property name="password" value="${jdbc.password02}" />  
7.     </bean>  

习惯使用c3p0数据库连接池，下次尝试试用下阿里DruidDataSource，据说功能更加强大，但是一直没有使用过-，-跑题了 新建三个数据库还是只有一个表user(id,name,age；id逐渐自增)。 
然后新建一个单例的线程ThreadLocal（这是一个很好用的东西，下次专门写个测试用例总结下，做项目总是能够不断的引出新的东西，这点真的很好），能够set/get/remove数据源

Java代码  

1. public class DatabaseContextHolder {  
2.     private static final ThreadLocal<String> contextHolder = new ThreadLocal<String>();  
3.   
4.     public static void setDateBaseType(String database) {  
5.         Assert.notNull(database, "DatabaseType cannot be null");  
6.         contextHolder.set(database);  
7.     }  
8.   
9.     public static String getDateBaseType() {  
10.         return contextHolder.get();  
11.     }  
12.   
13.     public static void clearDatabaseType() {  
14.         contextHolder.remove();  
15.     }  
16. }  

接下来是Spring支持多数据源切换的核心类AbstractRoutingDataSource，新建一个继承这个抽象类的子类DynamicDataSource，并实现determineCurrentLookupKey方法（抽象类与接口的相似性）

Java代码  

1. public class DynamicDataSource extends AbstractRoutingDataSource {  
2.   
3.     @Override  
4.     protected Object determineCurrentLookupKey() {  
5.         return DatabaseContextHolder.getDateBaseType();  
6.     }  
7. }  

继续回到spring的配置文件，通过DynamicDataSource管理多数据源，并设置默认的数据源

Java代码  

1. <bean id="dynamicDataSource" class="com.zh.demo.base.tool.DynamicDataSource">  
2.         <property name="targetDataSources">  
3.             <map key-type="java.lang.String">  
4.                 <entry value-ref="defaultDateSource" key="defaultDateSource"></entry>  
5.                 <entry value-ref="dataSource01" key="dataSource01"></entry>  
6.                 <entry value-ref="dataSource02" key="dataSource02"></entry>  
7.             </map>  
8.         </property>  
9.         <property name="defaultTargetDataSource" ref="defaultDateSource">  
10.         </property>  
11.     </bean>  

当然jdbctemplate中指向的数据源也需要切换成它

Java代码  

1. <bean id="jdbcTemplate" class="org.springframework.jdbc.core.JdbcTemplate">  
2.         <property name="dataSource" ref="dynamicDataSource" />  

最后新建一个DataBaseType类与配置中的数据源名称相关联

Java代码  

1. public class DataBaseType {  
2.     public static final String DATABASE_DEFAULT = "defaultDateSource";  
3.     public static final String DATABASE_01 = "dataSource01";  
4.     public static final String DATABASE_02 = "dataSource02";  
5. }  

这样就可以写测试用例了。 
简单的新增语句在测试类中进行数据库切换

Java代码  

1. public void test(){  
2.         userService.saveUser(("insert into user (name,age)values('张三',13)");  
3.         DatabaseContextHolder.clearDatabaseType();  
4.         DatabaseContextHolder.setDateBaseType(DataBaseType.DATABASE_01);  
5.         userService.saveUser("insert into user (name,age)values('张三',13)");  
6.         DatabaseContextHolder.clearDatabaseType();  
7.         DatabaseContextHolder.setDateBaseType(DataBaseType.DATABASE_02);  
8.         userService.saveUser("insert into user (name,age)values('张三',13)");  
9.     }  

结果是三个数据库中都插入了数据，多数据源配置成功。 
注意点与总结： 
一、实际项目中发现一个问题，就是在同一个事务中不能够进行数据库切换，该问题暂时未解决，有方法的朋友可以的话烦请指导下。 
二、在使用过程中扩展想到能否通过请求来动态的切换数据源，目前考虑到还是需要在请求中带参数来让后台识别进行数据库切换，不知道有没有更好的方法。 
以下内容如存在错误之处烦请斧正以达到共同进步的效果，多谢。

• SpringJDBC_多数据源_.rar (33.4 KB)
• 下载次数: 6