fixed problem with non-expired tokens that caused overhead

Author: dhilpipre

Kotlin-Coroutines_1.3.3/build/classes/java/main/com/nr/instrumentation/kotlin/coroutines/NRFunction1.class

@@ -6,6 +6,8 @@ kotlinx/coroutines/CompletionHandlerBase.java
  kotlinx.coroutines.CompletionHandlerBase
 kotlinx/coroutines/Job.java
  kotlinx.coroutines.Job
+com/nr/instrumentation/kotlin/coroutines/NRFunction2.java
+ com.nr.instrumentation.kotlin.coroutines.NRFunction2
 kotlin/coroutines/jvm/internal/BaseContinuationImpl.java
  kotlin.coroutines.jvm.internal.BaseContinuationImpl
 kotlinx/coroutines/InvokeOnCancelling.java
@@ -30,5 +32,7 @@ kotlinx/coroutines/CoroutineStart.java
  kotlinx.coroutines.CoroutineStart
 kotlinx/coroutines/CancelHandlerBase.java
  kotlinx.coroutines.CancelHandlerBase
+com/nr/instrumentation/kotlin/coroutines/NRFunction1.java
+ com.nr.instrumentation.kotlin.coroutines.NRFunction1
 kotlinx/coroutines/channels/ReceiveChannel.java
  kotlinx.coroutines.channels.ReceiveChannel

Kotlin-Coroutines_1.3.3/build/classes/java/main/com/nr/instrumentation/kotlin/coroutines/NRFunction2.class

@@ -0,0 +1,27 @@
+package com.nr.instrumentation.kotlin.coroutines;
+
+import com.newrelic.agent.bridge.AgentBridge;
+import com.newrelic.api.agent.Trace;
+
+import kotlin.jvm.functions.Function1;
+
+public class NRFunction1<P1, R> implements Function1<P1, R> {
+	
+	private Function1<P1, R> delegate = null;
+	private static boolean isTransformed = false;
+	
+	public NRFunction1(Function1<P1, R> d) {
+		delegate = d;
+		if(!isTransformed) {
+			isTransformed = true;
+			AgentBridge.instrumentation.retransformUninstrumentedClass(getClass());
+		}
+	}
+
+	@Override
+	@Trace(dispatcher=true)
+	public R invoke(P1 arg0) {
+		return delegate.invoke(arg0);
+	}
+
+}

Kotlin-Coroutines_1.3.3/build/classes/java/main/kotlin/coroutines/jvm/internal/BaseContinuationImpl.class

@@ -0,0 +1,28 @@
+package com.nr.instrumentation.kotlin.coroutines;
+
+import com.newrelic.agent.bridge.AgentBridge;
+import com.newrelic.api.agent.Trace;
+
+import kotlin.jvm.functions.Function2;
+
+public class NRFunction2<P1, P2, R> implements Function2<P1, P2, R> {
+	
+	Function2<P1, P2, R> delegate = null;
+	private static boolean isTransformed = false;
+	
+	public NRFunction2(Function2<P1, P2, R> d) {
+		delegate = d;
+		if(!isTransformed) {
+			isTransformed = true;
+			AgentBridge.instrumentation.retransformUninstrumentedClass(getClass());
+		}
+	}
+
+	@Override
+	@Trace(dispatcher=true)
+	public R invoke(P1 arg0, P2 arg1) {
+		
+		return delegate.invoke(arg0, arg1);
+	}
+
+}

Kotlin-Coroutines_1.3.3/build/classes/java/main/kotlinx/coroutines/AbstractCoroutine.class

@@ -1,14 +1,43 @@
 package kotlin.coroutines.jvm.internal;
 
 import com.newrelic.api.agent.NewRelic;
+import com.newrelic.api.agent.Token;
 import com.newrelic.api.agent.Trace;
 import com.newrelic.api.agent.weaver.MatchType;
+import com.newrelic.api.agent.weaver.NewField;
 import com.newrelic.api.agent.weaver.Weave;
 import com.newrelic.api.agent.weaver.Weaver;
 
+import kotlin.coroutines.Continuation;
+
 @Weave(type=MatchType.BaseClass)
 public abstract class BaseContinuationImpl {
 
+//	@NewField
+//	private Token token = null;
+	
+	public BaseContinuationImpl(Continuation<Object> c) {
+//		if(token == null) {
+//			token = NewRelic.getAgent().getTransaction().getToken();
+//		}
+	}
+	
+	@Trace(async=true)
+	protected Object invokeSuspend(java.lang.Object obj) {
+//		if(token != null) {
+//			token.linkAndExpire();
+//			token = null;
+//		}
+		String name =  null;
+		StackTraceElement element = getStackTraceElement();
+		if(element != null) {
+			name = element.getClassName() + "." + element.getMethodName();
+		}
+		if(name != null) {
+			NewRelic.getAgent().getTracedMethod().setMetricName("Custom","Continuation",name,"invokeSuspend");
+		}
+		return Weaver.callOriginal();
+	}
 
 	@Trace
 	public void resumeWith(Object obj) {