服务器之家

服务器之家 > 正文

谈谈Spring Boot 数据源加载及其多数据源简单实现(小结)

时间:2021-07-31 12:29     来源/作者:冷冷

业务需求

  • 提供所有微服务数据源的图形化维护功能
  • 代码生成可以根据选择的数据源加载表等源信息
  • 数据源管理要支持动态配置,实时生效

附录效果图

谈谈Spring Boot 数据源加载及其多数据源简单实现(小结)

谈谈Spring Boot 数据源加载及其多数据源简单实现(小结)

实现思路

本文提供方法仅供类似简单业务场景,在生产环境和复杂的业务场景 请使用分库分表的中间件(例如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 Boot 数据源加载及其多数据源简单实现(小结)

看下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.

谈谈Spring Boot 数据源加载及其多数据源简单实现(小结)

那我们只要获取当前的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

相关文章

热门资讯

2020微信伤感网名听哭了 让对方看到心疼的伤感网名大全
2020微信伤感网名听哭了 让对方看到心疼的伤感网名大全 2019-12-26
yue是什么意思 网络流行语yue了是什么梗
yue是什么意思 网络流行语yue了是什么梗 2020-10-11
背刺什么意思 网络词语背刺是什么梗
背刺什么意思 网络词语背刺是什么梗 2020-05-22
苹果12mini价格表官网报价 iPhone12mini全版本价格汇总
苹果12mini价格表官网报价 iPhone12mini全版本价格汇总 2020-11-13
2021德云社封箱演出完整版 2021年德云社封箱演出在线看
2021德云社封箱演出完整版 2021年德云社封箱演出在线看 2021-03-15
返回顶部