kamon.instrumentation.apache.httpclient.UriRequestAdvisor Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of kamon-apache-httpclient_3 Show documentation

kamon-apache-httpclient

The newest version!

package kamon.instrumentation.apache.httpclient;

import org.apache.http.HttpResponse;
import org.apache.http.client.methods.HttpUriRequest;

import kamon.Kamon;
import kamon.context.Context;
import kamon.context.Storage.Scope;
import kamon.instrumentation.context.HasContext;
import kamon.instrumentation.http.HttpClientInstrumentation.RequestHandler;
import kamon.instrumentation.http.HttpMessage.RequestBuilder;
import kamon.trace.Span;
import kanela.agent.libs.net.bytebuddy.asm.Advice;

public class UriRequestAdvisor {

    @Advice.OnMethodEnter(suppress = Throwable.class)
    public static void onEnter(@Advice.Argument(value = 0, readOnly = false) HttpUriRequest request,
            @Advice.Local("handler") RequestHandler handler,
            @Advice.Local("scope") Scope scope) {
        if (((HasContext) request).context().nonEmpty()) {
            // Request has been instrumented already
            return;
        }
        final Context parentContext = Kamon.currentContext();
        final RequestBuilder builder = ApacheHttpClientHelper.toRequestBuilder(request);
        handler = ApacheHttpClientInstrumentation.httpClientInstrumentation().createHandler(builder, parentContext);
        final Context ctx = parentContext.withEntry(Span.Key(), handler.span());
        scope = Kamon.storeContext(ctx);
        request = handler.request();
        ((HasContext) request).setContext(ctx);
    }

    @Advice.OnMethodExit(onThrowable = Throwable.class, suppress = Throwable.class)
    public static void onExit(@Advice.Return HttpResponse response,
            @Advice.Thrown Throwable t,
            @Advice.Local("handler") RequestHandler handler,
            @Advice.Local("scope") Scope scope) {
        if (scope == null) {
            return;
        }
        ApacheHttpClientInstrumentation.processResponse(handler, response, t);
        scope.close();
    }
}