业务需求
提供所有微服务数据源的图形化维护功能代码生成可以根据选择的数据源加载表等源信息数据源管理要支持动态配置,实时生效 附录效果图
实现思路
本文提供方法仅供类似简单业务场景,在生产环境和复杂的业务场景 请使用分库分表的中间件(例如mycat)或者框架 sharding-sphere (一直在用)等。
先来看Spring 默认的数据源注入策略,如下代码默认的事务管理器在初始化时回去加载数据源实现。这里就是我们动态数据源的入口:
通过注入一个新的DataSourceTransactionManager 实现,并且给它设置多个 DataSource 来实现多数据源实现 :
看下Spring 默认提供的路由数据源字段
开始动手
实现AbstractRoutingDataSource,定一个动态数据源实现,只需要实现他的路由key 查找方法即可。 这里的路由key 对应其实是resolvedDataSources Map 的key哟。
把我们动态数据源实现注入到Spring 的事务管理器,去数据库查询出来全部的数据源信息,定义一个个具体的数据源实现 我这里使用的HikariDataSource 给他赋值等等
怎么使用
只需要根据用户前台选择的数据源key ,在业务类保存到TTL 即可,会自动根据选择路由数据源
这里当然也可以根据AOP 自定义注解等实现。
如何动态数据源动态配置
上边其实已经完成了 我们想要的需求功能,但是有什么问题呢?
我们在数据源管理面维护了数据源,动态去修改这个 dataSourceMap 其实是无效的,不能做到实时刷新。
我们来看下 AbstractRoutingDataSource 的加载map 数据源的源码,只有在初始化的时候调用 afterPropertiesSet 去初始数据源map。
那我们只要获取当前的DynamicDataSource bean 手动调用afterPropertiesSet 即可。 整个代码如下:
举报/反馈

疯狂代码

3.7万获赞 5262粉丝
擅长并专注于计算机java编程文化领域!
科技领域爱好者
关注
0
0
收藏
分享