优化jaeger代码,使用otlp规范
This commit is contained in:
@@ -3,55 +3,36 @@ package jaeger
|
||||
import (
|
||||
"context"
|
||||
"encoding/json"
|
||||
"github.com/gogf/gf/contrib/trace/otlphttp/v2"
|
||||
"github.com/gogf/gf/v2/frame/g"
|
||||
"github.com/gogf/gf/v2/net/ghttp"
|
||||
"github.com/gogf/gf/v2/util/gconv"
|
||||
"go.opentelemetry.io/otel"
|
||||
"github.com/gogf/gf/v2/net/gtrace"
|
||||
"go.opentelemetry.io/otel/attribute"
|
||||
"go.opentelemetry.io/otel/exporters/jaeger"
|
||||
"go.opentelemetry.io/otel/sdk/resource"
|
||||
"go.opentelemetry.io/otel/sdk/trace"
|
||||
semconv "go.opentelemetry.io/otel/semconv/v1.21.0"
|
||||
"strconv"
|
||||
)
|
||||
|
||||
var Tp = new(trace.TracerProvider)
|
||||
var ShutDown func(ctx context.Context)
|
||||
|
||||
func init() {
|
||||
jaegerAgent, err := g.Cfg().Get(context.Background(), "jaeger.addr")
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
exp, err := jaeger.New(jaeger.WithCollectorEndpoint(jaeger.WithEndpoint(gconv.String(jaegerAgent))))
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
serverName, err := g.Cfg().Get(context.Background(), "server.Name")
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
// 创建一个 TracerProvider,并将 Jaeger exporter 设置为其处理器
|
||||
Tp = trace.NewTracerProvider(
|
||||
// 使用 BatchSpanProcessor 可以提高性能,它会批量发送 span
|
||||
trace.WithBatcher(exp),
|
||||
// 设置资源属性,这些属性会附加到所有导出的 span 上
|
||||
trace.WithResource(resource.NewWithAttributes(
|
||||
semconv.SchemaURL,
|
||||
semconv.ServiceName(gconv.String(serverName)), // 服务名称,在 Jaeger UI 中会显示
|
||||
)),
|
||||
)
|
||||
// 将 TracerProvider 设置为全局,方便在应用的任何地方通过 otel.Tracer() 获取
|
||||
otel.SetTracerProvider(Tp)
|
||||
ShutDown, err = otlphttp.Init(serverName.String(), jaegerAgent.String(), "/v1/traces")
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
}
|
||||
func NewTracer(r *ghttp.Request) {
|
||||
// 从传入的上下文中获取 Tracer,或者直接使用全局的
|
||||
tracer := otel.Tracer(r.GetServeHandler().GetMetaTag("summary"))
|
||||
_, span := tracer.Start(r.Context(), r.GetServeHandler().GetMetaTag("summary"))
|
||||
defer span.End() // 非常重要:确保 span 在函数结束时被关闭
|
||||
_, span := gtrace.NewSpan(r.Context(), r.GetServeHandler().GetMetaTag("summary"))
|
||||
defer span.End()
|
||||
span.SetAttributes(attribute.String("request", getParams(r)))
|
||||
r.Middleware.Next()
|
||||
span.SetAttributes(attribute.String("response", r.Response.BufferString()))
|
||||
//span.AddEvent("Saying hello is done")
|
||||
}
|
||||
func getParams(r *ghttp.Request) string {
|
||||
params := map[string]interface{}{}
|
||||
|
||||
Reference in New Issue
Block a user