This commit is contained in:
Marcin Grzejszczak
2016-11-06 20:02:28 +01:00
parent 3b14e6a1f7
commit 2ab2eb1747

View File

@@ -60,14 +60,10 @@ class MessagingTextMap implements SpanTextMap {
Message<?> initialMessage = this.delegate.build();
MessageHeaderAccessor accessor = MessageHeaderAccessor
.getMutableAccessor(initialMessage);
Map<String, String> headers = new HashMap<>();
headers.put(key, value);
accessor.copyHeaders(headers);
accessor.setHeader(key, value);
if (accessor instanceof NativeMessageHeaderAccessor) {
NativeMessageHeaderAccessor nativeAccessor = (NativeMessageHeaderAccessor) accessor;
for (String name : headers.keySet()) {
nativeAccessor.setNativeHeader(name, headers.get(name));
}
nativeAccessor.setNativeHeader(key, value);
}
this.delegate.copyHeaders(accessor.toMessageHeaders());
}