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();