SpringWebFlux 中 WebClient 怎么打印日志？

一、背景

去年高峰压测的时候，有个服务是专门调用其它系统的，在测试接口 http 请求的时候，那 TPS 唰唰的往下掉，还专门用 Arthas 看了一下方法执行时间，那家伙，我 sleep 2s，看时间都 3-4s 了，所以就想着后面优化一下。

所以，这不就到我们的主角SpringWebflux了，就想着用服务用 Webflux，Http 请求直接用 WebClient 了。但是有一个问题就不得不思考了，怎么记录请求的日志呢？这个日志很重要，必须的记录的清楚，不然不用系统间沟通(扯皮背锅)着实难搞。

二、怎么记录日志？

2.1 思考记录

打印日志，那第一步得去看看官网咯？看看官网提供了什么解决方案

2.1.1 官网寻答案

打开官网，翻到Filter这一页，发现着实有可以记录日志的，我把官网代码粘贴在下图。

WebClient client = WebClient.builder()        .filter((request, next) -> {            ClientRequest filtered = ClientRequest.from(request)                    .header("foo", "bar")                    .build();
            return next.exchange(filtered);        })        .build();

可以看到，我们是可以拿到 Request 的，这让我一顿高兴，那这操作起来不简单？ 但是呢，凡事不能高兴太早。

拿到这 ClientRequest 去代码一顿敲(clientRequest.xx)，这个点都被我按烂了，发现并没有可以获取 RequestBody 的方法，只有 Headers，URI... 可惜这都不是我要的啊。没办法，那就只有换别的方法了。

顺便点了一下 filter 的ExchangeFilterFunction，发现里面有方法还能拿到 Rresponse。代码如下：

static ExchangeFilterFunction ofResponseProcessor(Function<ClientResponse,                                                   Mono<ClientResponse>> processor) {  Assert.notNull(processor, "ClientResponse Function must not be null");  return (request, next) -> next.exchange(request).flatMap(processor);}

然后拿着 Response 又来一顿操作，可以拿到 Response Body 等，但是...但是...如果我们在 Response 里面把 Response Body 使用掉，就会报错:nested exception is java.lang.IllegalStateException: The client response body can only be consumed once。如图所示。

提前消费RespnseBody

没有办法，查看了 ClientResponse 实现类org.springframework.web.reactive.function.client.DefaultClientResponse，这个是个包权限的类，虽然很想直接用，比如：requestDescription（URL 路径），getBody, 但是是在拿不到啊~~难受。既然如此，那就只有从别的地方寻找。

2.1.2 其它方案

在官网上 WebFlux 有三种方案，分别是集成 Jetty, Netty, HttpComponent5。由于我使用的是 HttpComponent5，所以就直接从 http 这个寻找方案了。

寻寻觅觅，找到了了相似的方案，也可以添加 Interceptor。我就直接贴出代码，如下图。

HttpComponent添加拦截器

但是在实现时，发现了有两个问题。

问题一

打印 Request 日志的时候会出现两次。

问题二

打印 Response 的时候也会出现nested exception is java.lang.IllegalStateException: The client response body can only be consumed once这个异常。

出现这个问题，没有发现啥好用的解决方案。

2.1.3 Google

后来就想着想换成 Jetty 和 Netty，就 google 了一波，我直接把连接贴出来，就不过多描述了。 连接如下：

Logging Spring WebClient Calls | Baeldung 这个方案也尝试了一波，但是效果如上。会出现问题一，或者在获取 Body 的时候出现问题。

2.2 解决方案

在经历了好几天 Debug 和测试后，就选择了妥协方案，直接在 webClient 请求的时候打印日志。这还算是一个完美的解决方案。

我贴出一部分代码，如下。

    /**     * 打印全流程日志，需要外部传入参数     */    public static Function<ClientResponse, Mono<String>> logging(String url, String method, Object reqBody) {        return (clientResponse -> clientResponse.bodyToMono(String.class).doOnSuccess(body -> {            if (log.isDebugEnabled()) {                log.info("\n" +                                "TraceId      : {}, {}\n" +                                "URI          : {}, \n" +                                "Param        : {}, \n" +                                "RespHeader   : {}, \n" +                                "RespStatus   : {}, \n" +                                "Response     : {}", clientResponse.logPrefix(), method, url, JsonUtils.toJson(reqBody),                        clientResponse.headers().asHttpHeaders(), clientResponse.rawStatusCode(),                        StringUtils.abbreviate(body, 4000));            }        }));    }

在获取到 Body 的时候，直接转化成String.class，如果直接转成ParameterizedTypeReference就会失去部分原请求的数据，这样在(扯皮)查问题的时候不好找证据呀！（我们打印清晰的日志是为了更好的查询问题，不是扯皮︿(￣︶￣)︿ ）

这样打印出来的日志就是我们想要的了，舒服啊~ 这样看起来才舒服。

还写了一些其它的小工具，获取连接，使用方式如下：

小工具使用方式

三、总结

以上是探索 WebClient 的打印日志的方式，还有不足，欢迎大家讨论，提出更好的方式。谢谢！