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

Go to download

Show more of this group Show more artifacts with this name
Show all versions of brave-apache-http-interceptors Show documentation

Apache http client request and response interceptor implementations.

There is a newer version: 3.7.0

Show newest version

package com.github.kristofa.brave.httpclient;

import com.github.kristofa.brave.ClientTracer;
import com.github.kristofa.brave.client.ClientRequestInterceptor;
import com.google.common.base.Optional;
import org.apache.commons.lang.Validate;
import org.apache.http.HttpRequest;
import org.apache.http.HttpRequestInterceptor;
import org.apache.http.protocol.HttpContext;

/**
 * Apache HttpClient {@link HttpRequestInterceptor} that adds brave/zipkin annotations to outgoing client request.
 * 
 * We assume the first part of the URI is the context path. The context name will be used as service name in endpoint.
 * Remaining part of path will be used as span name unless X-B3-SpanName http header is set. For example, if we have URI:
 * 

 * /service/path/a/b
 * 
 * The service name will be 'service'. The span name will be '/path/a/b'.
 * 
 * @author kristof
 */
public class BraveHttpRequestInterceptor implements HttpRequestInterceptor {
    private final ClientRequestInterceptor clientRequestInterceptor;
    private final Optional serviceName;

    /**
     * Creates a new instance.
     * 
     * @param clientTracer ClientTracer should not be null.
     */
    public BraveHttpRequestInterceptor(final ClientTracer clientTracer, Optional serviceName) {
        Validate.notNull(clientTracer);
        Validate.notNull(serviceName);
        this.clientRequestInterceptor = new ClientRequestInterceptor(clientTracer);
        this.serviceName = serviceName;
    }

    /**
     * {@inheritDoc}
     */
    @Override
    public void process(final HttpRequest request, final HttpContext context) {
        clientRequestInterceptor.handle(new ApacheRequestAdapter(request), serviceName);
    }
}