-
Notifications
You must be signed in to change notification settings - Fork 376
/
tracer.go
122 lines (102 loc) · 2.63 KB
/
tracer.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
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
package tracer
import (
"context"
"fmt"
"log"
"runtime"
"go.opentelemetry.io/otel/api/global"
"go.opentelemetry.io/otel/api/kv"
"go.opentelemetry.io/otel/api/propagation"
"go.opentelemetry.io/otel/api/trace"
"go.opentelemetry.io/otel/exporters/trace/jaeger"
"go.opentelemetry.io/otel/exporters/trace/stdout"
sdktrace "go.opentelemetry.io/otel/sdk/trace"
)
type Cleanup func()
type ExporterType int
const (
ExporterTypeNone ExporterType = iota
ExporterTypeStdout
ExporterTypeJaeger
)
type Config struct {
ExporterType ExporterType
ServiceName string
RuntimeProvider bool
// Jaeger config
JaegerHost string
}
func InitTracer(flag, service string) func() {
cfg := &Config{
RuntimeProvider: true,
ServiceName: service,
}
switch flag {
case "": // None
return func() {}
case "stdout": // Stdout
cfg.ExporterType = ExporterTypeStdout
default: // Jaeger
cfg.ExporterType = ExporterTypeJaeger
cfg.JaegerHost = flag
}
pt, cl, err := ConfigureProvider(cfg)
if err != nil {
log.Fatalf("unable to init tracer: `%s`", err)
}
SetGlobalTraceProvider(pt)
return cl
}
func ConfigureProvider(cfg *Config) (pt trace.Provider, cl Cleanup, err error) {
switch cfg.ExporterType {
case ExporterTypeJaeger:
pt, cl, err = NewJaegerProvider(cfg.JaegerHost, cfg.ServiceName)
case ExporterTypeStdout:
pt, err = NewStdoutProvider()
default:
pt, cl, err = &trace.NoopProvider{}, func() {}, nil
return
}
if cfg.RuntimeProvider {
pt = NewRuntimeProvider(pt)
}
return
}
func SetGlobalTraceProvider(tp trace.Provider) {
global.SetTraceProvider(tp)
}
func NewStdoutProvider() (trace.Provider, error) {
exporter, err := stdout.NewExporter(stdout.Options{PrettyPrint: true})
if err != nil {
return nil, err
}
return sdktrace.NewProvider(sdktrace.WithConfig(
sdktrace.Config{DefaultSampler: sdktrace.AlwaysSample()}),
sdktrace.WithSyncer(exporter),
)
}
func NewJaegerProvider(host, service string) (trace.Provider, func(), error) {
host = fmt.Sprintf("https://%s/api/traces", host)
return jaeger.NewExportPipeline(
jaeger.WithCollectorEndpoint(host),
jaeger.WithProcess(jaeger.Process{
ServiceName: service,
Tags: []kv.KeyValue{
kv.String("exporter", "jaeger"),
kv.String("os", runtime.GOOS),
kv.String("arch", runtime.GOARCH),
kv.String("go", runtime.Version()),
},
}),
jaeger.WithSDK(&sdktrace.Config{DefaultSampler: sdktrace.AlwaysSample()}),
)
}
func New(name string) trace.Tracer {
return global.Tracer(name)
}
func From(ctx context.Context) trace.Tracer {
return trace.SpanFromContext(ctx).Tracer()
}
func Propagators() propagation.Propagators {
return global.Propagators()
}