io.github.jeqo.dropwizard.opentracing.OpenTracingFactory Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of dropwizard-opentracing Show documentation

There is a newer version: 0.1.2

Show newest version

package io.github.jeqo.dropwizard.opentracing;

import com.fasterxml.jackson.annotation.JsonProperty;
import io.github.jeqo.dropwizard.jaeger.JaegerClientFactory;
import io.opentracing.Tracer;
import io.opentracing.mock.MockTracer;
import io.opentracing.util.GlobalTracer;

/**
 * Dropwizard Factory for OpenTracing Tracers
 */
public class OpenTracingFactory {

  @JsonProperty("provider")
  private String provider = TracingProvider.MOCK.name();

  @JsonProperty("jaeger")
  private JaegerClientFactory jaegerClientFactory;

  /**
   * Builds a {@link Tracer} depending on {@link TracingProvider}.
   * Current supported providers: JEAGER, MOCK.
   *
   * @param moduleName
   * @return
   */
  public Tracer build(String moduleName) {
    switch (TracingProvider.valueOf(provider)) {
      case JAEGER:
        return jaegerClientFactory.build(moduleName);
      default:
        return new MockTracer();
    }
  }

  public void buildAndRegister(String moduleName) {
    if (!GlobalTracer.isRegistered()) {
      Tracer tracer = build(moduleName);
      GlobalTracer.register(tracer);
    }
  }

  enum TracingProvider {
    JAEGER, MOCK
  }
}