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

Set spanId in TracingResponderRSocketProxy (#1986)

Browse Source 
also respect traceIdHigh and debug flag in TracingMetadata
This commit is contained in:
Toshiaki Maki
2021-06-28 17:38:46 +09:00
committed by GitHub
parent d353e66556
commit 4f6eb04527
1 changed files with 9 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
spring-cloud-sleuth-instrumentation/src/main/java/org/springframework/cloud/sleuth/instrument/rsocket/TracingResponderRSocketProxy.java
View File

@@ -152,8 +152,16 @@ public class TracingResponderRSocketProxy extends RSocketProxy {
if (extract != null) {
TracingMetadata tracingMetadata = TracingMetadataCodec.decode(extract);
Span.Builder builder = this.tracer.spanBuilder();
String traceId = EncodingUtils.fromLong(tracingMetadata.traceId());
long traceIdHigh = tracingMetadata.traceIdHigh();
if (traceIdHigh != 0L) {
// ExtendedTraceId
traceId = EncodingUtils.fromLong(traceIdHigh) + traceId;
}
TraceContext.Builder parentBuilder = this.tracer.traceContextBuilder()
.sampled(tracingMetadata.isSampled()).traceId(EncodingUtils.fromLong(tracingMetadata.traceId()))
.sampled(tracingMetadata.isDebug() || tracingMetadata.isSampled())
.traceId(traceId)
.spanId(EncodingUtils.fromLong(tracingMetadata.spanId()))
.parentId(EncodingUtils.fromLong(tracingMetadata.parentId()));
return builder.setParent(parentBuilder.build());
}