com.github.kristofa.brave.httpclient.ApacheRequestAdapter Maven / Gradle / Ivy

Go to download
package com.github.kristofa.brave.httpclient;

import java.net.URI;

import org.apache.http.Header;
import org.apache.http.HttpRequest;

import com.github.kristofa.brave.BraveHttpHeaders;
import com.github.kristofa.brave.ClientRequestAdapter;
import com.google.common.base.Optional;

class ApacheRequestAdapter implements ClientRequestAdapter {

    private final HttpRequest request;

    public ApacheRequestAdapter(final HttpRequest request) {
        this.request = request;
    }

    @Override
    public URI getUri() {
        return URI.create(request.getRequestLine().getUri());
    }

    @Override
    public String getMethod() {
        return request.getRequestLine().getMethod();
    }

    @Override
    public Optional getSpanName() {
        Optional spanName = Optional.absent();
        final Header spanNameHeader = request.getFirstHeader(BraveHttpHeaders.SpanName.getName());
        if (spanNameHeader != null) {
            spanName = Optional.fromNullable(spanNameHeader.getValue());
        }
        return spanName;
    }

    @Override
    public void addHeader(final String header, final String value) {
        request.addHeader(header, value);
    }
}