/
trace.go
58 lines (46 loc) · 1.35 KB
/
trace.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 tracing
import (
"log"
"go.opentelemetry.io/otel"
"go.opentelemetry.io/otel/exporters/jaeger"
"go.opentelemetry.io/otel/propagation"
"go.opentelemetry.io/otel/sdk/resource"
sdktrace "go.opentelemetry.io/otel/sdk/trace"
semconv "go.opentelemetry.io/otel/semconv/v1.7.0"
"go.opentelemetry.io/otel/trace"
)
// nolint: ireturn
func New(cfg Config) trace.Tracer {
var exporter sdktrace.SpanExporter
if !cfg.Enabled {
return trace.NewNoopTracerProvider().Tracer("")
}
exporter, err := jaeger.New(
jaeger.WithAgentEndpoint(jaeger.WithAgentHost(cfg.Agent.Host), jaeger.WithAgentPort(cfg.Agent.Port)),
)
if err != nil {
log.Fatalf("failed to initialize export pipeline: %v", err)
}
res, err := resource.Merge(
resource.Default(),
resource.NewSchemaless(
semconv.ServiceNamespaceKey.String("snapp.dispatching"),
semconv.ServiceNameKey.String("natasha"),
),
)
if err != nil {
panic(err)
}
bsp := sdktrace.NewBatchSpanProcessor(exporter)
tp := sdktrace.NewTracerProvider(
sdktrace.WithSampler(sdktrace.ParentBased(sdktrace.TraceIDRatioBased(cfg.Ratio))),
sdktrace.WithSpanProcessor(bsp),
sdktrace.WithResource(res),
)
otel.SetTracerProvider(tp)
// register the TraceContext propagator globally.
var tc propagation.TraceContext
otel.SetTextMapPropagator(tc)
tracer := otel.Tracer("dispatching/natasha")
return tracer
}