小编典典

调用bodyToMono AFTER exchange()后,block()/ blockFirst()/ blockLast()发生阻塞错误

java

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

api调用(exchange())的响应是ClientResponse。从这里,我可以使用bodyToMono转换为ByteArrayResource并将其传输到文件中,或者,如果在创建文件时出错,那么我也可以使用bodyToMono转换为DTO。但是,我似乎既不执行任何操作,也不依赖于ClientResponse标头的内容。

在运行时,我收到了一个IllegalStateException引起的

block()/ blockFirst()/ blockLast()正在阻塞,线程反应器-http-client-epoll-12不支持

我认为我的问题是我无法在同一功能链中两次调用block()。

我的代码段如下所示:

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。

这可能吗?


阅读 4521

收藏
2020-09-26

共1个答案

小编典典

首先,一些事情将帮助您理解解决此用例的代码片段。

  1. 您永远不要在返回反应类型的方法中调用阻塞方法。您将阻塞应用程序的几个线程之一,这对应用程序来说是非常糟糕的
  2. 无论如何,从Reactor 3.2开始,在反应式管道内进行阻塞会引发错误
  3. subscribe如评论中所建议的那样,调用也不是一个好主意。它或多或少像在一个单独的线程中将该任务作为任务开始。完成后,您将获得一个回调(subscribe可以给这些方法指定lambdas),但实际上您正在将当前管道与该任务分离。在这种情况下,在您有机会读取完整的响应正文并将其写入文件之前,可能会关闭客户端HTTP响应并清理资源。
  4. 如果您不想将整个响应缓冲在内存中,Spring会提供DataBuffer(请考虑可以缓冲的ByteBuffer实例)。
  5. 如果要实现的方法本身是阻塞的(void例如返回),例如在测试用例中,则可以调用阻塞。

这是一个可用于执行此操作的代码段:

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在给定的时间在内存中保存几个实例(反应式运算符出于性能原因正在预取值),并将以反应式方式写入字节。

2020-09-26