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);
}
}