Golang日志中如何追踪请求链路

在Golang中，我们可以使用context包和log包来实现请求链路的追踪。以下是一个简单的示例，展示了如何在Golang中使用context和log包来追踪请求链路：

1. 首先，我们需要创建一个自定义的Key类型，用于在context中存储请求链路信息。

package main

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

type key int

const (
	traceIDKey key = iota
)

2. 然后，我们需要创建一个中间件函数，用于在每个请求的处理过程中生成和传递请求链路ID。

func traceMiddleware(next http.Handler) http.Handler {
	return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		traceID := generateTraceID()
		ctx := context.WithValue(r.Context(), traceIDKey, traceID)
		next.ServeHTTP(w, r.WithContext(ctx))
	})
}

func generateTraceID() string {
	// 这里可以使用任何生成唯一ID的方法，例如UUID或者时间戳+随机数等。
	return "trace-id"
}

3. 接下来，我们需要在日志记录器中使用请求链路ID。为此，我们可以创建一个自定义的Logger类型，它包含一个log.Logger实例，并添加了一个traceID字段。

type Logger struct {
	*log.Logger
	traceID string
}

func (l *Logger) Println(v ...interface{}) {
	log.SetPrefix(l.traceID + " ")
	l.Logger.Println(v...)
}

4. 现在，我们可以在处理请求时使用自定义的Logger类型来记录日志。

func handler(w http.ResponseWriter, r *http.Request) {
	logger := &Logger{
		Logger: log.Default(),
		traceID: r.Context().Value(traceIDKey).(string),
	}
	logger.Println("Handling request")
	// ...
}

5. 最后，我们需要将中间件和处理函数添加到HTTP服务器中。

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

现在，当您运行此程序并访问http://localhost:8080/时，您将在日志中看到每个请求的链路ID。这有助于您在分布式系统中追踪请求链路。