复用http.go里面的 g.client(),更新config.yml的注册方式, 更新jaeger链路追踪

This commit is contained in:
Cold
2025-12-15 17:27:10 +08:00
committed by 张斌
parent 8830091e7d
commit 8218487ed0
4 changed files with 102 additions and 59 deletions

View File

@@ -12,6 +12,8 @@ import (
"github.com/gogf/gf/v2/net/ghttp"
"github.com/gogf/gf/v2/net/gtrace"
"go.opentelemetry.io/otel/attribute"
"go.opentelemetry.io/otel/codes"
"go.opentelemetry.io/otel/trace"
)
var (
@@ -47,10 +49,75 @@ func init() {
// 默认自动初始化(保持向后兼容)
Init()
}
// NewSpan 创建新的链路追踪 Span
// spanName: Span 名称,用于在 Jaeger UI 中标识
// 返回带有 Span 的 context 和 Span 对象,调用方需 defer span.End()
func NewSpan(ctx context.Context, spanName string, opts ...trace.SpanStartOption) (context.Context, *gtrace.Span) {
return gtrace.NewSpan(ctx, spanName, opts...)
}
// RecordError 统一错误记录方法
// 功能:
// 1. 控制台输出错误(带完整堆栈 %+v
// 2. Jaeger 链路追踪记录错误
// 3. 设置 Span 错误状态
//
// 使用示例:
//
// jaeger.RecordError(ctx, err, "保存数据失败")
//
// 参数:
// - ctx: 包含 trace span 的上下文
// - err: 错误对象(支持 gerror 堆栈)
// - msg: 可选的错误描述(用于日志和 Jaeger 显示)
func RecordError(ctx context.Context, err error, msg ...string) {
if err == nil {
return
}
// 1. 控制台输出(%+v 打印完整堆栈)
if len(msg) > 0 && msg[0] != "" {
g.Log().Errorf(ctx, "%s: %+v", msg[0], err)
} else {
g.Log().Errorf(ctx, "%+v", err)
}
// 2. Jaeger 记录(从 context 获取当前 span
span := trace.SpanFromContext(ctx)
if span == nil || !span.IsRecording() {
return
}
// 3. 记录错误到 span
span.RecordError(err)
span.SetAttributes(
attribute.Bool("error", true),
attribute.String("error.message", err.Error()),
)
// 4. 设置 span 状态为错误
if len(msg) > 0 && msg[0] != "" {
span.SetAttributes(attribute.String("error.msg", msg[0]))
span.SetStatus(codes.Error, msg[0]+": "+err.Error())
return
}
span.SetStatus(codes.Error, err.Error())
}
// NewTracer HTTP 请求链路追踪中间件
// 功能:
// 1. 为每个 HTTP 请求创建 Span
// 2. 记录请求参数和响应内容
// 3. 自动捕获错误并记录到 Jaeger
//
// 使用方式:在路由组中注册为中间件
//
// group.Middleware(jaeger.NewTracer)
func NewTracer(r *ghttp.Request) {
_, span := gtrace.NewSpan(r.Context(), r.GetServeHandler().GetMetaTag("summary"))
// 创建 Span名称取自 controller 方法的 summary 标签)
ctx, span := gtrace.NewSpan(r.Context(), r.GetServeHandler().GetMetaTag("summary"))
r.SetCtx(ctx)
defer span.End()
// 记录请求参数
span.SetAttributes(attribute.String("request", getParams(r)))
// 执行后续中间件和 handler
r.Middleware.Next()
// 清理响应字符串,确保 UTF-8 有效(处理二进制数据如 ZIP 文件)
@@ -63,7 +130,19 @@ func NewTracer(r *ghttp.Request) {
}
span.SetAttributes(attribute.String("response", cleanResponse))
span.SetAttributes(attribute.Int("http.status_code", r.Response.Status))
if err := r.GetError(); err != nil {
RecordError(ctx, err)
return
}
if r.Response.Status >= 500 {
span.SetAttributes(attribute.Bool("error", true))
span.SetStatus(codes.Error, "http status "+strconv.Itoa(r.Response.Status))
}
}
// getParams 提取请求参数(用于 Jaeger 记录)
func getParams(r *ghttp.Request) string {
params := map[string]interface{}{}
if r.Method == "POST" {