526互联

提供一个SpringCloud Gateway获取body参数的方法

发布时间 2023-04-03 15:48:56作者: 三人成虎

在Spring Cloud Gateway中获取请求体（body）参数的方法与在普通的Spring应用程序中略有不同。Spring Cloud Gateway使用org.springframework.cloud.gateway.filter.factory.rewrite.ModifyRequestBodyGatewayFilterFactory过滤器工厂来修改请求体。下面是一个获取请求体参数的例子：

@Component
public class CustomGlobalFilter implements GlobalFilter, Ordered {

    @Override
    public Mono<Void> filter(ServerWebExchange exchange, GatewayFilterChain chain) {
        ServerHttpRequest request = exchange.getRequest();
        HttpHeaders headers = request.getHeaders();
        HttpMethod method = request.getMethod();

        // 检查请求是否包含请求体
        boolean hasBody = headers.getContentLength() > 0 || headers.getContentType() != null;

        if (hasBody) {
            // 从请求体中获取参数
            return DataBufferUtils.join(exchange.getRequest().getBody())
                    .flatMap(dataBuffer -> {
                        byte[] bytes = new byte[dataBuffer.readableByteCount()];
                        dataBuffer.read(bytes);
                        String body = new String(bytes, Charset.forName("UTF-8"));
                        // 此处可以对请求体进行处理
                        exchange.getAttributes().put("cachedRequestBodyObject", body);
                        return chain.filter(exchange.mutate().request(request.mutate()
                                .header("Content-Type", "application/json")
                                .body(Mono.just(DataBufferUtils.createByteBuffer(body.getBytes())))).build());
                    });
        }
        return chain.filter(exchange);
    }

    @Override
    public int getOrder() {
        return -1;
    }
}

这个例子中，我们首先检查请求是否包含请求体，如果包含，则从请求体中获取参数。然后，我们可以对请求体进行处理（例如，解析JSON），并将参数缓存到cachedRequestBodyObject属性中。最后，我们使用修改请求体的过滤器ModifyRequestBodyGatewayFilterFactory来更新请求体，并继续执行过滤器链。