block()/blockFirst()/blockLast() 在 exchange() 之后调用 bodyToMono 时出现阻塞错误

Question

我正在尝试使用 Webflux 将生成的文件流式传输到另一个位置，但是，如果文件的生成遇到错误，则 api 返回成功，但在生成文件时使用 DTO 详细说明错误，而不是文件本身。这是使用了一个非常古老且设计不佳的 api，所以请原谅使用 post 和 api 设计。

来自 api 调用 (exchange()) 的响应是 ClientResponse。从这里我可以使用 bodyToMono 转换为 ByteArrayResource，它可以流式传输到文件，或者，如果创建文件时出错，那么我也可以使用 bodyToMono 转换为 DTO。但是，我似乎无法根据 ClientResponse 标头的内容做任何事情。

在运行时我得到一个由

引起的 IllegalStateException

block()/blockFirst()/blockLast()是阻塞的，线程reactor-http-client-epoll-12不支持这个

我认为我的问题是我不能在同一个函数链中调用 block() 两次。

我的代码sn-p是这样的：

webClient.post()
        .uri(uriBuilder -> uriBuilder.path("/file/")
                                      .queryParams(params).build())
        .exchange()
        .doOnSuccess(cr -> {
                if (MediaType.APPLICATION_JSON_UTF8.equals(cr.headers().contentType().get())) {
                    NoPayloadResponseDto dto = cr.bodyToMono(NoPayloadResponseDto.class).block();
                    createErrorFile(dto);
                }
                else {
                    ByteArrayResource bAr = cr.bodyToMono(ByteArrayResource.class).block();
                    createSpreadsheet(bAr);
                }
            }
        )
        .block();

基本上，我想根据标头中定义的 MediaType 以不同方式处理 ClientResponse。

这可能吗？

Answer 1

首先，有几件事可以帮助您理解解决此用例的代码 sn-p。

1. 永远不要在返回反应类型的方法中调用阻塞方法；您将阻塞应用程序的少数线程之一，这对应用程序非常不利
2. 从 Reactor 3.2 开始，blocking within a reactive pipeline throws an error
3. 按照 cmets 的建议，调用 subscribe 也不是一个好主意。它或多或少像在单独的线程中将这项工作作为任务启动。完成后您会收到一个回调（可以为 subscribe 方法提供 lambdas），但实际上您正在将当前管道与该任务解耦。在这种情况下，可能会关闭客户端 HTTP 响应并清理资源，然后您才有机会读取完整的响应正文以将其写入文件
4. 如果您不想在内存中缓冲整个响应，Spring 提供了DataBuffer（想想可以池化的 ByteBuffer 实例）。
5. 如果您正在实现的方法本身是阻塞的（例如返回 void），您可以调用 block，例如在测试用例中。

这是一个代码 sn-p，您可以使用它来执行此操作：

Mono<Void> fileWritten = WebClient.create().post()
        .uri(uriBuilder -> uriBuilder.path("/file/").build())
        .exchange()
        .flatMap(response -> {
            if (MediaType.APPLICATION_JSON_UTF8.equals(response.headers().contentType().get())) {
                Mono<NoPayloadResponseDto> dto = response.bodyToMono(NoPayloadResponseDto.class);
                return createErrorFile(dto);
            }
            else {
                Flux<DataBuffer> body = response.bodyToFlux(DataBuffer.class);
                return createSpreadsheet(body);
            }
        });
// Once you get that Mono, you should give plug it into an existing
// reactive pipeline, or call block on it, depending on the situation

如您所见，我们不会在任何地方阻塞，处理 I/O 的方法会返回 Mono<Void>，这是一个 done(error) 回调的反应式等价物，它会在事情完成以及是否发生错误时发出信号。

由于我不确定createErrorFile 方法应该做什么，我为createSpreadsheet 提供了一个示例，它只是将正文字节写入文件。请注意，由于数据缓冲区可能会被回收/池化，因此我们需要在完成后释放它们。

private Mono<Void> createSpreadsheet(Flux<DataBuffer> body) {
    try {
        Path file = //...
        WritableByteChannel channel = Files.newByteChannel(file, StandardOpenOption.WRITE);
        return DataBufferUtils.write(body, channel).map(DataBufferUtils::release).then();
    } catch (IOException exc) {
        return Mono.error(exc);
    }
}

通过此实现，您的应用程序将在给定时间在内存中保存一些 DataBuffer 实例（反应式运算符出于性能原因预取值），并且会在字节以反应式方式进入时写入它们。

Answer 2

[2021 年 10 月 19 日更新]

toProcessor() 现已弃用。

考虑使用

myMono.toFuture().get();

---

正如投票最多的答案所述，永远不应该阻止。就我而言，这是唯一的选择，因为我们在命令式代码中使用了响应式库。拦截可以通过wrapping the mono in a processor:

myMono.toProcessor().block()

Answer 3

要在服务器请求池之外执行客户端请求，请使用myWebClientMono.share().block();

Answer 4

对我来说，添加网络依赖解决了这个问题。

        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-web</artifactId>
        </dependency>

Answer 5

RestResultMessage message= createWebClient()
                .get()
                .uri(uri)
                .exchange()
                .map(clientResponse -> {
                    //delegation
                    ClientResponseWrapper wrapper = new 
                                 ClientResponseWrapper(clientResponse);
                    return Mono.just(wrapper);
                })
                .block() //wait until request is not done
                .map(result -> {  
                    //convert to any data
                    if (!result.statusCode().isError()){
                       //extract the result from request
                        return create(RestResultMessage.Result.success, result.bodyToMono(String.class).block());}
                    } else {
                        return create(RestResultMessage.Result.error, result.statusCode().name());
                    }
                })
                .block();