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。這有助于您在分布式系統中追蹤請求鏈路。