com.github.kristofa.brave.resteasy.BraveClientExecutionInterceptor Maven / Gradle / Ivy

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

import javax.ws.rs.ext.Provider;

import com.github.kristofa.brave.*;
import com.github.kristofa.brave.http.*;

import org.jboss.resteasy.annotations.interception.ClientInterceptor;
import org.jboss.resteasy.client.ClientRequest;
import org.jboss.resteasy.client.ClientResponse;
import org.jboss.resteasy.spi.interception.ClientExecutionContext;
import org.jboss.resteasy.spi.interception.ClientExecutionInterceptor;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;


/**
 * {@link ClientExecutionInterceptor} that uses the {@link ClientTracer} to set up a new span.  It adds the necessary
 * HTTP header parameters to the request to propagate trace information. It also adds some span annotations:
 * 
 * Binary Annotation, key: request, value: http method and full request url.
 * Binary Annoration, key: response.code, value: http reponse code. This annotation is only submitted when response code
 * is unsuccessful
 * Annotation: failure. Only submitted when response code is unsuccessful. This allows us to filter on unsuccessful
 * requests.
 * 
 * If you add a http header with key: X-B3-SpanName, and with a custom span name as value this value will be used as span
 * name iso the path.
 * 
 * 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:
 * 

 * http://localhost:8080/service/path/a/b
 * 
 * The service name will be 'service. The span name will be '/path/a/b'.
 *
 * @author kristof
 */
@Component
@Provider
@ClientInterceptor
public class BraveClientExecutionInterceptor implements ClientExecutionInterceptor {

    private final ClientRequestInterceptor requestInterceptor;
    private final ClientResponseInterceptor responseInterceptor;
    private final ServiceNameProvider serviceNameProvider;
    private final SpanNameProvider spanNameProvider;

    /**
     * Create a new instance.
     *
     * @param serviceNameProvider Provides service name.
     * @param spanNameProvider Provides span name.
     * @param requestInterceptor Client request interceptor.
     * @param responseInterceptor Client response interceptor.
     */
    @Autowired
    public BraveClientExecutionInterceptor(ServiceNameProvider serviceNameProvider, SpanNameProvider spanNameProvider, ClientRequestInterceptor requestInterceptor, ClientResponseInterceptor responseInterceptor) {
        this.requestInterceptor = requestInterceptor;
        this.spanNameProvider = spanNameProvider;
        this.responseInterceptor = responseInterceptor;
        this.serviceNameProvider = serviceNameProvider;
    }

    /**
     * {@inheritDoc}
     */
    @Override
    public ClientResponse execute(final ClientExecutionContext ctx) throws Exception {

        final ClientRequest request = ctx.getRequest();

        final HttpClientRequest httpClientRequest = new RestEasyHttpClientRequest(request);
        final ClientRequestAdapter adapter = new HttpClientRequestAdapter(httpClientRequest, serviceNameProvider, spanNameProvider);
        requestInterceptor.handle(adapter);

        ClientResponse response = null;
        try {
            response = ctx.proceed();
        } catch (final Exception e) {
            throw e;
        }
        finally
        {
            if (response != null) {
                final HttpResponse httpResponse = new RestEasyHttpClientResponse(response);
                final ClientResponseAdapter responseAdapter = new HttpClientResponseAdapter(httpResponse);
                responseInterceptor.handle(responseAdapter);
            }
            else
            {
                responseInterceptor.handle(NoAnnotationsClientResponseAdapter.getInstance());
            }
        }
        return response;
    }
}