Spring Boot 3中@HttpExchange动态添加请求头的优化方案

针对你不想在每个@HttpExchange接口方法中添加@RequestHeader参数的问题，推荐两种更简洁的实现方式：

方案一：Reactor Context + WebClient过滤器（推荐） #

利用WebFlux的Reactor Context在响应式调用链中传递动态请求头，配合WebClient过滤器统一注入请求头，无需修改客户端接口方法。

1. 修改WebClient配置，添加请求头过滤器 #

WebClient webClient(String url) {
    return WebClient.builder()
            .baseUrl(url)
            .filter((request, next) -> {
                // 从Reactor Context中读取动态请求头并注入到请求中
                return Mono.deferContextual(contextView -> {
                    ClientRequest.Builder modifiedRequest = ClientRequest.from(request);
                    
                    // 从上下文取出预先存放的动态请求头
                    contextView.getOrEmpty("dynamicHeaders")
                            .ifPresent(headers -> {
                                if (headers instanceof Map<?, ?> headerMap) {
                                    headerMap.forEach((key, value) -> 
                                        modifiedRequest.header(key.toString(), value.toString())
                                    );
                                }
                            });
                    
                    return next.exchange(modifiedRequest.build());
                });
            })
            .build();
}

2. 保持原客户端接口不变 #

@HttpExchange("/blog")
public interface BlogClient {

    @PostExchange
    Mono<Course> create(@RequestBody BlogInfo blogInfo);
    
    @GetExchange
    Mono<Course> get(@PathVariable Long id);
}

3. 调用时通过Context传递动态请求头 #

在调用客户端方法时，通过contextWrite将动态请求头写入Reactor Context：

// 构造动态请求头（比如认证信息、用户数据等）
Map<String, String> dynamicHeaders = new HashMap<>();
dynamicHeaders.put("Authorization", "Bearer your-token");
dynamicHeaders.put("X-User-Id", "user-1001");

// 调用create接口并传递动态头
blogClient.create(new BlogInfo())
        .contextWrite(ctx -> ctx.put("dynamicHeaders", dynamicHeaders))
        .subscribe(result -> {
            // 处理返回结果
        });

// 调用get接口同理
blogClient.get(100L)
        .contextWrite(ctx -> ctx.put("dynamicHeaders", dynamicHeaders))
        .subscribe();

方案二：自定义参数解析器（进阶） #

如果你希望通过自定义注解的方式传递动态头，可以实现HttpServiceMethodArgumentResolver接口，自定义参数解析逻辑。不过这种方式相对复杂，适合需要更灵活扩展的场景：

1. 定义自定义注解 #

@Target(ElementType.PARAMETER)
@Retention(RetentionPolicy.RUNTIME)
public @interface DynamicHeaders {
}

2. 实现自定义参数解析器 #

public class DynamicHeadersArgumentResolver implements HttpServiceMethodArgumentResolver {

    @Override
    public boolean supportsParameter(Parameter parameter) {
        return parameter.isAnnotationPresent(DynamicHeaders.class);
    }

    @Override
    public Object resolveArgument(Parameter parameter, HttpRequestValues requestValues) {
        // 从Reactor Context读取动态头（可根据实际场景调整数据源）
        return Mono.deferContextual(contextView -> 
            Mono.justOrEmpty(contextView.getOrEmpty("dynamicHeaders"))
        ).block();
    }
}

3. 注册参数解析器到HttpServiceProxyFactory #

@Bean
BlogClient blogClient() {
    HttpServiceProxyFactory httpServiceProxyFactory = HttpServiceProxyFactory
            .builder(WebClientAdapter.forClient(webClient(blogBaseURL)))
            .argumentResolver(new DynamicHeadersArgumentResolver())
            .build();
    return httpServiceProxyFactory.createClient(BlogClient.class);
}

4. 修改客户端接口使用自定义注解 #

@HttpExchange("/blog")
public interface BlogClient {

    @PostExchange
    Mono<Course> create(@RequestBody BlogInfo blogInfo, @DynamicHeaders Map<String, String> headers);
    
    @GetExchange
    Mono<Course> get(@PathVariable Long id, @DynamicHeaders Map<String, String> headers);
}

方案对比 #

• 方案一无需修改客户端接口，通过响应式上下文传递数据，符合WebFlux的设计理念，代码简洁易维护，是优先推荐的方案。
• 方案二更贴近注解式开发，但需要额外编写解析器逻辑，适合有特殊扩展需求的场景。

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