问题描述：
这impl文件上已经注明了@DS("tms"),但是在时间执行过程中还是使用的别的数据源。
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;
}
}
在并发环境下，使用指定的数据源可能会出现问题，这通常与以下几个原因有关：
-
数据源上下文的线程本地变量：很多数据源切换机制依赖于线程本地变量（ThreadLocal）来存储当前的数据源信息。在并发环境下，线程池中的线程复用可能导致数据源上下文信息丢失或不一致。
-
CompletableFuture的线程池问题：
CompletableFuture.supplyAsync默认使用的是ForkJoinPool.commonPool，这个线程池中的线程是全局共享的，因此无法保证每个线程都使用正确的数据源上下文。 -
异步任务的AOP切面问题：
@DS注解依赖 AOP 切面来切换数据源，而异步任务中的切面可能不会被正确触发，导致数据源没有切换。
解决方法
- 在任务中手动设置数据源：在每个异步任务中显式地设置数据源。
通过以上配置和代码，你可以在并发环境下正确使用指定的数据源。希望这些步骤能够解决你的问题。
2. 在任务中手动设置数据源
在异步任务中手动切换数据源：
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") 注解。例如：
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" 数据源，例如：
@DS("tms")
public interface TradeLogMapper {
Map<Object, Object> querySlowInterfaceNumOnSevenDays(List<String> serviceNameList);
// 其他方法...
}
这样就不需要在每个方法上都添加 @DS("tms") 注解了。
确保在 Mapper 接口或方法上正确添加 @DS("tms") 注解，应该就可以解决实际使用的数据源与预期不一致的问题。
