项目场景:
项目场景:上周过需求,客户提了一个批量操作的功能。具体业务就不描述了,大概功能是批量新增数据,同时自动生成下游数据。我看了一下原来的屎山代码,改是改不了了,那就开始重新撸吧。
由于下游数据逻辑复杂,我使用线程池去执行weekPlanAutoDailyNew(我的业务方法);
于是定义了一个线程安全的集合(CopyOnWriteArrayList),用于收集下游数据,然后统一处理保存。部分代码如下;
问题描述
//根据租户参数判定客户是否是手动排程还是自动排程
if (b && isWeekPlanAutoDaily) {
//b && 自动排程
CountDownLatch countDownLatch = new CountDownLatch(weekPlans.size());
for (WeekPlan weekPlan : weekPlans) {
String tenantId = AuthUtil.getTenantId();
//这里使用线程池异步
CompletableFuture.supplyAsync(()->
weekPlanAutoDailyNew(weekPlan, tenantId),executor
).thenAccept(
workOrderProcessDaily -> {
dailies.add(workOrderProcessDaily);
countDownLatch.countDown();
}
).exceptionally(throwable -> {
log.info(throwable);
return null;
});
}
//计数器
return countDownLatch.await(10, TimeUnit.SECONDS);
}
在weekPlanAutoDailyNew方法中存在feign()调用,后期测试的时候发现一直出现服务调用失败的现象。
原因分析:
由于在weekPlanAutoDailyNew()方法中存在feign调用,而我线程池中线程是没有主线程token的,所以才会出现一直调用失败的结果。
解决方案:
因为项目中FeignConfig是已经配置好的 我的思路是让子线程拿到主线程中上下文的信息。
@Configuration
@EnableAsync
public class ThreadPoolConfig {
@Primary
@Bean("taskExecutor")
public Executor taskExecutor(){
ThreadPoolTaskExecutor taskExecutor = new ThreadPoolTaskExecutor();
//核心参数可用按照实际情况设置
taskExecutor.setCorePoolSize(10);
taskExecutor.setMaxPoolSize(50);
taskExecutor.setQueueCapacity(200);
taskExecutor.setKeepAliveSeconds(60);
taskExecutor.setThreadNamePrefix("taskExecutor--");
taskExecutor.setWaitForTasksToCompleteOnShutdown(true);
taskExecutor.setAwaitTerminationSeconds(60);
//这里设置了一个包装类,说白了就是AOP增强
taskExecutor.setTaskDecorator(new ContextCopyingDecorator());
return taskExecutor;
}
}
这样我们就可以在线程执行方法前后对他进行增强
public class ContextCopyingDecorator implements TaskDecorator {
@Override
public Runnable decorate(Runnable runnable) {
try {
//获取父线程的context
RequestAttributes context = RequestContextHolder.currentRequestAttributes();
HttpServletRequest request = ((ServletRequestAttributes) RequestContextHolder.getRequestAttributes()).getRequest();
//String header = request.getHeader("Your-Token");
return () -> {
try {
//在这进行增强
//将父线程的context设置进子线程里
RequestContextHolder.setRequestAttributes(context);
//子线程方法执行
runnable.run();
} finally {
//清除子线程context
RequestContextHolder.resetRequestAttributes();
}
};
} catch (IllegalStateException e) {
return runnable;
}
}
}
备注
具体如何设置线程信息,需要参考自己项目中Feign是如何设置token的(实现RequestInterceptor接口重新apply方法,对RequestTemplate增强)
是使用ThreadLocalUtil或者其他解决方案。