Move wrapRunnable to Tracer and avoid trace ref recycling

elastic · Dec 18, 2018 · 4457d4e · 4457d4e
1 parent 159f368
commit 4457d4e
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 19 deletions.
diff --git a/apm-agent-core/src/main/java/co/elastic/apm/agent/impl/ElasticApmTracer.java b/apm-agent-core/src/main/java/co/elastic/apm/agent/impl/ElasticApmTracer.java
@@ -304,8 +304,8 @@ public void recycle(ErrorCapture error) {
         errorPool.recycle(error);
     }
 
-    public InScopeRunnableWrapper allocateRunnableWrapper() {
-        return runnableWrapperObjectPool.createInstance();
+    public Runnable wrapRunnable(Runnable delegate, AbstractSpan<?> span) {
+        return runnableWrapperObjectPool.createInstance().wrap(delegate, span);
     }
 
     public void recycle(InScopeRunnableWrapper wrapper) {

diff --git a/apm-agent-core/src/main/java/co/elastic/apm/agent/impl/InScopeRunnableWrapper.java b/apm-agent-core/src/main/java/co/elastic/apm/agent/impl/InScopeRunnableWrapper.java
@@ -36,19 +36,15 @@ public class InScopeRunnableWrapper implements Runnable, Recyclable {
     private Runnable delegate;
     @Nullable
     private AbstractSpan<?> span;
-    @Nullable
-    private ElasticApmTracer tracer;
+
+    private final ElasticApmTracer tracer;
 
     InScopeRunnableWrapper(ElasticApmTracer tracer) {
         this.tracer = tracer;
     }
 
-    public InScopeRunnableWrapper wrap(Runnable delegate) {
+    public InScopeRunnableWrapper wrap(Runnable delegate, AbstractSpan<?> span) {
         this.delegate = delegate;
-        return this;
-    }
-
-    public InScopeRunnableWrapper withScope(AbstractSpan<?> span) {
         this.span = span;
         return this;
     }
@@ -75,7 +71,6 @@ public void run() {
                 if (span != null) {
                     span.deactivate();
                 }
-                //noinspection ConstantConditions
                 tracer.recycle(this);
             } catch (Throwable throwable) {
                 try {
@@ -91,6 +86,5 @@ public void run() {
     public void resetState() {
         delegate = null;
         span = null;
-        tracer = null;
     }
 }
diff --git a/...s/apm-servlet-plugin/src/main/java/co/elastic/apm/agent/servlet/AsyncInstrumentation.java b/...s/apm-servlet-plugin/src/main/java/co/elastic/apm/agent/servlet/AsyncInstrumentation.java
@@ -81,8 +81,6 @@ public Collection<String> getInstrumentationGroupNames() {
 
     public interface AsyncContextAdviceHelper<T> {
         void onExitStartAsync(T asyncContext);
-
-        Runnable wrapRunnable(Runnable delegate, Transaction transaction);
     }
 
     public static class StartAsyncInstrumentation extends AsyncInstrumentation {
@@ -174,7 +172,7 @@ private static void onEnterAsyncContextStart(@Advice.Argument(value = 0, readOnl
                 if (tracer != null) {
                     final Transaction transaction = tracer.currentTransaction();
                     if (transaction != null && runnable != null && asyncHelper != null) {
-                        runnable = asyncHelper.getForClassLoaderOfClass(AsyncContext.class).wrapRunnable(runnable, transaction);
+                        runnable = tracer.wrapRunnable(runnable, transaction);
                     }
                 }
             }

diff --git a/...lugin/src/main/java/co/elastic/apm/agent/servlet/helper/AsyncContextAdviceHelperImpl.java b/...lugin/src/main/java/co/elastic/apm/agent/servlet/helper/AsyncContextAdviceHelperImpl.java
@@ -60,9 +60,4 @@ public void onExitStartAsync(AsyncContext asyncContext) {
                 asyncContext.getRequest(), asyncContext.getResponse());
         }
     }
-
-    @Override
-    public Runnable wrapRunnable(Runnable delegate, Transaction transaction) {
-        return tracer.allocateRunnableWrapper().wrap(delegate).withScope(transaction);
-    }
 }