SpringCloud GateWay 网关 在GlobalFilter 拿出返回数据response(1)

WrapperResponseGlobalFilter.java

import com.alibaba.fastjson.JSONObject;

import com.google.common.base.Joiner;

import com.google.common.base.Throwables;

import com.google.common.collect.Lists;

import org.apache.commons.lang3.StringUtils;

import org.reactivestreams.Publisher;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

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.DataBufferFactory;

import org.springframework.core.io.buffer.DataBufferUtils;

import org.springframework.http.HttpStatus;

import org.springframework.http.server.reactive.ServerHttpResponse;

import org.springframework.http.server.reactive.ServerHttpResponseDecorator;

import org.springframework.stereotype.Component;

import org.springframework.web.server.ServerWebExchange;

import reactor.core.publisher.Flux;

import reactor.core.publisher.Mono;

import java.nio.charset.Charset;

import java.util.List;

import static org.springframework.cloud.gateway.support.ServerWebExchangeUtils.ORIGINAL_RESPONSE_CONTENT_TYPE_ATTR;

/**

• @Author JCccc

• @Description 拦截返回数据, 修改返回数据

• @Date 2021/8/16 19:22

*/

@Component

public class WrapperResponseGlobalFilter implements GlobalFilter, Ordered {

private static final Logger log = LoggerFactory.getLogger(WrapperResponseGlobalFilter.class);

@Override

public int getOrder() {

// -1 is response write filter, must be called before that

return -2;

}

private static Joiner joiner = Joiner.on(“”);

@Override

public Mono filter(ServerWebExchange exchange, GatewayFilterChain chain) {

ServerHttpResponse originalResponse = exchange.getResponse();

DataBufferFactory bufferFactory = originalResponse.bufferFactory();

ServerHttpResponseDecorator response = new ServerHttpResponseDecorator(originalResponse) {

@Override

public Mono writeWith(Publisher<? extends DataBuffer> body) {

if (getStatusCode().equals(HttpStatus.OK) && body instanceof Flux) {

// 获取ContentType，判断是否返回JSON格式数据

String originalResponseContentType = exchange.getAttribute(ORIGINAL_RESPONSE_CONTENT_TYPE_ATTR);

if (StringUtils.isNotBlank(originalResponseContentType) && originalResponseContentType.contains(“application/json”)) {

Flux<? extends DataBuffer> fluxBody = Flux.from(body);

//（返回数据内如果字符串过大，默认会切割）解决返回体分段传输

return super.writeWith(fluxBody.buffer().map(dataBuffers -> {

List list = Lists.newArrayList();

dataBuffers.forEach(dataBuffer -> {

try {

byte[] content = new byte[dataBuffer.readableByteCount()];

dataBuffer.read(content);

DataBufferUtils.release(dataBuffer);

list.add(new String(content, “utf-8”));

} catch (Exception e) {

log.info(“加载Response字节流异常，失败原因：{}”, Throwables.getStackTraceAsString(e));

}

});

String responseData = joiner.join(list);

System.out.println(“responseData：”+responseData);

byte[] uppedContent = new String(responseData.getBytes(), Charset.forName(“UTF-8”)).getBytes();

originalResponse.getHeaders().setContentLength(uppedContent.length);

return bufferFactory.wrap(uppedContent);

}));

}

}

return super.writeWith(body);

}

@Override

public Mono writeAndFlushWith(Publisher<? extends Publisher<? extends DataBuffer>> body) {

return writeWith(Flux.from(body).flatMapSequential(p -> p));

}

};

return chain.filter(exchange.mutate().response(response).build());

}

}

调用接口可以看到，response数据已经被我们拎出来了：

那么我们简单做个处理，

/**

• 返回数据处理

• @param responseData
  自我介绍一下，小编13年上海交大毕业，曾经在小公司待过，也去过华为、OPPO等大厂，18年进入阿里一直到现在。

深知大多数Java工程师，想要提升技能，往往是自己摸索成长或者是报班学习，但对于培训机构动则几千的学费，着实压力不小。自己不成体系的自学效果低效又漫长，而且极易碰到天花板技术停滞不前！

因此收集整理了一份《2024年Java开发全套学习资料》，初衷也很简单，就是希望能够帮助到想自学提升又不知道该从何学起的朋友，同时减轻大家的负担。

既有适合小白学习的零基础资料，也有适合3年以上经验的小伙伴深入学习提升的进阶课程，基本涵盖了95%以上Java开发知识点，真正体系化！

由于文件比较大，这里只是将部分目录截图出来，每个节点里面都包含大厂面经、学习笔记、源码讲义、实战项目、讲解视频，并且会持续更新！

如果你觉得这些内容对你有帮助，可以扫码获取！！（备注Java获取）

最后

看完美团、字节、腾讯这三家的面试问题，是不是感觉问的特别多，可能咱们又得开启面试造火箭、工作拧螺丝的模式去准备下一次的面试了。

开篇有提及我可是足足背下了1000道题目，多少还是有点用的呢，我看了下，上面这些问题大部分都能从我背的题里找到的，所以今天给大家分享一下互联网工程师必备的面试1000题。

注意不论是我说的互联网面试1000题，还是后面提及的算法与数据结构、设计模式以及更多的Java学习笔记等，皆可分享给各位朋友

互联网工程师必备的面试1000题

而且从上面三家来看，算法与数据结构是必备不可少的呀，因此我建议大家可以去刷刷这本左程云大佬著作的《程序员代码面试指南 IT名企算法与数据结构题目最优解》，里面近200道真实出现过的经典代码面试题。

《互联网大厂面试真题解析、进阶开发核心学习笔记、全套讲解视频、实战项目源码讲义》点击传送门即可获取！
结构是必备不可少的呀，因此我建议大家可以去刷刷这本左程云大佬著作的《程序员代码面试指南 IT名企算法与数据结构题目最优解》，里面近200道真实出现过的经典代码面试题**。

[外链图片转存中…(img-g82AAnGE-1713416899610)]

《互联网大厂面试真题解析、进阶开发核心学习笔记、全套讲解视频、实战项目源码讲义》点击传送门即可获取！