紧接上篇 - 实现
1 创建 DynamicDataSource 继承 AbstractRoutingDataSource 重写 determineCurrentLookupKey 方法
/**
*
* @Author: ll
* @Date: 2018年5月29日 上午10:32:32
* @Description: 多数据源
* @Version: 1.0
*/
public class DynamicDataSource extends AbstractRoutingDataSource {
@Override
protected Object determineCurrentLookupKey() {
return DynamicDataSourceHolder.getDatasource();
}
}
/**
*
* @Author: ll
* @Date: 2018年5月29日 上午10:32:32
* @Description: 多数据源
* @Version: 1.0
*/
public class DynamicDataSource extends AbstractRoutingDataSource {
@Override
protected Object determineCurrentLookupKey() {
return DynamicDataSourceHolder.getDatasource();
}
}
2 实现 DynamicDataSourceHolder 类
/**
*
* @Author: ll
* @Date: 2018年5月29日 下午10:32:32
* @Description: 多数据源
* @Version: 1.0
*/
public class DynamicDataSourceHolder {
/** 数据源标识保存在线程变量中,避免多线程操作数据源时互相干扰 */
private static final ThreadLocal<String> contextHolder = new ThreadLocal<String>();
/**
* 设置数据源名称
* @param dastsource 数据源名称
*/
public static void setDatasource(String datasource) {
contextHolder.set(datasource);
}
/**
* 获取数据源名称
* @return 数据源名称
*/
public static String getDatasource() {
return contextHolder.get();
}
/**
* 清除标志
*/
public static void clearDatasource() {
contextHolder.remove();
}
}
3 配置多数据源
由于代码较多 在此粘出关键代码
<bean id="dataSource" class="com.enter.net.frame.datasources.DynamicDataSource">
<property name="targetDataSources">
<map key-type="java.lang.String">
<entry key="fxdataSource" value-ref="fxdataSource"></entry>
<entry key="gblzdataSource" value-ref="gblzdataSource"></entry>
</map>
</property>
<!-- 默认使用的数据库-->
<property name="defaultTargetDataSource" ref="fxdataSource"></property>
</bean>
4 使用数据源
本人准备使用注解的方式同时结合spring的AOP
首先定义个一个注解 DataSource
/**
*
* @Author: ll
* @Date: 2018年5月29日 上午10:32:32
* @Description: 多数据源注解
* @Version: 1.0
*/
@Target({ElementType.TYPE, ElementType.METHOD})
@Retention(RetentionPolicy.RUNTIME)
public @interface DataSource {
String value();
}
2 定义一个切面
/**
*
* @Author: ll
* @Date: 2018年5月29日 上午10:32:32
* @Description: 多数据源切面
* @Version: 1.0
*/
@Aspect
@Component
public class DynamicDataSourceInterceptor {
/**
* 拦截目标方法,获取由@DataSource指定的数据源标识,设置到线程存储中以便切换数据源
* @param point
* @throws Exception
*/
@Before("execution(* com.enter.net.fhbusiness..*Controller.*(..)) or execution(* com.enter.net.system..*Controller.*(..))")
public void intercept(JoinPoint point) throws Exception {
Class<?> target = point.getTarget().getClass();
MethodSignature signature = (MethodSignature) point.getSignature();
resolveDataSource(target, signature.getMethod());
}
/**
* 提取目标对象方法注解和类型注解中的数据源标识
* @param clazz
* @param method
*/
private void resolveDataSource(Class<?> clazz, Method method) {
try {
Class<?>[] types = method.getParameterTypes();
// 默认使用类型注解
if (clazz.isAnnotationPresent(DataSource.class)) {
DataSource source = clazz.getAnnotation(DataSource.class);
DynamicDataSourceHolder.setDatasource(source.value());
}
// 方法注解可以覆盖类型注解
if(types.length!=0) {
Method m = clazz.getMethod(method.getName(), types);
if (m != null && m.isAnnotationPresent(DataSource.class)) {
DataSource source = m.getAnnotation(DataSource.class);
DynamicDataSourceHolder.setDatasource(source.value());
}
}
} catch (Exception e) {
e.printStackTrace();
}
}
}
这时候切面 会根据配置的表达式 进行拦截查看是否使用DataSource 注解 如果使用 则获取注解的值 来设定指定的s
下一篇 https://blog.csdn.net/lilongwangyamin/article/details/80507121 总结