多数据源配置-使用spring配置多个数据源实现读写分离

我们在很多的项目中经常会有用到多个数据源。比如数据库读写分离，读操作都去从库里读，写操作都往主库里写。那么这里主库和从库就是两个不同的数据源。再比如要做两个数据库之间的数据转换，从一个数据库读取数据写到另一个数据库中，等等这些情况都需要系统使用两个或多个数据源。那么该如何配置多个数据源呢？我这里使用的是spring管理，数据库连接池使用的是阿里的druid。具体步骤如下：
1.修改spring的配置文件， 配置多个不同的数据源。这里不同的数据源就是由spring维护多个DruidDataSource的实例，每个实例都指定不同的数据库链接信息。配置如下：

     <!-- 配置多数据源 ，druid -->    <bean id="masterDataSource" class="com.alibaba.druid.pool.DruidDataSource"        destroy-method="close">        <property name="url" value="${jdbc.master.url}" />        <property name="username" value="${jdbc.master.username}" />        <property name="password" value="${jdbc.master.password}" />        <property name="driverClassName" value="${jdbc.master.driver}" />        <!-- 最大连接池数量 -->        <property name="maxActive" value="${master.maxActive}" />        <!-- 获取连接时最大等待时间，单位毫秒。 配置了maxWait之后，缺省启用公平锁，并发效率会有所下降，如果需要可以通过配置useUnfairLock属性为true使用非公平锁。 -->        <property name="maxWait" value="60000" />        <!-- 建议配置为true，不影响性能，并且保证安全性。 申请连接的时候检测，如果空闲时间大于timeBetweenEvictionRunsMillis，执行validationQuery检测连接是否有效。 -->        <property name="testWhileIdle" value="true"></property>        <property name="filters" value="stat"></property>    </bean>    <!-- 配置多数据源 ，druid -->    <bean id="slaveDataSource" class="com.alibaba.druid.pool.DruidDataSource"        destroy-method="close">        <property name="url" value="${jdbc.slave1.url}" />        <property name="username" value="${jdbc.slave1.username}" />        <property name="password" value="${jdbc.slave1.password}" />        <property name="driverClassName" value="${jdbc.slave1.driver}" />        <!-- 最大连接池数量 -->        <property name="maxActive" value="${slave1.maxActive}" />        <!-- 获取连接时最大等待时间，单位毫秒。 配置了maxWait之后，缺省启用公平锁，并发效率会有所下降，如果需要可以通过配置useUnfairLock属性为true使用非公平锁。 -->        <property name="maxWait" value="60000" />        <!-- 建议配置为true，不影响性能，并且保证安全性。 申请连接的时候检测，如果空闲时间大于timeBetweenEvictionRunsMillis，执行validationQuery检测连接是否有效。 -->        <property name="testWhileIdle" value="true"></property>        <property name="filters" value="stat"></property>    </bean>

我这里主要配置了两个数据源，一个是主库的数据源，名字叫masterDataSource，还有一个是从库的数据源，名字叫slaveDataSource

2.选择要使用的数据源。我们有了数据源之后我们要使用哪一个呢，所以这个时候我们需要来定义一个类，指定选择哪一个数据源，这里也很简单，只要继承spring给我们提供的AbstractRoutingDataSource类就可以了，这是一个抽象类，需要重写determineCurrentLookupKey方法，这个方法就是用来选择哪一个数据源的。

我们先给两个数据源分别打个标记，比如，读库的源就叫slave，写库的叫master，在实现这个方法的时候，我们返回一个字符串'slave' 表示要找读库，返回一个master表示要找主库。既然要动态的返回，我们就需要有一个入口能够设置选择的标记。在多线程环境下，为了保证数据安全，可以使用ThreadLocal这个类。代码如下

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

DbContextHolder的代码如下

public class DbContextHolder {    public static final String master = "master";    public static final String slave = "slave";    private static ThreadLocal<String> ds = new ThreadLocal<>();    public static void setDataSource (String dataSource) {        ds.set(dataSource);    }    public static String getDataSource() {        return ds.get();    }}

3.继续配置文件，不管我们配置了几个数据源，同一时刻生效的只有一个，于是我们要通过DynamicDataSource这个类来告诉spring我们要使用哪个数据源

    <bean id="dataSource" class="com.hy.base.db.DynamicDataSource">        <property name="targetDataSources">            <map key-type="java.lang.String">                <!-- write -->                <entry key="master" value-ref="masterDataSource" />                <!-- read -->                <entry key="slave" value-ref="slaveDataSource" />            </map>        </property>        <property name="defaultTargetDataSource" ref="masterDataSource" />    </bean>

这里要注意，key就是我们上面说到的标记，需要和DbContextHolder里的常量配置的相同。

到这一步，我们就可以说配置完成了，代码中就可以指定数据源，来进行数据库操作。当然我们为了更加方便，我们可以使用注解的方式来选择使用哪一个数据源，首先定义注解类

@Retention(RetentionPolicy.RUNTIME)@Target(ElementType.METHOD)public @interface DataSource {    String value();}

我们希望呢，在方法上打上这个注解，当这个方法执行前，我们就调用setDataSource方法来设置注解中指定的值，这样也就不需要手动设置了。下面是定义切面

public class DataSourceAspect {    public void before(JoinPoint point)    {//        System.out.println("--------------------------");        Object target = point.getTarget();        String method = point.getSignature().getName();        Class<?>[] classz = target.getClass().getInterfaces();        Class<?>[] parameterTypes = ((MethodSignature) point.getSignature())                .getMethod().getParameterTypes();        try {            Method m = classz[0].getMethod(method, parameterTypes);            if (m != null && m.isAnnotationPresent(DataSource.class)) {                DataSource data = m                        .getAnnotation(DataSource.class);                DbContextHolder.setDataSource(data.value());                System.out.println(data.value());            }        } catch (Exception e) {//            e.printStackTrace();        }    }}

下面对切面进行配置

    <aop:aspectj-autoproxy> </aop:aspectj-autoproxy>    <bean id="manyDataSourceAspect" class="com.hy.base.db.DataSourceAspect" />    <aop:config>        <aop:aspect id="c" ref="manyDataSourceAspect">            <aop:pointcut id="tx1"                expression="execution(* *(..))" />            <aop:before pointcut-ref="tx1" method="before" />        </aop:aspect>    </aop:config>

OK，到现在就大功告成。下面进行测试。我们使用的是mybatis，这里的具体配置就不给大家贴了。执行两个操作，分别针对两个库，如果两个库都能正常的响应，就说明我们的配置成功了。下面的例子就是两个接口，一个去找主库，一个去找从库，分别查询在某个数据库中唯一的一张表，主库的t1表只有主库有，从库的t2表只有从库有，这样如果都能查出来数据，说名两个库都是可以连接的。下面是具体的代码。

1.mapper，两个接口，一个使用主库，一个使用从库

public interface AppMapper {

    @DataSource("master")

    List<Map<String, Object>> executeQuery(@Param("sql")String sql);

    @DataSource("slave")

    List<Map<String, Object>> executeQuery2(@Param("sql")String sql);

}

2.测试方法

     @Test

    public void testDatasource() throws Exception {

        ApplicationContext ac = new ClassPathXmlApplicationContext("classpath:spring/applicationContext-*.xml");

        AppMapper mapper = (AppMapper) ac.getBean("appMapper");

        String sql = "select * from t1 limit 0,10";

        System.out.println(mapper.executeQuery(sql));

        sql = "select * from t2";

        System.out.println(mapper.executeQuery2(sql));

    }

这两个表在主库和从库都是唯一的。我们发现都能查出数据，说明我们配置正确了。当然了，这里数据库并没有做主从。