Springboot 实现前台动态配置数据源（修改数据源之后自动重启）

1.将 db.properties 存放在classpath路径;

driverClassName=com.mysql.jdbc.Driver
url=jdbc:mysql://localhost:3306/ceshi?useUnicode=true&characterEncoding=utf8
username=root
password=root

2.使用Bean的方式配置 dataSource

package com.shulipeng.config;
 
import com.alibaba.druid.pool.DruidDataSource;
import com.shulipeng.AutocodeApplication;
import org.apache.commons.configuration.ConfigurationException;
import org.apache.commons.configuration.PropertiesConfiguration;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.context.annotation.Primary;
 
import javax.sql.DataSource;
import java.sql.SQLException;
 
/**
 * @author shulipeng
 * @Description:
 * @date 2018/4/2
 */
@Configuration
public class DataSourceConfig {
 
    private static Logger logger = LoggerFactory.getLogger(DataSourceConfig.class);
 
    //数据库类型
    private String dbType;
 
    @Bean(name = "dataSource")
    public DruidDataSource dataSource() {
        DruidDataSource dataSource = new DruidDataSource();
        try {
            PropertiesConfiguration properties = new PropertiesConfiguration("db.properties");
            String driverClassName = properties.getString("driverClassName");
            dataSource.setDriverClassName(driverClassName);
            dataSource.setUrl(properties.getString("url"));
            dataSource.setUsername(properties.getString("username"));
            dataSource.setPassword(properties.getString("password"));
 
            //判断数据库类型
            if(driverClassName.contains(Constant.DB_TYPE_MYSQL)){
                this.setDbType(Constant.DB_TYPE_MYSQL);
            }else if(driverClassName.contains(Constant.DB_TYPE_ORACLE)){
                this.setDbType(Constant.DB_TYPE_ORACLE);
            }
 
        } catch (ConfigurationException e) {
            logger.error("获取数据库配置文件失败" + e);
        }
 
        return dataSource;
    }
 
    /**
     * 动态修改数据库链接
     */
    public void changeDataSource () {
        Thread restartThread = new Thread(new Runnable() {
            @Override
            public void run() {
                try {
                    Thread.sleep(1000);
                    AutocodeApplication.restart();
                } catch (InterruptedException ignored) {
                }
            }
        });
        restartThread.setDaemon(false);
        restartThread.start();
    }
 
    public String getDbType() {
        return dbType;
    }
 
    public void setDbType(String dbType){
        this.dbType = dbType;
    }
}

3.定义一个前台改变数据库配置文件并且重启的方法

/**
   * 动态修改数据库地址，并且重新启动应用
   * @param dbMap
   * @return
   */
  @PostMapping("/db/update")
  @ResponseBody
  R changeDB(@RequestParam Map<String,Object> dbMap){
      try {
          PropertiesConfiguration conf = new PropertiesConfiguration("db.properties");
          conf.setProperty("driverClassName",dbMap.get("driverClassName"));
          conf.setProperty("url",dbMap.get("url"));
          conf.setProperty("username",dbMap.get("username"));
          conf.setProperty("password",dbMap.get("password"));
          conf.save();
          dataSourceConfig.changeDataSource();
      } catch (ConfigurationException e) {
          logger.error("获取数据库配置文件失败：" + e);
          return R.error("保存配置文件时出错" + e);
      }
      return  R.ok();
  }

4.修改Application.java

package com.shulipeng;
 
import org.mybatis.spring.annotation.MapperScan;
import org.springframework.boot.SpringApplication;
import org.springframework.boot.autoconfigure.SpringBootApplication;
import org.springframework.boot.web.servlet.ServletComponentScan;
import org.springframework.context.ConfigurableApplicationContext;
 
@ServletComponentScan
@MapperScan("com.shulipeng.dao")
@SpringBootApplication
public class AutocodeApplication {
 
    private static String[] args;
    private static ConfigurableApplicationContext context;
 
    public static void main(String[] args) {
        AutocodeApplication.args = args;
        AutocodeApplication.context = SpringApplication.run(AutocodeApplication.class, args);
    }
 
    public static void restart() {
        context.close();
        AutocodeApplication.context = SpringApplication.run(AutocodeApplication.class, args);
 
    }
}