SpringCloud/spring-cloud-sleuth
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Return span id from a controller, not from a deferred result

Browse Source
This commit is contained in:
Marcin Grzejszczak
2018-10-05 11:33:53 +02:00
parent 5140c50521
commit 5a14ffd957
1 changed files with 6 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
spring-cloud-sleuth-core/src/test/java/org/springframework/cloud/sleuth/annotation/SleuthSpanCreatorAspectWebFluxTests.java
View File

@@ -323,23 +323,26 @@ public class SleuthSpanCreatorAspectWebFluxTests {
@ContinueSpan
public Mono<Long> continueSpanInTraceContext() {
log.info("Continue");
return Mono.defer(() -> Mono.just(tracer.currentSpan().context().spanId()));
Long span = tracer.currentSpan().context().spanId();
return Mono.defer(() -> Mono.just(span));
}
@NewSpan(name = "newSpanInTraceContext")
public Mono<Long> newSpanInTraceContext() {
log.info("New Span in Trace Context");
return Mono.defer(() -> Mono.just(tracer.currentSpan().context().spanId()));
Long span = tracer.currentSpan().context().spanId();
return Mono.defer(() -> Mono.just(span));
}
@NewSpan(name = "newSpanInSubscriberContext")
public Mono<Long> newSpanInSubscriberContext() {
log.info("New Span in Subscriber Context");
Long span = tracer.currentSpan().context().spanId();
return Mono.subscriberContext()
.doOnSuccess(
context -> log.info("New Span in deferred Trace Context"))
.flatMap(context -> Mono.defer(
() -> Mono.just(tracer.currentSpan().context().spanId())));
() -> Mono.just(span)));
}
}