解决Spring Cloud Gateway获取body内容,不影响GET请求的操作
作者:seantdj
这篇文章主要介绍了解决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<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; } }
以上这篇解决Spring Cloud Gateway获取body内容,不影响GET请求的操作就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持脚本之家。