Spring MVC＋Mybatis 多數據源配置及發現的幾個問題

1.CustomerContextHolder 數據源管理類，負責管理當前的多個數據源，基於ThreadLocal實現，對每個線程設置不同的目標數據源

public class CustomerContextHolder {
    private static final ThreadLocal<String> contextHolder = new ThreadLocal<String>();

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

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

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

2.DynamicDataSource 數據注入類

public class DynamicDataSource extends AbstractRoutingDataSource {
    @Override
    protected Object determineCurrentLookupKey() {
        return CustomerContextHolder.getCustomerType();
    }
}

3.配置文件

　　a）spring-context-data-source.xml

　　

<!--統一的dataSource -->
    <bean id="dataSource" class="com.thinkgem.jeesite.common.config.DynamicDataSource">
        <property name="targetDataSources">
            <map key-type="java.lang.String">
                <entry value-ref="a" key="a"></entry>
                <entry value-ref="b" key="b"></entry>
            </map>
        </property>
        <property name="defaultTargetDataSource" ref="a">
        </property>
    </bean>


    <!-- 數據源配置, 使用 BoneCP 數據庫連接池 -->
    <bean id="a" class="com.alibaba.druid.pool.DruidDataSource" init-method="init" destroy-method="close">
        <!-- 數據源驅動類可不寫，Druid默認會自動根據URL識別DriverClass -->
        <property name="driverClassName" value="${jdbc.driver}" />
        <!-- 基本屬性 url、user、password -->
        <property name="url" value="${jdbc.url}" />
        <property name="username" value="${jdbc.username}" />
        <property name="password" value="${jdbc.password}" />
        <!-- 配置初始化大小、最小、最大 -->
        <property name="initialSize" value="${jdbc.pool.init}" />
        <property name="minIdle" value="${jdbc.pool.minIdle}" />
        <property name="maxActive" value="${jdbc.pool.maxActive}" />
        <!-- 配置獲取連接等待超時的時間 -->
        <property name="maxWait" value="60000" />
        <!-- 配置間隔多久才進行一次檢測，檢測需要關閉的空閑連接，單位是毫秒 -->
        <property name="timeBetweenEvictionRunsMillis" value="60000" />
        <!-- 配置一個連接在池中最小生存的時間，單位是毫秒 -->
        <property name="minEvictableIdleTimeMillis" value="300000" />
        <property name="validationQuery" value="${jdbc.testSql}" />
        <property name="testWhileIdle" value="true" />
        <property name="testOnBorrow" value="false" />
        <property name="testOnReturn" value="false" />
        <!-- 打開PSCache，並且指定每個連接上PSCache的大小（Oracle使用） <property name="poolPreparedStatements" value="true" /> <property name="maxPoolPreparedStatementPerConnectionSize" value="20" /> -->
        <!-- 配置監控統計攔截的filters -->
        <property name="filters" value="stat" />
    </bean>

    <!-- 數據源配置, 使用 BoneCP 數據庫連接池 -->
    <bean id="b" class="com.alibaba.druid.pool.DruidDataSource" init-method="init" destroy-method="close">
        <!-- 數據源驅動類可不寫，Druid默認會自動根據URL識別DriverClass -->
        <property name="driverClassName" value="${jdbc.driver}" />
        <!-- 基本屬性 url、user、password -->
        <property name="url" value="${jdbc.url.kpidata}" />
        <property name="username" value="${jdbc.username}" />
        <property name="password" value="${jdbc.password}" />
        <!-- 配置初始化大小、最小、最大 -->
        <property name="initialSize" value="${jdbc.pool.init}" />
        <property name="minIdle" value="${jdbc.pool.minIdle}" />
        <property name="maxActive" value="${jdbc.pool.maxActive}" />
        <!-- 配置獲取連接等待超時的時間 -->
        <property name="maxWait" value="60000" />
        <!-- 配置間隔多久才進行一次檢測，檢測需要關閉的空閑連接，單位是毫秒 -->
        <property name="timeBetweenEvictionRunsMillis" value="60000" />
        <!-- 配置一個連接在池中最小生存的時間，單位是毫秒 -->
        <property name="minEvictableIdleTimeMillis" value="300000" />
        <property name="validationQuery" value="${jdbc.testSql}" />
        <property name="testWhileIdle" value="true" />
        <property name="testOnBorrow" value="false" />
        <property name="testOnReturn" value="false" />
        <!-- 打開PSCache，並且指定每個連接上PSCache的大小（Oracle使用） <property name="poolPreparedStatements" value="true" /> <property name="maxPoolPreparedStatementPerConnectionSize" value="20" /> -->
        <!-- 配置監控統計攔截的filters -->
        <property name="filters" value="stat" />
    </bean>

　　b）spring-context.xml mybatis的配置

<!-- MyBatis begin -->
    <bean id="sqlSessionFactory" class="org.mybatis.spring.SqlSessionFactoryBean">
        <property name="dataSource" ref="dataSource"/>
        <property name="mapperLocations" value="classpath:/mappings/**/*.xml"/>
        <property name="configLocation" value="classpath:/mybatis-config.xml"></property>
    </bean>
    
    <!-- 掃描basePackage下所有以@MyBatisDao注解的接口 -->
    <bean id="mapperScannerConfigurer" class="org.mybatis.spring.mapper.MapperScannerConfigurer">
        <property name="sqlSessionFactoryBeanName" value="sqlSessionFactory" />
        <property name="basePackage" value="com.thinkgem.jeesite"/>
        <property name="annotationClass" value="com.thinkgem.jeesite.common.persistence.annotation.MyBatisDao"/>
    </bean>

4.使用 

　　@ResponseBody
    @RequestMapping(value = "jeesiteDb")
    public Map<String, Object> jeesiteDb() {
        Map<String, Object> returnMap = new HashMap<String, Object>();
        CustomerContextHolder.setCustomerType(CustomerContextHolder.DATA_SOURCE_JEESITE);
        List<Map<String, String>> list = testService.queryJeesite();
        returnMap.put("data", list);
        return returnMap;
    }

    @ResponseBody
    @RequestMapping(value = "myTestDb")
    public Map<String, Object> myTestDb() {
        Map<String, Object> returnMap = new HashMap<String, Object>();
        CustomerContextHolder.setCustomerType(CustomerContextHolder.DATA_SOURCE_MT_TEST);
        List<Map<String, String>> list = testService.queryMyTest();
        returnMap.put("data", list);
        return returnMap;
    }

5.問題

  a) 請求數據源的方法一定要寫在controller中，不要寫在dao中

CustomerContextHolder.setCustomerType(CustomerContextHolder.DATA_SOURCE_MT_TEST);

  b) 如果是同一個controller中，不同方法請求不同的數據源，上面的寫法足夠了，但是如果要統一個方法動態指定數據源，那么需要在使用之后加上，清除當前線程的數據源，否則數據切換會失敗

CustomerContextHolder.clearCustomerType();