如何在Armeria客户端装饰器中访问请求体以实现AWS v4签名请求拦截

我来帮你搞定这个在Armeria客户端里用AWS v4签名时获取请求体的问题～

问题的核心在于Armeria的HttpRequest默认是流式/一次性的，直接在装饰器里读取请求体的话，不仅可能读不到完整内容，还会导致后续请求执行时丢失body。解决思路是先把请求体完整缓存下来（也就是聚合请求），再进行签名计算，最后重新构建带签名的请求。

具体实现代码 #

wcb.decorator((delegate, ctx, req) -> {
    // 异步聚合请求，获取完整的请求方法、路径、headers和body
    return req.aggregate()
              .thenApply(aggregatedReq -> {
                  // 这里可以安全获取请求体：aggregatedReq.content()
                  String authHeader = constructAWS4SignHeader(aggregatedReq);
                  
                  // 重新构建带有签名头的HttpRequest，保留原请求的所有信息
                  HttpRequest signedReq = HttpRequest.of(
                      aggregatedReq.method(),
                      aggregatedReq.path(),
                      aggregatedReq.headers().toBuilder()
                                  .set("Authorization", authHeader)
                                  .build(),
                      aggregatedReq.content()
                  );
                  
                  // 执行签名后的请求
                  return delegate.execute(ctx, signedReq);
              })
              .exceptionally(e -> {
                  // 处理聚合失败的异常（比如超大请求、IO错误）
                  ctx.close();
                  return HttpResponse.of(
                      HttpStatus.INTERNAL_SERVER_ERROR,
                      MediaType.PLAIN_TEXT,
                      "Failed to prepare AWS signature: " + e.getMessage()
                  );
              });
});

关键细节说明 #

• 请求聚合的作用：aggregate()方法会异步将流式请求转换为AggregatedHttpRequest，它包含了完整的请求体（HttpData类型，可重复读取），这样你就能安全地获取签名所需的body内容，同时不会影响后续请求的执行。
• 适配你的签名方法：需要调整constructAWS4SignHeader方法，让它接收AggregatedHttpRequest参数，从中提取AWS v4签名需要的核心信息：HTTP方法、请求路径、请求头、请求体内容。
• 异常处理：添加exceptionally分支可以处理请求聚合失败的场景，比如请求体过大导致内存不足、IO异常等，避免客户端请求挂起。

特殊场景补充 #

如果你的业务涉及超大文件上传，直接聚合请求体可能会占用过多内存，这时候可以考虑AWS v4的分块签名（Chunked signing）机制。不过这种场景比较复杂，需要结合Armeria的流式处理能力，逐块计算签名并添加到请求头中。一般来说，普通业务场景用上面的聚合方案就足够了。

内容的提问来源于stack exchange，提问作者kash-hash