快捷導(dǎo)航

SpringBoot中使用異步線程導(dǎo)致Request請(qǐng)求頭丟失問(wèn)題的解決方法

更新時(shí)間：2025年07月23日 10:22:41 作者：就胡編亂碼

異步線程請(qǐng)求頭丟失的問(wèn)題通常發(fā)生在多線程環(huán)境中,特別是在使用 CompletableFuture 或其他異步編程模型時(shí),本文給大家詳細(xì)介紹了SpringBoot中使用異步線程導(dǎo)致Request請(qǐng)求頭丟失問(wèn)題的原因和解決方法,需要的朋友可以參考下

背景

異步線程請(qǐng)求頭丟失的問(wèn)題通常發(fā)生在多線程環(huán)境中，特別是在使用 CompletableFuture 或其他異步編程模型時(shí)。具體原因如下：

異步線程請(qǐng)求頭丟失的原因

當(dāng)主線程將任務(wù)提交到異步線程池執(zhí)行時(shí)，當(dāng)前線程中的 RequestAttributes（包括請(qǐng)求頭等上下文信息）不會(huì)自動(dòng)傳遞到異步線程中。這是因?yàn)?RequestAttributes 是基于當(dāng)前線程的本地存儲(chǔ)（ThreadLocal）實(shí)現(xiàn)的，而異步線程是在不同的線程中執(zhí)行任務(wù)，因此無(wú)法訪問(wèn)到主線程中的 RequestAttributes。這種情況下，異步線程執(zhí)行任務(wù)時(shí)會(huì)丟失請(qǐng)求頭等上下文信息。
為了確保異步線程能夠正確獲取請(qǐng)求上下文信息，需要在任務(wù)執(zhí)行前后顯式地設(shè)置和重置 RequestAttributes。

一、問(wèn)題描述

筆者在開(kāi)發(fā)中使用CompletableFuture去異步調(diào)用openFegin服務(wù)，但是在傳遞過(guò)程中發(fā)現(xiàn)了問(wèn)題，原來(lái)存儲(chǔ)在header中的信息，在進(jìn)入其他服務(wù)后header中存儲(chǔ)的user信息都丟失了

CompletableFuture<R<Boolean>> orderCheckFuture = CompletableFuture.supplyAsync(() -> remoteBusinessService.checkUnfinishedOrder(SecurityConstants.INNER, userId));
CompletableFuture<R<Boolean>> taskCheckFuture = CompletableFuture.supplyAsync(() -> remoteInspectionService.checkUnfinishedTask(SecurityConstants.INNER, userId));
CompletableFuture.allOf(orderCheckFuture, taskCheckFuture).join();
R<Boolean> orderCheck = orderCheckFuture.join();
R<Boolean> taskCheck = taskCheckFuture.join();

這種寫(xiě)法就導(dǎo)致了請(qǐng)求頭信息丟失，服務(wù)調(diào)用的發(fā)起方在system服務(wù)模塊，header中的租戶id為12361

請(qǐng)求在到達(dá)business服務(wù)模塊或者inspection服務(wù)模塊時(shí)，租戶id丟失，變成了默認(rèn)值9999

二、解決方案

RequestAttributes requestAttributes = RequestContextHolder.getRequestAttributes();

CompletableFuture<R<Boolean>> orderCheckFuture = CompletableFuture.supplyAsync(() -> {
    RequestContextHolder.setRequestAttributes(requestAttributes);
    return remoteBusinessService.checkUnfinishedOrder(SecurityConstants.INNER, userId);
});

CompletableFuture<R<Boolean>> taskCheckFuture = CompletableFuture.supplyAsync(() -> {
    RequestContextHolder.setRequestAttributes(requestAttributes);
    return remoteInspectionService.checkUnfinishedTask(SecurityConstants.INNER, userId);
});

CompletableFuture.allOf(orderCheckFuture, taskCheckFuture).join();

R<Boolean> orderCheck = orderCheckFuture.join();
R<Boolean> taskCheck = taskCheckFuture.join();

先通過(guò)如下代碼獲取請(qǐng)求頭屬性

RequestAttributes requestAttributes = RequestContextHolder.getRequestAttributes();

接著在異步線程中重新設(shè)置下異步線程所攜帶請(qǐng)求頭的信息，這樣就保證了請(qǐng)求頭的連續(xù)傳遞性

三、拓展

不想每次都在CompletableFuture中都寫(xiě)如下設(shè)置請(qǐng)求頭信息的代碼怎么辦？

RequestContextHolder.setRequestAttributes(requestAttributes);

筆者這里給出一個(gè)自認(rèn)為相對(duì)能減少點(diǎn)代碼量的解決方法(具體有無(wú)必要完全看個(gè)人)

自定義一個(gè)繼承CompletableFuture的類 CustomCompletableFuture

package com.zlbc.common.core.async;

import org.springframework.web.context.request.RequestContextHolder;
import org.springframework.web.context.request.RequestAttributes;

import java.util.concurrent.CompletableFuture;
import java.util.concurrent.Executor;
import java.util.function.Supplier;

/**
 * @author hulei
 * 異步線程傳遞請(qǐng)求頭,跨服務(wù)調(diào)用時(shí)需要傳遞請(qǐng)求頭信息
 */
public class CustomCompletableFuture<T> extends CompletableFuture<T> {

    public static <T> CustomCompletableFuture<T> supplyAsync(Supplier<T> supplier, Executor executor, RequestAttributes requestAttributes) {
        CustomCompletableFuture<T> future = new CustomCompletableFuture<>();
        executor.execute(() -> {
            try {
                RequestContextHolder.setRequestAttributes(requestAttributes);
                T result = supplier.get();
                future.complete(result);
            } catch (Exception e) {
                future.completeExceptionally(e);
            } finally {
                RequestContextHolder.resetRequestAttributes();
            }
        });
        return future;
    }

    public static CustomCompletableFuture<Void> runAsync(Runnable runnable, Executor executor, RequestAttributes requestAttributes) {
        CustomCompletableFuture<Void> future = new CustomCompletableFuture<>();
        executor.execute(() -> {
            try {
                RequestContextHolder.setRequestAttributes(requestAttributes);
                runnable.run();
                future.complete(null);
            } catch (Exception e) {
                future.completeExceptionally(e);
            } finally {
                RequestContextHolder.resetRequestAttributes();
            }
        });
        return future;
    }
}

使用方法如下，筆者設(shè)置了一個(gè)可以傳遞線程池的參數(shù)

RequestAttributes requestAttributes = RequestContextHolder.getRequestAttributes();
        
ExecutorService customExecutor = Executors.newFixedThreadPool(5);
        
CustomCompletableFuture<R<Boolean>> orderCheckFuture = CustomCompletableFuture.supplyAsync(
                () -> remoteBusinessService.checkUnfinishedOrder(SecurityConstants.INNER, userId),
                customExecutor,
                requestAttributes
);

CustomCompletableFuture<R<Boolean>> taskCheckFuture = CustomCompletableFuture.supplyAsync(
                () -> remoteInspectionService.checkUnfinishedTask(SecurityConstants.INNER, userId),
                customExecutor,
                requestAttributes
);

CompletableFuture.allOf(orderCheckFuture, taskCheckFuture).join();

R<Boolean> orderCheck = orderCheckFuture.join();
R<Boolean> taskCheck = taskCheckFuture.join();

這種寫(xiě)法就是減少了設(shè)置的代碼，傳一個(gè)參數(shù)進(jìn)去即可

RequestAttributes requestAttributes = RequestContextHolder.getRequestAttributes();

這個(gè)可以寫(xiě)在某些全局工具類如ServletUtils

使用時(shí)如下獲取即可

RequestAttributes requestAttributes = ServletUtils.getRequestAttributes();

四、總結(jié)

本篇文章主要描述了筆者實(shí)際開(kāi)發(fā)中遇到的采用異步線程導(dǎo)致請(qǐng)求頭丟失的問(wèn)題，本文詳細(xì)介紹了問(wèn)題發(fā)生的場(chǎng)景和解決方法。實(shí)際可以從以下幾個(gè)方面考慮

顯式傳遞 RequestAttributes：

在異步任務(wù)執(zhí)行前，獲取當(dāng)前線程的 RequestAttributes。
在異步任務(wù)執(zhí)行時(shí)，顯式地設(shè)置 RequestAttributes。
在異步任務(wù)執(zhí)行后，重置 RequestAttributes，以避免影響后續(xù)任務(wù)。

使用自定義 CompletableFuture 實(shí)現(xiàn)：

創(chuàng)建一個(gè)自定義的 CompletableFuture 實(shí)現(xiàn)，在任務(wù)執(zhí)行前后自動(dòng)設(shè)置和重置 RequestAttributes。
這樣可以確保每個(gè)異步任務(wù)都能訪問(wèn)到正確的請(qǐng)求上下文信息。