使用SpringBoot动态切换数据源的实现方式

# 数据源1
spring.datasource.url=jdbc:mysql://localhost:3306/db1
spring.datasource.username=user1
spring.datasource.password=password1

# 数据源2
spring.datasource.db2.url=jdbc:mysql://localhost:3306/db2
spring.datasource.db2.username=user2
spring.datasource.db2.password=password2

@Configuration
public class DataSource1Config {
	// 数据源1
    @Bean
    @ConfigurationProperties(prefix = "spring.datasource")
    public DataSource dataSource1() {
        return DataSourceBuilder.create().build();
    }

	// 数据源2
	@Bean
    @ConfigurationProperties(prefix = "spring.datasource.db2")
    public DataSource dataSource2() {
        return DataSourceBuilder.create().build();
    }
}

public class DataSourceContextHolder {
	// 数据源容器
    private static final ThreadLocal<String> contextHolder = new ThreadLocal<>();
	
	// 设置数据源
    public static void setDataSource(String dataSource) {
        contextHolder.set(dataSource);
    }
	
	// 获取数据源
    public static String getDataSource() {
        return contextHolder.get();
    }
	
	// 移除数据源
    public static void clearDataSource() {
        contextHolder.remove();
    }
}

import java.lang.annotation.ElementType;
import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
import java.lang.annotation.Target;

@Target(ElementType.METHOD)
@Retention(RetentionPolicy.RUNTIME)
public @interface DataSourceSwitch {
    String value() default "defaultDataSource";
}

@Aspect
@Component
public class DataSourceSwitchAspect {
    
    // 这里的注解路径根据你自己的路径填写
    @Before("@annotation(com.example.demo.annotation.DataSourceSwitch)")
    public void switchDataSource(JoinPoint joinPoint) {
        MethodSignature signature = (MethodSignature) joinPoint.getSignature();
        Method method = signature.getMethod();
        DataSourceSwitch dataSourceSwitch = method.getAnnotation(DataSourceSwitch.class);
        String dataSource = dataSourceSwitch.value();
        DataSourceContextHolder.setDataSource(dataSource);
    }

    // 这里的注解路径根据你自己的路径填写
    @After("@annotation(com.example.demo.annotation.DataSourceSwitch)")
    public void restoreDataSource(JoinPoint joinPoint) {
        DataSourceContextHolder.clearDataSource();
    }
}

@DataSourceSwitch("db2")
public void doSomething() {
    // 执行自己的业务逻辑
}