java - 如何在 Java 11 中读取 HttpRequest 的正文？

Question

在测试中，我想查看 HttpRequest 的主体内部。我想把 body 当作一根绳子。似乎唯一的方法就是订阅 BodyPublisher，但是这是如何工作的呢？

Answer 1

这是一个有趣的问题。您从哪里获取 HttpRequest ？最简单的方法是直接从创建 HttpRequest 的代码中获取正文。如果这是不可能的，那么下一步就是克隆该请求并将其正文发布者包装在您自己的 BodyPublisher 实现中，然后再通过 HttpClient 发送请求。编写 HttpRequest 的子类应该相对容易(如果乏味)，它包装 HttpRequest 的另一个实例并将每个调用委托(delegate)给包装的实例，但覆盖 HttpRequest::bodyPublisher 执行以下操作:

return request.bodyPublisher().map(this::wrapBodyPublisher);

否则，您也可以尝试订阅请求正文发布者并从中获取正文字节 - 但请注意，并非 BodyPublisher 的所有实现都可以支持多个订阅者(无论是并发还是顺序) 。

为了说明我上面的建议:类似于下面的内容可能会起作用，具体取决于正文发布者的具体实现，并且前提是您可以防止对正文发布者的并发订阅。也就是说，在一个受控的测试环境中，您了解所有各方，那么它可能是可行的。不要在生产中使用任何这样的东西:

public class HttpRequestBody {

    // adapt Flow.Subscriber<List<ByteBuffer>> to Flow.Subscriber<ByteBuffer>
    static final class StringSubscriber implements Flow.Subscriber<ByteBuffer> {
        final BodySubscriber<String> wrapped;
        StringSubscriber(BodySubscriber<String> wrapped) {
            this.wrapped = wrapped;
        }
        @Override
        public void onSubscribe(Flow.Subscription subscription) {
            wrapped.onSubscribe(subscription);
        }
        @Override
        public void onNext(ByteBuffer item) { wrapped.onNext(List.of(item)); }
        @Override
        public void onError(Throwable throwable) { wrapped.onError(throwable); }
        @Override
        public void onComplete() { wrapped.onComplete(); }
    }

    public static void main(String[] args) throws Exception {
        var request = HttpRequest.newBuilder(new URI("http://example.com/blah"))
                .POST(BodyPublishers.ofString("Lorem ipsum dolor sit amet"))
                .build();

        // you must be very sure that nobody else is concurrently 
        // subscribed to the body publisher when executing this code,
        // otherwise one of the subscribers is likely to fail.
        String reqbody = request.bodyPublisher().map(p -> {
            var bodySubscriber = BodySubscribers.ofString(StandardCharsets.UTF_8);
            var flowSubscriber = new StringSubscriber(bodySubscriber);
            p.subscribe(flowSubscriber);
            return bodySubscriber.getBody().toCompletableFuture().join();
        }).get();
        System.out.println(reqbody);
    }

}