Spring整合多数据源实现动态切换

Posted 2020-09-27

在实际项目中时常需要连接多个数据库，而且不同的业务需求在实现过程当中往往需要访问不同的数据库。

jdbc.properties配置文件,配置多个dataSource

##########################mysql#####################################
hibernate.dialect=org.hibernate.dialect.MySQLInnoDBDialect
connection.driver_class=com.mysql.jdbc.Driver
connection.url=jdbc:mysql://localhost:3306/test?useUnicode=true&characterEncoding=UTF-8
connection.username=yahu
connection.password=123456

##########################Oracle#####################################
connection1.driver_class=oracle.jdbc.driver.OracleDriver
connection1.url=jdbc\:oracle\:thin\:@localhost\:1521/MEDB
connection1.username=yahu
connection1.password=123456

##########################Sql Server2008#####################################
connection2.driver_class=net.sourceforge.jtds.jdbc.Driver
connection2.url=jdbc:jtds:sqlserver://localhost:1433;DatabaseName=test
connection2.username=yahu
connection2.password=123456

spring-config.xml配置文件如下,将DynamicDataSource Bean加入到Spring的上下文xml配置文件中去，同时配置DynamicDataSource的targetDataSources(多数据源目标)属性的Map映射,使用动态数据源DynamicDataSource是继承与AbstractRoutingDataSource，而AbstractRoutingDataSource又是继承于org.springframework.jdbc.datasource.AbstractDataSource，AbstractDataSource实现了统一的DataSource接口，所以DynamicDataSource同样可以当一个DataSource使用:

　　　<!-- 数据库连接池配置 -->
     <bean id="dataSource" class="org.apache.commons.dbcp.BasicDataSource">
        <property name="driverClassName" value="${connection.driver_class}" />
        <property name="url" value="${connection.url}" />
        <property name="username" value="${connection.username}" />
        <property name="password" value="${connection.password}" />
        
        <property name="initialSize" value="5" />
        <property name="minIdle" value="1" /> 
        <property name="maxActive" value="200" />
        
        <!-- 配置获取连接等待超时的时间 -->
        <property name="maxWait" value="30000" />
        
        <!-- 配置间隔多久才进行一次检测，检测需要关闭的空闲连接，单位是毫秒 -->
        <property name="timeBetweenEvictionRunsMillis" value="60000" />
        
        <!-- 配置一个连接在池中最小生存的时间，单位是毫秒 -->
        <property name="minEvictableIdleTimeMillis" value="300000" />
    </bean>
    
    <!-- Oracle  -->
    <bean id="dataSource1" class="org.apache.commons.dbcp.BasicDataSource">
        <property name="driverClassName" value="${connection1.driver_class}" />
        <property name="url" value="${connection1.url}" />
        <property name="username" value="${connection1.username}" />
        <property name="password" value="${connection1.password}" />
    </bean>

    <!-- Sql server 2008  -->
    <bean id="dataSource2" class="org.apache.commons.dbcp.BasicDataSource">
        <property name="driverClassName" value="${connection2.driver_class}" />
        <property name="url" value="${connection2.url}" />
        <property name="username" value="${connection2.username}" />
        <property name="password" value="${connection2.password}" />
    </bean>
    
     <!-- 动态数据源 -->
     <bean id="dynamicDataSource" class="com.yahu.core.dao.DynamicDataSource">  
        <!-- 通过key-value的形式来关联数据源 -->  
        <property name="targetDataSources">  
            <map>  
                <entry value-ref="dataSource" key="datasource" />  
                <entry value-ref="dataSource1" key="datasource1" />
                <entry value-ref="dataSource2" key="datasource2" />         
            </map>  
        </property>  
        <property name="defaultTargetDataSource" ref="dataSource" />  
    </bean> 

DynamicDataSource动态数据源类,扩展Spring的AbstractRoutingDataSource抽象类，实现动态数据源,AbstractRoutingDataSource中的抽象方法determineCurrentLookupKey是实现数据源的route的核心.这里对该方法进行Override:

package com.yahu.core.dao;
import org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource;

/*******************************************************************
 * @describe : 建立动态数据源类   必须继承AbstractRoutingDataSource
 ********************************************************************/
public class DynamicDataSource extends AbstractRoutingDataSource {
    //coverity 修改
    //private Log log = LogFactory.getLog(getClass());

    protected Object determineCurrentLookupKey() {
        String value = CustomerContextHolder.getCustomerType();
        //log.info(value);
        return value;
    }

}

获得和设置上下文环境,为一线程安全的ThreadLocal:

package com.yahu.core.dao;

/*******************************************************************
 * @describe : 获得和设置上下文环境
 ********************************************************************/
public class CustomerContextHolder {

    /**
     * mysql
     */
    public static final String DATASOURCE = "datasource";

    /**
     * oracle
     */
    public static final String DATASOURCE_1 = "datasource1";
    
    /**
     * sql server
     */
    public static final String DATASOURCE_2 = "datasource2";

    private static final ThreadLocal<String> contextHolder = new ThreadLocal<String>();

    public static void setCustomerType(String customerType) {
        contextHolder.set(customerType);
    }

    public static String getCustomerType() {
        return contextHolder.get();
    }

    public static void clearCustomerType() {
        contextHolder.remove();
    }
}

动态数据源的管理,如何选择控制每个业务中需要的具体数据源，可以使用手动控制,业务层通过加入以下代码

CustomerContextHolder.setCustomerType(CustomerContextHolder.DATASOURCE);

即可实现动态切换数据源,如果在service层有比较统一的规则的话，也可以使用aop设置数据源使用,这里一般都是一个service一个数据源，所以最好使用aop在service层执行完之后统一调用

CustomerContextHolder.clearCustomerType();

清空数据源信息。

当然，在上面配置里面有个参数defaultTargetDataSource为默认数据源，就是不设置数据源的话，就是用这个数据源。