sczx_group/sczx_gateway
4
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

再修改

Browse Source
This commit is contained in:
zhangli
2025-09-08 17:47:28 +08:00
parent 8b5648a3dc
commit 35b3cbfdc0
1 changed files with 9 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
src/main/java/com/sczx/gateway/filter/LoggingGlobalFilter.java
View File

@ -94,7 +94,7 @@ public class LoggingGlobalFilter implements GlobalFilter, Ordered {
@Override
public ServerHttpResponse getResponse() {
ServerHttpResponse originalResponse = super.getResponse();
// DataBufferFactory bufferFactory = originalResponse.bufferFactory();
DataBufferFactory bufferFactory = originalResponse.bufferFactory();
return new ServerHttpResponseDecorator(originalResponse) {
@Override
@ -102,14 +102,18 @@ public class LoggingGlobalFilter implements GlobalFilter, Ordered {
if (body instanceof Flux) {
Flux<? extends DataBuffer> flux = (Flux<? extends DataBuffer>) body;
return super.writeWith(flux.doOnNext(dataBuffer -> {
// 复制数据以避免修改原始缓冲区
DataBuffer duplicate = dataBuffer.factory().allocateBuffer(dataBuffer.readableByteCount());
duplicate.write(dataBuffer);
// 读取数据内容用于日志记录
byte[] bytes = new byte[dataBuffer.readableByteCount()];
dataBuffer.read(bytes);
byte[] bytes = new byte[duplicate.readableByteCount()];
duplicate.read(bytes);
String bodyContent = new String(bytes, StandardCharsets.UTF_8);
responseBodyBuilder.append(bodyContent);
// 重置读取位置，以便后续处理可以正常读取
dataBuffer.readPosition(0);
// 释放复制的缓冲区
DataBufferUtils.release(duplicate);
}));
}
return super.writeWith(body);