详细聊聊SpringBoot中动态切换数据源的方法
其实这个表示有点不太对,应该是 Druid 动态切换数据源的方法,只是应用在了 springboot 框架中,准备代码准备了半天,之前在一次数据库迁移中使用了,发现 Druid 还是很强大的,用来做动态数据源切换很方便。
首先这里的场景跟我原来用的有点点区别,在项目中使用的是通过配置中心控制数据源切换,统一切换,而这里的例子多加了个可以根据接口注解配置
第一部分是最核心的,如何基于 Spring JDBC 和 Druid 来实现数据源切换,是继承了org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource 这个类,他的determineCurrentLookupKey方法会被调用来获得用来决定选择那个数据源的对象,也就是 lookupKey,也可以通过这个类看到就是通过这个 lookupKey 来路由找到数据源。
public class DynamicDataSource extends AbstractRoutingDataSource { @Override protected Object determineCurrentLookupKey() { if (DatabaseContextHolder.getDatabaseType() != null) { return DatabaseContextHolder.getDatabaseType().getName(); } return DatabaseType.MASTER1.getName(); } }
而如何使用这个 lookupKey 呢,就涉及到我们的 DataSource 配置了,原来就是我们可以直接通过spring 的 jdbc 配置数据源,像这样
现在我们要使用 Druid 作为数据源了,然后配置 DynamicDataSource的参数,通过 key 来选择对应的 DataSource,也就是下面配的 master1 和 master2
<bean id="master1" class="com.alibaba.druid.pool.DruidDataSource" init-method="init" destroy-method="close" p:driverClassName="com.mysql.cj.jdbc.Driver" p:url="${master1.demo.datasource.url}" p:username="${master1.demo.datasource.username}" p:password="${master1.demo.datasource.password}" p:initialSize="5" p:minIdle="1" p:maxActive="10" p:maxWait="60000" p:timeBetweenEvictionRunsMillis="60000" p:minEvictableIdleTimeMillis="300000" p:validationQuery="SELECT 'x'" p:testWhileIdle="true" p:testOnBorrow="false" p:testOnReturn="false" p:poolPreparedStatements="false" p:maxPoolPreparedStatementPerConnectionSize="20" p:connectionProperties="config.decrypt=true" p:filters="stat,config"/> <bean id="master2" class="com.alibaba.druid.pool.DruidDataSource" init-method="init" destroy-method="close" p:driverClassName="com.mysql.cj.jdbc.Driver" p:url="${master2.demo.datasource.url}" p:username="${master2.demo.datasource.username}" p:password="${master2.demo.datasource.password}" p:initialSize="5" p:minIdle="1" p:maxActive="10" p:maxWait="60000" p:timeBetweenEvictionRunsMillis="60000" p:minEvictableIdleTimeMillis="300000" p:validationQuery="SELECT 'x'" p:testWhileIdle="true" p:testOnBorrow="false" p:testOnReturn="false" p:poolPreparedStatements="false" p:maxPoolPreparedStatementPerConnectionSize="20" p:connectionProperties="config.decrypt=true" p:filters="stat,config"/> <bean id="dataSource" class="com.nicksxs.springdemo.config.DynamicDataSource"> <property name="targetDataSources"> <map key-type="java.lang.String"> <!-- master --> <entry key="master1" value-ref="master1"/> <!-- slave --> <entry key="master2" value-ref="master2"/> </map> </property> <property name="defaultTargetDataSource" ref="master1"/> </bean>
现在就要回到头上,介绍下这个DatabaseContextHolder,这里使用了 ThreadLocal 存放这个 DatabaseType,为啥要用这个是因为前面说的我们想要让接口层面去配置不同的数据源,要把持相互隔离不受影响,就使用了 ThreadLocal,关于它也可以看我前面写的一篇文章聊聊传说中的 ThreadLocal,而 DatabaseType 就是个简单的枚举
public class DatabaseContextHolder { public static final ThreadLocal<DatabaseType> databaseTypeThreadLocal = new ThreadLocal<>(); public static DatabaseType getDatabaseType() { return databaseTypeThreadLocal.get(); } public static void putDatabaseType(DatabaseType databaseType) { databaseTypeThreadLocal.set(databaseType); } public static void clearDatabaseType() { databaseTypeThreadLocal.remove(); } } public enum DatabaseType { MASTER1("master1", "1"), MASTER2("master2", "2"); private final String name; private final String value; DatabaseType(String name, String value) { this.name = name; this.value = value; } public String getName() { return name; } public String getValue() { return value; } public static DatabaseType getDatabaseType(String name) { if (MASTER2.name.equals(name)) { return MASTER2; } return MASTER1; } }
这边可以看到就是通过动态地通过putDatabaseType设置lookupKey来进行数据源切换,要通过接口注解配置来进行设置的话,我们就需要一个注解
@Retention(RetentionPolicy.RUNTIME) @Target(ElementType.METHOD) public @interface DataSource { String value(); }
这个注解可以配置在我的接口方法上,比如这样
public interface StudentService { @DataSource("master1") public Student queryOne(); @DataSource("master2") public Student queryAnother(); }
通过切面来进行数据源的设置
@Aspect @Component @Order(-1) public class DataSourceAspect { @Pointcut("execution(* com.nicksxs.springdemo.service..*.*(..))") public void pointCut() { } @Before("pointCut()") public void before(JoinPoint point) { Object target = point.getTarget(); System.out.println(target.toString()); String method = point.getSignature().getName(); System.out.println(method); Class<?>[] classz = target.getClass().getInterfaces(); Class<?>[] parameterTypes = ((MethodSignature) point.getSignature()) .getMethod().getParameterTypes(); try { Method m = classz[0].getMethod(method, parameterTypes); System.out.println("method"+ m.getName()); if (m.isAnnotationPresent(DataSource.class)) { DataSource data = m.getAnnotation(DataSource.class); System.out.println("dataSource:"+data.value()); DatabaseContextHolder.putDatabaseType(DatabaseType.getDatabaseType(data.value())); } } catch (Exception e) { e.printStackTrace(); } } @After("pointCut()") public void after() { DatabaseContextHolder.clearDatabaseType(); } }
通过接口判断是否带有注解跟是注解的值,DatabaseType 的配置不太好,不过先忽略了,然后在切点后进行清理
这是我 master1 的数据,
master2 的数据
然后跑一下简单的 demo,
@Override public void run(String...args) { LOGGER.info("run here"); System.out.println(studentService.queryOne()); System.out.println(studentService.queryAnother()); }
看一下运行结果
其实这个方法应用场景不止可以用来迁移数据库,还能实现精细化的读写数据源分离之类的,算是做个简单记录和分享。
总结
到此这篇关于SpringBoot中动态切换数据源的文章就介绍到这了,更多相关SpringBoot动态切换数据源内容请搜索猪先飞以前的文章或继续浏览下面的相关文章希望大家以后多多支持猪先飞!
相关文章
- ps动态环绕动画效果是现在很多人都非常喜欢的,大多数人还不知道ps动态环绕动画效果怎么制作下面文章就给大家介绍下ps怎么制作科技感十足的动态环绕动画效果,一起来看看...2017-07-06
解决springboot使用logback日志出现LOG_PATH_IS_UNDEFINED文件夹的问题
这篇文章主要介绍了解决springboot使用logback日志出现LOG_PATH_IS_UNDEFINED文件夹的问题,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧...2021-04-28- 这篇文章主要介绍了vue 实现动态路由的方法,文中示例代码非常详细,帮助大家更好的理解和学习,感兴趣的朋友可以了解下...2020-07-06
- 这篇文章主要为大家详细介绍了SpringBoot实现excel文件生成和下载,文中示例代码介绍的非常详细,具有一定的参考价值,感兴趣的小伙伴们可以参考一下...2021-02-09
- 这篇文章主要介绍了详解springBoot启动时找不到或无法加载主类解决办法,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧...2020-09-16
- 今天我们来给大家介绍下在Vue开发中我们经常会碰到的一种需求场景,本文主要介绍了Vue动态查询规则生成组件,需要的朋友们下面随着小编来一起学习学习吧...2021-05-27
- 这篇文章主要介绍了SpringBoot集成Redis实现消息队列的方法,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧...2021-02-10
- 这篇文章主要介绍了解决Springboot get请求是参数过长的情况,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧...2020-09-17
- 这篇文章主要介绍了Springboot+TCP监听服务器搭建过程,本文通过图文并茂的形式给大家介绍的非常详细,对大家的学习或工作具有一定的参考借鉴价值,需要的朋友可以参考下...2020-10-28
- 这篇文章介绍了c#动态调用Webservice的两种方法实例,有需要的朋友可以参考一下...2020-06-25
Spring Boot项目@RestController使用重定向redirect方式
这篇文章主要介绍了Spring Boot项目@RestController使用重定向redirect方式,具有很好的参考价值,希望对大家有所帮助。如有错误或未考虑完全的地方,望不吝赐教...2021-09-02- 这篇文章主要介绍了springBoot 项目排除数据库启动方式,具有很好的参考价值,希望对大家有所帮助。如有错误或未考虑完全的地方,望不吝赐教...2021-09-10
- 本文详细讲解了SQLServer中执行动态SQL的方法,文中通过示例代码介绍的非常详细。对大家的学习或工作具有一定的参考借鉴价值,需要的朋友可以参考下...2022-05-19
- 想在网页中动态地显示当前系统的时间,找了好多,不过都是一些停在那里不动的。。。不过皇天不负有心人,终于让我找到了...2020-06-25
- 下面小编就为大家带来一篇jQuery为动态生成的select元素添加事件的方法。小编觉得挺不错的,现在就分享给大家,也给大家做个参考。一起跟随小编过来看看吧...2016-09-01
springboot中使用@Transactional注解事物不生效的坑
这篇文章主要介绍了springboot中使用@Transactional注解事物不生效的原因,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧...2021-01-26- 最近由于项目的需要,需要动态的添加和删除table中的tr,感觉用JS可以实现,但是在网上找了一下,单纯的自己写JS,感觉太麻烦,而且也不好维护。于是想到了最近学的jQuery。这篇文章给大家用实例介绍了jQuery动态添加与删除tr行的方法,有需要的朋友们可以参考借鉴。...2016-10-20
- 这篇文章主要介绍了SpringBoot接口接收json参数解析,具有很好的参考价值,希望对大家有所帮助。如有错误或未考虑完全的地方,望不吝赐教...2021-10-19
详解SpringBoot之访问静态资源(webapp...)
这篇文章主要介绍了详解SpringBoot之访问静态资源(webapp...),文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧...2020-09-14ShardingSphere jdbc集成多数据源的实现步骤
本文主要介绍了ShardingSphere jdbc集成多数据源的实现步骤,文中通过示例代码介绍的非常详细,具有一定的参考价值,感兴趣的小伙伴们可以参考一下...2021-10-21