Spring实现多数据源操作

spring中实现多数据源 只需要继承AbstractRoutingDataSource抽象类，实现determineCurrentLookupKey()方法即可。

1.该方法返回一个key。根据这个key

 protected DataSource determineTargetDataSource() {
        Assert.notNull(this.resolvedDataSources, "DataSource router not initialized");
        Object lookupKey = this.determineCurrentLookupKey();// 获取key值
        DataSource dataSource = (DataSource)this.resolvedDataSources.get(lookupKey);// 根据key 获取相应的DataSource
        if(dataSource == null && (this.lenientFallback || lookupKey == null)) {
            dataSource = this.resolvedDefaultDataSource;
        }

        if(dataSource == null) {
            throw new IllegalStateException("Cannot determine target DataSource for lookup key [" + lookupKey + "]");
        } else {
            return dataSource;
        }
    }

2.继承 AbstractRoutingDataSource

package com.mpp.web.datasource;

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

/**
 * Created by mapp on 2018/2/8.
 */
public class DynamicDataSource extends AbstractRoutingDataSource{


    /**
     * 实现该方法即可实现动态切换数据源
     *
     * @return
     */
    @Override
    protected Object determineCurrentLookupKey() {
        return DataSourceHolder.getDateSource();
    }
}

3.DataSourceHolder

package com.mpp.web.datasource;

/**
 * Created by mapp on 2018/2/8.
 * 存储当前线程中使用的数据源标识
 */
public class DataSourceHolder {


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

    public static String getDateSource() {
        return dateSource.get();
    }

    public static void setDataSource(String key) {
        dateSource.set(key);
    }

    public static void clearDateSource() {
        dateSource.remove();
    }
}

4.xml配置

<!-- 配置数据源 1 使用的是Druid数据源 -->
    <bean id="dataSourceOne" class="com.alibaba.druid.pool.DruidDataSource" init-method="init" destroy-method="close">
        <!-- 基本属性 url、user、password -->
        <property name="url" value="${jdbc.url}"/>
        <property name="username" value="${jdbc.username}"/>
        <property name="password" value="${jdbc.password}"/>
        <!--表明数据库类型-->
        <property name="dbType" value = "mysql" />
    </bean>

    <!-- 配置数据源 2 使用的是Druid数据源 -->
    <bean id="dataSourceTwo" class="com.alibaba.druid.pool.DruidDataSource" init-method="init" destroy-method="close">
        <!-- 基本属性 url、user、password -->
        <property name="url" value="${jdbc.url2}"/>
        <property name="username" value="${jdbc.username2}"/>
        <property name="password" value="${jdbc.password2}"/>
        <!--表明数据库类型-->
        <property name="dbType" value = "mysql" />
    </bean>

    <!--多数据源配置-->
    <bean id="dataSource" class="com.mpp.web.datasource.DynamicDataSource">
        <property name="targetDataSources">
            <map key-type="java.lang.String">
                <entry key="dataSourceOne" value-ref="dataSourceOne"/>
                <entry key="dataSourceTwo" value-ref="dataSourceTwo"/>
            </map>
        </property>
        <property name="defaultTargetDataSource" ref="dataSourceOne" />
    </bean>

5.测试

package com.mpp.test;

import com.mpp.web.dao.UserMapper;
import com.mpp.web.datasource.Constant;
import com.mpp.web.datasource.DataSourceHolder;
import com.mpp.web.entity.User;
import com.mysql.fabric.xmlrpc.base.Data;
import org.apache.log4j.Logger;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.test.context.ContextConfiguration;
import org.springframework.test.context.junit4.AbstractJUnit4SpringContextTests;
import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;
import org.springframework.transaction.annotation.Transactional;

/**
 * 测试类
 */
@RunWith(SpringJUnit4ClassRunner.class)
@ContextConfiguration(locations={"classpath:spring.xml","classpath:spring-mybatis.xml"})
//声明一个事务管理 每个单元测试都进行事务回滚 无论成功与否
@Transactional
public class SpringJunitTest extends AbstractJUnit4SpringContextTests {

    private final Logger logger = Logger.getLogger(SpringJunitTest.class);

    @Autowired
    private UserMapper userMapper;

    @Test
    public void testDateSourceHolder() {
        DataSourceHolder.setDataSource(Constant.DATEBASE_ONE);
        System.out.println("-------------------" + DataSourceHolder.getDateSource() + "--------------------------");
        DataSourceHolder.clearDateSource();
        System.out.println("-------------------" + DataSourceHolder.getDateSource() + "-------------------");
    }

    /**
     * 数据源切换测试
     */
    @Test
    public void testDBOne() {
        User user1 = userMapper.getByName("数据源1");
        System.out.println(user1);
        // 切换数据源2
        DataSourceHolder.setDataSource(Constant.DATEBASE_TWO);
        User user2 = userMapper.getByName("数据源2");
        System.out.println(user2);
        // 清空数据源
        DataSourceHolder.clearDateSource();
        User user3 = userMapper.getByName("数据源1");
        System.out.println(user3);
    }

}