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

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

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

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

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

1、配置文件 

properties 

 ds1.driverClassName=oracle.jdbc.OracleDriver
 ds1.url=jdbc:oracle:thin:@localhost:1521:ORCL
 ds1.username=SSM
 ds1.password=SSM

 ds2.driverClassName=oracle.jdbc.OracleDriver
 ds2.url=jdbc:oracle:thin:@10.27.192.43:1522:ORCL
 ds2.username=FRAMEWORK_DEV
 ds2.password=123456

xml 

 "dataSource" class="com.cnpc.framework.db.DynamicDataSource">
         "targetDataSources">
             "java.lang.String">
                 "ds1" value-ref="ds1" />
                 "ds2" value-ref="ds2" />


         "defaultTargetDataSource" ref="ds1" />


     "ds1" class="org.apache.commons.dbcp.BasicDataSource"
         destroy-method="close">
         "driverClassName" value="${ds1.driverClassName}" />
         "url" value="${ds1.url}" />
         "username" value="${ds1.username}" />
         "password" value="${ds1.password}" />

     "ds2" class="org.apache.commons.dbcp.BasicDataSource"
         destroy-method="close">
         "driverClassName" value="${ds2.driverClassName}" />
         "url" value="${ds2.url}" />
         "username" value="${ds2.username}" />
         "password" value="${ds2.password}" />

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

 public class DynamicDataSource extends AbstractRoutingDataSource {

     /**
      * 取得当前使用那个数据源。
      */
     @Override
     protected Object determineCurrentLookupKey() {
         return DbContextHolder.getDbType();
     }


     public Logger getParentLogger() {
         // TODO Auto-generated method stub
         return null;
     }

 }

DbContextHolder 源码 

 public class DbContextHolder
 {
     private static final ThreadLocal contextHolder = new ThreadLocal();

     /**
      * 设置当前数据库。
      * @param dbType
      */
     public static void setDbType(String dbType)
     {
         contextHolder.set(dbType);
     }

     /**
      * 取得当前数据源。
      * @return
      */
     public static String getDbType()
     {
         String str = (String) contextHolder.get();
         return str;
     }

     /**
      * 清除上下文数据
      */
     public static void clearDbType()
     {
         contextHolder.remove();
     }

 }

3、测试Code 

 public ModelAndView list(HttpServletRequest request, HttpServletResponse response) throws Exception {

         DbContextHolder.setDbType("ds1");
         List list1 = demoSysUserService.getAll();
         System.out.println(DbContextHolder.getDbType() + ".list.size()=" + list1.size());

         DbContextHolder.setDbType("ds2");
         List list2 = demoSysUserService.getAll();
         System.out.println(DbContextHolder.getDbType() + ".list.size()=" + list2.size());
         DbContextHolder.clearDbType();

         List list = demoSysUserService.getAll();
         ModelAndView mv = this.getAutoView().addObject("sysUserList", list);
         return mv;
     }

获取【下载地址】