@DS和@Async 共同使用出现的bug

在前端发起的异步请求场景下,由于数据库分库导致设置线程请求头失败,从而影响入库操作。问题在于异步执行时新线程无法继承父线程的头信息。为解决此问题,采取手动在任务执行前设置线程上下文和头信息的方法,通过自定义`TaskDecorator`确保子线程获取到正确的请求头和上下文。这样保证了异步处理过程中线程头信息的正确传递。
摘要由CSDN通过智能技术生成

项目场景:

项目相关背景:

项目场景:前端发送了请求,由于响应时间较长,所以要采用异步,但是异步时由于数据库分库的原因,设置当前线程的头信息失败,导致入库失败


问题描述

这样设置线程的请求头会失败

    @DS("#dynamic")
    @Async("transaction")
    @Override
    public void doSomting(Entity vos, HttpHeaders headers) {
        CurrentHeadersUtil.setHeaders(headers);
        //这是设置线程的请求头会失败
        }

原因分析:

因为异步时会重新起一个线程,子线程跟父线程不在同一个线程池中,在
CurrentHeadersUtil.setHeaders时,并没有把当前的线程头信息赋值给子线程


解决方案:

在线程池中主动手动把线程头信息赋值给子线程

   //当前用到的线程池
    @Bean("transaction")
    public TaskExecutor icbctaskExecutor() {
        ThreadPoolTaskExecutor executor = new ThreadPoolTaskExecutor();
        //一些常规参数
        executor.setCorePoolSize(8);
        executor.setMaxPoolSize(30);
        executor.setQueueCapacity(30);
        executor.setThreadNamePrefix("newThreadPool_");
        //主要解决问题点就是这里,手动赋值头信息
        executor.setTaskDecorator(new ThreadPoolConfig.ContextDecorator());
        executor.setRejectedExecutionHandler(new ThreadPoolExecutor.CallerRunsPolicy());
        executor.setKeepAliveSeconds(60);
        executor.setWaitForTasksToCompleteOnShutdown(true);
        executor.setAwaitTerminationSeconds(60);
        executor.initialize();
        return executor;
    }
  //这里是具体操作
    static class ContextDecorator implements TaskDecorator {
        ContextDecorator() {
        }
    
        public Runnable decorate(Runnable runnable) {
            try {
                RequestAttributes context = RequestContextHolder.getRequestAttributes();
                HttpHeaders headers = CurrentHeadersUtil.getHeaders();
                return () -> {
                    try {
                    //上下文和头信息都要赋值
                        if (context != null) {
                            RequestContextHolder.setRequestAttributes(context);
                        }

                        if (headers != null) {
                            CurrentHeadersUtil.setHeaders(headers);
                        }

                        runnable.run();
                    } finally {
                        RequestContextHolder.resetRequestAttributes();
                    }

                };
            } catch (Exception var4) {
                var4.printStackTrace();
                return runnable;
            }
        }
    }

这是我的理解,有问题请指出

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值