19
19
import java .nio .charset .StandardCharsets ;
20
20
import java .util .Collections ;
21
21
import java .util .HashMap ;
22
- import java .util .LinkedHashSet ;
23
22
import java .util .List ;
24
23
import java .util .Map ;
25
- import java .util .Set ;
26
- import java .util .concurrent .atomic .AtomicBoolean ;
27
24
28
25
import brave .propagation .Propagation ;
29
26
import org .apache .commons .logging .Log ;
@@ -48,21 +45,6 @@ enum MessageHeaderPropagation
48
45
49
46
private static final Log log = LogFactory .getLog (MessageHeaderPropagation .class );
50
47
51
- private static final Set <String > DEPRECATED_HEADERS = new LinkedHashSet <>();
52
-
53
- private static final Map <String , String > LEGACY_HEADER_MAPPING = new HashMap <>();
54
-
55
- static {
56
- LEGACY_HEADER_MAPPING .put ("X-B3-TraceId" , TraceMessageHeaders .TRACE_ID_NAME );
57
- LEGACY_HEADER_MAPPING .put ("X-B3-SpanId" , TraceMessageHeaders .SPAN_ID_NAME );
58
- LEGACY_HEADER_MAPPING .put ("X-B3-ParentSpanId" ,
59
- TraceMessageHeaders .PARENT_ID_NAME );
60
- LEGACY_HEADER_MAPPING .put ("X-B3-Sampled" , TraceMessageHeaders .SAMPLED_NAME );
61
- LEGACY_HEADER_MAPPING .put ("X-B3-Flags" , TraceMessageHeaders .SPAN_FLAGS_NAME );
62
- DEPRECATED_HEADERS .addAll (LEGACY_HEADER_MAPPING .keySet ());
63
- DEPRECATED_HEADERS .addAll (LEGACY_HEADER_MAPPING .values ());
64
- }
65
-
66
48
static Map <String , ?> propagationHeaders (Map <String , ?> headers ,
67
49
List <String > propagationHeaders ) {
68
50
Map <String , Object > headersToCopy = new HashMap <>();
@@ -116,15 +98,8 @@ private static NativeMessageHeaderAccessor ensureNativeHeadersAreMutable(
116
98
return nativeAccessor ;
117
99
}
118
100
119
- static final AtomicBoolean LOGGED_PUT_DEPRECATED_HEADER = new AtomicBoolean ();
120
-
121
101
@ Override
122
102
public void put (MessageHeaderAccessor accessor , String key , String value ) {
123
- if (DEPRECATED_HEADERS .contains (key )
124
- && LOGGED_PUT_DEPRECATED_HEADER .compareAndSet (false , true )) {
125
- log .warn ("Please update your code so that it uses the 'b3' header "
126
- + "instead of " + key );
127
- }
128
103
try {
129
104
doPut (accessor , key , value );
130
105
}
@@ -177,25 +152,10 @@ public String get(MessageHeaderAccessor accessor, String key) {
177
152
+ "] from message" , ex );
178
153
}
179
154
}
180
- return legacyValue (accessor , key );
181
- }
182
-
183
- private String legacyValue (MessageHeaderAccessor accessor , String key ) {
184
- String legacyKey = LEGACY_HEADER_MAPPING .get (key );
185
- if (legacyKey != null ) {
186
- return doGet (accessor , legacyKey );
187
- }
188
155
return null ;
189
156
}
190
157
191
- static final AtomicBoolean LOGGED_GET_DEPRECATED_HEADER = new AtomicBoolean ();
192
-
193
158
private String doGet (MessageHeaderAccessor accessor , String key ) {
194
- if (DEPRECATED_HEADERS .contains (key )
195
- && LOGGED_GET_DEPRECATED_HEADER .compareAndSet (false , true )) {
196
- log .warn ("Please update your code so that it uses the 'b3' header "
197
- + "instead of " + key );
198
- }
199
159
if (accessor instanceof NativeMessageHeaderAccessor ) {
200
160
NativeMessageHeaderAccessor nativeAccessor = (NativeMessageHeaderAccessor ) accessor ;
201
161
String result = nativeAccessor .getFirstNativeHeader (key );
0 commit comments