/
requestHeaderCarrier.go
58 lines (48 loc) · 1.44 KB
/
requestHeaderCarrier.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
package tracingutil
import (
"net/textproto"
http "github.com/valyala/fasthttp"
"go.opentelemetry.io/otel/propagation"
)
var (
_ propagation.TextMapCarrier = RequestHeaderCarrier{}
)
// RequestHeaderCarrier adapts fasthttp.RequestHeader to satisfy the TextMapCarrier interface.
type RequestHeaderCarrier struct {
header *http.RequestHeader
}
func NewRequestHeaderCarrier(header *http.RequestHeader) *RequestHeaderCarrier {
return &RequestHeaderCarrier{
header: header,
}
}
// Get returns the value associated with the passed key.
func (hc RequestHeaderCarrier) Get(key string) string {
hc.header.DisableNormalizing()
var value = string(hc.header.Peek(key))
if len(value) == 0 {
// NOTE: patch compatibility
// https://www.w3.org/TR/trace-context/#header-name
canonicalHeaderKey := textproto.CanonicalMIMEHeaderKey(key)
value = string(hc.header.Peek(canonicalHeaderKey))
}
hc.header.EnableNormalizing()
return value
}
// Set stores the key-value pair.
func (hc RequestHeaderCarrier) Set(key string, value string) {
// NOTE: patch compatibility
// https://www.w3.org/TR/trace-context/#header-name
hc.header.DisableNormalizing()
hc.header.Set(key, value)
hc.header.EnableNormalizing()
}
// Keys lists the keys stored in this carrier.
func (hc RequestHeaderCarrier) Keys() []string {
header := hc.header
keys := make([]string, 0, header.Len())
header.VisitAll(func(key, value []byte) {
keys = append(keys, string(key))
})
return keys
}