首先,一些有助于您理解解决此用例的代码段的事情。
你永远不应该在返回一个被动类型的方法中调用阻塞方法; 你将阻止应用程序的少数线程之一,这对应用程序来说非常糟糕
无论如何,从Reactor 3.2开始, 在反应式管道中阻塞会引发错误
如评论中所建议的,调用subscribe也不是一个好主意。 它或多或少像在单独的线程中将该作业作为任务启动。 完成后你会得到一个回调( subscribe方法可以给出lambdas),但实际上你正在将当前的管道与该任务分离。 在这种情况下,在您有机会读取完整的响应主体以将其写入文件之前,可以关闭客户端HTTP响应并清理资源
如果你不想在内存中缓冲整个响应,Spring提供了DataBuffer (想想可以合并的ByteBuffer实例)。
如果您正在实现的方法本身阻塞(例如返回void ),则可以调用block,例如在测试用例中。
这是您可以用来执行此操作的代码段:
Mono fileWritten = WebClient.create().post()
.uri(uriBuilder -> uriBuilder.path("/file/").build())
.exchange()
.flatMap(response -> {
if (MediaType.APPLICATION_JSON_UTF8.equals(response.headers().contentType().get())) {
Mono dto = response.bodyToMono(NoPayloadResponseDto.class);
return createErrorFile(dto);
}
else {
Flux 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 ,这是done(error)回调的反应等价物,它在事情完成时发出信号并且发生错误。
由于我不确定createErrorFile方法应该做什么,我已经为createSpreadsheet提供了一个示例,它只是将body字节写入文件。 请注意,由于数据库可能被回收/合并,我们需要在完成后释放它们。
private Mono createSpreadsheet(Flux 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实例(由于性能原因,响应式运算符正在预取值),并且会以反应方式写入字节。