参考: https://www.cnblogs.com/liujiduo/p/5004691.html
参考: https://blog.csdn.net/ll535299/article/details/78203634
1. 需求和原理
有个项目需求需要使用 多数据源,但是之前没有使用过相关的经验,在网上搜索发现一些用jndi 时时获取数据源,和配置多数据源,然后根据特征去切换的
配置多数据源实现原理: 在spring配置文件中配置多个数据源, 通过 继承
AbstractRoutingDataSource 类, 然后可以在配置文件中将 多个数据源配置进来, 怎么动态切换,看了网上的大多数是根据 aop 拦截Dao 类的方法, 然后在对应的方法上加上自定义注解,指定需要使用那个数据源,在方法成功后,清除原数据源
2. 具体过程即实现
2.1 实现 AbstractRoutingDataSource , 创建一个 DataSourceHolder 类,用于持有当前线程中使用的数据源标识,代码如下:
package com.common.data;
import org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource;
import org.springframework.jdbc.datasource.lookup.DataSourceLookup;
public class MyDataSource extends AbstractRoutingDataSource {
@Override
protected Object determineCurrentLookupKey() {
return DataSourceHolder.getDataSourceType();
}
}
package com.common.data;
public class DataSourceHolder {
private static final ThreadLocal<String> contextHolder = new ThreadLocal<String>();
/**
* @Description: 设置数据源类型
* @param dataSourceType 数据库类型
* @return void
* @throws
*/
public static void setDataSourceType(String dataSourceType) {
contextHolder.set(dataSourceType);
}
/**
* @Description: 获取数据源类型
* @param
* @return String
* @throws
*/
public static String getDataSourceType() {
return contextHolder.get();
}
/**
* @Description: 清除数据源类型
* @param
* @return void
* @throws
*/
public static void clearDataSourceType() {
contextHolder.remove();
}
}
2.2 application 配置修改
<!--创建数据源1,连接数据库db1 -->
<bean id="dataSource1" class="org.apache.commons.dbcp.BasicDataSource" destroy-method="close">
<property name="driverClassName" value="${db1.driver}" />
<property name="url" value="${db1.url}" />
<property name="username" value="${db1.username}" />
<property name="password" value="${db1.password}" />
</bean>
<!--创建数据源2,连接数据库db2 -->
<bean id="dataSource2" class="org.apache.commons.dbcp.BasicDataSource" destroy-method="close">
<property name="driverClassName" value="${db2.driver}" />
<property name="url" value="${db2.url}" />
<property name="username" value="${db2.username}" />
<property name="password" value="${db2.password}" />
</bean>
<!--创建数据源3,连接数据库db3 -->
<bean id="dataSource3" class="org.apache.commons.dbcp.BasicDataSource" destroy-method="close">
<property name="driverClassName" value="${db3.driver}" />
<property name="url" value="${db3.url}" />
<property name="username" value="${db3.username}" />
<property name="password" value="${db3.password}" />
</bean>
<bean id="dynamicDataSource" class="com.test.context.datasource.DynamicDataSource">
<property name="targetDataSources">
<map key-type="java.lang.String">
<!-- 指定lookupKey和与之对应的数据源 -->
<entry key="dataSource1" value-ref="dataSource1"></entry>
<entry key="dataSource2" value-ref="dataSource2"></entry>
<entry key="dataSource3 " value-ref="dataSource3"></entry>
</map>
</property>
<!-- 这里可以指定默认的数据源 -->
<property name="defaultTargetDataSource" ref="dataSource1" />
</bean>
2.3 手动切换数据源,没有指定数据源时,使用的是默认数据源,当指定了数据源之后,使用的是指定的数据源
@Service
public class DataServiceImpl implements DataService {
@Autowired
private DataMapper dataMapper;
@Override
public List<Map<String, Object>> getList1() {
// 没有指定,则默认使用数据源1
return dataMapper.getList1();
}
@Override
public List<Map<String, Object>> getList2() {
// 指定切换到数据源2
DynamicDataSourceHolder.setDataSource("dataSource2");
return dataMapper.getList2();
}
@Override
public List<Map<String, Object>> getList3() {
// 指定切换到数据源3
DynamicDataSourceHolder.setDataSource("dataSource3");
return dataMapper.getList3();
}
}
2.4 创建自定义注解 和AOP 拦截类,实现 自动切换
@Target(ElementType.METHOD)
@Retention(RetentionPolicy.RUNTIME)
@Component
public @interface DataSource {
String value() default "";
}
import java.lang.reflect.Method;
import org.aspectj.lang.JoinPoint;
import org.aspectj.lang.annotation.AfterReturning;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.annotation.Before;
import org.aspectj.lang.annotation.Pointcut;
import org.aspectj.lang.reflect.MethodSignature;
import org.springframework.stereotype.Repository;
import com.coe.exp.core.dataSource.DataSourceHolder;
@Aspect
@Repository
public class DataSourceAspect {
// 这里切点指定, 拦截所有 com.*.dao 包的 Dao..Impl 类的所有方法
@Pointcut("execution(* com..dao..*Impl.*(..))")
private void anyMethod() {
}
@AfterReturning(value = "anyMethod()", returning = "result")
public void afterReturning(JoinPoint joinPoint,Object result){
DataSourceHolder.clearDataSourceType();//方法执行成功后清除数据源信息
}
@Before(value="anyMethod()")
public void before(JoinPoint joinPoint) throws Throwable {
MethodSignature methodSignature = (MethodSignature) joinPoint.getSignature();
Method method = methodSignature.getMethod();
//如果方法体上使用了DataSource注解
if (method.isAnnotationPresent(DataSource.class)) {
//获取该方法上的注解名
DataSource datasource = method.getAnnotation(DataSource.class);
//将方法体上的注解的值赋予给DataSourceHolder数据源持有类
DataSourceHolder.setDataSourceType(datasource.value());
}
}
}