Follow-up fix for #5b711a

Not only do we need to preserve Mono in writeWith in the base class
but also in ReactorServerHttpResponse where buffers are mapped.

See gh-22800
This commit is contained in:
Rossen Stoyanchev
2019-04-17 13:18:01 -04:00
parent 52b6f9b56d
commit 13c746ae9d

View File

@@ -122,7 +122,9 @@ class ReactorServerHttpResponse extends AbstractServerHttpResponse implements Ze
}
private Publisher<ByteBuf> toByteBufs(Publisher<? extends DataBuffer> dataBuffers) {
return Flux.from(dataBuffers).map(NettyDataBufferFactory::toByteBuf);
return dataBuffers instanceof Mono ?
Mono.from(dataBuffers).map(NettyDataBufferFactory::toByteBuf) :
Flux.from(dataBuffers).map(NettyDataBufferFactory::toByteBuf);
}
}