67 lines
1.5 KiB
Go
67 lines
1.5 KiB
Go
package module
|
|
|
|
import (
|
|
"context"
|
|
"git.hlsq.asia/mmorpg/service-common/config"
|
|
"git.hlsq.asia/mmorpg/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
|
|
tp *sdktrace.TracerProvider
|
|
metricCfg *config.MetricConfig
|
|
serviceName string
|
|
}
|
|
|
|
func (m *Tracer) Init() error {
|
|
exporter, err := otlptracegrpc.New(
|
|
context.Background(),
|
|
otlptracegrpc.WithInsecure(),
|
|
otlptracegrpc.WithEndpoint(m.metricCfg.Jaeger.Endpoint),
|
|
)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
m.tp = sdktrace.NewTracerProvider(
|
|
sdktrace.WithBatcher(exporter),
|
|
sdktrace.WithResource(resource.NewWithAttributes(
|
|
semconv.SchemaURL,
|
|
semconv.ServiceNameKey.String(m.serviceName),
|
|
)),
|
|
)
|
|
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
|
|
}
|
|
|
|
func (m *Tracer) Bind(data ...any) Module {
|
|
if data == nil || len(data) == 0 {
|
|
return m
|
|
}
|
|
if mc, ok := data[0].(*config.MetricConfig); ok {
|
|
m.metricCfg = mc
|
|
}
|
|
if name, ok := data[1].(string); ok {
|
|
m.serviceName = name
|
|
}
|
|
return m
|
|
}
|