当前位置: 首页 > news >正文

Spring Boot 多数据源切换:AbstractRoutingDataSource

在实际项目中,我们常常会用到多个数据库,比如一个主数据库(master)专门用来写入一个从数据库(slave)专门用来读取。这种场景非常常见,那么问题来了:

我该如何在项目中动态地切换主从数据库?
Spring 是怎么实现“在运行时自动选择数据库”的?

什么是动态数据源?

动态数据源就是在运行时,Spring 动态决定使用哪个数据库连接。

也就是说:

  • 有多个数据库(比如 master、slave)

  • 程序运行过程中自动切换用哪个数据库

  • 你不用手动写 if...else 来判断

第一步:我们需要哪些数据源?

我们以最常见的主从结构举例:

spring:
datasource:
druid:
master:
url: jdbc:mysql://localhost:3306/db_master
username: root
password: 123456
slave:
enabled: true
url: jdbc:mysql://localhost:3306/db_slave
username: root
password: 123456

我们准备两个数据源:

  • 主数据源 master

  • 从数据源 slave(注意加了一个 enabled=true)

第二步:配置两个数据源 Bean

在 Spring Boot 中,我们通过 @Bean 创建连接池:

@Bean
@ConfigurationProperties("spring.datasource.druid.master")
public DataSource masterDataSource(DruidProperties druidProperties) {DruidDataSource dataSource = DruidDataSourceBuilder.create().build();return druidProperties.dataSource(dataSource);
}@Bean
@ConfigurationProperties("spring.datasource.druid.slave")
@ConditionalOnProperty(prefix = "spring.datasource.druid.slave", name = "enabled", havingValue = "true")
public DataSource slaveDataSource(DruidProperties druidProperties) {DruidDataSource dataSource = DruidDataSourceBuilder.create().build();return druidProperties.dataSource(dataSource);
}

这样我们就有了两个数据库连接池。

第三步:创建一个“数据源路由器”

我们不能在代码中每次都自己写 if 判断用哪个数据源,这太麻烦了。

Spring 给我们提供了抽象类AbstractRoutingDataSource,它是一个“动态路由器”,且实现了

DataSource接口
public abstract class AbstractRoutingDataSource

它是怎么工作的?

假设你调用了:

dataSource.getConnection();

如果这个 dataSource 是继承了 AbstractRoutingDataSource 的类,它就会:

  1. 自动调用 determineCurrentLookupKey()
  2. 返回一个 key(比如 "MASTER" 或 "SLAVE")
  3. 根据 key 从 map 中找到对应的数据源
  4. 最终返回真正的数据库连接

源码也是非常通俗易懂:

可以看到在getConnection方法中,它会先去获取一下DataSource。其中就会先去获取key,如果为空就是默认的DataSource。

第四步:定义我们的 DynamicDataSource 类

public class DynamicDataSource extends AbstractRoutingDataSource {@Overrideprotected Object determineCurrentLookupKey() {return DataSourceContextHolder.getDataSourceType(); // 返回 master 或 slave}
}

它的作用就是:

每次执行数据库操作前,去上下文 ThreadLocal 中取出当前线程应该使用哪个数据源。

 

第五步:设置 dynamicDataSource Bean

@Bean(name = "dynamicDataSource")
@Primary
public DynamicDataSource dataSource(DataSource masterDataSource) {Map<Object, Object> targetDataSources = new HashMap<>();targetDataSources.put("MASTER", masterDataSource);setDataSource(targetDataSources, "SLAVE", "slaveDataSource");return new DynamicDataSource(masterDataSource, targetDataSources);
}

注意:

  • @Primary 让 Spring 默认使用这个数据源

  • setDataSource() 会从 Spring 容器中动态拿到 slaveDataSource

第六步:用 ThreadLocal 保存当前线程的数据源类型

public class DataSourceContextHolder {private static final ThreadLocal<String> contextHolder = new ThreadLocal<>();public static void set(String key) {contextHolder.set(key);}public static String get() {return contextHolder.get();}public static void clear() {contextHolder.remove();}
}

这样每个线程就可以独立地记录当前要使用哪个数据源了!

第七步:通过 AOP 自动切换数据源

你可以加一个注解,比如:

@DS("SLAVE")
public List<User> getUserList() {...
}

AOP 在方法执行前会调用:

DataSourceContextHolder.set("SLAVE");

这样你的方法执行时,就自动使用了从数据库!

整体流程图

最后

多数据源非常适合读写分离、分库分表、租户隔离场景,还是很值得研究一下的。

http://www.dtcms.com/a/269819.html

相关文章:

  • Elasticsearch Kibana 使用 原理
  • 用基础模型构建应用(第七章)AI Engineering: Building Applications with Foundation Models学习笔记
  • Linux基础篇、第五章_01利用 Cobbler 实现 CentOS 7 与 Rocky 9.5 自动化安装全攻略
  • 记录一次在 centos 虚拟机 中 安装 Java环境
  • windows内核研究(系统调用 1)
  • 从传统项目管理到敏捷DevOps:如何转向使用DevOps看板工具进行工作流管理
  • 谁主沉浮:人工智能对未来信息技术发展路径的影响研究
  • 优化提示词提升VLLM准确率
  • K8s——配置管理(1)
  • 构建高效分布式系统:bRPC组合Channels与HTTP/H2访问指南
  • 从单体到微服务:Spring Cloud 开篇与微服务设计
  • 微前端框架对比
  • 无缝矩阵支持音频分离带画面分割功能的全面解析
  • ​AI赋能的自动驾驶革命:从安全架构到世界模型的系统性突破
  • 【操作系统】磁盘调度
  • hmall学习
  • 2025年模型与机器学习国际会议 (ICMML 2025)
  • BM9 删除链表的倒数第n个节点
  • 计算机网络4层架构怎么理解,分别把协议和对应的层用一些生活的例子形象说明一下
  • MyBatis完全学习指南
  • 算法题练习3-判定链表是否是回文串
  • 【踩坑随笔】PlatformIO导入Arduino项目出现的问题
  • STM32第十八天 ESP8266-01S和电脑实现串口通信
  • HTTP/3.x协议详解:基于QUIC的下一代Web传输协议
  • 小红书APP品牌升级,启用新品牌口号“你的生活兴趣社区”
  • 2025年社会学与安全科学国际会议 (ICSSS 2025)
  • 【AI News | 20250707】每日AI进展
  • C++ dijkstra 最短路径算法
  • c语言学习_函数递归
  • 数学建模:非线性规划:凸规划问题