Spring上设置mysql数据源动态切换（master写、slaver读）

一般情况下，很多项目（spring）由于对数据库（mysql）的并发量上来之后，都会配置主从机数据库，写数据使用主机数据库，而读数据使用从机数据库。那么如何使得对主从机数据源的使用能动态根据业务切换？

一、一些配置和依赖

1、一些需要的jar包

由于我这边使用了maven，也用到了aspect、mysql，so附上maven依赖，方便copy。

<properties>
    <spring.version>4.2.2.RELEASE</spring.version>
    <aspectj.version>1.6.11</aspectj.version>
    <mysql_connect_version>5.1.35</mysql_connect_version>
</properties>
<dependencies>
    <dependency>
        <groupId>mysql</groupId>
        <artifactId>mysql-connector-java</artifactId>
        <version>${mysql_connect_version}</version>
    </dependency>
    <dependency>
        <groupId>org.springframework</groupId>
        <artifactId>spring-aop</artifactId>
        <version>${spring.version}</version>
    </dependency>
     <dependency>
        <groupId>org.springframework</groupId>
        <artifactId>spring-aspects</artifactId>
        <version>${spring.version}</version>
    </dependency>
    <!--使用AspectJ方式注解需要相应的包-->  
    <dependency>  
        <groupId>org.aspectj</groupId>  
        <artifactId>aspectjrt</artifactId>  
        <version>${aspectj.version}</version>  
    </dependency>  
     <!--使用AspectJ方式注解需要相应的包-->  
    <dependency>  
        <groupId>org.aspectj</groupId>  
        <artifactId>aspectjweaver</artifactId>  
        <version>${aspectj.version}</version>  
    </dependency>
</dependencies>

2、spring配置（数据源和动态选择类）

 <!-- 主机数据库数据源配置 -->
<bean id="dataSourceMaster">
    <!-- 略过若干配置，我相信来看这篇文章的兄弟应该不会不懂吧 -->
</bean>

<bean id="dataSourceSlave">
       <!-- 略过若干配置，我相信来看这篇文章的兄弟应该不会不懂吧 -->
</bean>
 <!-- 配置动态分配的读写 数据源 -->
 <!-- 该ChooseDataSource类需要继承AbstractRoutingDataSource类并重写determineCurrentLookupKey方法，determineCurrentLookupKey方法主要用于从ThreadLocal变量中获取数据源的key -->
<bean id="dataSource" class="ChooseDataSource" lazy-init="true">
    <property name="targetDataSources">
        <map key-type="java.lang.String" value-type="javax.sql.DataSource">
            <!-- write -->
            <entry key="master" value-ref="dataSourceMaster"/>
            <!-- read -->
            <entry key="slave" value-ref="dataSourceSlave"/>
        </map>
    </property>
    <!-- 配置一个默认的使用数据源 -->
    <property name="defaultTargetDataSource" ref="dataSourceMaster"/>
    <!-- 给一个Map<String,String>注入选择源策略 -->
    <!-- 后面的切面方法（aspect()）会判断执行的Dao的方法名字是否包含Map的value中的字符子串，包含即把对应的key（数据源名称）使用ThreadLocal把key存到该Dao的方法执行线程的ThreadMap里，那么ChooseDataSource类重写的determineCurrentLookupKey方法就会调用ThreadLocal把key从执行线程的ThreadMap取出来，就实现了一种动态切换的效果了 -->
    <property name="methodType">
        <map key-type="java.lang.String">
            <!-- read -->
            <entry key="slave">
                <list>
                    <value>get</value>
                    <value>select</value>
                    <value>find</value>
                </list>
            </entry>
            <!-- write -->
            <entry key="master">
                <list>
                    <value>add</value>
                    <value>update</value>
                    <value>delete</value>
                    <value>insert</value>
                </list>
            </entry>
        </map>
    </property>
</bean>

二、代码实现

1、数据源handle

在后面的aspect切面方法里，可以看到会给当前执行方法选择一个数据源的名称，把该名称put到执行该方法的线程的ThreadMap（通过holder），然后在继承AbstractRoutingDataSource的类中determineCurrentLookupKey方法调用holder的get方法返回当前线程选择的数据源名称。

/**
 * 数据源的Handler类
 */
public class DataSourceHandler {

    // 数据源名称holder
    public static final ThreadLocal<String> dataSourceHolder = new ThreadLocal<String>();

    /**
     * 将读、写数据源加到dataSourceHolder中
     */
    public static void putDataSource(String datasource) {
        dataSourceHolder.set(datasource);
    }

    /**
     * 从holer中获取数据源字符串
     */
    public static String getDataSource() {
        return dataSourceHolder.get();
    }
     /** 
     * @Description: 清除数据源类型 
     * @param  
     * @return void 
     */  
    public static void clearDataSourceType() {  
        dataSourceHolder.remove();  
    }  
}

2、继承AbstractRoutingDataSource实现自定义数据源ChooseDataSource

在spring配置文件中会注入map，该Map存储对应方法包含某些字符映射某个数据源，在切面方法中根据该Map给执行方法的线程选择数据源。

/**
 * 获取数据源，用于动态切换数据源
 */
public class ChooseDataSource extends AbstractRoutingDataSource {

    public static Map<String, List<String>> map = new HashMap<String, List<String>>();

    /**
     * 实现父类中的抽象方法，获取数据源名称
     * @return
     */
    protected Object determineCurrentLookupKey() {
        return DataSourceHandler.getDataSource();
    }


}

3、切面方法根据DAO类执行方法名称选择数据源

PS:很多人可能会使用失败，原因很可能就是下面的Asprct配置的问题

/**
 * 切换数据源(不同方法调用不同数据源)
 */
@Aspect
@Component
@EnableAspectJAutoProxy(proxyTargetClass = true)
public class DataSourceAspect {


    @Pointcut("execution(*.*.*(..))")
    public void aspect() {
    }

    /**
     * 配置前置通知,使用在方法aspect()上注册的切入点
     */
     /**
     *该方法主要判断执行的Dao类方法名是否包含指定字符子**串，从而选择对应设定的数据源名。
     */
    @Before("aspect()")
    public void before(JoinPoint point) {
        String className = point.getTarget().getClass().getName();
        String method = point.getSignature().getName();
        try {
            for (String key : ChooseDataSource.map.keySet()) {
                for (String type : ChooseDataSource.map.get(key)) {
                    if (method.startsWith(type)) {
                        DataSourceHandler.putDataSource(key);
                    }
                }
            }
        } catch (Exception e) {
            e.printStackTrace();
        }
    }
}

好了，到这里就大功告成了，体验体验效果吧。。。