Spring WebFlux：如何将WebClient响应头纳入流式ServerResponse

你遇到的核心问题是响应式编程的异步特性导致的时序问题：构建ServerResponse头的代码是同步执行的，而WebClient的请求是异步触发的——当你设置ServerResponse头时，WebClient还没完成上游请求，AtomicReference自然还是null。

要正确传递上游响应头，必须把ServerResponse的构建逻辑和WebClient的响应处理逻辑串联在同一个响应式流中，不能用外部变量捕获异步值。

正确实现代码 #

@Override
public Mono<ServerResponse> exportAsStreaming(
    ExportExpensesRequest body,
    ForwardableHttpHeaders requestHeaders) {
    return webClient
        .post()
        .uri(uriBuilder -> uriBuilder.path("/api/exports/file").build())
        .headers(ServerRequestUtility.transferHttpHeaders(requestHeaders))
        .bodyValue(body)
        // 用exchangeToMono获取完整的上游响应（头+体）
        .exchangeToMono(clientResponse -> {
            HttpHeaders upstreamHeaders = clientResponse.headers().asHttpHeaders();
            
            // 初始化ServerResponse构建器
            ServerResponse.BodyBuilder responseBuilder = ServerResponse.ok()
                .contentType(MediaType.APPLICATION_OCTET_STREAM);
            
            // 按需传递Content-Disposition头（处理上游无该头的情况）
            Optional.ofNullable(upstreamHeaders.getFirst(HttpHeaders.CONTENT_DISPOSITION))
                .ifPresent(disposition -> responseBuilder.header(HttpHeaders.CONTENT_DISPOSITION, disposition));
            
            // 直接复用上游的响应体流，避免重复消费
            return responseBuilder.body(clientResponse.bodyToFlux(DataBuffer.class), DataBuffer.class);
        });
}

关键说明 #

1. 用exchangeToMono替代原逻辑：exchangeToMono会在WebClient收到上游响应后触发回调，此时可以安全获取响应头和响应体，确保头和体的处理处于同一个异步流程中。
2. 避免外部变量捕获异步值：响应式编程中，不要用AtomicReference这类同步容器来存储异步生成的值，必须通过响应式操作符（如exchangeToMono、flatMap）串联逻辑。
3. 响应体复用：直接使用clientResponse.bodyToFlux(DataBuffer.class)作为ServerResponse的响应体，确保响应流只被消费一次（响应式流的体只能被消费一次）。

内容的提问来源于stack exchange，提问作者tyb9900