SpringCloud Gateway 网关的请求体body的读取和修改

本文涉及的产品
服务治理 MSE Sentinel/OpenSergo,Agent数量 不受限
可观测可视化 Grafana 版,10个用户账号 1个月
简介: SpringCloud Gateway 框架中,为了处理请求体body,实现多次读取与修改,创建了一个名为`RequestParamGlobalFilter`的全局过滤器。这个过滤器使用`@Component`和`@Slf4j`注解,实现了`GlobalFilter`和`Ordered`接口,设置最高优先级以首先读取body。它通过缓存请求体并创建装饰过的`ServerHttpRequest`来实现body的动态获取。

SpringCloud Gateway 网关的请求体body的读取和修改

getway需要多次对body 进行操作,需要对body 进行缓存

缓存body 动态多次获取

新建顶层filter,对body 进行缓存

 

import lombok.extern.slf4j.Slf4j;

import org.springframework.cloud.gateway.filter.GatewayFilterChain;

import org.springframework.cloud.gateway.filter.GlobalFilter;

import org.springframework.core.Ordered;

import org.springframework.core.io.buffer.DataBuffer;

import org.springframework.core.io.buffer.DataBufferUtils;

import org.springframework.core.io.buffer.DefaultDataBuffer;

import org.springframework.core.io.buffer.DefaultDataBufferFactory;

import org.springframework.http.HttpHeaders;

import org.springframework.http.MediaType;

import org.springframework.http.codec.HttpMessageReader;

import org.springframework.http.server.reactive.ServerHttpRequest;

import org.springframework.http.server.reactive.ServerHttpRequestDecorator;

import org.springframework.stereotype.Component;

import org.springframework.web.reactive.function.server.HandlerStrategies;

import org.springframework.web.reactive.function.server.ServerRequest;

import org.springframework.web.server.ServerWebExchange;

import reactor.core.publisher.Flux;

import reactor.core.publisher.Mono;

import java.util.List;

/**

* @author: zhoumo

* @descriptions:

*/

@Component

@Slf4j

public class RequestParamGlobalFilter implements GlobalFilter, Ordered {

   @Override

   public Mono<Void> filter(ServerWebExchange exchange, GatewayFilterChain chain) {

       /**

        * save request path and serviceId into gateway context

        */

       ServerHttpRequest request = exchange.getRequest();

       HttpHeaders headers = request.getHeaders();

       // 处理参数

       MediaType contentType = headers.getContentType();

       long contentLength = headers.getContentLength();

       if (contentLength > 0) {

                 return readBody(exchange, chain);

       }

       return chain.filter(exchange);

   }

   /**

    * default HttpMessageReader

    */

   private static final List<HttpMessageReader<?>> messageReaders = HandlerStrategies.withDefaults().messageReaders();

   /**

    * ReadJsonBody

    *

    * @param exchange

    * @param chain

    * @return

    */

   private Mono<Void> readBody(ServerWebExchange exchange, GatewayFilterChain chain) {

       /**

        * join the body

        */

       return DataBufferUtils.join(exchange.getRequest().getBody()).flatMap(dataBuffer -> {

           byte[] bytes = new byte[dataBuffer.readableByteCount()];

           dataBuffer.read(bytes);

           DataBufferUtils.release(dataBuffer);

           Flux<DataBuffer> cachedFlux = Flux.defer(() -> {

               DataBuffer buffer = exchange.getResponse().bufferFactory().wrap(bytes);

               DataBufferUtils.retain(buffer);

               return Mono.just(buffer);

           });

           /**

            * repackage ServerHttpRequest

            */

           ServerHttpRequest mutatedRequest = new ServerHttpRequestDecorator(exchange.getRequest()) {

               @Override

               public Flux<DataBuffer> getBody() {

                   return cachedFlux;

               }

           };

           /**

            * mutate exchage with new ServerHttpRequest

            */

           ServerWebExchange mutatedExchange = exchange.mutate().request(mutatedRequest).build();

           /**

            * read body string with default messageReaders

            */

           return ServerRequest.create(mutatedExchange, messageReaders).bodyToMono(String.class)

                   .doOnNext(objectValue -> {

                       log.debug("[GatewayContext]Read JsonBody:{}", objectValue);

                   }).then(chain.filter(mutatedExchange));

       });

   }

   @Override

   public int getOrder() {

       return HIGHEST_PRECEDENCE;

   }

}

在子节点层获取body




AtomicReference<String> requestBody = new AtomicReference<>("");                RecorderServerHttpRequestDecorator requestDecorator = new RecorderServerHttpRequestDecorator(request);                Flux<DataBuffer> body = requestDecorator.getBody();                body.subscribe(buffer -> {                    CharBuffer charBuffer = StandardCharsets.UTF_8.decode(buffer.asByteBuffer());                    requestBody.set(charBuffer.toString());                });                String body= requestBody.get();

重写获取body方法



  public class RecorderServerHttpRequestDecorator  extends ServerHttpRequestDecorator {        private final List<DataBuffer> dataBuffers = new ArrayList<>();        public RecorderServerHttpRequestDecorator(ServerHttpRequest delegate) {            super(delegate);            super.getBody().map(dataBuffer -> {                dataBuffers.add(dataBuffer);                return dataBuffer;            }).subscribe();        }        @Override        public Flux<DataBuffer> getBody() {            return copy();        }        private Flux<DataBuffer> copy() {            return Flux.fromIterable(dataBuffers)                    .map(buf -> buf.factory().wrap(buf.asByteBuffer()));        }    }


对body 进行修改重新封装



               String str=""+encodedDecryptedParam;                DataBuffer bodyDataBuffer = stringBuffer(str);                Flux<DataBuffer> bodyFlux = Flux.just(bodyDataBuffer);                MediaType contentType = request.getHeaders().getContentType();                ServerHttpRequest mutatedRequest = new ServerHttpRequestDecorator(                        exchange.getRequest()) {                    @Override                    public HttpHeaders getHeaders() {                        HttpHeaders httpHeaders = new HttpHeaders();                        int length = str.getBytes().length;                        httpHeaders.putAll(super.getHeaders());                        httpHeaders.remove(HttpHeaders.CONTENT_TYPE);                        httpHeaders.remove(HttpHeaders.CONTENT_LENGTH);                        httpHeaders.setContentLength(length);                        httpHeaders.set(HttpHeaders.CONTENT_TYPE, contentType.toString());                        // 设置CONTENT_TYPE                        return httpHeaders;                    }                    @Override                    public Flux<DataBuffer> getBody() {                        return bodyFlux;                    }                };                return chain.filter(exchange.mutate().request(mutatedRequest).build());



一定必须加上 public HttpHeaders getHeaders()对header 重新封装,否则接口层会卡死,request 无限大

目录
相关文章
|
1月前
|
算法 NoSQL API
SpringCloud&Gateway网关限流
SpringCloud&Gateway网关限流
82 7
|
1月前
|
负载均衡 Nacos 数据安全/隐私保护
SpringCloud GateWay 使用
SpringCloud GateWay 使用
31 0
|
1月前
|
负载均衡 Java 网络架构
在SpringCloud2023中快速集成SpringCloudGateway网关
本文主要简单介绍SpringCloud2023实战中SpringCoudGateway的搭建。后续的文章将会介绍在微服务中使用熔断Sentinel、鉴权OAuth2、SSO等技术。
66 2
在SpringCloud2023中快速集成SpringCloudGateway网关
|
5天前
|
Java API 开发者
Java一分钟之-Spring Cloud Gateway:API网关
【6月更文挑战第10天】Spring Cloud Gateway是Spring Cloud生态中的API网关组件,基于Spring Framework 5、Reactor和Spring Boot 2.0,支持响应式编程。它提供路由转发、过滤器链(包括预处理、路由和后处理)和断言功能。快速入门涉及添加相关依赖和配置路由规则。常见问题包括路由冲突、过滤器顺序和性能瓶颈。通过动态路由和过滤器示例,展示了其灵活性。Spring Cloud Gateway是微服务架构的有力工具,可提升系统稳定性和开发效率。
111 0
|
1月前
|
Java 微服务 Spring
SpringCloud&Gateway全局过滤器
SpringCloud&Gateway全局过滤器
16 1
|
1月前
|
监控 Java API
第七章 Spring Cloud 之 GateWay
第七章 Spring Cloud 之 GateWay
34 0
|
1月前
|
JSON 安全 关系型数据库
SpringCloud Gateway 实现自定义全局过滤器 + JWT权限验证
SpringCloud Gateway 实现自定义全局过滤器 + JWT权限验证
|
1月前
|
运维 网络协议 安全
长连接网关技术专题(十):百度基于Go的千万级统一长连接服务架构实践
本文将介绍百度基于golang实现的统一长连接服务,从统一长连接功能实现和性能优化等角度,描述了其在设计、开发和维护过程中面临的问题和挑战,并重点介绍了解决相关问题和挑战的方案和实践经验。
135 1
|
7月前
|
负载均衡 应用服务中间件 API
微服务技术系列教程(25) - SpringCloud- 接口网关服务Zuul
微服务技术系列教程(25) - SpringCloud- 接口网关服务Zuul
74 0
|
6月前
|
负载均衡 Cloud Native Java
【云原生】Spring Cloud Alibaba 之 Gateway 服务网关实战开发
【云原生】Spring Cloud Alibaba 之 Gateway 服务网关实战开发
678 0