SpringBoot集成nacos动态刷新数据源

2023-10-29

前言

因为项目需要,需要在项目运行过程中能够动态修改数据源(即:数据源的热更新)。这里以com.alibaba.druid.pool.DruidDataSource数据源为例

第一步:重写DruidAbstractDataSource类

这里为什么要重写这个类:因为DruidDataSource数据源在初始化后,就不允许再重新设置数据库的url和userName

public void setUrl(String jdbcUrl) {
        if (StringUtils.equals(this.jdbcUrl, jdbcUrl)) {
            return;
        }
        // 重写的时候,需要将这个判断注释掉,否则会报错
        // if (inited) {
        //     throw new UnsupportedOperationException();
        // }

        if (jdbcUrl != null) {
            jdbcUrl = jdbcUrl.trim();
        }

        this.jdbcUrl = jdbcUrl;

        // if (jdbcUrl.startsWith(ConfigFilter.URL_PREFIX)) {
        // this.filters.add(new ConfigFilter());
        // }
    }
    
    public void setUsername(String username) {
        if (StringUtils.equals(this.username, username)) {
            return;
        }
		// 重写的时候,需要将这个判断注释掉,否则会报错
        // if (inited) {
        //     throw new UnsupportedOperationException();
        // }

        this.username = username;
    }

重写的时候包路径不能变,只有这样类加载的时候才会优先加载重写后的类
在这里插入图片描述

第二步:配置动态获取nacos配置信息

package com.mp.demo.config;

import com.alibaba.druid.pool.DruidDataSource;
import lombok.Data;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.cloud.context.config.annotation.RefreshScope;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

@Slf4j
@Configuration
@RefreshScope
@Data
public class DruidConfiguration
{
    @Value("${spring.datasource.url}")
    private String dbUrl;

    @Value("${spring.datasource.username}")
    private String username;

    @Value("${spring.datasource.password}")
    private String password;

    @Value("${spring.datasource.driver-class-name}")
    private String driverClassName;

    @Bean
    @RefreshScope
    public DruidDataSource dataSource()
    {
        DruidDataSource datasource = new DruidDataSource();
        datasource.setUrl(this.dbUrl);
        datasource.setUsername(username);
        datasource.setPassword(password);
        datasource.setDriverClassName(driverClassName);
        return datasource;
    }
}

这里要注意增加@RefreshScope注解

第三步:手动刷新数据源

    @GetMapping("/refresh")
    public String refresh() throws SQLException
    {
        DruidDataSource master = SpringUtils.getBean("dataSource");
        master.setUrl(druidConfiguration.getDbUrl());
        master.setUsername(druidConfiguration.getUsername());
        master.setPassword(druidConfiguration.getPassword());
        master.setDriverClassName(druidConfiguration.getDriverClassName());
        master.restart();
        return userName + "<>" + jdbcUrl+"----------"+druidConfiguration.getDbUrl();
    }

源码地址:https://gitee.com/jackson_hou/RefreshDataSource.git

本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)

SpringBoot集成nacos动态刷新数据源 的相关文章

随机推荐