spring cloud gateway 网关,在请求过程中,如果请求方主动断开连接的话,网关不会执行重写response的方法。
@Override
public int getOrder() {
// -1 is response write filter, must be called before that
return NettyWriteResponseFilter.WRITE_RESPONSE_FILTER_ORDER - 1;
}
@Override
public Mono filter(ServerWebExchange exchange, GatewayFilterChain chain) {
// log.info("order:{}", getOrder());
GatewayContext gatewayContext = (GatewayContext) exchange.getAttributes().get(GatewayContext.CACHE_GATEWAY_CONTEXT);
ServerHttpResponse originalResponse = exchange.getResponse();
// HttpHeaders headers = originalResponse.getHeaders();
DataBufferFactory bufferFactory = originalResponse.bufferFactory();
ServerHttpResponseDecorator decoratedResponse = new ServerHttpResponseDecorator(originalResponse) {
@Override
public Mono writeWith(Publisher body) {
if (body instanceof Flux) {
Fluxextends DataBuffer> fluxBody = (Fluxextends DataBuffer>) body;
return super.writeWith(fluxBody.buffer().map(dataBuffer -> {
DataBufferFactory dataBufferFactory = new DefaultDataBufferFactory();
DataBuffer join = dataBufferFactory.join(dataBuffer);
byte[] content = new byte[join.readableByteCount()];
join.read(content);
//释放掉内存
DataBufferUtils.release(join);
String responseString = new String(content, StandardCharsets.UTF_8);
List strings = exchange.getResponse().getHeaders().get(HttpHeaders.CONTENT_ENCODING);
if (!CollectionUtils.isEmpty(strings) && strings.contains("gzip")) {
GZIPInputStream gzipInputStream = null;
try {
gzipInputStream = new GZIPInputStream(new ByteArrayInputStream(content), content.length);
StringWriter writer = new StringWriter();
IOUtils.copy(gzipInputStream, writer, "UTF-8");
responseString = writer.toString();
} catch (IOException e) {
e.printStackTrace();
}
}
// 此处为实际业务逻辑
return bufferFactory.wrap(bytes);
}));
}
// if body is not a flux. never got there.
return super.writeWith(body);
}
};
// replace response with decorator
return chain.filter(exchange.mutate().response(decoratedResponse).build());
}
业务逻辑可以正常运行,但是因为请求方主动断开连接之后,不执行writeWith,不能将response重写。
使用链路追踪之后,返现filter前后都是可以正常运行,排除了代码问题。
寻找网关有无其他配置项,没有找到,还在寻找。
打开了网关监控日志,没有发现什么指定的问题。
请求方主动断开连接之后还可以正常进入 writeWith方法。
关于该问题,我找了一篇非常好的博客,你可以看看是否有帮助,链接:Spring Cloud Gateway