@@ -39,13 +39,13 @@ final class InstrumentTests: XCTestCase {
39
39
}
40
40
}
41
41
42
- private struct DictionaryInjector : InjectorProtocol {
42
+ private struct DictionaryInjector : Injector {
43
43
func inject( _ value: String , forKey key: String , into dictionary: inout [ String : String ] ) {
44
44
dictionary [ key] = value
45
45
}
46
46
}
47
47
48
- private struct DictionaryExtractor : ExtractorProtocol {
48
+ private struct DictionaryExtractor : Extractor {
49
49
func extract( key: String , from dictionary: [ String : String ] ) -> String ? {
50
50
return dictionary [ key]
51
51
}
@@ -61,22 +61,14 @@ private final class FirstFakeTracer: Instrument {
61
61
static let headerName = " first-fake-trace-id "
62
62
static let defaultTraceID = UUID ( ) . uuidString
63
63
64
- func inject< Carrier, Injector> (
65
- _ baggage: Baggage , into carrier: inout Carrier , using injector: Injector
66
- )
67
- where
68
- Injector: InjectorProtocol ,
69
- Carrier == Injector . Carrier {
64
+ func inject< Carrier, Inject> ( _ baggage: Baggage , into carrier: inout Carrier , using injector: Inject )
65
+ where Inject: Injector , Carrier == Inject . Carrier {
70
66
guard let traceID = baggage [ TraceIDKey . self] else { return }
71
67
injector. inject ( traceID, forKey: FirstFakeTracer . headerName, into: & carrier)
72
68
}
73
69
74
- func extract< Carrier, Extractor> (
75
- _ carrier: Carrier , into baggage: inout Baggage , using extractor: Extractor
76
- )
77
- where
78
- Extractor: ExtractorProtocol ,
79
- Carrier == Extractor . Carrier {
70
+ func extract< Carrier, Extract> ( _ carrier: Carrier , into baggage: inout Baggage , using extractor: Extract )
71
+ where Extract: Extractor , Carrier == Extract . Carrier {
80
72
let traceID = extractor. extract ( key: FirstFakeTracer . headerName, from: carrier) ?? FirstFakeTracer . defaultTraceID
81
73
baggage [ TraceIDKey . self] = traceID
82
74
}
@@ -92,22 +84,22 @@ private final class SecondFakeTracer: Instrument {
92
84
static let headerName = " second-fake-trace-id "
93
85
static let defaultTraceID = UUID ( ) . uuidString
94
86
95
- func inject< Carrier, Injector > (
96
- _ baggage: Baggage , into carrier: inout Carrier , using injector: Injector
87
+ func inject< Carrier, Inject > (
88
+ _ baggage: Baggage , into carrier: inout Carrier , using injector: Inject
97
89
)
98
90
where
99
- Injector : InjectorProtocol ,
100
- Carrier == Injector . Carrier {
91
+ Inject : Injector ,
92
+ Carrier == Inject . Carrier {
101
93
guard let traceID = baggage [ TraceIDKey . self] else { return }
102
94
injector. inject ( traceID, forKey: SecondFakeTracer . headerName, into: & carrier)
103
95
}
104
96
105
- func extract< Carrier, Extractor > (
106
- _ carrier: Carrier , into baggage: inout Baggage , using extractor: Extractor
97
+ func extract< Carrier, Extract > (
98
+ _ carrier: Carrier , into baggage: inout Baggage , using extractor: Extract
107
99
)
108
100
where
109
- Extractor : ExtractorProtocol ,
110
- Carrier == Extractor . Carrier {
101
+ Extract : Extractor ,
102
+ Carrier == Extract . Carrier {
111
103
let traceID = extractor. extract ( key: SecondFakeTracer . headerName, from: carrier) ?? SecondFakeTracer . defaultTraceID
112
104
baggage [ TraceIDKey . self] = traceID
113
105
}
0 commit comments