io.opentracing.contrib.httpcomponents.SpanHttpAsyncResponseConsumer 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 java.io.IOException;
import org.apache.http.HttpException;
import org.apache.http.HttpResponse;
import org.apache.http.nio.ContentDecoder;
import org.apache.http.nio.IOControl;
import org.apache.http.nio.protocol.HttpAsyncResponseConsumer;
import org.apache.http.protocol.HttpContext;

public class SpanHttpAsyncResponseConsumer implements HttpAsyncResponseConsumer {

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

    public SpanHttpAsyncResponseConsumer(HttpAsyncResponseConsumer delegate, Span span, HttpTagger tagger) {
        this.delegate = delegate;
        this.span = span;
        this.tagger = tagger;
    }

    public void responseReceived(HttpResponse response) throws IOException, HttpException {
        tagger.tagResponse(response);
        delegate.responseReceived(response);
    }

    public void consumeContent(ContentDecoder decoder, IOControl ioctrl) throws IOException {
        delegate.consumeContent(decoder, ioctrl);
    }

    public void responseCompleted(HttpContext context) {
        delegate.responseCompleted(context);
    }

    public void failed(Exception ex) {
        Tags.ERROR.set(span, true);
        delegate.failed(ex);
    }

    public Exception getException() {
        return delegate.getException();
    }

    public T getResult() {
        return delegate.getResult();
    }

    public boolean isDone() {
        return delegate.isDone();
    }

    public void close() throws IOException {
        delegate.close();
    }

    public boolean cancel() {
        span.log("response cancel");
        return delegate.cancel();
    }

}