springboot+mybatis实现动态切换数据源

一个小白开发

已于 2022-08-18 11:10:08 修改

阅读量598

点赞数

分类专栏： springboot 文章标签： mybatis spring boot java

于 2022-08-18 11:00:51 首次发布

本文链接：https://blog.csdn.net/mxasd/article/details/126401427

版权

springboot 专栏收录该内容

9 篇文章 0 订阅

订阅专栏

本文介绍了在Spring Boot项目中配置多个数据源的方法，包括直接连接数据库和通过JNDI查找DataSource。然后，创建了一个DynamicDataSource类作为默认数据源，并实现了动态数据源切换。在Controller层，可以根据需求指定使用哪个数据源。动态数据源的切换是通过ThreadLocal变量实现的。

摘要由CSDN通过智能技术生成

由于目前项目中需要用到项目中的两个库，研究了一下发现一个比较轻松的写法。

首先配置多个数据源并且交给bean管理写在 Config中

  @Bean(name = "db1")
    public DataSource druidDataSource() {
        DruidDataSource datasource = null;
        //直连数据库
        if ("0".equals(jndiflag)) {
            datasource = new DruidDataSource();
            datasource.setUrl(dbUrl);
            datasource.setUsername(username);
            datasource.setPassword(password);
            datasource.setDriverClassName(driverClassName);
            datasource.setInitialSize(initialSize);
            datasource.setMinIdle(minIdle);
            datasource.setMaxActive(maxActive);
            datasource.setMaxWait(maxWait);
            datasource.setTimeBetweenEvictionRunsMillis(timeBetweenEvictionRunsMillis);
            datasource.setMinEvictableIdleTimeMillis(minEvictableIdleTimeMillis);
            datasource.setValidationQuery(validationQuery);
            datasource.setTestWhileIdle(testWhileIdle);
            datasource.setTestOnBorrow(testOnBorrow);
            datasource.setTestOnReturn(testOnReturn);
            try {
                datasource.setFilters(filters);
            } catch (SQLException e) {
                log.error("druid configuration initialization filter", e);
            }
        } else {
            try {
                Context ctx = new InitialContext();
                datasource = (DruidDataSource) ctx.lookup("");
            } catch (NamingException e) {
                log.error("通过JNDI名称找到DataSource 异常", e);
            }
        }
        return datasource;
    }

    @Bean(name = "db2")
    public DataSource druidDataSourceTwo() {
        DruidDataSource datasource = null;
        //直连数据库
        if ("0".equals(jndiflag)) {
            datasource = new DruidDataSource();
            datasource.setUrl(dbUrlTwo);
            datasource.setUsername(usernameTwo);
            datasource.setPassword(passwordTwo);
            datasource.setDriverClassName(driverClassNameTwo);
            datasource.setInitialSize(initialSize);
            datasource.setMinIdle(minIdle);
            datasource.setMaxActive(maxActive);
            datasource.setMaxWait(maxWait);
            datasource.setTimeBetweenEvictionRunsMillis(timeBetweenEvictionRunsMillis);
            datasource.setMinEvictableIdleTimeMillis(minEvictableIdleTimeMillis);
            datasource.setValidationQuery(validationQuery);
            datasource.setTestWhileIdle(testWhileIdle);
            datasource.setTestOnBorrow(testOnBorrow);
            datasource.setTestOnReturn(testOnReturn);
            try {
                datasource.setFilters(filters);
            } catch (SQLException e) {
                log.error("druid configuration initialization filter", e);
            }
        } else {
            try {
                Context ctx = new InitialContext();
                datasource = (DruidDataSource) ctx.lookup("");
            } catch (NamingException e) {
                log.error("通过JNDI名称找到DataSource 异常", e);
            }
        }
        return datasource;
    }

目前配置了两个数据源后，我们也应该在Config 中指定一个默认数据源

@Bean(name = "dynamicDataSource")
    @Primary
    public DataSource dynamicDataSource(@Qualifier("db1") DataSource db1, @Qualifier("db2") DataSource db2) {
        DynamicDataSource dynamicDataSource = new DynamicDataSource();
        // 默认数据源
        dynamicDataSource.setDefaultTargetDataSource(db1);
        // 配置多数据源
        Map<Object, Object> dsMap = new HashMap<>();
        dsMap.put("db1", db1);
        dsMap.put("db2", db2);

        dynamicDataSource.setTargetDataSources(dsMap);
        return dynamicDataSource;
    }

以下是DynamicDataSource 类

package com.guobaojingrong.salesupport.config;/*
 * 消息描述
 * @Description: $
 * @return: $
 * @Author: 李
 * @Date: $ $
 */

import lombok.extern.slf4j.Slf4j;
import org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource;

@Slf4j
public class DynamicDataSource extends AbstractRoutingDataSource {
    @Override
    protected Object determineCurrentLookupKey() {
        log.info("动态获取数据源——{}",DataSourceUtil.getDB());
        return DataSourceUtil.getDB();
    }
}

DataSourceUtil类

package com.guobaojingrong.salesupport.config;/*
 * 消息描述
 * @Description: $
 * @return: $
 * @Author: 李
 * @Date: $ $
 */

public class DataSourceUtil {

    private static final ThreadLocal<String> contextHolder = new ThreadLocal();

    /**
     * 设置数据源名
     * @param dbType
     */
    public static void setDB(String dbType) {
        contextHolder.set(dbType);
    }

    /**
     * 获取数据源名
     * @return
     */
    public static String getDB() {
        return contextHolder.get();
    }

    /**
     * 清除数据源名
     */
    public static void clearDB() {
        contextHolder.remove();
    }
}

在Controller 指定对应的库