动态数据源-spring-AbstractRoutingDataSource

AbstractRoutingDataSource是Spring框架中用于实现动态数据源路由的一个抽象类。它可以在运行时根据某种键值动态切换数据源。

以下是一个简单的使用AbstractRoutingDataSource的例子：

import org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource;
 
public class DynamicDataSource extends AbstractRoutingDataSource {
    @Override
    protected Object determineCurrentLookupKey() {
        // DynamicDataSourceContextHolder用于存储当前线程使用的数据源标识
        return DynamicDataSourceContextHolder.getDataSourceType();
    }
}
 
// 配置数据源
@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", secondaryDataSource());
        
        dynamicDataSource.setTargetDataSources(dataSourceMap);
        
        return dynamicDataSource;
    }
 
    @Bean
    public DataSource primaryDataSource() {
        // 创建并配置主数据源
        return new HikariDataSource(HikariConfig config);
    }
 
    @Bean
    public DataSource secondaryDataSource() {
        // 创建并配置副数据源
        return new HikariDataSource(HikariConfig config);
    }
}
 
// 使用DynamicDataSourceContextHolder来设置当前线程使用的数据源
public class DynamicDataSourceContextHolder {
    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();
    }
}
 
// 在需要切换数据源的地方调用
DynamicDataSourceContextHolder.setDataSourceType("secondary");
// 执行数据库操作
DynamicDataSourceContextHolder.clearDataSourceType(); // 清除数据源标识

在这个例子中，我们定义了一个DynamicDataSource类，它继承自AbstractRoutingDataSource并重写了determineCurrentLookupKey方法。然后我们配置了两个数据源，并通过DynamicDataSourceContextHolder在运行时动态切换。这样，我们可以在不同的业务场景下使用不同的数据源。