本文介紹了WebClient-如何獲取請求正文?的處理方法,對大家解決問題具有一定的參考價值,需要的朋友們下面隨著小編來一起學習吧!
問題描述
我已經開始使用WebClient,正在添加請求/響應的日志記錄,并在構建WebClient時使用Filter方法:
WebClient.builder()
.baseUrl(properties.getEndpoint())
.filter((request, next) -> {
// logging
request.body()
})
.build();
我可以訪問url、http方法、標頭,但獲取原始請求正文時遇到問題,因為body()
請求的方法返回BodyInserter
(BodyInserter<?, ? super ClientHttpRequest> body()
。
如何將請求Body的BodyInserter
表示轉換為String
表示?或者,如何正確記錄整個請求/響應,同時還能夠對其中的潛在憑據進行散列?
推薦答案
您可以圍繞JSON編碼器創建自己的包裝器/代理類,并在序列化后的正文被發送到內部管道之前將其攔截。
此blog post顯示如何記錄WebClient請求和響應的JSON負載
具體地說,您將擴展Jackson2JsonEncoder
的encodeValue
方法(如果是流數據,則encodeValues
)。然后,您可以隨心所欲地處理這些數據,如日志記錄等。您甚至可以根據環境/配置文件有條件地執行此操作
此自定義日志編碼器可以在創建WebClient
時由編解碼器指定:
CustomBodyLoggingEncoder bodyLoggingEncoder = new CustomBodyLoggingEncoder();
WebClient.builder()
.codecs(clientDefaultCodecsConfigurer -> {
clientDefaultCodecsConfigurer.defaultCodecs().jackson2JsonEncoder(bodyLoggingEncoder);
clientDefaultCodecsConfigurer.defaultCodecs().jackson2JsonDecoder(new Jackson2JsonDecoder(new ObjectMapper(), MediaType.APPLICATION_JSON));
})
...
更新2020/7/3:
這里是一個匆忙的例子,應用了相同的原則,但對于解碼器:
public class LoggingJsonDecoder extends Jackson2JsonDecoder {
private final Consumer<byte[]> payloadConsumer;
public LoggingJsonEncoder(final Consumer<byte[]> payloadConsumer) {
this.payloadConsumer = payloadConsumer;
}
@Override
public Mono<Object> decodeToMono(final Publisher<DataBuffer> input, final ResolvableType elementType, final MimeType mimeType, final Map<String, Object> hints) {
// Buffer for bytes from each published DataBuffer
final ByteArrayOutputStream payload = new ByteArrayOutputStream();
// Augment the Flux, and intercept each group of bytes buffered
final Flux<DataBuffer> interceptor = Flux.from(input)
.doOnNext(buffer -> bufferBytes(payload, buffer))
.doOnComplete(() -> payloadConsumer.accept(payload.toByteArray()));
// Return the original method, giving our augmented Publisher
return super.decodeToMono(interceptor, elementType, mimeType, hints);
}
private void bufferBytes(final ByteArrayOutputStream bao, final DataBuffer buffer) {
try {
bao.write(ByteUtils.extractBytesAndReset(buffer));
} catch (IOException e) {
throw new RuntimeException(e);
}
}
}
您將使用WebClient
上的codecs
生成器方法將其與編碼器一起配置。
當然,上面的方法只有在您的數據被反序列化為Mono的情況下才有效。但如果需要,可以覆蓋其他方法。另外,我只是在那里標準輸出結果JSON,但是您可以傳遞一個Consumer<String>
或其他東西,讓解碼器將字符串發送到,或者只是從那里記錄;由您決定。
警告一句,在當前形式下,這將使您的內存使用量翻倍,因為它緩沖了整個響應。如果您可以立即將該字節數據發送到另一個進程/線程以寫入日志文件或某個輸出流(甚至是Flux),則可以避免在內存中緩沖整個有效負載。
這篇關于WebClient-如何獲取請求正文?的文章就介紹到這了,希望我們推薦的答案對大家有所幫助,