本文介紹了Java Web Flux:如何從ServerHttpResponse獲取正文?的處理方法,對大家解決問題具有一定的參考價值,需要的朋友們下面隨著小編來一起學習吧!
問題描述
我已實現自定義ServerHttpRequestDecorator來獲取請求正文,但無法對響應執行相同的操作。
@Override
public Mono<Void> filter(ServerWebExchange exchange, WebFilterChain chain)
{
long start = System.currentTimeMillis();
LogServerWebExchangeDecorator exchangeDecorator = new LogServerWebExchangeDecorator(
exchange);
exchangeDecorator.getResponse().beforeCommit(() ->
{
logRequest(exchangeDecorator.getRequest(), exchangeDecorator.getRequestBody());
logResponse(exchangeDecorator.getRequest(), exchangeDecorator.getResponse(), start);
return Mono.empty();
});
return chain.filter(exchangeDecorator);
}
如何實現同樣的目標?
推薦答案
private class LogServerHttpResponseDecorator extends ServerHttpResponseDecorator
{
private final FastByteArrayOutputStream fastByteArrayOutputStream;
public LogServerHttpResponseDecorator(ServerHttpResponse response, LoggingContext loggingContext)
{
super(response);
fastByteArrayOutputStream = new FastByteArrayOutputStream();
response.beforeCommit(LoggingContextUtils.withContext(loggingContext, (Supplier<Mono<Void>>) () ->
{
logResponse(byteArrayToString(fastByteArrayOutputStream.toByteArray(),
getHeaders()));
return Mono.empty();
}));
}
@Override
public Mono<Void> writeWith(Publisher<? extends DataBuffer> body)
{
return super.writeWith(Flux.from(body).map(buffer ->
{
readBuffer(fastByteArrayOutputStream, buffer);
return buffer;
}));
}
@Override
public Mono<Void> writeAndFlushWith(
Publisher<? extends Publisher<? extends DataBuffer>> body)
{
return super.writeAndFlushWith(Flux.from(body).map(publisher ->
Flux.from(publisher).map(buffer ->
{
readBuffer(fastByteArrayOutputStream, buffer);
return buffer;
})));
}
}
private void readBuffer(FastByteArrayOutputStream fastByteArrayOutputStream, DataBuffer buffer)
{
try
{
Channels.newChannel(fastByteArrayOutputStream)
.write(buffer.asByteBuffer().asReadOnlyBuffer());
}
catch (IOException e)
{
if (LoggingUtils.isLoggable(LOGGER, Level.ERROR))
{
LoggingUtils.log(LOGGER, Level.ERROR, null, "Unable to log", e);
}
}
}
這篇關于Java Web Flux:如何從ServerHttpResponse獲取正文?的文章就介紹到這了,希望我們推薦的答案對大家有所幫助,