Springboot如何使用JdbcTemplate 来实现多数据源切换

当使用 @DataSource 注解结合 JdbcTemplate 来实现多数据源切换时，你需要定义注解、配置数据源、创建 JdbcTemplate 实例、实现 AOP 切面以及编写一个类来根据注解的值选择正确的数据源。下面是一个简化的示例来展示这个过程：

首先，定义 @DataSource 注解：

java
@Target({ElementType.METHOD, ElementType.TYPE})
@Retention(RetentionPolicy.RUNTIME)
public @interface DataSource {
String value() default “default”;
}
接着，配置数据源和 JdbcTemplate 实例（这里以 Java 配置为例）：

java
@Configuration
public class DataSourceConfig {

@Bean(name = "dataSourceMaster")  
@ConfigurationProperties(prefix = "spring.datasource.master")  
public DataSource dataSourceMaster() {  
    return DataSourceBuilder.create().build();  
}  

@Bean(name = "dataSourceSlave")  
@ConfigurationProperties(prefix = "spring.datasource.slave")  
public DataSource dataSourceSlave() {  
    return DataSourceBuilder.create().build();  
}  

@Bean(name = "jdbcTemplateMaster")  
public JdbcTemplate jdbcTemplateMaster(@Qualifier("dataSourceMaster") DataSource dataSource) {  
    return new JdbcTemplate(dataSource);  
}  

@Bean(name = "jdbcTemplateSlave")  
public JdbcTemplate jdbcTemplateSlave(@Qualifier("dataSourceSlave") DataSource dataSource) {  
    return new JdbcTemplate(dataSource);  
}  

// 其他数据源配置...  

}
然后，实现 AOP 切面来根据 @DataSource 注解的值切换数据源：

java
@Aspect
@Component
public class DataSourceAspect {

@Autowired  
private Map<Object, Object> jdbcTemplateMap;  

@Pointcut("@annotation(com.example.DataSource)")  
public void dataSourcePointCut() {}  

@Before("dataSourcePointCut()")  
public void switchDataSource(JoinPoint point) {  
    MethodSignature signature = (MethodSignature) point.getSignature();  
    Method method = signature.getMethod();  

    DataSource dataSource = method.getAnnotation(DataSource.class);  
    if (dataSource != null) {  
        String dataSourceName = dataSource.value();  
        DataSourceContextHolder.setDataSource(dataSourceName);  
    }  
}  

@After("dataSourcePointCut()")  
public void restoreDataSource() {  
    DataSourceContextHolder.clearDataSource();  
}  

}
注意这里我们使用了 DataSourceContextHolder 来保存和切换当前线程的数据源。这个类需要你自己实现，它通常是一个 ThreadLocal 变量来保存数据源的信息。

java
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();  
}  

}
最后，在 Service 或 DAO 层中使用 @DataSource 注解和 JdbcTemplate：

java
@Service
public class MyService {

@Autowired  
@Qualifier("jdbcTemplateMap")  
private Map<Object, Object> jdbcTemplateMap;  

public void doSomethingOnMaster() {  
    JdbcTemplate jdbcTemplate = (JdbcTemplate) jdbcTemplateMap.get("jdbcTemplateMaster");  
    // 使用 jdbcTemplate 进行数据库操作...  
}  

@DataSource("slave")  
public void doSomethingOnSlave() {  
    String dataSourceName = DataSourceContextHolder.getDataSource();  
    JdbcTemplate jdbcTemplate = (JdbcTemplate) jdbcTemplateMap.get(dataSourceName + "jdbcTemplate");  
    // 使用 jdbcTemplate 进行数据库操作...  
}  

}
注意这里我们使用了 jdbcTemplateMap 来根据数据源名称获取对应的 JdbcTemplate 实例。你可以通过 Spring 的 @Autowired 和 @Qualifier 注解来注入这个 Map，其中 key 是你定义的 bean 名称，value 是对应的 JdbcTemplate 实例。

请注意，上述代码只是一个简化的示例，并且你可能需要根据你的实际需求进行适当的调整。特别是 DataSourceContextHolder 和 DataSourceAspect 的实现细节，以及 JdbcTemplate 的使用方式，都可能因你的具体需求而有所不同。