如何使用spring webflux读取请求体

Question

我正在使用Spring5、Netty和Spring webflux开发和API网关。有时，我希望网关停止请求，但我也希望读取请求的主体以记录它，例如，并向客户机返回一个错误。

我尝试在WebFilter中通过订阅body来实现这一点。

@Override
public Mono<Void> filter(ServerWebExchange exchange, GatewayFilterChain chain) {
    if (enabled) {
        logger.debug("gateway is enabled. The Request is routed.");
        return chain.filter(exchange);
    } else {
        logger.debug("gateway is disabled. A 404 error is returned.");

        exchange.getRequest().getBody().subscribe();
        exchange.getResponse().setStatusCode(HttpStatus.NOT_FOUND);
        return exchange.getResponse().writeWith(Mono.just(exchange.getResponse().bufferFactory().allocateBuffer(0)));
    }
}

当我这样做的时候，它在身体的内容很少的时候起作用。但当我有一个大的身体时，只有第一个元素的通量被读取，所以我不能有整个身体。知道怎么做吗？

Answer 1

这里的问题是您在筛选器中手动订阅，这意味着您将请求的读取与管道的其余部分断开。调用subscribe()将为您提供一个disposable，帮助您管理基础的订阅。

因此您需要将整个过程转换为单个管道连接起来，有点像：

Flux<DataBuffer> requestBody = exchange.getRequest().getBody();
// decode the request body as a Mono or a Flux
Mono<String> decodedBody = decodeBody(requestBody); 
exchange.getResponse().setStatusCode(HttpStatus.NOT_FOUND);
return decodedBody.doOnNext(s -> logger.info(s))
                  .then(exchange.getResponse().setComplete());

请注意，将整个请求主体解码为mono意味着网关必须将整个请求主体缓冲在内存中。

databuffer是一种低级别类型。如果希望将其解码（即实现示例DecodeBody方法）为字符串，则可以使用Spring中的各种Decoder实现之一，如StringDecoder。

现在，由于这是一个相当大和复杂的空间，您可以使用和/或看一看Spring Cloud Gateway，它只做了这一点，而且做得更多。