SpringBoot中的双数据源切换

╰半橙微兮° 2022-04-25 06:44 463阅读 0赞

在我们的项目中涉及到Oracle，Mysql数据库切换；

数据放在了Oracle上，而配置和权限管理放在了Mysql上；所以经常要在两者之间切换。

数据源切换总共有三种方法：

1  在任何地方切换。

例如我们的项目：DBContext.changeDB(DBName.IMS.getKey())  //切换Oracle

DBContext.changeDB(DBName.DBVS.getKey())                          //切换Mysql

2  在方法上加注解。

例如：@DBAspect("dbvs")  //切换Mysql

@DBAspect("ims")  //切换Oracle

3 在dao层之间分开，一个mapper文件专门去访问Mysql,一个专门访问Oracle

接下来我们主要将第一种：在任何地方切换(我们项目代码无法拷贝，纯手敲)

类DBContext

public class DBContext \{

private static ThreadLocal<String> dbName = new ThreadLocal<>();

public static String getTargetDataSource()\{

return dbName.get();

public static void changeDB(String name)\{

dbName.set(name);

//清除数据源

public static void clearDB()\{

dbName.remove();

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl80MTEyNjg0Mg_size_16_color_FFFFFF_t_70][]

数据源枚举类DBName

public enum DBName \{

DBVS("dbvs"),

IMS("ims");

private String key;

DBName(String key)\{

this.key = key;

public String getKey()\{

return key;

public void setKey(String key)\{

this.key  = key;

DynamicDataSource类继承AbstractRoutingDataSource

public class DynamicDataSource extends AbstractRoutingDataSource\{

protected Object determineCurrentLookupKey()\{

return DBContext.getRargetDataSource();

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl80MTEyNjg0Mg_size_16_color_FFFFFF_t_70 1][]配置类ContextConfig

@SpringBootConfiguration

@ComponentScan(basePackages = \{"com.dbvs.\*"\})

@MapperScan(basePackages = \{"com.dbvs.\*.dao"\})

@EnableTransactionManagement

@EnableAsync

Public class ContextConfig \{

@ConfigurationProperties(prefix = "spring.datasource.druid.dbvs")

@Bean("dbvs.mysql")

@Qualifier("dbvs.mysql")'

public DataSource mysqlDataSource()  throws Exception\{

DruidDataSource dataSource = DruidDataSourceBuilder.create().build();

dataSource.setUsername(userName);//userName，password 配置在核心配置文件中

dataSource.setPassword(password);

return dataSource;

@ConfigurationProperties(prefix = "spring.datasource.druid.ims")

@Bean("ims.oracle")

@Qualifier("ims.oracle")'

public DataSource oracleDataSource()  throws Exception\{

DruidDataSource dataSource = DruidDataSourceBuilder.create().build();

dataSource.setUsername(userName);//userName，password 配置在核心配置文件中

dataSource.setPassword(password);

return dataSource;

@Bean("dynamicDataSource")

@Qualifier("dynamicDataSource")

@Primary

public DataSource dynamicDataSource() throws Exception\{

DynamicDataSource dataSource = new DynamicDataSource();

HashMap<Object,Object> targetDataSource = new HashMap<>();

targetDataSource.put(DBName.DBVS.getKey(),mysqlDataSource());

targetDataSource.put(DBName.IMS.getKey(),oracleDataSource());

dataSource.setTargetDataSources(targetDataSource);

dataSource.setDefaultTargetDataSource(targetDataSource.get(DBName.DBVS.getKey()));

return dataSource;

dataSource的拦截器

public class DataSourceInterceptor implements HandlerInterceptor\{

public boolean preHandle(HttpServletRequest request,HttpServletResponse response,Object handler) throws Exception\{

DBContext.clearDB();

DBContext.changeDB(DBName.DBVS.getKey());

return HandlerInterceptor.super.preHandle(request,response,handler);

总拦截器类

@SpringBootConfiguration

public class InterceptorConfig implements WebMvConfigurer \{

//拦截器配置，与业务相关，有着强烈的前后依赖顺序

//其他拦截器在datasourceInterceptor之前添加

public void addInterceptors(InterceptorRegistry registry)\{

WebMvcConfigurer.super.addInterceptors(registry);

// registry.addInterceptor(其他拦截器，例如：日志，权限);

registry.addInterceptor(datasourceInterceptor);

@Bean("datasourceInterceptor")

public DatasourceInterceptor datasourceInterceptor()\{

DatasourceInterceptor  interceptor = new DatasourceInterceptor ();

return interceptor;

配置文件

\#-----------datasource-------mysql------------

spring.datasource.druid.dbvs.url=

spring.datasource.druid.dbvs.driver-class-name=

spring.datasource.druid.dbvs.username=

spring.datasource.druid.dbvs.password=

......

\#-----------------datasource-------oracle-----

spring.datasource.druid.dbvs.url=

spring.datasource.druid.ims.driver-class-name=

spring.datasource.druid.ims.username=

spring.datasource.druid.ims.password=

........

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl80MTEyNjg0Mg_size_16_color_FFFFFF_t_70]: /images/20220126/9ea4e7c4635d4228bcc71455e1c0e509.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl80MTEyNjg0Mg_size_16_color_FFFFFF_t_70 1]: /images/20220126/0a1ff7a0f2b34628af0f923209673f28.png