|
13 | 13 | import io.opentelemetry.instrumentation.netty.common.internal.NettyErrorHolder; |
14 | 14 | import io.opentelemetry.javaagent.bootstrap.http.HttpServerResponseCustomizerHolder; |
15 | 15 | import io.opentelemetry.javaagent.instrumentation.netty.v3_8.NettyRequest; |
| 16 | +import javax.annotation.Nullable; |
16 | 17 | import org.jboss.netty.channel.Channel; |
17 | 18 | import org.jboss.netty.channel.ChannelHandlerContext; |
18 | 19 | import org.jboss.netty.channel.MessageEvent; |
@@ -41,10 +42,20 @@ public void writeRequested(ChannelHandlerContext ctx, MessageEvent msg) throws E |
41 | 42 | try (Scope ignored = context.makeCurrent()) { |
42 | 43 | super.writeRequested(ctx, msg); |
43 | 44 | } catch (Throwable t) { |
44 | | - instrumenter().end(context, request, response, NettyErrorHolder.getOrDefault(context, t)); |
| 45 | + end(ctx.getChannel(), context, request, response, t); |
45 | 46 | throw t; |
46 | 47 | } |
47 | | - instrumenter().end(context, request, response, NettyErrorHolder.getOrDefault(context, null)); |
| 48 | + end(ctx.getChannel(), context, request, response, null); |
| 49 | + } |
| 50 | + |
| 51 | + private static void end( |
| 52 | + Channel channel, |
| 53 | + Context context, |
| 54 | + NettyRequest request, |
| 55 | + HttpResponse response, |
| 56 | + @Nullable Throwable error) { |
| 57 | + requestAndContextField.set(channel, null); |
| 58 | + instrumenter().end(context, request, response, NettyErrorHolder.getOrDefault(context, error)); |
48 | 59 | } |
49 | 60 |
|
50 | 61 | private static void customizeResponse(Context context, HttpResponse response) { |
|
0 commit comments