springboot 在拦截器中根据参数动态切换数据源

在 Spring Boot 中，可以使用拦截器（Interceptor）来实现在请求处理过程中进行数据源的动态切换。以下是一个简单的实现方式：

1. 首先需要定义一个拦截器，在其中实现数据源的动态切换逻辑，例如：

public class DataSourceInterceptor implements HandlerInterceptor {

    @Override
    public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {
        // 获取请求中的参数，根据参数值选择不同的数据源
        String dataSourceName = request.getParameter("dataSource");
        DataSourceContextHolder.setDataSource(dataSourceName);
        return true;
    }

    @Override
    public void afterCompletion(HttpServletRequest request, HttpServletResponse response, Object handler, Exception ex) throws Exception {
        // 清除数据源上下文
        DataSourceContextHolder.clearDataSource();
    }
}

2. 在拦截器中获取请求中的参数，并根据参数值选择相应的数据源。这里使用了一个 `DataSourceContextHolder` 类来保存当前数据源的名称，代码如下：

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

3. 接着，在 Spring Boot 的配置类中注册拦截器，并设置拦截器的顺序，例如：

@Configuration
public class WebMvcConfig implements WebMvcConfigurer {

    @Autowired
    private DataSourceInterceptor dataSourceInterceptor;

    @Override
    public void addInterceptors(InterceptorRegistry registry) {
        registry.addInterceptor(dataSourceInterceptor)
            .addPathPatterns("/**")
            .order(Ordered.HIGHEST_PRECEDENCE);
    }
}

4. 最后，需要在应用程序中配置多个数据源，并为每个数据源指定一个名称。这里使用了 Spring Boot 的自动配置来配置数据源，代码如下：

@Configuration
public class DataSourceConfig {

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

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

    @Bean
    public RoutingDataSource routingDataSource(@Qualifier("dataSource1") DataSource dataSource1,
                                                @Qualifier("dataSource2") DataSource dataSource2) {
        Map<Object, Object> targetDataSources = new HashMap<>();
        targetDataSources.put("dataSource1", dataSource1);
        targetDataSources.put("dataSource2", dataSource2);

        RoutingDataSource routingDataSource = new RoutingDataSource();
        routingDataSource.setTargetDataSources(targetDataSources);
        routingDataSource.setDefaultTargetDataSource(dataSource1);

        return routingDataSource;
    }
}

在这个示例中，我们通过定义两个数据源 `dataSource1` 和 `dataSource2`，并为每个数据源指定一个名称。然后，我们创建一个 `RoutingDataSource`，并将这两个数据源添加到目标数据源列表中。在拦截器中，我们可以根据请求参数的值选择不同的数据源，并将选择的数据源名称保存在 `DataSourceContextHolder