如何解决线程池Feign调用失败的

最新推荐文章于 2024-04-30 08:30:00 发布

dyk121

最新推荐文章于 2024-04-30 08:30:00 发布

阅读量529

点赞数 4

文章标签： java

本文链接：https://blog.csdn.net/dyk121/article/details/137923695

版权

项目场景：

项目场景：上周过需求，客户提了一个批量操作的功能。具体业务就不描述了，大概功能是批量新增数据，同时自动生成下游数据。我看了一下原来的屎山代码，改是改不了了，那就开始重新撸吧。

由于下游数据逻辑复杂，我使用线程池去执行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;
        }
    }
}