业务需求
提供所有微服务数据源的图形化维护功能 代码生成可以根据选择的数据源加载表等源信息 数据源管理要支持动态配置，实时生效附录效果图
实现思路
本文提供方法仅供类似简单业务场景，在生产环境和复杂的业务场景 请使用分库分表的中间件（例如mycat）或者框架 sharding-sphere （一直在用）等
先来看spring 默认的数据源注入策略，如下代码默认的事务管理器在初始化时回去加载数据源实现。这里就是我们动态数据源的入口
|
1 2 3 4 5 6 7 8 9 10 11 12 13 14 |
// 默认的事务管理器 ppublic class datasourcetransactionmanager extends abstractplatformtransactionmanager implements resourcetransactionmanager, initializingbean {
// 启动时候注入一个数据源 public void setdatasource( @nullable datasource datasource) { if (datasource instanceof transactionawaredatasourceproxy) { this .datasource = ((transactionawaredatasourceproxy) datasource).gettargetdatasource(); } else { this .datasource = datasource; } } 」 |
通过注入一个新的datasourcetransactionmanager 实现，并且给它设置多个 datasource 来实现多数据源实现
看下spring 默认提供的路由数据源字段
|
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 |
public abstract class abstractroutingdatasource extends abstractdatasource implements initializingbean {
// 用户设置的全部的数据源配置 @nullable private map<object, object> targetdatasources; // 为空默认的数据源配置 @nullable private object defaulttargetdatasource;
// 路由键查找实现 private datasourcelookup datasourcelookup = new jndidatasourcelookup();
// 最终有效的数据源配置（一般清空对应上边用户的设置） @nullable private map<object, datasource> resolveddatasources; } |
开始动手
实现abstractroutingdatasource，定一个动态数据源实现，只需要实现他的路由key 查找方法即可。
这里的路由key 对应其实是resolveddatasources map 的key哟
|
1 2 3 4 5 6 7 8 9 10 11 12 13 |
@slf4j public class dynamicdatasource extends abstractroutingdatasource {
/** * 指定路由key,这里很简单 获取 threadlocal 中目标key 即可 * * @return */ @override protected object determinecurrentlookupkey() { return dynamicdatasourcecontextholder.getdatasourcetype(); } } |
把我们动态数据源实现注入到spring 的事务管理器,去数据库查询出来全部的数据源信息，定义一个个具体的数据源实现 我这里使用的hikaridatasource 给他赋值等等
|
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 |
@slf4j @configuration @allargsconstructor public class dynamicdatasourceconfig implements transactionmanagementconfigurer { private final map<object, object> datasourcemap = new hashmap<>( 8 ); private final datasourceproperties datasourceproperties;
@bean ( "dynamicdatasource" ) public dynamicdatasource datasource() { jdbctemplate(dds).queryforlist(datasourceconstant.query_ds_sql); log.info( "开始 -> 初始化动态数据源" ); optional.of(dblist).ifpresent(list -> list.foreach(db -> { log.info( "数据源:{}" , db.get(datasourceconstant.ds_name)); hikaridatasource ds = new hikaridatasource(); datasourcemap.put(db.get(datasourceconstant.ds_route_key), ds); }));
dynamicdatasource ds = new dynamicdatasource(); ds.settargetdatasources(datasourcemap); return ds; }
@bean public platformtransactionmanager txmanager() { return new datasourcetransactionmanager(datasource()); }
@override public platformtransactionmanager annotationdriventransactionmanager() { return txmanager(); }
} |
怎么使用
只需要根据用户前台选择的数据源key ，在业务类保存到ttl 即可，会自动根据选择路由数据源
|
1 |
dynamicdatasourcecontextholder.setdatasourcetype（key） |
这里当然也可以根据aop 自定义注解等实现。
如何动态数据源动态配置
上边其实已经完成了 我们想要的需求功能，但是有什么问题呢？
我们在数据源管理面维护了数据源，动态去修改这个 datasourcemap 其实是无效的，不能做到实时刷新
我们来看下 abstractroutingdatasource 的加载map 数据源的源码，只有在初始化的时候调用 afterpropertiesset 去初始数据源map.
那我们只要获取当前的dynamicdatasource bean 手动调用afterpropertiesset 即可。
整个代码如下
|
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 |
public class dynamicdatasourceconfig implements transactionmanagementconfigurer { private final map<object, object> datasourcemap = new hashmap<>( 8 ); private final datasourceproperties datasourceproperties; private final stringencryptor stringencryptor;
@bean ( "dynamicdatasource" ) public dynamicdatasource datasource() { dynamicdatasource ds = new dynamicdatasource(); hikaridatasource cads = new hikaridatasource(); cads.setjdbcurl(datasourceproperties.geturl()); cads.setdriverclassname(datasourceproperties.getdriverclassname()); cads.setusername(datasourceproperties.getusername()); cads.setpassword(datasourceproperties.getpassword()); ds.setdefaulttargetdatasource(cads); datasourcemap.put( 0 , cads); ds.settargetdatasources(datasourcemap); return ds; }
/** * 组装默认配置的数据源，查询数据库配置 */ @postconstruct public void init() { drivermanagerdatasource dds = new drivermanagerdatasource(); dds.seturl(datasourceproperties.geturl()); dds.setdriverclassname(datasourceproperties.getdriverclassname()); dds.setusername(datasourceproperties.getusername()); dds.setpassword(datasourceproperties.getpassword());
list<map<string, object>> dblist = new jdbctemplate(dds).queryforlist(datasourceconstant.query_ds_sql); log.info( "开始 -> 初始化动态数据源" ); optional.of(dblist).ifpresent(list -> list.foreach(db -> { log.info( "数据源:{}" , db.get(datasourceconstant.ds_name)); hikaridatasource ds = new hikaridatasource(); ds.setjdbcurl(string.valueof(db.get(datasourceconstant.ds_jdbc_url))); ds.setdriverclassname(driver. class .getname()); ds.setusername((string) db.get(datasourceconstant.ds_user_name));
string decpwd = stringencryptor.decrypt((string) db.get(datasourceconstant.ds_user_pwd)); ds.setpassword(decpwd); datasourcemap.put(db.get(datasourceconstant.ds_route_key), ds); }));
log.info( "完毕 -> 初始化动态数据源,共计 {} 条" , datasourcemap.size()); }
/** * 重新加载数据源配置 */ public boolean reload() { init(); dynamicdatasource datasource = datasource(); datasource.settargetdatasources(datasourcemap); datasource.afterpropertiesset(); return boolean . false ; }
@bean public platformtransactionmanager txmanager() { return new datasourcetransactionmanager(datasource()); }
@override public platformtransactionmanager annotationdriventransactionmanager() { return txmanager(); } |
以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持。
原文链接：https://segmentfault测试数据/a/1190000018844625
查看更多关于谈谈Spring Boot 数据源加载及其多数据源简单实现(小结)的详细内容...