问题遇到的现象和发生背景
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<Void> 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<Void> writeWith(Publisher<? extends DataBuffer> body) {
if (body instanceof Flux) {
Flux<? extends DataBuffer> fluxBody = (Flux<? extends 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<String> 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方法。