Skip to content

Commit 2056f71

Browse files
shichanglin5mr3
authored andcommitted
fix: netty v3 request recording may be incomplete due to lack of handling of httpChunk
1 parent 6fe1343 commit 2056f71

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

arex-instrumentation/netty/arex-netty-v3/src/main/java/io/arex/inst/netty/v3/server/RequestTracingHandler.java

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -47,6 +47,11 @@ public void messageReceived(ChannelHandlerContext ctx, MessageEvent event) throw
4747
// cache mocker for writeComplete and writeRequested, not use ctx.setAttachment(mocker), because it maybe used by user handler
4848
ContextManager.currentContext().setAttachment("arex-netty-server-mocker", mocker);
4949
}
50+
} else if (event.getMessage() instanceof HttpChunk) {
51+
Mocker mocker = (Mocker) ContextManager.currentContext().getAttachment("arex-netty-server-mocker");
52+
if (mocker != null) {
53+
setContent(((HttpChunk) event.getMessage()).getContent(), mocker);
54+
}
5055
}
5156
} catch (Throwable e) {
5257
LogManager.warn("netty messageReceived error", e);

0 commit comments

Comments
 (0)