问题描述:
这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 面试专题
需要全套面试笔记【点击此处】即可免费获取
在并发环境下,使用指定的数据源可能会出现问题,这通常与以下几个原因有关:
-
数据源上下文的线程本地变量:很多数据源切换机制依赖于线程本地变量(ThreadLocal)来存储当前的数据源信息。在并发环境下,线程池中的线程复用可能导致数据源上下文信息丢失或不一致。
-
CompletableFuture的线程池问题:
CompletableFuture.supplyAsync
默认使用的是ForkJoinPool.commonPool
,这个线程池中的线程是全局共享的,因此无法保证每个线程都使用正确的数据源上下文。 -
异步任务的AOP切面问题:
@DS
注解依赖 AOP 切面来切换数据源,而异步任务中的切面可能不会被正确触发,导致数据源没有切换。
解决方法
- 在任务中手动设置数据源:在每个异步任务中显式地设置数据源。
通过以上配置和代码,你可以在并发环境下正确使用指定的数据源。希望这些步骤能够解决你的问题。
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")
注解,应该就可以解决实际使用的数据源与预期不一致的问题。