时间戳

首页 > 解决方案 > 如何从 WebFlux 客户端正确排出/释放响应正文？

java - 如何从 WebFlux 客户端正确排出/释放响应正文？

问题描述

我正在使用 Spring 5 反应堆栈中的 WebFlux HTTP 客户端来访问外部 REST 服务。我想根据 HTTP 状态处理响应：

  1. 如果状态是 2xx 我想返回Mono反序列化的响应正文。

  2. 如果状态为 404 我想删除响应正文并立即返回 empty Mono。

  3. 对于任何其他状态，我想删除响应正文并返回错误 Mono with MyBusinessException。

我的代码如下所示：

webClient.get()
    .uri("/search")
    .syncBody(request)
    .exchange()
    .flatMap { response ->
        when {
            response.statusCode().is2xxSuccessful -> response.bodyToMono(MyResponse::class.java)
            response.statusCode() == NOT_FOUND -> Mono.empty()
            else -> MyBusinessException().toMono<MyResponse>()
        }
     }

我不想在不需要的时候浪费时间接收和处理响应正文。exchange()用于方法状态的 JavaDoc

您必须始终使用响应的主体或实体方法之一来确保释放资源。

如果我想排空响应正文并立即返回结果，我应该怎么做？

标签： javaspring-webfluxproject-reactor

解决方案

需要排空响应，以便可以将连接重用于未来的请求（即 http 保持活动/持久连接）。

要返回一个Mono在主体耗尽后完成的空（忽略错误）：

// Using WebFlux >= 5.2
response.releaseBody()
    // ignore errors
    .onErrorResume(exception -> Mono.empty());


// Using WebFlux < 5.2
response.body(BodyExtractors.toDataBuffers())
    // release DataBuffers
    .doOnNext(DataBufferUtils::release)
    // ignore errors
    .onErrorResume(exception -> Mono.empty())
    // return an empty Mono
    .then();

要返回一个立即完成的空Mono，并在后台异步排出主体（忽略错误）：

// Using WebFlux >= 5.2
Mono.<Void>empty()
    .doOnSubscribe(s ->
        // drain the body
        response.releaseBody()
            // initiate drain on a separate Scheduler
            .subscribeOn(Schedulers.parallel())
            // subscribe, and ignore errors
            .subscribe())

// Using WebFlux < 5.2
Mono.<Void>empty()
    .doOnSubscribe(s ->
        // drain the body
        response.body(BodyExtractors.toDataBuffers())
            // release DataBuffers
            .doOnNext(DataBufferUtils::release)
            // initiate drain on a separate Scheduler
            .subscribeOn(Schedulers.parallel())
            // subscribe, and ignore errors
            .subscribe())

我仍然推荐第一个选项，因为它会立即释放资源，并且可能是WebClient开发人员在编写它并记录其使用情况时所考虑的。

我从未在生产系统中使用过第二个选项，因此请自行测试以确保 http 连接池的行为符合预期。如果使用 reactor-netty，您可以启用调试日志记录reactor.netty.resources.PooledConnectionProvider以比较两种方法。

推荐阅读