spring中怎么通过注解切换多数据源，针对这个问题，这篇文章详细介绍了相对应的分析和解答，希望可以帮助更多想解决这个问题的小伙伴找到更简单易行的方法。
第一步，配置数据源
<bean id="masterDataSource" class="com.alibaba.druid.pool.DruidDataSource"> <property name="username" value="root" /> <property name="password" value="spring" /> <property name="url" value="jdbc:mysql://localhost:3306/taotao?characterEncoding=utf-8" /> <!-- 最大并发连接数 --> <property name="maxActive" value="30" /> <!-- 最小空闲连接数 --> <property name="minIdle" value="5" /> <!-- 用于显示数据源监控中的sql语句监控 --> <property name="filters" value="stat" /> </bean> <bean id="slaveDataSource" class="com.alibaba.druid.pool.DruidDataSource"> <property name="username" value="root" /> <property name="password" value="spring" /> <property name="url" value="jdbc:mysql://localhost:3306/taobao?characterEncoding=utf-8" /> <!-- 最大并发连接数 --> <property name="maxActive" value="30" /> <!-- 最小空闲连接数 --> <property name="minIdle" value="5" /> <!-- 用于显示数据源监控中的sql语句监控 --> <property name="filters" value="stat" /> </bean>
第二步，定义用来切库的注解，和枚举类
@Target({ElementType.METHOD})
@Retention(RetentionPolicy.RUNTIME)
public @interface Dataswitch {
Datatype value() default Datatype.master;
}
public enum Datatype {
master("masterDataSource"),slave("slaveDataSource");
private String value;
Datatype(String name){
this.value = name;
}
public String getValue() {
return value;
}
public void setValue(String value) {
this.value = value;
}
}第三步，定义一个当前线程的变量的工具类，用于设置对应的数据源名称
public class DynamicDataSourceHolder {
private static final ThreadLocal<String> threadLocal = new ThreadLocal<String>();
public static String getThreadLocal() {
return threadLocal.get();
}
public static void setThreadLocal(String name) {
threadLocal.set(name);
}
public static void clear(){
threadLocal.remove();
}
}第四步，创建AbstactRoutingDataSource的子类，重写determineCurrentLockupKey方法
public class DynamicDataSource extends AbstractRoutingDataSource{
protected Object determineCurrentLookupKey() {
System.out.println(DynamicDataSourceHolder.getThreadLocal());
return DynamicDataSourceHolder.getThreadLocal();
}
}第五步，将多数据源配置到用我们创建的DynamicDataSource
<bean id="dynamicDataSource" class="xin.youhuila.sorceswitch.process.DynamicDataSource"> <property name="targetDataSources"> <map key-type="java.lang.String"> <entry key="masterDataSource" value-ref="masterDataSource"></entry> <entry key="slaveDataSource" value-ref="slaveDataSource"></entry> </map> </property> <property name="defaultTargetDataSource" ref="masterDataSource"></property> </bean>
第六步，配置切面，在操作数据库方法之前，获取注解配置的数据源名称，返回
@Component
@Aspect
@Order(0)
public class DataSourceAspect {
@Pointcut("execution (* xin.youhuila.sorceswitch.service..*(..))")
public void aspect(){
}
@Before("aspect()")
public void before(JoinPoint joinPoint){
Class<?> clazz = joinPoint.getTarget().getClass();
Method[] method = clazz.getMethods();
Dataswitch dataswitch = null;
boolean is = false;
for(Method m:method){//此处最好改成通过签名获取调用方法是否含有注解，而不是遍历每个方法，可参考http://www.gitout.cn/?p=2398
if(m.isAnnotationPresent(Dataswitch.class)){
dataswitch = m.getAnnotation(Dataswitch.class);
DynamicDataSourceHolder.setThreadLocal(dataswitch.value().getValue());
is = true;
}
}
if(!is){
DynamicDataSourceHolder.setThreadLocal(Datatype.master.getValue());
}
}
@After("aspect()")
public void after(){
DynamicDataSourceHolder.clear();
}
}第七步，使用
@Service
public class DemoService {
@Autowired
DemoMapper demoMapper;
@Dataswitch(Datatype.master)
public void select(){
List<Demo> d = demoMapper.select();
for(Demo demo:d){
System.out.println(demo);
}
}
}--------------------------------http://www.gitout.cn/?p=2398文章实现-----------------------
/**
* 数据源切面
*/
@Aspect
@Component
public class DynamicDataSourceAspect {
@Pointcut("@annotation(com...datasource.DynamicDataSourceAnnotation)")
public void pointCut() {
}
@Before("pointCut()")
public void testBefore(JoinPoint point) {
// 获得当前访问的class
Class<?> className = point.getTarget().getClass();
DynamicDataSourceAnnotation dataSourceAnnotation = className.getAnnotation(DynamicDataSourceAnnotation.class);
String dataSource = DataSourceConst.DB_ACTIVITY;
// 优先级: 方法 > 类 > DB_ACTIVITY
if(dataSourceAnnotation != null) {
dataSource = dataSourceAnnotation.dataSource();
}
String methodName = point.getSignature().getName();
// 得到方法的参数的类型
Class<?>[] argClass = ((MethodSignature) point.getSignature()).getParameterTypes();
try {
Method method = className.getMethod(methodName, argClass);
if (method.isAnnotationPresent(DynamicDataSourceAnnotation.class)) {
DynamicDataSourceAnnotation annotation = method.getAnnotation(DynamicDataSourceAnnotation.class);
dataSource = annotation.dataSource();
}
} catch (Exception e) {
e.printStackTrace();
}
DataSourceContextHolder.setDataSourceType(dataSource);
}
@After("pointCut()")
public void testAfter(JoinPoint point) {
// 获得当前访问的class
Class<?> className = point.getTarget().getClass();
DynamicDataSourceAnnotation dataSourceAnnotation = className.getAnnotation(DynamicDataSourceAnnotation.class);
if (dataSourceAnnotation != null) {
// 获得访问的方法名
String methodName = point.getSignature().getName();
// 得到方法的参数的类型
Class<?>[] argClass = ((MethodSignature) point.getSignature()).getParameterTypes();
String dataSource = DataSourceConst.DB_ACTIVITY;
try {
Method method = className.getMethod(methodName, argClass);
if (method.isAnnotationPresent(DynamicDataSourceAnnotation.class)) {
DynamicDataSourceAnnotation annotation = method.getAnnotation(DynamicDataSourceAnnotation.class);
dataSource = annotation.dataSource();
}
} catch (Exception e) {
e.printStackTrace();
}
if (dataSource != null && !DataSourceConst.DB_ACTIVITY.equals(dataSource)) {
DataSourceContextHolder.clearDataSourceType();
}
}
}
}关于spring中怎么通过注解切换多数据源问题的解答就分享到这里了，希望以上内容可以对大家有一定的帮助，如果你还有很多疑惑没有解开，可以关注亿速云行业资讯频道了解更多相关知识。
免责声明：本站发布的内容（图片、视频和文字）以原创、转载和分享为主，文章观点不代表本网站立场，如果涉及侵权请联系站长邮箱：is@yisu.com进行举报，并提供相关证据，一经查实，将立刻删除涉嫌侵权内容。
