springCloud在不重启服务的情况下切换数据源 ,不使用git钩子的方式refresh配置文件

需求:只启动一套代码,要求在用户登录时选择不同的数据库...

 

在网上看了很多参考,大多都是使用git push配置文件,触发刷新客服端配置的方法,,

但是这个项目要在生产上使用,肯定不能有程序员更新git上的配置文件这种操作.

因此想到了一个类似的方法:

1.将所需要切换的数据库配置全部写好在配置中心的配置文件中

2.在客户端增加一个变量,用于判断前端传递的参数 用户选用那个数据库

3.使用此参数修改数据库配置

 

配置中心的数据库配置如下:

  datasource:
    url: jdbc:oracle:thin:@xx
    url2: jdbc:oracle:thin:@xx
    username: xx
    password: xx
    driver-class-name: oracle.jdbc.OracleDriver

 

客户端数源配置如下:

数据

import com.alibaba.druid.pool.DruidDataSource;
import lombok.extern.slf4j.Slf4j;
import org.apache.ibatis.session.SqlSessionFactory;
import org.mybatis.spring.SqlSessionFactoryBean;
import org.mybatis.spring.SqlSessionTemplate;
import org.mybatis.spring.annotation.MapperScan;
import org.springframework.beans.factory.annotation.Qualifier;
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;
import org.springframework.context.annotation.Primary;
import org.springframework.core.io.support.PathMatchingResourcePatternResolver;
import org.springframework.jdbc.core.JdbcTemplate;
import org.springframework.jdbc.datasource.DataSourceTransactionManager;

import javax.sql.DataSource;

import static com.yss.db.rest.DataSourceConfigController.dbChoose;


@Slf4j
@Configuration
@RefreshScope
@MapperScan(basePackages = "xxx.mapper", sqlSessionFactoryRef = "sqlSessionFactory", sqlSessionTemplateRef = "sqlSessionTemplate")
public class DataSourceConfig {
    @Value("${spring.datasource.url}")
    private String dbUrl;

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

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

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

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

    @Value("${spring.datasource.druid.initial-size}")
    private int initialSize;

    @Value("${spring.datasource.druid.min-idle}")
    private int minIdle;

    @Value("${spring.datasource.druid.max-active}")
    private int maxActive;

    @Value("${spring.datasource.druid.test-on-borrow}")
    private boolean testOnBorrow;


    @Bean(name = "dataSource")
    @RefreshScope
    public DataSource dataSource() {
        DruidDataSource datasource = new DruidDataSource();

        if(dbChoose==0){
            datasource.setUrl(this.dbUrl);
            log.info("加载数据库配置 dbUrl {}",dbUrl);
        }else {
            datasource.setUrl(this.dbUrl2);
            log.info("加载数据库配置 dbUrl {}",dbUrl2);
        }
        datasource.setUsername(username);
        datasource.setPassword(password);
        datasource.setDriverClassName(driverClassName);

        datasource.setInitialSize(initialSize);
        datasource.setMinIdle(minIdle);
        datasource.setMaxActive(maxActive);
        datasource.setTestOnBorrow(testOnBorrow);

        return datasource;
    }

    @Bean(name = "transactionManager")
    @Primary
    public DataSourceTransactionManager setTransactionManager(@Qualifier("dataSource") DataSource dataSource) {
        return new DataSourceTransactionManager(dataSource);
    }

    @Bean(name = "sqlSessionFactory")
    @Primary
    public SqlSessionFactory setSqlSessionFactory(@Qualifier("dataSource") DataSource dataSource) throws Exception {
        SqlSessionFactoryBean bean = new SqlSessionFactoryBean();
        bean.setDataSource(dataSource);
        bean.setMapperLocations(new PathMatchingResourcePatternResolver().getResources("classpath:mapper/*.xml"));
        return bean.getObject();
    }

    @Bean(name = "sqlSessionTemplate")
    @Primary
    public SqlSessionTemplate setSqlSessionTemplate(@Qualifier("sqlSessionFactory") SqlSessionFactory sqlSessionFactory) throws Exception {
        return new SqlSessionTemplate(sqlSessionFactory);
    }

    @Bean("jdbcTemplate")
    @Primary
    public JdbcTemplate jdbcTemplate() throws Exception {
        return new JdbcTemplate(dataSource());
    }

}

 

接受前端发送的请求,切换数据库

package com.yss.db.rest;

import com.yss.base.common.exception.BaseException;
import lombok.extern.slf4j.Slf4j;
import org.springframework.http.HttpEntity;
import org.springframework.http.HttpHeaders;
import org.springframework.http.MediaType;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.PathVariable;
import org.springframework.web.bind.annotation.RestController;
import org.springframework.web.client.RestTemplate;

/**
 * 手动刷新数据库配置
 *
 * @author fuzhi 2019-04-24
 */
@RestController
@Slf4j
public class DataSourceConfigController {
    /**
     * 用户选择数据库 默认为0
     * dbChoose 0  1
     */
    public static int dbChoose = 0;


    @GetMapping(value = "/test/{dbChoose}")
    public void chooseDB(@PathVariable(value = "dbChoose") int dbChooses) {
        if (dbChooses == 0) {
            if (dbChoose != 0) {
                dbChoose = 0;
                sendRefresh();
            }
        } else {
            if (dbChoose != 1) {
                dbChoose = 1;
                sendRefresh();
            }
        }
    }

//restTemplate 发送post请求,告诉客户端刷新配置
    private void sendRefresh() {
        RestTemplate restTemplate = new RestTemplate();
        HttpHeaders headers = new HttpHeaders();
        MediaType type = MediaType.parseMediaType("application/json; charset=UTF-8");
        try {
            headers.setContentType(type);
            HttpEntity<String> requestEntity = new HttpEntity<>(null, headers);
            restTemplate.postForLocation("http://localhost:8083/refresh", requestEntity);
        } catch (Exception e) {
            log.info("切换数据库成功失败" + e.getMessage());
            throw new BaseException("切换数据库成功失败" + e.getMessage());
        }
        log.info("开始切换数据库 数据库标记-dbChoose: {}", dbChoose);
    }
}

有其他好的方法么,,求教