谈谈Spring Boot 数据源加载及其多数据源简单实现(小结)
业务需求[*]提供所有微服务数据源的图形化维护功能
[*]代码生成可以根据选择的数据源加载表等源信息
[*]数据源管理要支持动态配置，实时生效
附录效果图
实现思路
本文提供方法仅供类似简单业务场景，在生产环境和复杂的业务场景 请使用分库分表的中间件（例如mycat）或者框架 sharding-sphere （一直在用）等
先来看spring 默认的数据源注入策略，如下代码默认的事务管理器在初始化时回去加载数据源实现。这里就是我们动态数据源的入口
// 默认的事务管理器
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 默认提供的路由数据源字段
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哟
@slf4j
public class dynamicdatasource extends abstractroutingdatasource {
/**
* 指定路由key,这里很简单 获取 threadlocal 中目标key 即可
*
* @return
*/
@override
protected object determinecurrentlookupkey() {
return dynamicdatasourcecontextholder.getdatasourcetype();
}
}
把我们动态数据源实现注入到spring 的事务管理器,去数据库查询出来全部的数据源信息，定义一个个具体的数据源实现 我这里使用的hikaridatasource 给他赋值等等
@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 即可，会自动根据选择路由数据源
dynamicdatasourcecontextholder.setdatasourcetype（key）
这里当然也可以根据aop 自定义注解等实现。
如何动态数据源动态配置
上边其实已经完成了 我们想要的需求功能，但是有什么问题呢？
我们在数据源管理面维护了数据源，动态去修改这个 datasourcemap 其实是无效的，不能做到实时刷新
我们来看下 abstractroutingdatasource 的加载map 数据源的源码，只有在初始化的时候调用 afterpropertiesset 去初始数据源map.
那我们只要获取当前的dynamicdatasource bean 手动调用afterpropertiesset 即可。
整个代码如下
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();
}
以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持CodeAE代码之家。
原文链接：https://segmentfault.com/a/1190000018844625
文档来源：http://www.zzvips.com/article/180136.html
页:
[1]