在Golang的Web服务开发中，实现请求追踪可以帮助我们快速定位请求处理过程中的异常、统计各环节耗时，是提升服务可维护性的重要手段。核心思路是为每个进入服务的请求生成唯一标识，再通过上下文将标识传递到请求处理的全链路中，最后在日志、下游调用中携带该标识实现关联。
核心实现思路
实现Web请求追踪主要包含三个核心步骤：
- 请求入口生成唯一请求ID，通常放在Web框架的中间件中处理
- 将请求ID存入context上下文，在后续的业务逻辑、下游调用中从context中取出使用
- 在日志输出、HTTP调用、数据库操作等环节主动携带请求ID，实现全链路关联
请求ID生成与中间件实现
首先我们需要一个生成唯一请求ID的方法，这里使用UUID作为示例，也可以根据业务需求自定义生成规则。然后编写中间件，在请求进入时生成ID并注入到context中，同时设置到响应头方便前端排查问题。
package middleware
import (
"context"
"github.com/google/uuid"
"net/http"
)
// 定义context的key类型，避免冲突
type contextKey string
const RequestIDKey contextKey = "request_id"
// 生成请求ID的中间件
func RequestIDMiddleware(next http.Handler) http.Handler {
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
// 优先从请求头获取已有的请求ID，没有则生成新的
requestID := r.Header.Get("X-Request-ID")
if requestID == "" {
requestID = uuid.New().String()
}
// 将请求ID存入context
ctx := context.WithValue(r.Context(), RequestIDKey, requestID)
// 设置到响应头
w.Header().Set("X-Request-ID", requestID)
// 传递带有请求ID的context到下一个处理环节
next.ServeHTTP(w, r.WithContext(ctx))
})
}
// 从context中获取请求ID的辅助函数
func GetRequestID(ctx context.Context) string {
if id, ok := ctx.Value(RequestIDKey).(string); ok {
return id
}
return ""
}
业务逻辑中携带请求ID
在业务逻辑处理时，我们可以从context中取出请求ID，在输出日志时主动携带，这样排查问题时可以通过请求ID快速筛选所有相关日志。
package service
import (
"context"
"fmt"
"log"
"time"
"your_project/middleware"
)
// 模拟业务处理函数
func HandleUserRequest(ctx context.Context, userID int) error {
requestID := middleware.GetRequestID(ctx)
// 日志中携带请求ID
log.Printf("[request_id:%s] 开始处理用户%d的请求", requestID, userID)
start := time.Now()
// 模拟业务逻辑处理
err := queryUserFromDB(ctx, userID)
if err != nil {
log.Printf("[request_id:%s] 查询用户%d失败，错误：%v", requestID, userID, err)
return err
}
log.Printf("[request_id:%s] 处理用户%d的请求完成，耗时：%v", requestID, userID, time.Since(start))
return nil
}
下游调用中传递请求ID
如果业务中存在调用其他HTTP服务、消息队列等下游场景，也需要将请求ID传递过去，保证全链路可追踪。以下是HTTP调用的示例：
package client
import (
"context"
"net/http"
"your_project/middleware"
)
// 调用下游服务的HTTP客户端
func CallDownstreamService(ctx context.Context, url string) (*http.Response, error) {
requestID := middleware.GetRequestID(ctx)
req, err := http.NewRequestWithContext(ctx, "GET", url, nil)
if err != nil {
return nil, err
}
// 将请求ID设置到下游请求的头部
req.Header.Set("X-Request-ID", requestID)
return http.DefaultClient.Do(req)
}
常见问题与优化
context传递注意事项
在Golang中，context是需要显式传递的，不要存储在结构体中。如果业务中有异步处理场景，需要手动将携带请求ID的context传递到异步协程中，避免异步任务丢失请求ID。
请求ID的格式选择
如果服务需要对接外部的链路追踪系统，可以参考W3C的Trace Context规范生成请求ID，方便和开源的链路追踪组件如Jaeger、Zipkin等对接，实现更完整的链路可视化能力。
性能优化
UUID生成虽然简单，但如果是高并发场景，可以考虑使用更轻量的ID生成方式，比如结合时间戳、机器ID、序列号的方式生成唯一ID，减少UUID生成的开销。