如何使用Nacos实现数据库连接的自动切换？(1)

Nacos配置在触发事件时，只会传输更新值的key/value值，所以，需要根据更新属性的key值，判断是否修改的数据库配置。其源码如下：

@Slf4j

public class DynamicRoutingAndSwitchingDataSource extends AbstractRoutingDataSource implements ApplicationListener, EnvironmentAware,

InitializingBean, DisposableBean {

/**

* mybatis plus 动态数据源Properties

*/

private DynamicDataSourceProperties dynamicDataSourceProperties;

/**

* 多数据源加载接口

*/

private YmlDynamicDataSourceProvider dynamicDataSourceProvider;

/**

* 动态数据源Properties 属性key集合

*/

private Set propertyKeys;

/**

* mybatis plus动态路由数据源，使用装饰模式修饰

*/

private DynamicRoutingDataSource dynamicRoutingDataSource;

/**

* spring Environment 配置对象

*/

private Environment environment;

public DynamicRoutingAndSwitchingDataSource(DynamicDataSourceProperties properties, DynamicDataSourceProvider provider) {

this.dynamicDataSourceProperties = properties;

this.dynamicDataSourceProvider = (YmlDynamicDataSourceProvider) provider;

this.dynamicRoutingDataSource = this.buildDynamicRoutingDataSource(properties, provider);

this.propertyKeys = Sets.newHashSet(

ConfigProperty.MASTER_USERNAME_KEY, ConfigProperty.MASTER_PASSWORD_KEY,

ConfigProperty.MASTER_DRIVER_CLASS_NAME_KEY, ConfigProperty.MASTER_RL_KEY,

ConfigProperty.SLAVE_1_USERNAME_KEY, ConfigProperty.SLAVE_1_PASSWORD_KEY,

ConfigProperty.SLAVE_1_DRIVER_CLASS_NAME_KEY, ConfigProperty.SLAVE_1_RL_KEY);

}

private DynamicRoutingDataSource buildDynamicRoutingDataSource(DynamicDataSourceProperties dynamicDataSourceProperties, DynamicDataSourceProvider provider) {

DynamicRoutingDataSource dataSource = new DynamicRoutingDataSource();

dataSource.setPrimary(dynamicDataSourceProperties.getPrimary());

dataSource.setStrategy(dynamicDataSourceProperties.getStrategy());

dataSource.setProvider(provider);

dataSource.setP6spy(dynamicDataSourceProperties.getP6spy());

dataSource.setStrict(dynamicDataSourceProperties.getStrict());

return dataSource;

}

/**

* 监听配置修改的EnvironmentChangeEvent事件

*/

@Override

public void onApplicationEvent(EnvironmentChangeEvent event) {

Set keys;

if (isNotDatasourceConfigChanged(keys = event.getKeys())) {

log.info(“keys {} are not database config!”, JSON.toJSONString(keys));

return;

}

DynamicDataSourceProperties refreshProperties = this.dynamicDataSourceProperties;

Map<String, DataSourceProperty> datasource = refreshProperties.getDatasource();

Map<String, Properties> changedPropertiesMap = this.getChangedProperties();

if (log.isDebugEnabled()) {

log.debug(“changed properties {}”, JSON.toJSONString(changedPropertiesMap));

}

for (Map.Entry<String, DataSourceProperty> entry : datasource.entrySet()) {

String nodeName = entry.getKey();

Properties properties;

if (Objects.isNull(properties = changedPropertiesMap.get(nodeName))) {

log.info(“node {} has no properties!changedPropertiesMap {}”, nodeName, JSON.toJSONString(changedPropertiesMap));

continue;

}

DataSourceProperty dataSourceProperty = entry.getValue();

String userName;

if (StringUtils.isNotBlank(userName = properties.getProperty(ConfigProperty.USERNAME))) {

dataSourceProperty.setUsername(userName);

}

String password;

if (StringUtils.isNotBlank(password = properties.getProperty(ConfigProperty.PASSWORD))) {

dataSourceProperty.setPassword(password);

}

String driverClassName;

if (StringUtils.isNotBlank(driverClassName = properties.getProperty(ConfigProperty.DRIVER_CLASS_NAME))) {

dataSourceProperty.setDriverClassName(driverClassName);

}

String url;

if (StringUtils.isNotBlank(url = properties.getProperty(ConfigProperty.URL))) {

dataSourceProperty.setUrl(url);

}

}

try {

// 使用反射设置properties，避免初始化后，spring自动注入的DynamicDataSourceCreator无法获取

// @Autowired

// private DynamicDataSourceCreator dynamicDataSourceCreator;

Reflections.setFieldValue(this.dynamicDataSourceProvider, “properties”, refreshProperties);

this.dynamicRoutingDataSource = this.buildDynamicRoutingDataSource(refreshProperties, this.dynamicDataSourceProvider);

// 重置数据源

this.dynamicRoutingDataSource.afterPropertiesSet();

} catch (Exception e) {

throw new RuntimeException(“refresh dynamic routing switch datasource error!”, e);

}

}

private boolean isNotDatasourceConfigChanged(Set keys) {

for (String key : keys) {

if (propertyKeys.contains(key)) {

return false;

}

}

return true;

}

private boolean isMasterKey(String key) {

return key.contains(ConfigProperty.MASTER);

}

private boolean isSlave1Key(String key) {

return key.contains(ConfigProperty.SLAVE_1);

}

private boolean isUsernameKey(String key) {

return key.contains(ConfigProperty.USERNAME);

}

private boolean isPasswordKey(String key) {

return key.contains(ConfigProperty.PASSWORD);

}

private boolean isDriverClassNameKey(String key) {

return key.contains(ConfigProperty.DRIVER_CLASS_NAME);

}

private boolean isUrlKey(String key) {

return key.contains(ConfigProperty.URL);

}

private Map<String, Properties> getChangedProperties() {

Map<String, Properties> propertiesMap = ImmutableMap.of(ConfigProperty.MASTER, new Properties(), ConfigProperty.SLAVE_1, new Properties());

for (String key : this.propertyKeys) {

String value;

if (StringUtils.isBlank(value = this.environment.getProperty(key))) {

continue;

}

String propertyKey = this.convert2propertyKey(key);

if (this.isMasterKey(key)) {

propertiesMap.get(ConfigProperty.MASTER).setProperty(propertyKey, value);

} else if (isSlave1Key(key)) {

propertiesMap.get(ConfigProperty.SLAVE_1).setProperty(propertyKey, value);

}

}

return propertiesMap;

}

private String convert2propertyKey(String key) {

String propertyKey = key;

if (this.isUsernameKey(key)) {

propertyKey = ConfigProperty.USERNAME;

} else if (this.isPasswordKey(key)) {

propertyKey = ConfigProperty.PASSWORD;

} else if (this.isDriverClassNameKey(key)) {

propertyKey = ConfigProperty.DRIVER_CLASS_NAME;

} else if (this.isUrlKey(key)) {

propertyKey = ConfigProperty.URL;

}

return propertyKey;

}

/**

* 设置Environment 属性

*/

@Override

public void setEnvironment(Environment environment) {

this.environment = environment;

}

/**

* 调用被装饰的dynamicRoutingDataSource的方法，获取指定的数据源

*/

@Override

protected DataSource determineDataSource() {

return this.dynamicRoutingDataSource.determineDataSource();

}

/**

* 调用被装饰的dynamicRoutingDataSource的方法，初始化数据源

*/

@Override

public void afterPropertiesSet() throws Exception {

this.dynamicRoutingDataSource.afterPropertiesSet();

}

@Override

public void destroy() throws Exception {

this.dynamicRoutingDataSource.destroy();

}

}

public interface ConfigProperty {

//数据源属性配置

String MASTER = “master”;

String SLAVE_1 = “slave_1”;

String USERNAME = “username”;

String PASSWORD = “password”;

String DRIVER_CLASS_NAME = “driver-class-name”;

String URL = “url”;

//数据源属性key配置

String MASTER_USERNAME_KEY = “spring.datasource.dynamic.datasource.master.” + USERNAME;

String MASTER_PASSWORD_KEY = “spring.datasource.dynamic.datasource.master.” + PASSWORD;

String MASTER_DRIVER_CLASS_NAME_KEY = “spring.datasource.dynamic.datasource.master.” + DRIVER_CLASS_NAME;

String MASTER_RL_KEY = “spring.datasource.dynamic.datasource.master.” + URL;

String SLAVE_1_USERNAME_KEY = “spring.datasource.dynamic.datasource.slave_1.” + USERNAME;

String SLAVE_1_PASSWORD_KEY = “spring.datasource.dynamic.datasource.slave_1.” + PASSWORD;

String SLAVE_1_DRIVER_CLASS_NAME_KEY = “spring.datasource.dynamic.datasource.slave_1.” + DRIVER_CLASS_NAME;

String SLAVE_1_RL_KEY = “spring.datasource.dynamic.datasource.slave_1.” + URL;

}

自定义DynamicDataSourceAutoConfiguration，在spring中注入定义的DynamicRoutingAndSwitchingDataSource，其源码如下：

@Configuration

@EnableConfigurationProperties(DynamicDataSourceProperties.class)

@AutoConfigureBefore(DataSourceAutoConfiguration.class)

@Import(DruidDynamicDataSourceConfiguration.class)

@ConditionalOnProperty(prefix = DynamicDataSourceProperties.PREFIX, name = “enabled”, havingValue = “true”, matchIfMissing = true)

public class DynamicDataSourceAutoConfiguration {

@Autowired

private DynamicDataSourceProperties properties;

@Bean

public DataSource dataSource(DynamicDataSourceProvider dynamicDataSourceProvider) {

DynamicRoutingAndSwitchingDataSource dataSource = new DynamicRoutingAndSwitchingDataSource(properties, dynamicDataSourceProvider);

return dataSource;

}

}

结果测试，在nacos服务器修改从数据库的配置，如图：

自我介绍一下，小编13年上海交大毕业，曾经在小公司待过，也去过华为、OPPO等大厂，18年进入阿里一直到现在。

深知大多数Java工程师，想要提升技能，往往是自己摸索成长或者是报班学习，但对于培训机构动则几千的学费，着实压力不小。自己不成体系的自学效果低效又漫长，而且极易碰到天花板技术停滞不前！

因此收集整理了一份《2024年Java开发全套学习资料》，初衷也很简单，就是希望能够帮助到想自学提升又不知道该从何学起的朋友，同时减轻大家的负担。

既有适合小白学习的零基础资料，也有适合3年以上经验的小伙伴深入学习提升的进阶课程，基本涵盖了95%以上Java开发知识点，真正体系化！

由于文件比较大，这里只是将部分目录截图出来，每个节点里面都包含大厂面经、学习笔记、源码讲义、实战项目、讲解视频，并且会持续更新！

如果你觉得这些内容对你有帮助，可以扫码获取！！（备注Java获取）

复习的面试资料

这些面试全部出自大厂面试真题和面试合集当中，小编已经为大家整理完毕（PDF版）

• 第一部分：Java基础-中级-高级

• 第二部分：开源框架（SSM：Spring+SpringMVC+MyBatis）

• 第三部分：性能调优（JVM+MySQL+Tomcat）

• 第四部分：分布式（限流：ZK+Nginx；缓存：Redis+MongoDB+Memcached；通讯：MQ+kafka）

• 第五部分：微服务（SpringBoot+SpringCloud+Dubbo）

• 第六部分：其他：并发编程+设计模式+数据结构与算法+网络

进阶学习笔记pdf

• Java架构进阶之架构筑基篇（Java基础+并发编程+JVM+MySQL+Tomcat+网络+数据结构与算法）

• Java架构进阶之开源框架篇（设计模式+Spring+SpringMVC+MyBatis）

• Java架构进阶之分布式架构篇 （限流（ZK/Nginx）+缓存（Redis/MongoDB/Memcached）+通讯（MQ/kafka））

• Java架构进阶之微服务架构篇（RPC+SpringBoot+SpringCloud+Dubbo+K8s）

《互联网大厂面试真题解析、进阶开发核心学习笔记、全套讲解视频、实战项目源码讲义》点击传送门即可获取！
664167)]

[外链图片转存中…(img-vLzhklOd-1713632664167)]

[外链图片转存中…(img-G7QRK24U-1713632664167)]

• Java架构进阶之分布式架构篇 （限流（ZK/Nginx）+缓存（Redis/MongoDB/Memcached）+通讯（MQ/kafka））

[外链图片转存中…(img-9Uj1LrVV-1713632664167)]

[外链图片转存中…(img-E6QTfIyd-1713632664168)]

[外链图片转存中…(img-lOl9HlGO-1713632664168)]

• Java架构进阶之微服务架构篇（RPC+SpringBoot+SpringCloud+Dubbo+K8s）

[外链图片转存中…(img-F9Xi4mBe-1713632664168)]

[外链图片转存中…(img-56ueHe7o-1713632664169)]

《互联网大厂面试真题解析、进阶开发核心学习笔记、全套讲解视频、实战项目源码讲义》点击传送门即可获取！