org.drjekyll.sentry.apachehttpclient4.SentryHttpRequestInterceptor Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of sentry-http-interceptors Show documentation
Sends tracing information about Apache HttpClient 4 calls to Sentry
There is a newer version: 7.0.3
Show newest version
package org.drjekyll.sentry.apachehttpclient4;

import org.apache.http.Header;
import org.apache.http.HttpRequest;
import org.apache.http.HttpRequestInterceptor;
import org.apache.http.client.methods.HttpRequestWrapper;
import org.apache.http.client.methods.HttpUriRequest;
import org.apache.http.protocol.HttpContext;
import org.apache.http.util.Args;

import java.util.Arrays;
import java.util.stream.Collectors;

import javax.annotation.Nonnull;

import io.sentry.BaggageHeader;
import io.sentry.Breadcrumb;
import io.sentry.HubAdapter;
import io.sentry.IHub;
import io.sentry.ISpan;
import io.sentry.SentryTraceHeader;
import io.sentry.util.PropagationTargetsUtils;

/**
 * An Apache HttpClient request interceptor that creates a Sentry span and adds Sentry tracing information to the HTTP
 * request headers.
 * 
 * Add this interceptor as first request interceptor with
 * {@link org.apache.http.impl.client.HttpClientBuilder#addInterceptorFirst(HttpRequestInterceptor)} to measure the span
 * duration precisely as possible:
 * 
 * HttpClientBuilder.create()
 *   .addInterceptorFirst(new SentryHttpRequestInterceptor(HubAdapter.getInstance()))
 *   .addInterceptorLast(new SentryHttpResponseInterceptor(HubAdapter.getInstance()))
 *   .build();
 * 
 * This interceptor alone won't finish the span and won't add a breadcrumb, so you need
 * {@link SentryHttpResponseInterceptor} as well.  The hub needs to be the same hub in both interceptors.
 */
public class SentryHttpRequestInterceptor implements HttpRequestInterceptor {

  private final IHub hub;

  /**
   * Initializes this HTTP request interceptor with the given Sentry hub.
   *
   * @param hub A Sentry hub, e.g. {@link HubAdapter#getInstance()}. Must not be null.
   */
  public SentryHttpRequestInterceptor(@Nonnull IHub hub) {
    this.hub = Args.notNull(hub, "Hub");
  }

  @Override
  public void process(HttpRequest request, HttpContext context) {
    Args.notNull(request, "HTTP request");
    if (!(request instanceof HttpRequestWrapper)) {
      return;
    }
    HttpRequestWrapper requestWrapper = (HttpRequestWrapper) request;
    if (!(requestWrapper.getOriginal() instanceof HttpUriRequest)) {
      return;
    }
    HttpUriRequest originalRequest = (HttpUriRequest) requestWrapper.getOriginal();
    ISpan activeSpan = hub.getSpan();
    String method = originalRequest.getMethod();
    String url = originalRequest.getURI().toString();
    if (activeSpan == null) {
      hub.addBreadcrumb(Breadcrumb.http(url, method));
      return;
    }
    ISpan childSpan = activeSpan.startChild("http.client");
    childSpan.setData(RequestHash.SPAN_DATA_KEY, RequestHash.create(requestWrapper));
    childSpan.setDescription(String.format("%s %s", method, url));
    final SentryTraceHeader sentryTraceHeader = childSpan.toSentryTrace();
    if (PropagationTargetsUtils.contain(hub.getOptions().getTracePropagationTargets(), url)) {
      request.addHeader(sentryTraceHeader.getName(), sentryTraceHeader.getValue());
      Header[] headers = request.getHeaders(BaggageHeader.BAGGAGE_HEADER);
      if (headers != null) {
        BaggageHeader baggageHeader = childSpan.toBaggageHeader(Arrays.stream(headers)
          .sequential()
          .map(Header::getValue)
          .collect(Collectors.toList()));
        if (baggageHeader != null) {
          request.addHeader(baggageHeader.getName(), baggageHeader.getValue());
        }
      }
    }

  }

}