Spring Boot 整合 MyBatis-Plus 实现多数据源配置

三、application.yml 多数据源配置

spring:
  datasource:
    master:
      url: jdbc:mysql://localhost:3306/master_db?useSSL=false&serverTimezone=UTC
      username: root
      password: root
      driver-class-name: com.mysql.cj.jdbc.Driver
    slave:
      url: jdbc:mysql://localhost:3306/slave_db?useSSL=false&serverTimezone=UTC
      username: root
      password: root
      driver-class-name: com.mysql.cj.jdbc.Driver

四、核心代码

定义数据源枚举

public enum DataSourceType {
    MASTER, SLAVE
}

数据源上下文管理（ThreadLocal）

public class DataSourceContextHolder {
    private static final ThreadLocal<DataSourceType> CONTEXT = new ThreadLocal<>();

    public static void set(DataSourceType type) {
        CONTEXT.set(type);
    }

    public static DataSourceType get() {
        return CONTEXT.get();
    }

    public static void clear() {
        CONTEXT.remove();
    }
}

动态数据源路由实现

继承 AbstractRoutingDataSource

import org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource;

public class DynamicDataSource extends AbstractRoutingDataSource {
    @Override
    protected Object determineCurrentLookupKey() {
        // 核心：返回当前线程的数据源标识
        return DataSourceContextHolder.get();
    }
}

数据源配置类

import com.zaxxer.hikari.HikariDataSource;
import org.springframework.boot.context.properties.ConfigurationProperties;
import org.springframework.context.annotation.*;
import org.springframework.jdbc.datasource.DataSourceTransactionManager;
import javax.sql.DataSource;
import java.util.HashMap;
import java.util.Map;

@Configuration
public class DataSourceConfig {
    // 主数据源
    @Bean
    @ConfigurationProperties("spring.datasource.master")
    public DataSource masterDataSource() {
        return new HikariDataSource();
    }

    // 从数据源
    @Bean
    @ConfigurationProperties("spring.datasource.slave")
    public DataSource slaveDataSource() {
        return new HikariDataSource();
    }

    // 动态数据源
    @Primary
    @Bean
    public DataSource dynamicDataSource() {
        DynamicDataSource dynamicDataSource = new DynamicDataSource();
        Map<Object, Object> targetDataSources = new HashMap<>();
        targetDataSources.put(DataSourceType.MASTER, masterDataSource());
        targetDataSources.put(DataSourceType.SLAVE, slaveDataSource());
        dynamicDataSource.setTargetDataSources(targetDataSources);
        // 默认数据源
        dynamicDataSource.setDefaultTargetDataSource(masterDataSource());
        return dynamicDataSource;
    }

    // 事务管理器
    @Bean
    public DataSourceTransactionManager transactionManager(DataSource dynamicDataSource) {
        return new DataSourceTransactionManager(dynamicDataSource);
    }
}

MyBatis-Plus 配置

import org.mybatis.spring.annotation.MapperScan;
import org.springframework.context.annotation.Configuration;

@Configuration
@MapperScan("com.example.mapper")
public class MyBatisPlusConfig {}

五、测试示例

实体类

@Data
@TableName("user")
public class User {
    private Long id;
    private String name;
}

Mapper

public interface UserMapper extends BaseMapper<User> {}

Service 示例

@Service
public class UserService {
    @Autowired
    private UserMapper userMapper;

    // 写操作（主库）
    public void saveUser(User user) {
        DataSourceContextHolder.set(DataSourceType.MASTER);
        userMapper.insert(user);
        DataSourceContextHolder.clear();
    }

    // 读操作（从库）
    public List<User> listUser() {
        DataSourceContextHolder.set(DataSourceType.SLAVE);
        List<User> list = userMapper.selectList(null);
        DataSourceContextHolder.clear();
        return list;
    }
}

Controller 测试

@RestController
@RequestMapping("/user")
public class UserController {
    @Autowired
    private UserService userService;

    @GetMapping("/save")
    public String save() {
        User user = new User();
        user.setName("test");
        userService.saveUser(user);
        return "ok";
    }

    @GetMapping("/list")
    public List<User> list() {
        return userService.listUser();
    }
}

六、常见坑 & 解决方案

坑 1：事务失效

问题原因

Spring 事务基于 AOP 代理。

如果：

this.saveUser();

会导致事务失效。

解决方案

• 不要 self-invocation
• 或通过 Spring 容器获取代理对象

坑 2：事务内切换数据源无效

AbstractRoutingDataSource 在获取连接时就确定数据源。

如果：

@Transactional
public void test() {
    setSlave();
    select();
    setMaster();
    insert();
}

⚠️ 会全部使用同一个数据源。

解决方案

• 不要在同一个事务中切换数据源
• 读写分离通常不加事务

坑 3：循环依赖

如果配置错误：

dynamicDataSource() 里又注入自己 

会导致循环依赖。

正确写法

让 Spring 自动注入参数：

@Bean
public DataSourceTransactionManager transactionManager(DataSource dataSource)

坑 4：ThreadLocal 未清理

如果忘记：

DataSourceContextHolder.clear();

可能导致线程复用时污染数据源。

建议使用 AOP 自动切换（生产推荐）。

七、生产级优化建议

使用注解 + AOP 自动切换

定义：

@Target(ElementType.METHOD)
@Retention(RetentionPolicy.RUNTIME)
public @interface DS {
    DataSourceType value();
}

通过切面自动设置数据源，比手动 set 更优雅。

连接池优化

spring:
  datasource:
    master:
      hikari:
        maximum-pool-size: 20
        minimum-idle: 5
        idle-timeout: 600000

避免连接池爆掉。

八、总结

实现多数据源的核心原理只有一句话：

利用 AbstractRoutingDataSource + ThreadLocal 实现动态路由。

完整流程：

1. 定义数据源
2. ThreadLocal 保存当前数据源
3. 动态路由
4. MyBatis-Plus 使用统一 DataSource

最终效果

• 写操作走主库
• 读操作走从库
• 可扩展多个业务库
• 可生产使用