io.crnk.brave.internal.OkHttpBraveIntegration Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of crnk-brave Show documentation

JSON API framework for Java

There is a newer version: 2.6.20180430102904

package io.crnk.brave.internal;

import com.github.kristofa.brave.Brave;
import com.github.kristofa.brave.BraveExecutorService;
import com.github.kristofa.brave.okhttp.BraveTracingInterceptor;
import io.crnk.client.http.okhttp.OkHttpAdapterListener;
import okhttp3.Dispatcher;
import okhttp3.OkHttpClient;

public class OkHttpBraveIntegration implements OkHttpAdapterListener {

	private Brave brave;

	public OkHttpBraveIntegration(Brave brave) {
		this.brave = brave;
	}

	@Override
	public void onBuild(OkHttpClient.Builder builder) {
		BraveTracingInterceptor interceptor = buildInterceptor();

		BraveExecutorService tracePropagatingExecutor = buildExecutor();

		builder.addInterceptor(interceptor);
		builder.addNetworkInterceptor(interceptor);
		builder.dispatcher(new Dispatcher(tracePropagatingExecutor));
	}

	protected BraveExecutorService buildExecutor() {
		return new BraveExecutorService(new Dispatcher().executorService(), brave.serverSpanThreadBinder());
	}

	protected BraveTracingInterceptor buildInterceptor() {
		BraveTracingInterceptor.Builder tracingBuilder = BraveTracingInterceptor.builder(brave);
		return tracingBuilder.build();
	}
}