io.opentracing.contrib.httpcomponents.SpanFutureCallback Maven / Gradle / Ivy

Go to download

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

httpasyncclient

There is a newer version: 0.12.3

Show newest version

package io.opentracing.contrib.httpcomponents;

import io.opentracing.Span;
import io.opentracing.tag.Tags;
import org.apache.http.concurrent.FutureCallback;

public class SpanFutureCallback implements FutureCallback {

    private final FutureCallback delegate;
    private final Span span;
    private final HttpTagger tagger;

    public SpanFutureCallback(final FutureCallback delegate, final Span span, final HttpTagger tagger) {
        this.delegate = delegate;
        this.span = span;
        this.tagger = tagger;
    }

    public void completed(T result) {
        tagger.tagContext();
        span.finish();
        if (delegate != null) {
            delegate.completed(result);
        }
    }

    public void failed(Exception ex) {
        Tags.ERROR.set(span, true);
        span.finish();
        if (delegate != null) {
            delegate.failed(ex);
        }
    }

    public void cancelled() {
        Tags.ERROR.set(span, true);
        span.log("client cancel");
        span.finish();
        if (delegate != null) {
            delegate.cancelled();
        }
    }

}