Spring动态添加数据源（全自动）

news2024/11/27 0:26:05

一、使用场景

1、现在网上所有写的动态数据源，都是假动态数据，都是预先知道要连接几个数据库，但这不满足具体真正需要动态添加连接数据库的需求；

2、在很多业务场景下，如项目默认连一个主数据库，服务启动后，因业务不停的变更，需要动态连接其它数据库处理数据，关键点是其它数据库，在服务启动前是不知道连接地址的，所以其它数据库的地址是无法预先配置在application.yml文件中

二、因项目需求，需要真正动态连接其它库处理数据

1、做之前，百度网上搜，都是假动态数据源，为实现上面的真正动态添加数据，也花了几天时间才集成起，废话不多说

三、准备工作

1、准备三个数据，db_main，db_add01，db_add02，其中db_main为项目起动，默认连的主数据库，为了下面测试方便，我在三个数据库中创建了不同的三张表sys_company、sys_role、sys_user，以表示查询的数据不同，数据来源于不同的数据库，截图如下

application.yml中配置主数据库接信息

server:
  port: 8085
spring:
  application:
    name: mydata
  datasource:
    username: root
    password: 123456
    url: jdbc:mysql://192.168.137.202:3306/db_main?useUnicode=true&characterEncoding=utf-8&useSSL=false&serverTimezone=GMT%2B8
    druid:
      initialSize: 5
      minIdle: 5
      maxActive: 20
      maxWait: 60000
      #配置监控属性： 在druid-starter的： com.alibaba.druid.spring.boot.autoconfigure.stat包下进行的逻辑配置
      # WebStatFilter配置，
      stat-view-servlet:
        #配置DruidStatViewServlet的访问地址。后台监控页面的访问地址
        url-pattern: /druid/*
        #禁用HTML页面上的“重置”功能，会把所有监控的数据全部清空，一般不使用
        reset-enable: false
        #监控页面登录的用户名
        login-username: admin
        #监控页面登录的密码
        login-password: 123456
        #白名单
        allow:
        #黑名单
        deny:

2、动态添加数据库核心代码

/**
     * 添加私有数据源
     * @param dbName 数据库名
     * @param url
     * @param username
     * @param password
     */
    public void addDataSource(String dbName,String url,String username,String password){
        if(!StringUtils.isEmpty(dbName)) {
            if (dataSourcesMap.get(dbName) == null) {
                DataSource dataSource = druidDataSource(url, username, password);
                dataSourcesMap.put(dbName, dataSource);
                super.setTargetDataSources(dataSourcesMap);
                super.afterPropertiesSet();
            } else {
                DruidDataSource dataSource = (DruidDataSource) dataSourcesMap.get(dbName);
                if (!dataSource.isEnable()) {
                    try {
                        dataSource.restart();
                    } catch (SQLException e) {
                        e.printStackTrace();
                    }
                }
            }
        }
    }

    /**
     * 移出对应数据源
     * @param dbName
     */
    public void removeDataSource(String dbName){
        if(dataSourcesMap.get(dbName)!=null) {
            DruidDataSource dataSource= (DruidDataSource)dataSourcesMap.get(dbName);
            dataSourcesMap.remove(dbName);
            dataSource.close();
            super.afterPropertiesSet();
        }
    }

3、动态切换数据库

//线程本地环境
    private static final ThreadLocal<String> dataSources = new ThreadLocal<String>();
    //设置数据源,动态切换,就是调用这个setDataSource方法
    public static void setDataSource(String customerType) {
        dataSources.set(customerType);
    }
    //获取数据源
    public static String getDataSource() {
        return (String) dataSources.get();
    }
    //清除数据源
    public static void clearDataSource() {
        dataSources.remove();
    }

4、controller接口测试，代码如下

/**
     * 查询主库数据
     */
    @GetMapping (value="/findMain")
    public SysCompany findMain(@RequestParam(defaultValue = "C190829X5R")String id) {
        DataSourceHolder.setDataSource("db_main");
        SysCompany company = (SysCompany)dataService.findTById(SysCompany.class,id);
        return company;
    }

    /**
     * 查询db_add01号库数据
     * dbName 对应数据库名称
     */
    @GetMapping (value="/addDB")
    public String addDB(@RequestParam(defaultValue = "db_add01") String dbName, @RequestParam(defaultValue = "jdbc:mysql://192.168.137.202:3306/db_add01") String url,
                        @RequestParam(defaultValue = "root")String username, @RequestParam(defaultValue = "123456")String password) {
        dynamicDataSource.addDataSource(dbName,url,username,password);
        return "添加成功";
    }

    /**
     *
     * 查询db_add01号库数据
     * dbName 对应数据库名称
     */
    @GetMapping (value="/findDb01")
    public SysUser findDb01(@RequestParam(defaultValue = "db_add01")String dbName,@RequestParam(defaultValue = "5")String id) {
        DataSourceHolder.setDataSource(dbName);
        SysUser sysUser = (SysUser)dataService.findTById(SysUser.class,id);
        return sysUser;
    }

    /**
     * 查询db_add02号库数据
     * dbName 对应数据库名称
     */
    @GetMapping (value="/findDb02")
    public SysRole findDb02(@RequestParam(defaultValue = "db_add02")String dbName,@RequestParam(defaultValue = "1")String id) {
        DataSourceHolder.setDataSource(dbName);
        SysRole sysRole = (SysRole)dataService.findTById(SysRole.class,id);
        return sysRole;
    }