在复杂的企业系统中，经常会遇到以下场景：

一个系统连接多个业务数据库（如订单库、用户库、商品库）
实现主从数据库的读写分离
多租户 SaaS 系统中，每个租户一个库

如果继续使用默认的 @Primary + @Qualifier 方式手动注入，会导致代码复杂，耦合严重。为此，构建一个“ 可自动识别上下文、支持注解方式的多数据源切换机制”就显得尤为重要。

一、核心思路概览

使用 AbstractRoutingDataSource 动态决定当前数据库连接
定义 ThreadLocal 作为数据源上下文
通过自定义注解 @DS("order") 动态切换数据源
AOP 拦截注解并切换上下文

二、数据库准备

准备两个数据库（可用同一 MySQL 服务下的不同库）：

db_user：保存用户信息
db_order：保存订单数据

初始化建表 SQL 示例：

sql复制编辑CREATE DATABASE db_user;CREATE DATABASE db_order;-- 用户表CREATE TABLE user_info (
  id BIGINT PRIMARY KEY AUTO_INCREMENT,
  username VARCHAR(100)
);-- 订单表CREATE TABLE order_info (
  id BIGINT PRIMARY KEY AUTO_INCREMENT,
  user_id BIGINT,
  product VARCHAR(100)
);

三、POM 依赖引入

xml复制编辑      org.springframework.boot    spring-boot-starter-jdbc        mysql    mysql-connector-j        org.springframework.boot    spring-boot-starter-aop

四、配置多数据源

4.1 application.yml 配置

yaml复制编辑spring:
  datasource:
    user:
      url: jdbc:mysql://localhost:3306/db_user
      username: root
      password: 123456
      driver-class-name: com.mysql.cj.jdbc.Driver
    order:
      url: jdbc:mysql://localhost:3306/db_order
      username: root
      password: 123456
      driver-class-name: com.mysql.cj.jdbc.Driver

4.2 创建枚举标识

java复制编辑public enum DataSourceType {
    USER, ORDER
}

4.3 数据源上下文工具类

java复制编辑public class DynamicDataSourceContextHolder {    private static final ThreadLocal contextHolder = new ThreadLocal<>();    public static void set(String dsKey) {
        contextHolder.set(dsKey);
    }    public static String get() {        return contextHolder.get();
    }    public static void clear() {
        contextHolder.remove();
    }
}

五、实现动态数据源路由

java复制编辑public class DynamicRoutingDataSource extends AbstractRoutingDataSource {    @Override
    protected Object determineCurrentLookupKey() {        return DynamicDataSourceContextHolder.get();
    }
}

六、数据源配置类

java复制编辑@Configurationpublic class DataSourceConfig {    @Bean
    @Primary
    public DataSource dataSource(@Qualifier("userDataSource") DataSource userDs,                                 @Qualifier("orderDataSource") DataSource orderDs) {
        Map map = new HashMap<>();
        map.put("USER", userDs);
        map.put("ORDER", orderDs);        DynamicRoutingDataSource ds = new DynamicRoutingDataSource();
        ds.setTargetDataSources(map);
        ds.setDefaultTargetDataSource(userDs);        return ds;
    }    @Bean("userDataSource")
    @ConfigurationProperties(prefix = "spring.datasource.user")
    public DataSource userDataSource() {        return DataSourceBuilder.create().build();
    }    @Bean("orderDataSource")
    @ConfigurationProperties(prefix = "spring.datasource.order")
    public DataSource orderDataSource() {        return DataSourceBuilder.create().build();
    }
}

七、自定义注解实现切换

7.1 定义注解 `@DS`

java复制编辑@Target({ElementType.METHOD, ElementType.TYPE})@Retention(RetentionPolicy.RUNTIME)public @interface DS {
    String value();
}

7.2 AOP 实现注解拦截

java复制编辑@Aspect@Componentpublic class DynamicDataSourceAspect {    @Pointcut("@annotation(ds)")
    public void dsPointcut(DS ds) {}    @Around("dsPointcut(ds)")
    public Object around(ProceedingJoinPoint point, DS ds) throws Throwable {        try {
            DynamicDataSourceContextHolder.set(ds.value().toUpperCase());            return point.proceed();
        } finally {
            DynamicDataSourceContextHolder.clear();
        }
    }
}

八、DAO 与 Service 层调用示例

8.1 Repository 示例（使用 JdbcTemplate）

java复制编辑@Repositorypublic class UserDao {    @Autowired
    private JdbcTemplate jdbcTemplate;    public List> findAllUsers() {        return jdbcTemplate.queryForList("SELECT * FROM user_info");
    }
}

java复制编辑@Repositorypublic class OrderDao {    @Autowired
    private JdbcTemplate jdbcTemplate;    public List> findAllOrders() {        return jdbcTemplate.queryForList("SELECT * FROM order_info");
    }
}

8.2 Service 示例

java复制编辑@Servicepublic class DataService {    @Autowired private UserDao userDao;    @Autowired private OrderDao orderDao;    @DS("user")
    public List> queryUsers() {        return userDao.findAllUsers();
    }    @DS("order")
    public List> queryOrders() {        return orderDao.findAllOrders();
    }
}

九、测试 Controller

java复制编辑@RestController@RequestMapping("/api")public class TestController {    @Autowired
    private DataService dataService;    @GetMapping("/users")
    public Result getUsers() {        return Result.success(dataService.queryUsers());
    }    @GetMapping("/orders")
    public Result getOrders() {        return Result.success(dataService.queryOrders());
    }
}

十、扩展建议

场景	解决方案
多租户动态数据库连接	结合租户 ID + 连接池缓存机制，实现动态构建 DataSource
多模块共享数据源配置	使用 spring.factories 统一注册配置
Mapper 层集成	MyBatis 支持 SQLSessionFactory 动态注册
读写分离场景支持	配合 AOP 或 SQL 注解区分读写策略
使用 Spring Cloud + Nacos	多数据源配置可从 Nacos 动态拉取

十一、总结

本文完整实现了：

支持多数据源的配置与初始化
自定义注解 + AOP 实现动态切换
使用 ThreadLocal 管理上下文数据源
Service 层零侵入式调用多个数据库
可扩展到多租户、读写分离等高级场景

这是中大型系统中常见但复杂度较高的一种实践，通过合理抽象可以极大提升系统的灵活性与扩展性。

Python 实战：构建支持模块热插拔的本地 WebSocket 实时日志面板