并发条件下-数据源切换失效问题及解决方案

问题描述：

这impl文件上已经注明了@DS("tms"),但是在时间执行过程中还是使用的别的数据源。

dart

代码解读

复制代码

public Map<String, List<String>> queryMonitoringMetricsOnSevenDays(QueryResourceTo projectInfo) { if(projectInfo.getServiceNameList()==null ||projectInfo.getServiceNameList().isEmpty()){ return null; } // 创建三个 CompletableFuture 任务 CompletableFuture<Map<Object, Object>> slowInterfaceFuture = CompletableFuture.supplyAsync( () -> tradeLogMapper.querySlowInterfaceNumOnSevenDays(projectInfo.getServiceNameList()), taskExecutor); CompletableFuture<Map<Object, Object>> slowSqlFuture = CompletableFuture.supplyAsync( () -> sqlLogMapper.querySlowSqlNumOnSevenDays(projectInfo.getServiceNameList()), taskExecutor); CompletableFuture<Map<Object, Object>> errorInfoFuture = CompletableFuture.supplyAsync( () -> errorLogMapper.queryErrorNumOnSevenDays(projectInfo.getServiceNameList()), taskExecutor); // 在需要打印数据源的地方添加以下代码 DataSource dataSource = applicationContext.getBean(DataSource.class); System.out.println("当前使用的数据源为：" + dataSource); // 等待所有任务完成 CompletableFuture.allOf(slowInterfaceFuture).join(); try { // 获取三个任务的结果 Map<Object, Object> slowInterface = slowInterfaceFuture.get(); Map<Object, Object> slowSql = slowSqlFuture.get(); Map<Object, Object> errorInfo = errorInfoFuture.get(); HashMap<String, List<String>> objectObjectHashMap = new HashMap<>(); // 在这里处理获取到的结果 // ... return null; } catch (InterruptedException | ExecutionException e) { // 处理异常 e.printStackTrace(); return null; } }

整理了这份Java面试笔记包括了：Java面试、Spring、JVM、MyBatis、Redis、MySQL、并发编程、微服务、Linux、Springboot、SpringCloud、MQ、Kafka 面试专题

需要全套面试笔记【点击此处】即可免费获取

在并发环境下，使用指定的数据源可能会出现问题，这通常与以下几个原因有关：

1. 数据源上下文的线程本地变量：很多数据源切换机制依赖于线程本地变量（ThreadLocal）来存储当前的数据源信息。在并发环境下，线程池中的线程复用可能导致数据源上下文信息丢失或不一致。

2. CompletableFuture的线程池问题：CompletableFuture.supplyAsync 默认使用的是 ForkJoinPool.commonPool，这个线程池中的线程是全局共享的，因此无法保证每个线程都使用正确的数据源上下文。

3. 异步任务的AOP切面问题：@DS 注解依赖 AOP 切面来切换数据源，而异步任务中的切面可能不会被正确触发，导致数据源没有切换。

解决方法

1. 在任务中手动设置数据源：在每个异步任务中显式地设置数据源。

通过以上配置和代码，你可以在并发环境下正确使用指定的数据源。希望这些步骤能够解决你的问题。

2. 在任务中手动设置数据源

在异步任务中手动切换数据源：

java

代码解读

复制代码

CompletableFuture<Map<Object, Object>> slowInterfaceFuture = CompletableFuture.supplyAsync(() -> { DynamicDataSourceContextHolder.push("tms"); try { return tradeLogMapper.querySlowInterfaceNumOnSevenDays(projectInfo.getServiceNameList()); } finally { DynamicDataSourceContextHolder.poll(); } }, taskExecutor); CompletableFuture<Map<Object, Object>> slowSqlFuture = CompletableFuture.supplyAsync(() -> { DynamicDataSourceContextHolder.push("tms"); try { return sqlLogMapper.querySlowSqlNumOnSevenDays(projectInfo.getServiceNameList()); } finally { DynamicDataSourceContextHolder.poll(); } }, taskExecutor); CompletableFuture<Map<Object, Object>> errorInfoFuture = CompletableFuture.supplyAsync(() -> { DynamicDataSourceContextHolder.push("tms"); try { return errorLogMapper.queryErrorNumOnSevenDays(projectInfo.getServiceNameList()); } finally { DynamicDataSourceContextHolder.poll(); } }, taskExecutor);

通过以上方式，可以确保在并发环境下也能正确使用指定的数据源。你可以选择其中一种方式来解决你的问题，具体选择取决于你的项目结构和上下文管理方式。

提供的代码中，虽然使用了 @DS("tms") 注解来指定数据源为 "tms"，但实际执行查询操作的是 tradeLogMapper、sqlLogMapper 和 errorLogMapper 这三个 Mapper。

如果要确保这三个 Mapper 使用的是 "tms" 数据源，需要在它们对应的方法上也添加 @DS("tms") 注解。例如：

java

代码解读

复制代码

public interface TradeLogMapper { @DS("tms") Map<Object, Object> querySlowInterfaceNumOnSevenDays(List<String> serviceNameList); // 其他方法... } public interface SqlLogMapper { @DS("tms") Map<Object, Object> querySlowSqlNumOnSevenDays(List<String> serviceNameList); // 其他方法... } public interface ErrorLogMapper { @DS("tms") Map<Object, Object> queryErrorNumOnSevenDays(List<String> serviceNameList); // 其他方法... }

通过在 Mapper 接口的方法上添加 @DS("tms") 注解，可以确保这些方法执行时使用的是 "tms" 数据源。

另外，你可以在 Mapper 接口上添加 @DS("tms") 注解，表示该 Mapper 接口中的所有方法都使用 "tms" 数据源，例如：

java

代码解读

复制代码

@DS("tms") public interface TradeLogMapper { Map<Object, Object> querySlowInterfaceNumOnSevenDays(List<String> serviceNameList); // 其他方法... }

这样就不需要在每个方法上都添加 @DS("tms") 注解了。

确保在 Mapper 接口或方法上正确添加 @DS("tms") 注解，应该就可以解决实际使用的数据源与预期不一致的问题。