动态切换数据源原理详解，以及Spring Boot + MyBatis-Plus 在实际开发中的快速切换方法

在Spring Boot + MyBatis-Plus 中实现动态数据源切换，可以使用AbstractRoutingDataSource和AOP。

以下是核心步骤和示例代码：

1. 创建动态数据源（DynamicDataSource）类，继承AbstractRoutingDataSource，并重写determineCurrentLookupKey方法。

public class DynamicDataSource extends AbstractRoutingDataSource {
    @Override
    protected Object determineCurrentLookupKey() {
        return DataSourceContextHolder.getDataSourceType();
    }
}

2. 使用ThreadLocal保存当前数据源标识。

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

3. 创建AOP切面，在切面方法执行前切换数据源。

@Aspect
@Component
public class DataSourceAspect {
    @Before("@annotation(targetDataSource)")
    public void switchDataSource(JoinPoint point, TargetDataSource targetDataSource) {
        DataSourceContextHolder.setDataSourceType(targetDataSource.value());
    }
 
    @After("@annotation(targetDataSource)")
    public void restoreDataSource(JoinPoint point, TargetDataSource targetDataSource) {
        DataSourceContextHolder.clearDataSourceType();
    }
}

4. 创建自定义注解TargetDataSource用于标注方法的数据源。

@Target({ElementType.METHOD, ElementType.TYPE})
@Retention(RetentionPolicy.RUNTIME)
public @interface TargetDataSource {
    String value();
}

5. 配置动态数据源，并将其注册到Spring上下文中。

@Configuration
public class DataSourceConfig {
    @Bean
    public DataSource dataSource() {
        DynamicDataSource dynamicDataSource = new DynamicDataSource();
        dynamicDataSource.setDefaultTargetDataSource(primaryDataSource()); // 默认数据源
        
        // 配置多数据源
        Map<Object, Object> dataSourceMap = new HashMap<>();
        dataSourceMap.put("primary", primaryDataSource());
        dataSourceMap.put("secondary", secondar