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

*/

protected abstract DataSource determineDataSource();

@Override

public Connection getConnection() throws SQLException {

return determineDataSource().getConnection();

}

@Override

public Connection getConnection(String username, String password) throws SQLException {

return determineDataSource().getConnection(username, password);

}

}

分析mybatis plus配置类DynamicDataSourceAutoConfiguration，可以知道，只需要实现可以监听nacos参数配置，可以动态切换数据源配置的DataSource即可，DynamicDataSourceAutoConfiguration的部分源码如下：

@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

@ConditionalOnMissingBean

public DynamicDataSourceProvider dynamicDataSourceProvider() {

return new YmlDynamicDataSourceProvider(properties);

}

/**

* 在spring中注入动态数据源，注入自定义的DataSource即可扩展

*/

@Bean

@ConditionalOnMissingBean

public DataSource dataSource(DynamicDataSourceProvider dynamicDataSourceProvider) {

DynamicRoutingDataSource dataSource = new DynamicRoutingDataSource();

dataSource.setPrimary(properties.getPrimary());

dataSource.setStrategy(properties.getStrategy());

dataSource.setProvider(dynamicDataSourceProvider);

dataSource.setP6spy(properties.getP6spy());

dataSource.setStrict(properties.getStrict());

return dataSource;

}

}

定义DynamicRoutingAndSwitchingDataSource，实现数据源动态路由，数据库配置自动切换。采用装饰模式扩展mybatis plus的DynamicRoutingDataSource的功能，实现EnvironmentChangeEvent事件的监听器ApplicationListener，在nacos配置修改的时候，监听配置的修改。

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();

}

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

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

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

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

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

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

最后

针对最近很多人都在面试，我这边也整理了相当多的面试专题资料，也有其他大厂的面经。希望可以帮助到大家。

最新整理面试题

上述的面试题答案都整理成文档笔记。也还整理了一些面试资料&最新2021收集的一些大厂的面试真题

最新整理电子书

最新整理大厂面试文档

以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持。
《互联网大厂面试真题解析、进阶开发核心学习笔记、全套讲解视频、实战项目源码讲义》点击传送门即可获取！
//img-community.csdnimg.cn/images/e5c14a7895254671a72faed303032d36.jpg" alt=“img” style=“zoom: 33%;” />

最后

针对最近很多人都在面试，我这边也整理了相当多的面试专题资料，也有其他大厂的面经。希望可以帮助到大家。

最新整理面试题
[外链图片转存中…(img-riE3yUxm-1713632697315)]

上述的面试题答案都整理成文档笔记。也还整理了一些面试资料&最新2021收集的一些大厂的面试真题

最新整理电子书

[外链图片转存中…(img-yE5uAg5a-1713632697315)]

最新整理大厂面试文档

[外链图片转存中…(img-GaiuGtzJ-1713632697316)]

以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持。
《互联网大厂面试真题解析、进阶开发核心学习笔记、全套讲解视频、实战项目源码讲义》点击传送门即可获取！