文章详情

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

请输入下面的图形验证码

提交验证

短信预约提醒成功

Spring Cloud Gateway怎么修改HTTP响应信息

2023-06-20 16:45

关注

这篇文章主要讲解了“Spring Cloud Gateway怎么修改HTTP响应信息”,文中的讲解内容简单清晰,易于学习与理解,下面请大家跟着小编的思路慢慢深入,一起来研究和学习“Spring Cloud Gateway怎么修改HTTP响应信息”吧!

Gateway 修改HTTP响应信息

实践Spring Cloud的过程中,使用Gateway作为路由组件,并且基于Gateway实现权限的验证、拦截、过滤,对于下游微服务的响应结果,我们总会有需要修改以统一数据格式,或者修改过滤用户没有权限看到的数据信息,这时候就需要有一个能够修改响应体的Filter。

Spring Cloud Gateway 版本为2.1.0

在当前版本,ModifyRequestBodyGatewayFilterFactory是官方提供的修改响应体的参考类,This filter is BETA and may be subject to change in a future release.,类的注释中说明这个类在以后版本中会改进,实际使用可以参考实现功能,但是性能影响较大,不过没有别的选择还是得选择这个。

官方文档:

实现

最终代码

先贴最终代码

public class ResponseDecryptionGlobalFilter implements GlobalFilter, Ordered {    private static Logger log = LoggerFactory.getLogger(ResponseDecryptionGlobalFilter.class);    @Override    public int getOrder() {    // 控制在NettyWriteResponseFilter后执行        return NettyWriteResponseFilter.WRITE_RESPONSE_FILTER_ORDER - 1;    }    @Override    public Mono<Void> filter(ServerWebExchange exchange, GatewayFilterChain chain) {        return processResponse(exchange, chain);    }    private Mono<Void> processResponse(ServerWebExchange exchange, GatewayFilterChain chain) {        // 路由中如果不需要过滤则不进行过滤        if (!BooleanUtils.isTrue()) {            return chain.filter(exchange);        }        ServerHttpResponseDecorator responseDecorator = new ServerHttpResponseDecorator(exchange.getResponse()) {            @Override            public Mono<Void> writeWith(Publisher<? extends DataBuffer> body) {                String originalResponseContentType = exchange.getAttribute(ORIGINAL_RESPONSE_CONTENT_TYPE_ATTR);                HttpHeaders httpHeaders = new HttpHeaders();                httpHeaders.add(HttpHeaders.CONTENT_TYPE, originalResponseContentType);                ResponseAdapter responseAdapter = new ResponseAdapter(body, httpHeaders);                DefaultClientResponse clientResponse = new DefaultClientResponse(responseAdapter, ExchangeStrategies.withDefaults());                Mono<String> rawBody = clientResponse.bodyToMono(String.class).map(s -> s);                BodyInserter<Mono<String>, ReactiveHttpOutputMessage> bodyInserter = BodyInserters.fromPublisher(rawBody, String.class);                CachedBodyOutputMessage outputMessage = new CachedBodyOutputMessage(exchange, exchange.getResponse().getHeaders());                return bodyInserter.insert(outputMessage, new BodyInserterContext())                        .then(Mono.defer(() -> {                            Flux<DataBuffer> messageBody = outputMessage.getBody();                            Flux<DataBuffer> flux = messageBody.map(buffer -> {                                CharBuffer charBuffer = StandardCharsets.UTF_8.decode(buffer.asByteBuffer());                                DataBufferUtils.release(buffer);// 将响应信息转化为字符串                                String responseStr = charBuffer.toString();                                if (StringUtils.isNotBlank(responseStr)) {                                    try {                                        JSONObject result = JSONObject.parseObject(responseStr);                                        System.out.println(dataFilter(result));                                        if (result.containsKey("data")) {                                            responseStr = dataFilter(result);                                        } else {                                            log.error("响应结果序列化异常:{}", responseStr);                                        }                                    } catch (JSONException e) {                                        log.error("响应结果序列化异常:{}", responseStr);                                    }                                }                                return getDelegate().bufferFactory().wrap(responseStr.getBytes(StandardCharsets.UTF_8));                            });                            HttpHeaders headers = getDelegate().getHeaders();                            // 修改响应包的大小,不修改会因为包大小不同被浏览器丢掉                            flux = flux.doOnNext(data -> headers.setContentLength(data.readableByteCount()));                            return getDelegate().writeWith(flux);                        }));            }        };        return chain.filter(exchange.mutate().response(responseDecorator).build());    }        private String dataFilter(JSONObject result) {        Object data = result.get("data");        return result.toJSONString();    }    private class ResponseAdapter implements ClientHttpResponse {        private final Flux<DataBuffer> flux;        private final HttpHeaders headers;        @SuppressWarnings("unchecked")        private ResponseAdapter(Publisher<? extends DataBuffer> body, HttpHeaders headers) {            this.headers = headers;            if (body instanceof Flux) {                flux = (Flux) body;            } else {                flux = ((Mono) body).flux();            }        }        @Override        public Flux<DataBuffer> getBody() {            return flux;        }        @Override        public HttpHeaders getHeaders() {            return headers;        }        @Override        public HttpStatus getStatusCode() {            return null;        }        @Override        public int getRawStatusCode() {            return 0;        }        @Override        public MultiValueMap<String, ResponseCookie> getCookies() {            return null;        }    }}

踩过的坑

flux = flux.doOnNext(data -> headers.setContentLength(data.readableByteCount()));

理解ServerWebExchange

先看ServerWebExchange的注释:

Contract for an HTTP request-response interaction. Provides access to the HTTP request and response and also exposes additional server-side processing related properties and features such as request attributes.

翻译一下大概是:

ServerWebExchange是一个**HTTP请求-响应交互的契约。**提供对HTTP请求和响应的访问,并公开额外的服务器端处理相关属性和特性,如请求属性。

ServerWebExchange有点像Context的角色,我把它理解为http请求信息在Filter透传的容器,之所以称之为容器,因为它可以存储我们像放进去的数据。

注意:

ServerHttpRequest是一个只读类,因此需要通过下面例子的方法来进行修改,对于读多写少的场景,这种设计模式是值得借鉴的

ServerHttpRequest newRequest = request.mutate().headers("key","value").path("/myPath").build();ServerWebExchange newExchange = exchange.mutate().response(responseDecorator).build();

Gateway 修改返回的响应体

问题描述:

在gateway中修改返回的响应体,在全局Filter中添加如下代码:

import org.springframework.core.Ordered;import org.springframework.cloud.gateway.filter.GlobalFilter;import org.springframework.cloud.gateway.filter.GatewayFilterChain;import org.springframework.stereotype.Component;import org.springframework.web.server.ServerWebExchange;import reactor.core.publisher.Mono;@Componentpublic class RequestGlobalFilter implements GlobalFilter, Ordered { //...  @Override public Mono<Void> filter(ServerWebExchange exchange, GatewayFilterChain chain) {  //...  ResponseDecorator decorator = new ResponseDecorator(exchange.getResponse());  return chain.filter(exchange.mutate().response(decorator).build()); } @Override public int getOrder() {  return -1000; }}

通过.response(decorator)设置一个响应装饰器(自定义),以下是装饰器具体实现:

import cn.hutool.json.JSONObject;import org.reactivestreams.Publisher;import org.springframework.core.io.buffer.DataBuffer;import org.springframework.core.io.buffer.DataBufferFactory;import org.springframework.core.io.buffer.DataBufferUtils;import org.springframework.core.io.buffer.DefaultDataBufferFactory;import org.springframework.http.server.reactive.ServerHttpResponse;import org.springframework.http.server.reactive.ServerHttpResponseDecorator;import reactor.core.publisher.Flux;import reactor.core.publisher.Mono;import java.nio.charset.Charset;public class ResponseDecorator extends ServerHttpResponseDecorator{ public ResponseDecorator(ServerHttpResponse delegate){  super(delegate); } @Override @SuppressWarnings(value = "unchecked") public Mono<Void> writeWith(Publisher<? extends DataBuffer> body) {  if(body instanceof Flux) {   Flux<DataBuffer> fluxBody = (Flux<DataBuffer>) body;   return super.writeWith(fluxBody.buffer().map(dataBuffers -> {    DataBufferFactory dataBufferFactory = new DefaultDataBufferFactory();    DataBuffer join = dataBufferFactory.join(dataBuffers);    byte[] content = new byte[join.readableByteCount()];    join.read(content);    DataBufferUtils.release(join);// 释放掉内存        String bodyStr = new String(content, Charset.forName("UTF-8"));                //修改响应体    bodyStr = modifyBody(bodyStr);    getDelegate().getHeaders().setContentLength(bodyStr.getBytes().length);    return bufferFactory().wrap(bodyStr.getBytes());   }));  }  return super.writeWith(body); }    //重写这个函数即可 private String modifyBody(String jsonStr){  JSONObject json = new JSONObject(jsonStr);        //TODO...修改响应体  return json.toString(); }}

感谢各位的阅读,以上就是“Spring Cloud Gateway怎么修改HTTP响应信息”的内容了,经过本文的学习后,相信大家对Spring Cloud Gateway怎么修改HTTP响应信息这一问题有了更深刻的体会,具体使用情况还需要大家实践验证。这里是编程网,小编将为大家推送更多相关知识点的文章,欢迎关注!

阅读原文内容投诉

免责声明:

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

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

软考中级精品资料免费领

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

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

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

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

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

    难度     221人已做
    查看

相关文章

发现更多好内容

猜你喜欢

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