-
Notifications
You must be signed in to change notification settings - Fork 376
/
bertytypes_header.go
47 lines (36 loc) · 1.27 KB
/
bertytypes_header.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
package tracer
import (
"context"
"go.opentelemetry.io/otel/api/kv"
"go.opentelemetry.io/otel/api/propagation"
"go.opentelemetry.io/otel/api/trace"
"berty.tech/berty/v2/go/pkg/protocoltypes"
)
type metadataSupplier struct {
header *protocoltypes.MessageHeaders
}
func (m *metadataSupplier) Get(key string) string {
if meta := m.header.GetMetadata(); meta != nil {
if v, ok := meta[key]; ok {
return v
}
}
return ""
}
func (m *metadataSupplier) Set(key string, val string) {
if m.header.Metadata == nil {
m.header.Metadata = make(map[string]string)
}
m.header.Metadata[key] = val
}
func InjectSpanContextToMessageHeaders(ctx context.Context, h *protocoltypes.MessageHeaders) {
propagation.InjectHTTP(ctx, Propagators(), &metadataSupplier{h})
}
func ExtractSpanContextFromMessageHeaders(ctx context.Context, h *protocoltypes.MessageHeaders) context.Context {
return propagation.ExtractHTTP(ctx, Propagators(), &metadataSupplier{h})
}
func SpanFromMessageHeaders(ctx context.Context, h *protocoltypes.MessageHeaders, name string, attrs ...kv.KeyValue) (context.Context, trace.Span) {
hctx := ExtractSpanContextFromMessageHeaders(context.Background(), h)
sctx := trace.RemoteSpanContextFromContext(hctx)
return From(ctx).Start(hctx, name, trace.LinkedTo(sctx))
}