From c5701bcdd29588a111e5642a48e2fe38eb1067bd Mon Sep 17 00:00:00 2001 From: Lauri Tulmin Date: Tue, 13 Apr 2021 02:03:05 +0300 Subject: [PATCH] Akka: clean up state on exception (#2782) --- .../akkaactor/AkkaDispatcherInstrumentation.java | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/instrumentation/akka-actor-2.5/javaagent/src/main/java/io/opentelemetry/javaagent/instrumentation/akkaactor/AkkaDispatcherInstrumentation.java b/instrumentation/akka-actor-2.5/javaagent/src/main/java/io/opentelemetry/javaagent/instrumentation/akkaactor/AkkaDispatcherInstrumentation.java index 85d72cec52..7336fe22e8 100644 --- a/instrumentation/akka-actor-2.5/javaagent/src/main/java/io/opentelemetry/javaagent/instrumentation/akkaactor/AkkaDispatcherInstrumentation.java +++ b/instrumentation/akka-actor-2.5/javaagent/src/main/java/io/opentelemetry/javaagent/instrumentation/akkaactor/AkkaDispatcherInstrumentation.java @@ -50,5 +50,10 @@ public class AkkaDispatcherInstrumentation implements TypeInstrumentation { } return null; } + + @Advice.OnMethodExit(onThrowable = Throwable.class, suppress = Throwable.class) + public static void exitDispatch(@Advice.Enter State state, @Advice.Thrown Throwable throwable) { + ExecutorInstrumentationUtils.cleanUpOnMethodExit(state, throwable); + } } }