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

废话

这几天换了新工作，需要重新开发一套系统，技术选用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 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 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 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;
  }
}

以上这篇解决Spring Cloud Gateway获取body内容,不影响GET请求的操作就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持考高分网。