基于Spring+Mybatis的多数据源动态切换

Spring单数据源直接在<bean id="dataSource">下配置数据源的各种连接参数。但动态数据源需要配置个各个数据源例如ds1、ds2等。然后在dataSource中动态根据传递过来的参数动态调用不同的数据源。 

1、当进行访问时，首先通过DbContextHolder.setDbType("ds1");设置需要使用的数据源。DbContextHolder就是一个用来存储数据源信息的类，其中通过ThreadLocal来记录数据源信息。 

2、DynamicDataSource类集成Spring的AbstractRoutingDataSource类，通过determineCurrentLookupKey方法来获取数据源类型，如果没有对应的数据源则使用defaultTargetDataSource配置。 

注意：当设置了数据源之后会一直使用该数据源进行连接，除非使用 DbContextHolder.setDbType重新设置数据源或使用DbContextHolder.clearDbType()清除，清除后使用defaultTargetDataSource进行连接。 

1、配置文件 

properties 

1. ds1.driverClassName=oracle.jdbc.OracleDriver  
2. ds1.url=jdbc:oracle:thin:@localhost:1521:ORCL  
3. ds1.username=SSM  
4. ds1.password=SSM  
5.   
6. ds2.driverClassName=oracle.jdbc.OracleDriver  
7. ds2.url=jdbc:oracle:thin:@10.27.192.43:1522:ORCL  
8. ds2.username=FRAMEWORK_DEV  
9. ds2.password=123456  

xml 

1. <bean id="dataSource" class="com.cnpc.framework.db.DynamicDataSource">  
2.         <property name="targetDataSources">  
3.             <map key-type="java.lang.String">  
4.                 <entry key="ds1" value-ref="ds1" />  
5.                 <entry key="ds2" value-ref="ds2" />  
6.             </map>  
7.         </property>  
8.         <property name="defaultTargetDataSource" ref="ds1" />  
9.     </bean>  
10.       
11.     <bean id="ds1" class="org.apache.commons.dbcp.BasicDataSource"  
12.         destroy-method="close">  
13.         <property name="driverClassName" value="${ds1.driverClassName}" />  
14.         <property name="url" value="${ds1.url}" />  
15.         <property name="username" value="${ds1.username}" />  
16.         <property name="password" value="${ds1.password}" />  
17.     </bean>  
18.     <bean id="ds2" class="org.apache.commons.dbcp.BasicDataSource"  
19.         destroy-method="close">  
20.         <property name="driverClassName" value="${ds2.driverClassName}" />  
21.         <property name="url" value="${ds2.url}" />  
22.         <property name="username" value="${ds2.username}" />  
23.         <property name="password" value="${ds2.password}" />  
24.     </bean>  

2、Java文件 
com.cnpc.framework.db.DynamicDataSource 源码 

1. public class DynamicDataSource extends AbstractRoutingDataSource {  
2.   
3.     /** 
4.      * 取得当前使用那个数据源。 
5.      */  
6.     @Override  
7.     protected Object determineCurrentLookupKey() {  
8.         return DbContextHolder.getDbType();    
9.     }  
10.   
11.       
12.     public Logger getParentLogger() {  
13.         // TODO Auto-generated method stub  
14.         return null;  
15.     }  
16.       
17. }  

DbContextHolder 源码 

1. public class DbContextHolder  
2. {  
3.     private static final ThreadLocal<String> contextHolder = new ThreadLocal<String>();  
4.   
5.     /** 
6.      * 设置当前数据库。 
7.      * @param dbType 
8.      */  
9.     public static void setDbType(String dbType)  
10.     {  
11.         contextHolder.set(dbType);  
12.     }  
13.   
14.     /** 
15.      * 取得当前数据源。 
16.      * @return 
17.      */  
18.     public static String getDbType()  
19.     {  
20.         String str = (String) contextHolder.get();  
21.         return str;  
22.     }  
23.       
24.     /** 
25.      * 清除上下文数据 
26.      */  
27.     public static void clearDbType()  
28.     {  
29.         contextHolder.remove();  
30.     }  
31.       
32. }  

3、测试Code 

1. public ModelAndView list(HttpServletRequest request, HttpServletResponse response) throws Exception {  
2.   
3.         DbContextHolder.setDbType("ds1");  
4.         List<DemoSysUser> list1 = demoSysUserService.getAll();  
5.         System.out.println(DbContextHolder.getDbType() + ".list.size()=" + list1.size());  
6.   
7.         DbContextHolder.setDbType("ds2");  
8.         List<DemoSysUser> list2 = demoSysUserService.getAll();  
9.         System.out.println(DbContextHolder.getDbType() + ".list.size()=" + list2.size());  
10.         DbContextHolder.clearDbType();  
11.           
12.         List<DemoSysUser> list = demoSysUserService.getAll();  
13.         ModelAndView mv = this.getAutoView().addObject("sysUserList", list);  
14.         return mv;  
15.     }  

获取【下载地址】