brave.jaxrs2.TracingFeature Maven / Gradle / Ivy

Go to download
package brave.jaxrs2;

import brave.Tracing;
import brave.http.HttpTracing;
import javax.inject.Inject;
import javax.ws.rs.core.Feature;
import javax.ws.rs.core.FeatureContext;
import javax.ws.rs.ext.Provider;

@Provider
public final class TracingFeature implements Feature {
  public static Feature create(Tracing tracing) {
    return new TracingFeature(HttpTracing.create(tracing));
  }

  public static Feature create(HttpTracing httpTracing) {
    return new TracingFeature(httpTracing);
  }

  final HttpTracing httpTracing;

  @Inject TracingFeature(HttpTracing httpTracing) { // intentionally hidden
    this.httpTracing = httpTracing;
  }

  // TODO: figure out how to deal with when the client or server impl is also traced
  @Override public boolean configure(FeatureContext context) {
    context.register(new TracingClientFilter(httpTracing));
    context.register(new TracingContainerFilter(httpTracing));
    return true;
  }
}