欢迎来到代码驿站!

JAVA代码

当前位置:首页 > 软件编程 > JAVA代码

解决Spring Cloud Gateway获取body内容,不影响GET请求的操作

时间:2021-03-05 12:40:49|栏目:JAVA代码|点击:

废话

这几天换了新工作,需要重新开发一套系统,技术选用Spring Cloud。在对接终端接口的时候要做验签,就涉及到在网关做拦截器,然后取出BODY里面的数据。

网上找了几个方法,有的拿不到数据,有的拿到数据之后不支持GET请求了。没有一个合理的解决办法,最后想到在动态路由构建的时候可以指定METHOD,于是有了如下解决办法

解决

@Bean
  public RouteLocator vmRouteLocator(RouteLocatorBuilder builder) {
    return builder.routes()
        .route(r -> r.method(HttpMethod.POST).and()
            .readBody(Object.class, requestBody -> {
              //相当于缓存了body信息,在filter 中可以这么获取 exchange.getAttribute("cachedRequestBodyObject");
              log.info("requestBody is {}", requestBody);
              return true;
            })
            .and().path("/terminal/**")
            .filters(f -> f.filter(terminalSignFilter()))
            .uri("lb://TERMINAL-SERVICE")
            .order(0)
            .id("terminal-service")
        )
        .route(r -> r.method(HttpMethod.GET).and()
            .path("/terminal/**")
            .filters(f -> f.filter(terminalSignFilter()))
            .uri("lb://TERMINAL-SERVICE")
            .order(1)
            .id("terminal-service")
        )
        .build();
  }

关键代码:

r.method(HttpMethod.POST)

r.method(HttpMethod.GET)

分别指定了不同请求METHOD对应的路由策略

在POST请求中需要缓存BODY信息,在Filter中便可以获取到

GET请求因为没有BODY,所以如果不指定GET的路由便会报错

可能会有更通用的方法,但是目前只想到这么多,以后有好的解决办法会继续更新

补充知识:Spring Cloud Gateway 2.x 获取body中的数据并缓存在请求中

场景

因为http请求中的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.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;

@Component
@Slf4j
public class CacheBodyParamsFilter implements GlobalFilter, Ordered {

  @Override
  public Mono<Void> filter(ServerWebExchange exchange, GatewayFilterChain chain) {
    if (ParamsUtil.logBody(exchange)) {
      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);
            });
            ServerHttpRequest mutatedRequest = new ServerHttpRequestDecorator(exchange.getRequest()) {
              @Override
              public Flux<DataBuffer> getBody() {
                return cachedFlux;
              }
            };
            ServerWebExchange mutatedExchange = exchange.mutate().request(mutatedRequest).build();
            return ServerRequest.create(mutatedExchange, HandlerStrategies.withDefaults().messageReaders())
                .bodyToMono(String.class)
                .doOnNext(objectValue -> {
                  //在此处,将body中的params值获取到,并存放在本次请求的attributes属性中,这样就可以在本次请求中的所有地方进行使用了              

                  mutatedExchange.getAttributes().put(CommonConstant.PARAMS, ParamsUtil.buildParams(mutatedRequest,objectValue));
                                  }).then(chain.filter(mutatedExchange));
          });
    }
    return chain.filter(exchange);
  }

  @Override
  public int getOrder() {
    return Ordered.HIGHEST_PRECEDENCE;
  }
}

上一篇:Java wait和notify虚假唤醒原理

栏    目:JAVA代码

下一篇:Mapper批量插入Oracle数据@InsertProvider注解

本文标题:解决Spring Cloud Gateway获取body内容,不影响GET请求的操作

本文地址:http://www.codeinn.net/misctech/74522.html

推荐教程

广告投放 | 联系我们 | 版权申明

重要申明:本站所有的文章、图片、评论等,均由网友发表或上传并维护或收集自网络,属个人行为,与本站立场无关。

如果侵犯了您的权利,请与我们联系,我们将在24小时内进行处理、任何非本站因素导致的法律后果,本站均不负任何责任。

联系QQ:914707363 | 邮箱:codeinn#126.com(#换成@)

Copyright © 2020 代码驿站 版权所有