业务需求
- 提供所有微服务数据源的图形化维护功能
- 代码生成可以根据选择的数据源加载表等源信息
- 数据源管理要支持动态配置，实时生效
附录效果图
实现思路
本文提供方法仅供类似简单业务场景，在生产环境和复杂的业务场景 请使用分库分表的中间件（例如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
|
@slf4jpublic 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@allargsconstructorpublic 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.com/a/1190000018844625
