springmvc+mybatis多数据源切换

前文：之前练习SSM时配置的都是单数据源，现在项目需求多数据源，所以自己查阅了以下资料，在controller中手动切换数据源成功，以下的配置细节。

           实际上应该在dao层进行注解的方式切换，使用AOP代理数据源。

1、步骤：

        a、编写多数据源property文件

        b、配置applicationContext-mybatis.xml

        c、编写动态切换数据源类 ：DynamicDataSource.java

        d、controller中切换数据源

2、编写多数据源properties文件

###  DB_Mysql 
mysql_driver=com.mysql.jdbc.Driver
mysql_url=jdbc:mysql://XXX:3306/ofs?useUnicode=true&characterEncoding=utf-8
mysql_username=XXX
mysql_password=XXX
###  DB_MSSQL
sqlserver_driver=com.microsoft.sqlserver.jdbc.SQLServerDriver
sqlserver_url=jdbc:sqlserver://XXX;DatabaseName=XXX;
sqlserver_username=XXX
sqlserver_password=XXX

###  default setting 
minIdle=45
maxIdle=50
initialSize=5
maxActive=100
maxWait=100
removeAbandonedTimeout=180
removeAbandoned=true

　　

3、  配置applicationContext-mybatis.xml，主要添加两个地方

我使用的是‘dbcp连接池’

        <!-- 读取数据库配置文件 -->
	<context:property-placeholder location="classpath:database.properties" />

	<!-- 使用JNDI数据源 dbcp连接池(Mysql)-->
	<bean id="dataSource_mysql" class="org.apache.commons.dbcp.BasicDataSource" destroy-method="close" >
		<property name="driverClassName" value="${mysql_driver}" />
		<property name="url" value="${mysql_url}" />
		<property name="username" value="${mysql_username}" />
		<property name="password" value="${mysql_password}" />
		
		<!-- 初始化数据库连接时第一次要创建的个数    默认值为0 -->
		<property name="initialSize" value="${initialSize}" />
		<property name="maxActive" value="${maxActive}" />
		<property name="maxIdle" value="${maxIdle}" />
		<property name="minIdle" value="${minIdle}" />
		<property name="maxWait" value="${maxWait}" />
		<!-- 开启无用连接的回收机制之后，如果连接池超出removeAbandoned所设置时间之后回收无用的连接   -->
		<property name="removeAbandonedTimeout" value="${removeAbandonedTimeout}" />
		<!-- 告诉连接池是否开启无用连接的回收机制 -->
		<property name="removeAbandoned" value="${removeAbandoned}" />
		<!-- 配置一个连接在池中最小生存的时间，单位是毫秒 -->
                <property name="minEvictableIdleTimeMillis" value="300000" />
		<!-- SQL心跳 -->
		<!-- 开启定时校验 -->
		<property name="testWhileIdle" value="true" />
		<property name="testOnBorrow" value="false" />
		<property name="testOnReturn" value="false" />
		<property name="validationQuery" value="select 1" />
		<property name="timeBetweenEvictionRunsMillis" value="60000" />
		<!-- 定义每次校验的连接数  -->
		<property name="numTestsPerEvictionRun" value="${maxActive}" />
	</bean>
	
	<!-- 使用JNDI数据源 dbcp连接池(MSSQL)-->
	<bean id="dataSource_sqlServer" class="org.apache.commons.dbcp.BasicDataSource" destroy-method="close" >
		<property name="driverClassName" value="${sqlserver_driver}" />
		<property name="url" value="${sqlserver_url}" />
		<property name="username" value="${sqlserver_username}" />
		<property name="password" value="${sqlserver_password}" />
		
		<!-- 初始化数据库连接时第一次要创建的个数    默认值为0 -->
		<property name="initialSize" value="${initialSize}" />
		<property name="maxActive" value="${maxActive}" />
		<property name="maxIdle" value="${maxIdle}" />
		<property name="minIdle" value="${minIdle}" />
		<property name="maxWait" value="${maxWait}" />
		<!-- 开启无用连接的回收机制之后，如果连接池超出removeAbandoned所设置时间之后回收无用的连接   -->
		<property name="removeAbandonedTimeout" value="${removeAbandonedTimeout}" />
		<!-- 告诉连接池是否开启无用连接的回收机制 -->
		<property name="removeAbandoned" value="${removeAbandoned}" />
		 <!-- 配置一个连接在池中最小生存的时间，单位是毫秒 -->
        <property name="minEvictableIdleTimeMillis" value="300000" />
		<!-- SQL心跳 -->
		<!-- 开启定时校验 -->
		<property name="testWhileIdle" value="true" />
		<property name="testOnBorrow" value="false" />
		<property name="testOnReturn" value="false" />
		<property name="validationQuery" value="select 1" />
		<property name="timeBetweenEvictionRunsMillis" value="60000" />
		<!-- 定
		义每次校验的连接数  -->
		<property name="numTestsPerEvictionRun" value="${maxActive}" />
	</bean>

配置默认使用MySQL数据库，主次数据库分别是：Mysql和MSSql

<bean id="dataSource" class="utils.DynamicDataSource">  <!-- utils.DynamicDataSource为手动切换数据源的工具类-->
        <property name="targetDataSources">
            <map key-type="java.lang.String">
                <entry key="dataSource_mysql" value-ref="dataSource_mysql"/>
                <entry key="dataSource_sqlServer" value-ref="dataSource_sqlServer"/>
            </map>
        </property>
        <!--默认数据源-->
        <property name="defaultTargetDataSource" ref="dataSource_mysql"/>
    </bean>
    
    
	<!-- 会话工厂 -->
	<bean id="sqlSessionFactory" class="org.mybatis.spring.SqlSessionFactoryBean">
		<property name="dataSource" ref="dataSource" />
		<!-- 引入mybatis配置文件 -->
		<property name="configLocation" value="classpath:mybatis-config.xml" />
	</bean>

　　

4、编写动态切换数据源类 ：DynamicDataSource.java

import org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource;
 
import java.sql.SQLFeatureNotSupportedException;
import java.util.logging.Logger;
 
/**
 * Created by LB on 2017/6/14.
 * 配置多数据源
 */
public class DynamicDataSource extends AbstractRoutingDataSource {
    //配置文件 applicationContext-mybatis.xml 中设置的两个"key"名称
    public static final String DATASOURCE_MYSQL = "dataSource_mysql"; 
    public static final String DATASOURCE_MSSQL = "dataSource_sqlServer";
    //本地线程，获取当前正在执行的currentThread
    public static final ThreadLocal<String> contextHolder = new ThreadLocal<String>();
 
    public static void setCustomerType(String customerType) {
        //把当前请求的参数，存入当前线程，这个参数是我们定义的DATASOURCE_A或者DATASOURCE_B
        System.out.println("当前切换的数据源="+customerType);
        contextHolder.set(customerType);
    }
    public static String getCustomerType() {
 
        return contextHolder.get();
    }
    //切换默认数据源
    public static void clearCustomerType() {
        contextHolder.remove();
    }
 
    protected Object determineCurrentLookupKey() {
        return getCustomerType();
    }
 
   
}

5、controller中切换数据源

// 从数据库查询是否有client
//切换Mssql数据源
DynamicDataSource.setCustomerType(DynamicDataSource.DATASOURCE_MSSQL);
String testing=clientService.findTest(customerId, appkey, format);
List<Client> client = clientService.findByClient(customerId, appkey, format);
//切回默认Mysql数据源
DynamicDataSource.clearCustomerType();
Client cownerCode = clientService.findOwnerCode(customerId,ownerCode);

 

6、效果：

7、注意事项：当在同一个controller中来回切换数据源时，记得切换 ：DynamicDataSource.clearCustomerType();