io.opentracing.contrib.dropwizard.DropWizardTracer 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

Support for integrating OpenTracing in DropWizard projects

The newest version!

package io.opentracing.contrib.dropwizard;

import io.opentracing.Tracer;
import io.opentracing.Span;

import java.util.concurrent.ConcurrentHashMap;
import java.util.Map;
import javax.ws.rs.core.Context;
import javax.ws.rs.core.Request;
import javax.ws.rs.client.ClientRequestContext;

/**
 * 
 *
 */
public class DropWizardTracer {

    private Tracer tracer;
    private Map serverSpans;
    private Map clientSpans;

    /**
     * Create a tracer for DropWizard applications.
     * @param tracer an io.opentracing.Tracer to trace requests with
     */
    public DropWizardTracer(Tracer tracer) {
        this.serverSpans = new ConcurrentHashMap();
        this.clientSpans = new ConcurrentHashMap();
        this.tracer = tracer;
    }

    /**
     * @return the underlying io.opentracing.Tracer
     */
    public Tracer getTracer() {
        return this.tracer;
    }

    /**
     * @param request for which we want to find the associated span 
     * @return the span for this server request, if it exists and isn't 
     *  finished. Otherwise returns null.
     */
    public Span getSpan(Request request) {
        return this.serverSpans.get(request);
    }

    /**
     * @param requestCtx context for which we want to find the associated span
     * @return the span for this client request, if it exists and is 
     *  not finished. Otherwise returns null.
     */
    public Span getSpan(ClientRequestContext requestCtx) {
        return this.clientSpans.get(requestCtx);
    }

    protected void addServerSpan(Request request, Span span) {
        this.serverSpans.put(request, span);
    }

    protected void addClientSpan(ClientRequestContext requestCtx, Span span) {
        this.clientSpans.put(requestCtx, span);
    }

    protected void finishServerSpan(Request request) {
        Span span = this.serverSpans.get(request);
        if (span != null) {
            this.serverSpans.remove(request);
            span.finish();
        }
    }

    protected void finishClientSpan(ClientRequestContext requestCtx) {
        Span span = this.clientSpans.get(requestCtx);
        if (span != null) {
            this.clientSpans.remove(requestCtx);
            span.finish();
        }
    }
}