58 lines
1.5 KiB
Go
58 lines
1.5 KiB
Go
package module
|
|
|
|
import (
|
|
"context"
|
|
"git.hlsq.asia/alex/service-common/config"
|
|
"git.hlsq.asia/alex/service-common/log"
|
|
"go.opentelemetry.io/otel"
|
|
"go.opentelemetry.io/otel/exporters/otlp/otlptrace/otlptracegrpc"
|
|
"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.24.0"
|
|
)
|
|
|
|
// Tracer 链路追踪模块
|
|
type Tracer struct {
|
|
DefaultModule
|
|
Cfg *config.MetricConfig
|
|
ServiceName string
|
|
tp *sdktrace.TracerProvider
|
|
}
|
|
|
|
func (m *Tracer) Init() error {
|
|
exporter, err := otlptracegrpc.New(
|
|
context.Background(),
|
|
otlptracegrpc.WithInsecure(),
|
|
otlptracegrpc.WithEndpoint(m.Cfg.Jaeger.Endpoint),
|
|
)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
samplingRate := 1.0
|
|
if 0 <= m.Cfg.Jaeger.SamplingRate && m.Cfg.Jaeger.SamplingRate <= 1 {
|
|
samplingRate = m.Cfg.Jaeger.SamplingRate
|
|
}
|
|
m.tp = sdktrace.NewTracerProvider(
|
|
sdktrace.WithBatcher(exporter),
|
|
sdktrace.WithResource(resource.NewWithAttributes(
|
|
semconv.SchemaURL,
|
|
semconv.ServiceNameKey.String(m.ServiceName),
|
|
)),
|
|
sdktrace.WithSampler(sdktrace.ParentBased(sdktrace.TraceIDRatioBased(samplingRate))),
|
|
)
|
|
otel.SetTracerProvider(m.tp)
|
|
otel.SetTextMapPropagator(propagation.TraceContext{})
|
|
|
|
return nil
|
|
}
|
|
|
|
func (m *Tracer) Stop() error {
|
|
if m.tp != nil {
|
|
if err := m.tp.Shutdown(context.Background()); err != nil {
|
|
log.Errorf("stop tracer provider failed: %v", err)
|
|
}
|
|
}
|
|
return nil
|
|
}
|