forked from jaegertracing/jaeger-client-go
-
Notifications
You must be signed in to change notification settings - Fork 0
/
zipkin.go
62 lines (54 loc) · 1.59 KB
/
zipkin.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
59
60
61
62
package jaeger
import (
"github.com/opentracing/opentracing-go"
)
// ZipkinSpanFormat is an OpenTracing carrier format constant
const ZipkinSpanFormat = "zipkin-span-format"
// ExtractableZipkinSpan is a type of Carrier used for integration with Zipkin-aware
// RPC frameworks (like TChannel). It does not support baggage, only trace IDs.
type ExtractableZipkinSpan interface {
TraceID() uint64
SpanID() uint64
ParentID() uint64
Flags() byte
}
// InjectableZipkinSpan is a type of Carrier used for integration with Zipkin-aware
// RPC frameworks (like TChannel). It does not support baggage, only trace IDs.
type InjectableZipkinSpan interface {
SetTraceID(traceID uint64)
SetSpanID(spanID uint64)
SetParentID(parentID uint64)
SetFlags(flags byte)
}
type zipkinPropagator struct {
tracer *tracer
}
func (p *zipkinPropagator) Inject(
ctx SpanContext,
abstractCarrier interface{},
) error {
carrier, ok := abstractCarrier.(InjectableZipkinSpan)
if !ok {
return opentracing.ErrInvalidCarrier
}
carrier.SetTraceID(ctx.TraceID())
carrier.SetSpanID(ctx.SpanID())
carrier.SetParentID(ctx.ParentID())
carrier.SetFlags(ctx.flags)
return nil
}
func (p *zipkinPropagator) Extract(abstractCarrier interface{}) (SpanContext, error) {
carrier, ok := abstractCarrier.(ExtractableZipkinSpan)
if !ok {
return emptyContext, opentracing.ErrInvalidCarrier
}
if carrier.TraceID() == 0 {
return emptyContext, opentracing.ErrSpanContextNotFound
}
var ctx SpanContext
ctx.traceID = carrier.TraceID()
ctx.spanID = carrier.SpanID()
ctx.parentID = carrier.ParentID()
ctx.flags = carrier.Flags()
return ctx, nil
}