如何通过Golang日志追踪请求

在Go中实现请求追踪与日志记录

当我们在构建Web服务时，一个清晰、可追踪的日志系统往往是排查问题的生命线。在Go语言生态中，除了标准库自带的log包，社区也提供了像logrus、zap这样功能更强大的第三方选择。那么，如何有效地记录每一次请求的关键信息——比如请求ID、时间戳、客户端IP——以便后续追踪呢？一个常见的做法是使用中间件。下面，我们就通过一个具体的例子，来看看如何用标准库搭建一个基础的请求日志追踪模块。

第一步：构建基础日志中间件

首先，我们来创建一个最核心的日志中间件。它的任务很明确：在请求进入和离开时，分别记录下关键信息。

package main

import (
    "log"
    "net/http"
    "time"
)

func loggingMiddleware(next http.Handler) http.Handler {
    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        start := time.Now()

        // 记录请求信息
        log.Printf("Request: %s %s %s", r.RemoteAddr, r.Method, r.URL.Path)

        // 调用下一个处理器
        next.ServeHTTP(w, r)

        duration := time.Since(start)
        log.Printf("Response: %s %s %s %v", r.RemoteAddr, r.Method, r.URL.Path, duration)
    })
}

func mainHandler(w http.ResponseWriter, r *http.Request) {
    w.Write([]byte("Hello, World!"))
}

func main() {
    http.Handle("/", loggingMiddleware(http.HandlerFunc(mainHandler)))
    log.Fatal(http.ListenAndServe(":8080", nil))
}

看这段代码，loggingMiddleware这个中间件的工作流程非常清晰：它包裹了实际的处理函数。在请求开始时，记下客户端地址、HTTP方法和请求路径；等业务逻辑处理完毕，再记录一次，并额外计算出整个请求的耗时。这样一来，服务控制台的输出就不再是杂乱无章的信息，而是有了明确的开始和结束标记，对于监控请求生命周期非常直观。

第二步：引入请求ID，实现精准追踪

不过，上面的方案在并发量高的时候会面临一个问题：日志是有了，但怎么从海量的记录里，精准地挑出属于“某一次特定请求”的所有日志行呢？这时候，为每个请求分配一个唯一的ID就显得至关重要了。

我们可以对中间件进行升级，在请求进入时就生成一个唯一标识（比如UUID），并把它贯穿于整个请求上下文和后续的日志中。

package main

import (
    "context"
    "log"
    "net/http"
    "time"

    "github.com/google/uuid"
)

const requestIDKey = "requestID"

func loggingMiddleware(next http.Handler) http.Handler {
    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        requestID := uuid.New().String()
        ctx := context.WithValue(r.Context(), requestIDKey, requestID)
        start := time.Now()

        // 记录请求信息
        log.Printf("Request ID: %s %s %s", requestID, r.RemoteAddr, r.Method, r.URL.Path)

        // 调用下一个处理器
        next.ServeHTTP(w, r.WithContext(ctx))

        duration := time.Since(start)
        log.Printf("Request ID: %s %s %s %v", requestID, r.RemoteAddr, r.URL.Path, duration)
    })
}

func mainHandler(w http.ResponseWriter, r *http.Request) {
    requestID := r.Context().Value(requestIDKey).(string)
    w.Write([]byte("Hello, World! Request ID: " + requestID))
}

func main() {
    http.Handle("/", loggingMiddleware(http.HandlerFunc(mainHandler)))
    log.Fatal(http.ListenAndServe(":8080", nil))
}

这个改进版做了几件关键事：首先，它利用github.com/google/uuid库为每个请求生成了一个独一无二的ID。然后，这个ID被存入了请求的上下文（Context）中，这意味着在本次请求链路的任何地方，只要你能拿到这个Context，就能取出这个ID。最后，无论是在入口日志还是出口日志里，这个请求ID都被打印了出来。这样一来，无论日志多么繁忙，你都可以通过这个ID像穿珠子一样，把一次请求的所有行为轨迹完整地串联起来，实现精准的问题定位和请求追踪。