common/jaeger/jaeger.go

package jaeger

import (
	"context"
	"encoding/json"
	"strconv"
	"strings"

	"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/net/gtrace"
	"go.opentelemetry.io/otel/attribute"
)

var ShutDown func(ctx context.Context)

func init() {
	jaegerAgent := g.Cfg().MustGet(context.Background(), "jaeger.addr").String()
	serverName := g.Cfg().MustGet(context.Background(), "server.name").String()
	shutdown, err := otlphttp.Init(serverName, jaegerAgent, "/v1/traces")
	if err != nil {
		panic(err)
	}
	ShutDown = shutdown
}
func NewTracer(r *ghttp.Request) {
	_, span := gtrace.NewSpan(r.Context(), r.GetServeHandler().GetMetaTag("summary"))
	defer span.End()
	span.SetAttributes(attribute.String("request", getParams(r)))
	r.Middleware.Next()

	// 清理响应字符串，确保 UTF-8 有效（处理二进制数据如 ZIP 文件）
	response := r.Response.BufferString()
	cleanResponse := strings.ToValidUTF8(response, "")

	// 如果响应太大（如文件下载），只记录前 1000 字符
	if len(cleanResponse) > 1000 {
		cleanResponse = cleanResponse[:1000] + "... (truncated)"
	}

	span.SetAttributes(attribute.String("response", cleanResponse))
}
func getParams(r *ghttp.Request) string {
	params := map[string]interface{}{}
	if r.Method == "POST" {
		json.Unmarshal(r.GetBody(), &params) //获取raw传参
	}
	if r.Method == "GET" {
		r.Request.ParseForm()
		form := r.Form
		for k, v := range form {
			if vl, e := strconv.Atoi(v[0]); e == nil {
				params[k] = vl
			} else {
				params[k] = v[0]
			}
		}
	}
	rp, _ := json.Marshal(&params)
	return string(rp)
}