spring+mybatis多数据源切换

在实际的公司项目中，很可能会遇到一个问题就是，一个java项目，但是项目中涉及两个数据库，这两个数据库还在不同IP的机子上。

遇到这种情况的时候，我们有两个选择

1、不走spring的aop方式，直接去多做两个dataSource

2、用spring进行管理，灵活地进行数据源切换

 

现在就来对第2种方式进行笔记：

spring.xml配置文件：

<?xml version="1.0" encoding="UTF-8"?>  
<beans xmlns="http://www.springframework.org/schema/beans"
xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xmlns:aop="http://www.springframework.org/schema/aop"
xmlns:task="http://www.springframework.org/schema/task" xmlns:tx="http://www.springframework.org/schema/tx"
xmlns:jdbc="http://www.springframework.org/schema/jdbc" xmlns:context="http://www.springframework.org/schema/context"
xsi:schemaLocation="
     http://www.springframework.org/schema/context http://www.springframework.org/schema/context/spring-context-3.0.xsd
     http://www.springframework.org/schema/beans http://www.springframework.org/schema/beans/spring-beans-3.0.xsd
     http://www.springframework.org/schema/jdbc http://www.springframework.org/schema/jdbc/spring-jdbc-3.0.xsd
     http://www.springframework.org/schema/tx http://www.springframework.org/schema/tx/spring-tx-3.0.xsd
 http://www.springframework.org/schema/task http://www.springframework.org/schema/task/spring-task.xsd
     http://www.springframework.org/schema/aop http://www.springframework.org/schema/aop/spring-aop-3.0.xsd ">  
  
       <!--配置mysql数据库参数 -->
    <context:property-placeholder location="mysql.properties" />
    
    <!--多个数据源配置  -->
    <bean id="db1" class="org.apache.commons.dbcp.BasicDataSource"
        destroy-method="close">
        <!-- 连接数据库参数 -->
        <property name="driverClassName" value="${db1.driver}" />
        <property name="url" value="${db1.url}" />
        <property name="username" value="${db1.username}" />
        <property name="password" value="${db1.password}" />
     
      <!--连接池参数 -->
<property name="initialSize" value="10" />
<property name="maxActive" value="500" />
<property name="maxIdle" value="40" />
<property name="minIdle" value="10" />
    </bean>
    
    <bean id="db2" class="org.apache.commons.dbcp.BasicDataSource"
        destroy-method="close">
        <!-- 连接数据库参数 -->
        <property name="driverClassName" value="${db2.driver}" />
        <property name="url" value="${db2.url}" />
        <property name="username" value="${db2.username}" />
        <property name="password" value="${db2.password}" />
     
      <!--连接池参数 -->
<property name="initialSize" value="10" />
<property name="maxActive" value="500" />
<property name="maxIdle" value="40" />
<property name="minIdle" value="10" />
    </bean>
    
    <bean id="dataSource" class="com.ckd.datasource.mybatis.DynamicDataSource">
        <property name="targetDataSources">
            <map key-type="java.lang.String">
                <entry key="db1" value-ref="db1" />
                <entry key="db2" value-ref="db2" />
            </map>
        </property>
        <property name="defaultTargetDataSource" ref="db2" />
    </bean>
    
    <!-- mybatis配置 -->
       <bean id="sqlSessionFactory" class="org.mybatis.spring.SqlSessionFactoryBean">
<property name="dataSource" ref="dataSource" />
<property name="configLocation" value="classpath:mybatis/mybatis-config.xml" />
<property name="mapperLocations" value="classpath:mybatis/mapper/*.xml"/>
</bean>
<!--获取sqlSession-->
<bean id="sqlSession" class="org.mybatis.spring.SqlSessionTemplate">
<constructor-arg index="0">
         <ref bean="sqlSessionFactory"/>
        </constructor-arg>
</bean> 
<!-- 事务管理器配置,单数据源事务 -->
<bean id="transactionManager"
class="org.springframework.jdbc.datasource.DataSourceTransactionManager">
<property name="dataSource" ref="dataSource" />
</bean>
     
</beans>  

具体实现切换的操作类：

/**
 * @fileName DynamicDataSource.java
 * @author chenkaideng
 * @date 2015年8月27日
 * @describe 动态获取数据源
 */
public class DynamicDataSource extends AbstractRoutingDataSource{
 @Override
 protected Object determineCurrentLookupKey() {
  // TODO Auto-generated method stub
  return DataSourceContextHolder.getDbType();
 }
}

/**
 * @fileName DataSourceContextHolder.java
 * @author chenkaideng
 * @date 2015年8月27日
 * @describe 数据源设值Holder类
 */
public class DataSourceContextHolder {
 
  private static final ThreadLocal<String> contextHolder = new ThreadLocal<String>();  
   
  public static void setDbType(String dbType) {  
   contextHolder.set(dbType);  
  }  
   
  public static String getDbType() {  
   return ((String) contextHolder.get());  
  }  
   
  public static void clearDbType() {  
   contextHolder.remove();  
  }  
} 

弄完以上的事情，剩下的事情就简单了

    

    -》先是加载spring.xml文件applicationContext = new ClassPathXmlApplicationContext("spring.xml");

    -》然后设置数据源DataSourceContextHolder.setDbType("db1");

    -》接着从applicationContext 中获取sqlSession = (SqlSession) applicationContext.getBean("sqlSession");

    -》最后就可以拿这个sqlSession去做增删改查的操作

 

注意：不用对这个sqlSession做close和comit的操作，因为都已经由spring自己管理了，不用手动做这些操作。