spring+mybatis数据源切换【service层以及controller层】

最近项目需要切换数据源，查询了相关资料后做了一个简单的数据源的切换。

一、基本的简单配置

首先是我认为比较好的切换是在service层进行切换，基本思想是利用spring的AbstractRoutingDataSource类进行datasource的选择，就像map的key-value,AbstractRoutingDataSource是有个determineCurrentLookupKey选择datasource的bean进行切换。

下面是java类的配置

import org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource;

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

/**数据源的管理类**/

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

    public static final String DEFAULT_DATA_SOURCE="defaultDataSource"; //默认数据源

    public static final String READ_DATA_SOURCE="readDataSource";       //第二个数据源，因为这个数据源是只提供读取，所以叫read...

    /**
     * 设置数据源类型
     * @param dataSource 数据源名称
     */
    public static void setDataSource(String dataSource) {
        contextHolder.set(dataSource);
    }

    /**
     * 获取数据源名称
     */
    public static String getDataSource() {
        return contextHolder.get();
    }

    /**
     * 清除数据源名称
     */
    public static void clearDataSource() {
        contextHolder.remove();
    }
}

接下来要在xml文件中配置动态的数据源

部分配置文件如下：

<bean id="sqlSessionFactory" class="org.mybatis.spring.SqlSessionFactoryBean">
        <property name="dataSource" ref="dynamicDataSource"/>
        <property name="configLocation" value="classpath:mybatis-config.xml"/>
        <property name="mapperLocations" value="classpath*:/com/test/core/domain/mapper/**/*.xml"/>
    </bean>

    <bean class="org.springframework.beans.factory.config.PropertyPlaceholderConfigurer">
        <property name="locations">
            <list>
                <value>classpath:/properties/app-${env}.properties</value>
            </list>
        </property>
    </bean>
    
    <!-- 数据源1-->
    <bean id="defaultDataSource" class="com.zaxxer.hikari.HikariDataSource" destroy-method="close">
        <property name="driverClassName" value="${datasource.classname}"/>
        <property name="jdbcUrl" value="${jdbc.url}"/>
        <property name="username" value="${username}"/>
        <property name="password" value="${password}"/>
        <property name="connectionTimeout" value="${connection.timeout}"/>
        <property name="maximumPoolSize" value="${max.pool.size}"/>
        <property name="leakDetectionThreshold" value="${leak.detection.threshold}"/>
    </bean>
    <!-- 数据源2-->
    <bean id="readDataSource" class="com.zaxxer.hikari.HikariDataSource" destroy-method="close">
        <property name="driverClassName" value="${read.datasource.classname}"/>
        <property name="jdbcUrl" value="${read.jdbc.url}"/>
        <property name="username" value="${read.username}"/>
        <property name="password" value="${read.password}"/>
        <property name="connectionTimeout" value="${read.connection.timeout}"/>
        <property name="maximumPoolSize" value="${read.max.pool.size}"/>
        <property name="leakDetectionThreshold" value="${read.leak.detection.threshold}"/>
    </bean>

    <!-- 动态数据源-->
    <bean id="dynamicDataSource" class="com.test.datasource.DynamicDataSource">
        <property name="targetDataSources">
            <map key-type="java.lang.String">
                <entry value-ref="defaultDataSource" key="defaultDataSource"/>
                <entry value-ref="readDataSource" key="readDataSource"/>
            </map>
        </property>
        <!-- 默认数据源为defaultDataSource-->
        <property name="defaultTargetDataSource" ref="defaultDataSource"/>
    </bean>

    <bean id="transactionManager" class="org.springframework.jdbc.datasource.DataSourceTransactionManager">
        <property name="dataSource" ref="dynamicDataSource"/>
    </bean>

    <tx:annotation-driven mode="proxy" transaction-manager="transactionManager"/>

    <bean id="mapperScannerConfigurer" class="org.mybatis.spring.mapper.MapperScannerConfigurer">
        <property name="basePackage" value="com.test.core.domain.mapper"/>
        <property name="annotationClass" value="org.springframework.stereotype.Repository"/>
    </bean>

做好了这些工作就可以在dao层进行数据源的切换了，

如：DataSourceContextHolder.setDataSource(DataSourceContextHolder.READ_DATA_SOURCE);

二、利用spring的aop实现在service切换数据源

然而我们需要在service层或者controller层进行切换，而在进入service时已经开启了一个事务，这时候直接切换数据源是无效的，所以要在service前就切换数据源。

利用spring 的aop来实现我们的想法吧

切换数据源的java类：

/**切面**/

import lombok.extern.slf4j.Slf4j;
import org.springframework.aop.AfterReturningAdvice;
import org.springframework.aop.MethodBeforeAdvice;

import java.lang.reflect.Method;
@Slf4j
public class DataSourceExchange implements MethodBeforeAdvice,AfterReturningAdvice{
    @Override
    public void afterReturning(Object returnValue, Method method, Object[] args, Object target) throws Throwable {
        DataSourceContextHolder.clearDataSource();
        log.info("del datasource");
    }

    @Override
    public void before(Method method, Object[] args, Object target) throws Throwable {
       
        if (method.isAnnotationPresent(DataSource.class))
        {
            DataSource datasource = method.getAnnotation(DataSource.class);
            DataSourceContextHolder.setDataSource(datasource.name());
        }
        else
        {
            DataSourceContextHolder.setDataSource(DataSourceContextHolder.DEFAULT_DATA_SOURCE);
        }

        log.info("now datasource:"+DataSourceContextHolder.getDataSource());
    }
}

/**为了更好的体验，直接用注解来定义数据源**/

@Target({ElementType.METHOD, ElementType.TYPE})
@Retention(RetentionPolicy.RUNTIME)
@Documented
public @interface DataSource {
    String name() default "defaultDataSource";
}

xml的相关配置

<!--1.配置Spring框架自身提供的切面类-->
    <tx:advice id="userTxAdvice" transaction-manager="transactionManager">
        <tx:attributes>
            <tx:method name="delete*" propagation="REQUIRED" read-only="false"
                       rollback-for="java.lang.Exception" no-rollback-for="java.lang.RuntimeException" />
            <tx:method name="insert*" propagation="REQUIRED" read-only="false"
                       rollback-for="java.lang.Exception" />
            <tx:method name="update*" propagation="REQUIRED" read-only="false"
                       rollback-for="java.lang.Exception" />
            <tx:method name="find*" propagation="SUPPORTS" />
            <tx:method name="get*" propagation="SUPPORTS" />
            <tx:method name="select*" propagation="SUPPORTS" />
        </tx:attributes>
    </tx:advice>
    <!-- 2.配置自己定义的切面类 -->
    <bean id="dataSourceExchange" class="com.chejianer.admin.datasource.DataSourceExchange"/>
    <!-- 3.根据切面动态切换数据源-->
    <aop:config>
        <aop:pointcut id="service" expression="execution(* com.test.service.*.*(..))" />
        <aop:advisor pointcut-ref="service" advice-ref="userTxAdvice" order="2" />
        <aop:advisor pointcut-ref="service" advice-ref="dataSourceExchange" order="1" />
    </aop:config>

配置完成后就可以进行测试了，在serviceImpl的的方法上面加上@DataSource(name=你要切换的数据源)就ok了，这里就不贴代码了。

三、直接在controller进行切换，继续探索中

因为项目需要切数据源的地方特别多，在service层切换很费时间，所以准备在controller层进行切换，而尝试了下spring的切面在这里直接切不了，具体原因还没找到，所以暂时只能采用笨的方法，在controller的方法开头加上：

DataSourceContextHolder.setDataSource(DataSourceContextHolder.READ_DATA_SOURCE);

方法结尾加上：

DataSourceContextHolder.clearDataSource();

2017-03-25