文章详情

短信预约-IT技能 免费直播动态提醒

请输入下面的图形验证码

提交验证

短信预约提醒成功

SpringCloud gateway request的body验证或修改方式是什么

2023-06-20 16:27

关注

这篇文章主要介绍“SpringCloud gateway request的body验证或修改方式是什么”,在日常操作中,相信很多人在SpringCloud gateway request的body验证或修改方式是什么问题上存在疑惑,小编查阅了各式资料,整理出简单好用的操作方法,希望对大家解答”SpringCloud gateway request的body验证或修改方式是什么”的疑惑有所帮助!接下来,请跟着小编一起来学习吧!

SpringCloud gateway request的body验证或修改

后续版本新增了以下过滤器

org.springframework.cloud.gateway.filter.headers.RemoveHopByHopHeadersFilter

默认会把以下头部移除(暂不了解这做法的目的)

- connection

- keep-alive

- te

- transfer-encoding

- trailer

- proxy-authorization

- proxy-authenticate

- x-application-context

- upgrade

从而导致下面我们重写getHeaders方法时添加的transfer-encoding头部移除,导致无法解析body。

解决办法:

在yml文件中配置自定义的头部移除列表

spring:  cloud:      filter:        remove-hop-by-hop:          headers:            - connection            - keep-alive            - te            - trailer            - proxy-authorization            - proxy-authenticate            - x-application-context            - upgrade

源码可见链接,且可实现动态路由配置:https://github.com/SingleTigger/SpringCloudGateway-Nacos-Demo

------------原文------------

往往业务中我们需要在网关对请求参数作修改操作(注意以下只针对带有body的请求),springcloud gateway中有提供一个

ModifyRequestBodyGatewayFilterFactory的filter,看了一下它的实现,需要指定输入类型和输出类型,比较局限。

我就参考它自己实现了一个拦截器

注意:上传文件也带有请求body,需特殊处理。

以下是主要代码

 import org.springframework.cloud.gateway.filter.GatewayFilter;import org.springframework.cloud.gateway.filter.factory.AbstractGatewayFilterFactory;import org.springframework.cloud.gateway.support.BodyInserterContext;import org.springframework.core.io.buffer.DataBuffer;import org.springframework.http.HttpHeaders;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.BodyInserter;import org.springframework.web.reactive.function.BodyInserters;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;import java.util.concurrent.atomic.AtomicReference;import java.util.function.BiFunction; @Componentpublic class CModifyRequestBodyGatewayFilterFactory extends AbstractGatewayFilterFactory {     private final List<HttpMessageReader<?>> messageReaders;     public CModifyRequestBodyGatewayFilterFactory() {        this.messageReaders = HandlerStrategies.withDefaults().messageReaders();    }     @Override    @SuppressWarnings("unchecked")    public GatewayFilter apply(Object config) {        return (exchange, chain) -> {            ServerRequest serverRequest = ServerRequest.create(exchange,                    this.messageReaders);             Mono<String> modifiedBody = serverRequest.bodyToMono(String.class)                    .flatMap(originalBody -> modifyBody()                            .apply(exchange,Mono.just(originalBody)));             BodyInserter bodyInserter = BodyInserters.fromPublisher(modifiedBody,                    String.class);            HttpHeaders headers = new HttpHeaders();            headers.putAll(exchange.getRequest().getHeaders());            headers.remove(HttpHeaders.CONTENT_LENGTH);             CachedBodyOutputMessage outputMessage = new CachedBodyOutputMessage(exchange,                    headers);            return bodyInserter.insert(outputMessage, new BodyInserterContext())                    .then(Mono.defer(() -> {                        ServerHttpRequest decorator = decorate(exchange, headers,                                outputMessage);                        return chain.filter(exchange.mutate().request(decorator).build());                    }));         };    }         private BiFunction<ServerWebExchange,Mono<String>,Mono<String>> modifyBody(){        return (exchange,json)-> {            AtomicReference<String> result = new AtomicReference<>();            json.subscribe(                    value -> {                        //value 即为请求body,在此处修改                        result.set(value);                        System.out.println(result.get());                    },                    Throwable::printStackTrace            );            return Mono.just(result.get());        };    }     private ServerHttpRequestDecorator decorate(ServerWebExchange exchange, HttpHeaders headers,                                        CachedBodyOutputMessage outputMessage) {        return new ServerHttpRequestDecorator(exchange.getRequest()) {            @Override            public HttpHeaders getHeaders() {                long contentLength = headers.getContentLength();                HttpHeaders httpHeaders = new HttpHeaders();                httpHeaders.putAll(super.getHeaders());                if (contentLength > 0) {                    httpHeaders.setContentLength(contentLength);                }                else {                    httpHeaders.set(HttpHeaders.TRANSFER_ENCODING, "chunked");                }                return httpHeaders;            }             @Override            public Flux<DataBuffer> getBody() {                return outputMessage.getBody();            }        };    }}

SpringCloud Gateway获取post请求体(request body)不完整解决方案

Spring Cloud Gateway做为网关服务,通过gateway进行请求转发,在请求到达后端服务前我们可以通过filter进行一些预处理如:请求的合法性,商户验证等。

如我们在请求体中添加商户ID(merId)和商户KEY(merkey),通过此来验证请求的合法性。但是如果我们请求内容太长如转为base64的文件存储请求。此时我们在filter获取body内容就会被截取(太长的 Body 会被截断)。目前网上也没有好的解决方式。

springboot及Cloud版本如下;


版本
springboot2.0.8.RELEASE
springcloudFinchley.SR2

这里提供一种解决方式,相关代码如下:

1.Requestfilter

我们采用Gateway网关的Gobalfilter,建立我们的第一个过滤器过滤所有请求。

1).通过Spring 5 的 WebFlux我们使用bodyToMono方法把响应内容转换成类 String的对象,最终得到的结果是 Mono对象

2).bodyToMono方法我们可以拿到完整的body内容,并返回String。

3).我们生成唯一的token(通过UUID),并将token放入请求的header中。

4).将获取到的完整body内容,存放到redis中。

@Componentpublic class RequestFilter implements GlobalFilter, Ordered {@Autowiredprivate RedisClientTemplate redisClientTemplate;@Overridepublic Mono<Void> filter(ServerWebExchange exchange, GatewayFilterChain chain) {DefaultServerRequest req = new DefaultServerRequest( exchange );String token = UUID.randomUUID().toString();//向headers中放入token信息ServerHttpRequest serverHttpRequest =exchange.getRequest().mutate().header("token", token).build();//将现在的request变成change对象ServerWebExchange build = exchange.mutate().request( serverHttpRequest ).build();return req.bodyToMono( String.class ).map( str -> {redisClientTemplate.setObjex( "microservice:gateway:".concat( token ), 180, str );MySlf4j.textInfo( "请求参数:{0}", str );return str;} ).then( chain.filter( build ) );}@Overridepublic int getOrder() {return 0;}}
2.MerchantAuthFilter

建立商户认证过滤器,相关代码如下:

1).获取存储在headers中的token。

2).通过token获取我们存储在redis中的body内容(WebFlux 中不能使用阻塞的操作,目前想到的是通过这种方式实现)。

3).获取到完整的body内容后我们就可以进行相应的商户认证操作。

4).认证通过,将信息重新写入,不通过则返回异常信息。

@Overridepublic Mono<Void> filter(ServerWebExchange exchange, GatewayFilterChain chain) {ServerHttpRequest serverHttpRequest = exchange.getRequest();String token = serverHttpRequest.getHeaders().get( "token" ).get( 0 );       String bodyStr = (String) redisClientTemplate.getObj("microservice:gateway:".concat(token));BaseReqVo baseReqVo = JsonUtil.fromJson( bodyStr, BaseReqVo.class );try {// 商户认证BaseRespVo<?> baseRespVo = merchantAuthService.checkMerchantAuth( baseReqVo );if (MicroserviceConstantParamUtils.RESULT_CODE_SUCC.equals( baseRespVo.getCode() )) {               // 若验证成功,将信息重新写入避免request信息消费后后续无法从request获取信息的问题               URI uri = serverHttpRequest.getURI();               ServerHttpRequest request = serverHttpRequest.mutate().uri(uri).build();               DataBuffer bodyDataBuffer = stringBuffer(bodyStr);               Flux<DataBuffer> bodyFlux = Flux.just(bodyDataBuffer);               request = new ServerHttpRequestDecorator(request) {                   @Override                   public Flux<DataBuffer> getBody() {                       return bodyFlux;                   }               };// 封装request,传给下一级               return chain.filter(exchange.mutate().request(request).build());} else {// 若验证不成功,返回提示信息return gatewayResponse( baseRespVo.getCode(), baseRespVo.getMessage(), exchange );}} catch (MicroserviceServiceException ex) {// 若验证不成功,返回提示信息MySlf4j.textError( "商户访问权限验证异常,异常代码:{0},异常信息:{1}, 异常{2}", ex.getCode(), ex.getMessage(), ex );return gatewayResponse( ex.getCode(), ex.getMessage(), exchange );} catch (Exception ex) {MySlf4j.textError( "商户访问权限验证服务异常:{0}", LogUtil.ExceptionToString( ex ) );return gatewayResponse( MicroserviceException.ERR_100000, "系统异常", exchange );} finally {redisClientTemplate.del( "microservice:gateway:".concat( token ) );}}private DataBuffer stringBuffer(String value) {byte[] bytes = value.getBytes( StandardCharsets.UTF_8 );NettyDataBufferFactory nettyDataBufferFactory = new NettyDataBufferFactory( ByteBufAllocator.DEFAULT );DataBuffer buffer = nettyDataBufferFactory.allocateBuffer( bytes.length );buffer.write( bytes );return buffer;}private Mono<Void> gatewayResponse(String code, String message, ServerWebExchange exchange) {// 若验证不成功,返回提示信息ServerHttpResponse response = exchange.getResponse();BaseRespVo<T> baseRespVo = ResponseUtils.responseMsg( code, message, null );byte[] bits = JsonUtil.toJson( baseRespVo ).getBytes( StandardCharsets.UTF_8 );DataBuffer buffer = response.bufferFactory().wrap( bits );response.setStatusCode( HttpStatus.UNAUTHORIZED );// 指定编码,否则在浏览器中会中文乱码response.getHeaders().add( "Content-Type", "text/plain;charset=UTF-8" );return response.writeWith( Mono.just( buffer ) );}@Overridepublic int getOrder() {return 1;}

另外我们还可以通过GlobalFilter实现请求过滤,OAUTH授权,相关代码如下:

请求方式验证过滤器(RequestAuthFilter):

@Overridepublic Mono<Void> filter(ServerWebExchange exchange, GatewayFilterChain chain) { ServerHttpRequest serverHttpRequest = exchange.getRequest(); String method = serverHttpRequest.getMethodValue(); if (!"POST".equals(method)) {  ServerHttpResponse response = exchange.getResponse();  BaseRespVo<T> baseRespVo = ResponseUtils.responseMsg(MicroserviceException.ERR_100008, "非法请求", null);  byte[] bits = JsonUtil.toJson(baseRespVo).getBytes(StandardCharsets.UTF_8);  DataBuffer buffer = response.bufferFactory().wrap(bits);  response.setStatusCode(HttpStatus.UNAUTHORIZED);  //指定编码,否则在浏览器中会中文乱码  response.getHeaders().add("Content-Type", "text/plain;charset=UTF-8");  return response.writeWith(Mono.just(buffer)); } return chain.filter(exchange); }

OAUTH授权过滤器(OAuthSignatureFilter):

@Value("${spring.security.user.name}")private String securityUserName;@Value("${spring.security.user.password}")private String securityUserPassword;@Overridepublic Mono<Void> filter(ServerWebExchange exchange, GatewayFilterChain chain) {  String auth = securityUserName.concat(":").concat(securityUserPassword); String encodedAuth = new sun.misc.BASE64Encoder().encode(auth.getBytes(Charset.forName("US-ASCII"))); String authHeader = "Basic " + encodedAuth; //向headers中放授权信息 ServerHttpRequest serverHttpRequest = exchange.getRequest().mutate().header("Authorization", authHeader)   .build(); //将现在的request变成change对象 ServerWebExchange build = exchange.mutate().request(serverHttpRequest).build(); return chain.filter(build);}

到此,关于“SpringCloud gateway request的body验证或修改方式是什么”的学习就结束了,希望能够解决大家的疑惑。理论与实践的搭配能更好的帮助大家学习,快去试试吧!若想继续学习更多相关知识,请继续关注编程网网站,小编会继续努力为大家带来更多实用的文章!

阅读原文内容投诉

免责声明:

① 本站未注明“稿件来源”的信息均来自网络整理。其文字、图片和音视频稿件的所属权归原作者所有。本站收集整理出于非商业性的教育和科研之目的,并不意味着本站赞同其观点或证实其内容的真实性。仅作为临时的测试数据,供内部测试之用。本站并未授权任何人以任何方式主动获取本站任何信息。

② 本站未注明“稿件来源”的临时测试数据将在测试完成后最终做删除处理。有问题或投稿请发送至: 邮箱/279061341@qq.com QQ/279061341

软考中级精品资料免费领

  • 历年真题答案解析
  • 备考技巧名师总结
  • 高频考点精准押题
  • 2024年上半年信息系统项目管理师第二批次真题及答案解析(完整版)

    难度     813人已做
    查看
  • 【考后总结】2024年5月26日信息系统项目管理师第2批次考情分析

    难度     354人已做
    查看
  • 【考后总结】2024年5月25日信息系统项目管理师第1批次考情分析

    难度     318人已做
    查看
  • 2024年上半年软考高项第一、二批次真题考点汇总(完整版)

    难度     435人已做
    查看
  • 2024年上半年系统架构设计师考试综合知识真题

    难度     224人已做
    查看

相关文章

发现更多好内容

猜你喜欢

AI推送时光机
位置:首页-资讯-后端开发
咦!没有更多了?去看看其它编程学习网 内容吧
首页课程
资料下载
问答资讯