Java Tracer类代码示例

import org.springframework.cloud.sleuth.Tracer; //导入依赖的package包/类
SleuthRxJavaSchedulersHook(Tracer tracer, TraceKeys traceKeys,
		List<String> threadsToSample) {
	this.tracer = tracer;
	this.traceKeys = traceKeys;
	this.threadsToSample = threadsToSample;
	try {
		this.delegate = RxJavaPlugins.getInstance().getSchedulersHook();
		if (this.delegate instanceof SleuthRxJavaSchedulersHook) {
			return;
		}
		RxJavaErrorHandler errorHandler = RxJavaPlugins.getInstance().getErrorHandler();
		RxJavaObservableExecutionHook observableExecutionHook
			= RxJavaPlugins.getInstance().getObservableExecutionHook();
		logCurrentStateOfRxJavaPlugins(errorHandler, observableExecutionHook);
		RxJavaPlugins.getInstance().reset();
		RxJavaPlugins.getInstance().registerSchedulersHook(this);
		RxJavaPlugins.getInstance().registerErrorHandler(errorHandler);
		RxJavaPlugins.getInstance().registerObservableExecutionHook(observableExecutionHook);
	} catch (Exception e) {
		log.error("Failed to register Sleuth RxJava SchedulersHook", e);
	}
}
 

import org.springframework.cloud.sleuth.Tracer; //导入依赖的package包/类
@Test
public void should_pass_tracing_information_when_using_Hystrix_commands() {
	Tracer tracer = new DefaultTracer(new AlwaysSampler(), new Random(),
			new DefaultSpanNamer(), new NoOpSpanLogger(), new NoOpSpanReporter());
	TraceKeys traceKeys = new TraceKeys();
	HystrixCommand.Setter setter = HystrixCommand.Setter
			.withGroupKey(HystrixCommandGroupKey.Factory.asKey("group"))
			.andCommandKey(HystrixCommandKey.Factory.asKey("command"));
	// tag::hystrix_command[]
	HystrixCommand<String> hystrixCommand = new HystrixCommand<String>(setter) {
		@Override
		protected String run() throws Exception {
			return someLogic();
		}
	};
	// end::hystrix_command[]
	// tag::trace_hystrix_command[]
	TraceCommand<String> traceCommand = new TraceCommand<String>(tracer, traceKeys, setter) {
		@Override
		public String doRun() throws Exception {
			return someLogic();
		}
	};
	// end::trace_hystrix_command[]

	String resultFromHystrixCommand = hystrixCommand.execute();
	String resultFromTraceCommand = traceCommand.execute();

	then(resultFromHystrixCommand).isEqualTo(resultFromTraceCommand);
	then(tracer.getCurrentSpan()).isNull();
}
 

import org.springframework.cloud.sleuth.Tracer; //导入依赖的package包/类
@Test
public void should_set_runnable_name_to_annotated_value()
		throws ExecutionException, InterruptedException {
	ExecutorService executorService = Executors.newSingleThreadExecutor();
	SpanNamer spanNamer = new DefaultSpanNamer();
	Tracer tracer = Mockito.mock(Tracer.class);

	// tag::span_name_annotated_runnable_execution[]
	Runnable runnable = new TraceRunnable(tracer, spanNamer, new TaxCountingRunnable());
	Future<?> future = executorService.submit(runnable);
	// ... some additional logic ...
	future.get();
	// end::span_name_annotated_runnable_execution[]

	BDDMockito.then(tracer).should().createSpan(BDDMockito.eq("calculateTax"), BDDMockito.any(Span.class));
}
 

import org.springframework.cloud.sleuth.Tracer; //导入依赖的package包/类
@Test
public void should_set_runnable_name_to_to_string_value()
		throws ExecutionException, InterruptedException {
	ExecutorService executorService = Executors.newSingleThreadExecutor();
	SpanNamer spanNamer = new DefaultSpanNamer();
	Tracer tracer = Mockito.mock(Tracer.class);

	// tag::span_name_to_string_runnable_execution[]
	Runnable runnable = new TraceRunnable(tracer, spanNamer, new Runnable() {
		@Override public void run() {
			// perform logic
		}

		@Override public String toString() {
			return "calculateTax";
		}
	});
	Future<?> future = executorService.submit(runnable);
	// ... some additional logic ...
	future.get();
	// end::span_name_to_string_runnable_execution[]

	BDDMockito.then(tracer).should().createSpan(BDDMockito.eq("calculateTax"), BDDMockito.any(Span.class));
	executorService.shutdown();
}
 

import org.springframework.cloud.sleuth.Tracer; //导入依赖的package包/类
@Test
public void optInto128BitTraceId() {
	EnvironmentTestUtils.addEnvironment(context, "spring.sleuth.traceId128:true");
	context.register(PropertyPlaceholderAutoConfiguration.class,
			SleuthLogAutoConfiguration.class, TraceAutoConfiguration.class);
	context.refresh();
	Tracer tracer = context.getBean(Tracer.class);

	Span span = null;
	try {
		span = tracer.createSpan("foo", NeverSampler.INSTANCE);
		assertThat(span.getTraceIdHigh()).isNotEqualTo(0L);
		assertThat(span.getTraceId()).isNotEqualTo(0L);
	}
	finally {
		if (span != null) {
			tracer.close(span);
		}
	}
}
 

import org.springframework.cloud.sleuth.Tracer; //导入依赖的package包/类
SpanSubscriber(Subscriber<? super T> subscriber, Context ctx, Tracer tracer,
		String name) {
	this.subscriber = subscriber;
	this.tracer = tracer;
	Span root = ctx.getOrDefault(Span.class, tracer.getCurrentSpan());
	if (log.isTraceEnabled()) {
		log.trace("Span from context [{}]", root);
	}
	this.rootSpan = root;
	if (log.isTraceEnabled()) {
		log.trace("Stored context root span [{}]", this.rootSpan);
	}
	this.span = tracer.createSpan(name, root);
	if (log.isTraceEnabled()) {
		log.trace("Created span [{}], with name [{}]", this.span, name);
	}
	this.context = ctx.put(Span.class, this.span);
}
 

import org.springframework.cloud.sleuth.Tracer; //导入依赖的package包/类
@Test
public void should_pass_tracing_information_when_using_Hystrix_commands() {
	Tracer tracer = new DefaultTracer(new AlwaysSampler(), new Random(),
			new DefaultSpanNamer(), new NoOpSpanLogger(), new NoOpSpanReporter(), new TraceKeys());
	TraceKeys traceKeys = new TraceKeys();
	HystrixCommand.Setter setter = withGroupKey(asKey("group"))
			.andCommandKey(HystrixCommandKey.Factory.asKey("command"));
	// tag::hystrix_command[]
	HystrixCommand<String> hystrixCommand = new HystrixCommand<String>(setter) {
		@Override
		protected String run() throws Exception {
			return someLogic();
		}
	};
	// end::hystrix_command[]
	// tag::trace_hystrix_command[]
	TraceCommand<String> traceCommand = new TraceCommand<String>(tracer, traceKeys, setter) {
		@Override
		public String doRun() throws Exception {
			return someLogic();
		}
	};
	// end::trace_hystrix_command[]

	String resultFromHystrixCommand = hystrixCommand.execute();
	String resultFromTraceCommand = traceCommand.execute();

	then(resultFromHystrixCommand).isEqualTo(resultFromTraceCommand);
	then(tracer.getCurrentSpan()).isNull();
}
 

import org.springframework.cloud.sleuth.Tracer; //导入依赖的package包/类
public Result invoke(Invoker<?> invoker, Invocation invocation) throws RpcException {
    boolean isTraceDubbo = false;
    Tracer tracer = null;
    SpanInjector spanInjector = null;
    try {
        tracer = ApplicationContextAwareBean.CONTEXT.getBean(Tracer.class);
        spanInjector = ApplicationContextAwareBean.CONTEXT.getBean(DubboSpanInjector.class);
        isTraceDubbo = (tracer != null && spanInjector != null);
        if (isTraceDubbo) {
            String spanName = invoker.getUrl().getParameter("interface") + ":" + invocation.getMethodName() + ":" + invoker.getUrl().getParameter("version") + "(" + invoker.getUrl().getHost() + ")";
            Span newSpan = tracer.createSpan(spanName);
            spanInjector.inject(newSpan, RpcContext.getContext());
            newSpan.logEvent(Span.CLIENT_SEND);

        }

        Result result = invoker.invoke(invocation);
        return result;

    } finally {
        if (isTraceDubbo) {
            if (tracer.isTracing()) {
                tracer.getCurrentSpan().logEvent(Span.CLIENT_RECV);
                tracer.close(tracer.getCurrentSpan());
            }

        }
    }
}
 

import org.springframework.cloud.sleuth.Tracer; //导入依赖的package包/类
@Bean
@ConditionalOnMissingBean(Tracer.class)
public Tracer sleuthTracer(Sampler sampler, Random random,
		SpanNamer spanNamer, SpanLogger spanLogger,
		SpanReporter spanReporter, TraceKeys traceKeys) {
	return new DefaultTracer(sampler, random, spanNamer, spanLogger,
			spanReporter, true, traceKeys);
}
 

import org.springframework.cloud.sleuth.Tracer; //导入依赖的package包/类
@Bean
@ConditionalOnProperty(value = "spring.sleuth.amqp.enabled", matchIfMissing = true)
@ConditionalOnMissingBean(AmqpMessagingSpanManager.class)
public AmqpMessagingSpanManager amqpMessagingSpanManager(
    AmqpMessagingSpanInjector amqpMessagingSpanInjector,
    AmqpMessagingSpanExtractor amqpMessagingSpanExtractor,
    Tracer tracer) {
  return new DefaultAmqpMessagingSpanManager(
      amqpMessagingSpanInjector, amqpMessagingSpanExtractor, tracer);
}
 

import org.springframework.cloud.sleuth.Tracer; //导入依赖的package包/类
@Autowired
public SpringOneController(RestTemplate restTemplate, Tracer tracer,
		SpanNamer spanNamer, SpringOneWorker springOneWorker,
		AsyncRestTemplate asyncRestTemplate, Service1Client service1Client,
		TraceKeys traceKeys) {
	this.restTemplate = restTemplate;
	this.tracer = tracer;
	this.spanNamer = spanNamer;
	this.springOneWorker = springOneWorker;
	this.asyncRestTemplate = asyncRestTemplate;
	this.service1Client = service1Client;
	this.traceKeys = traceKeys;
}
 

import org.springframework.cloud.sleuth.Tracer; //导入依赖的package包/类
@Bean
public FilterRegistrationBean traceWebFilter(Tracer tracer, TraceKeys traceKeys,
		SkipPatternProvider skipPatternProvider, SpanReporter spanReporter,
		SpanExtractor<HttpServletRequest> spanExtractor,
		SpanInjector<HttpServletResponse> spanInjector,
		HttpTraceKeysInjector httpTraceKeysInjector, TraceFilter traceFilter) {
	FilterRegistrationBean filterRegistrationBean = new FilterRegistrationBean(traceFilter);
	filterRegistrationBean.setDispatcherTypes(ASYNC, ERROR, FORWARD, INCLUDE, REQUEST);
	return filterRegistrationBean;
}
 

import org.springframework.cloud.sleuth.Tracer; //导入依赖的package包/类
@Bean
public TraceFilter traceFilter(Tracer tracer, TraceKeys traceKeys,
		SkipPatternProvider skipPatternProvider, SpanReporter spanReporter,
		SpanExtractor<HttpServletRequest> spanExtractor,
		SpanInjector<HttpServletResponse> spanInjector,
		HttpTraceKeysInjector httpTraceKeysInjector) {
	return new TraceFilter(tracer, traceKeys, skipPatternProvider.skipPattern(),
			spanReporter, spanExtractor, spanInjector, httpTraceKeysInjector);
}
 

import org.springframework.cloud.sleuth.Tracer; //导入依赖的package包/类
@Bean
@ConditionalOnMissingBean
public TraceRestTemplateInterceptor traceRestTemplateInterceptor(Tracer tracer,
		SpanInjector<HttpRequest> spanInjector,
		HttpTraceKeysInjector httpTraceKeysInjector) {
	return new TraceRestTemplateInterceptor(tracer, spanInjector, httpTraceKeysInjector);
}
 

import org.springframework.cloud.sleuth.Tracer; //导入依赖的package包/类
/**
 * According to the JavaDocs all Spring {@link AsyncClientHttpRequestFactory} implement
 * the {@link ClientHttpRequestFactory} interface.
 *
 * In case that it's not true we're setting the {@link SimpleClientHttpRequestFactory}
 * as a default for sync request processing.
 *
 * @see org.springframework.web.client.AsyncRestTemplate#AsyncRestTemplate(AsyncClientHttpRequestFactory)
 */
public TraceAsyncClientHttpRequestFactoryWrapper(Tracer tracer,
		SpanInjector<HttpRequest> spanInjector,
		AsyncClientHttpRequestFactory asyncDelegate,
		HttpTraceKeysInjector httpTraceKeysInjector) {
	super(tracer, spanInjector, httpTraceKeysInjector);
	this.asyncDelegate = asyncDelegate;
	this.syncDelegate = asyncDelegate instanceof ClientHttpRequestFactory ?
			(ClientHttpRequestFactory) asyncDelegate : defaultClientHttpRequestFactory();
}