@Async注解导致请求头参数丢失


import org.apache.poi.ss.formula.functions.T;
import org.jetbrains.annotations.NotNull;
import org.springframework.context.annotation.Configuration;
import org.springframework.scheduling.annotation.AsyncConfigurer;
import org.springframework.scheduling.annotation.EnableAsync;
import org.springframework.scheduling.concurrent.ThreadPoolTaskExecutor;
import org.springframework.util.concurrent.ListenableFuture;
import org.springframework.web.context.request.RequestAttributes;
import org.springframework.web.context.request.RequestContextHolder;

import java.util.concurrent.Callable;
import java.util.concurrent.Executor;
import java.util.concurrent.Future;

/**
 * @Author: hrd
 * @CreateTime: 2024/4/15 9:42
 * @Description:
 */
@Configuration
@EnableAsync
public class AsyncConfig implements AsyncConfigurer {
    // 实现 getAsyncExecutor 方法来配置任务执行器
    @Override
    public Executor getAsyncExecutor() {
        ContextAwarePoolExecutor executor = new ContextAwarePoolExecutor();
        // 配置线程池参数
        executor.setCorePoolSize(5);
        executor.setMaxPoolSize(10);
        executor.setQueueCapacity(100);
        executor.initialize();
        return executor;
    }

    public static class ContextAwareCallable<T> implements Callable<T> {
        private final Callable<T> task;
        private final RequestAttributes context;

        public ContextAwareCallable(Callable<T> task, RequestAttributes context) {
            this.task = task;
            this.context = context;
        }

        @Override
        public T call() throws Exception {
            if (context != null) {
                RequestContextHolder.setRequestAttributes(context);
            }
            try {
                return task.call();
            } finally {
                RequestContextHolder.resetRequestAttributes();
            }
        }

    }


    public static class ContextAwarePoolExecutor extends ThreadPoolTaskExecutor {
        static <T> ContextAwareCallable<T> of(Callable<T> task, RequestAttributes context) {
            return new ContextAwareCallable(task, context);

        }

        @NotNull
        @Override
        public <T> Future<T> submit(@NotNull Callable<T> task) {
            return super.submit(of(task, RequestContextHolder.currentRequestAttributes()));
        }

        @NotNull
        @Override
        public <T> ListenableFuture<T> submitListenable(@NotNull Callable<T> task) {
            return super.submitListenable(of(task, RequestContextHolder.currentRequestAttributes()));
        }
    }
}

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值