如何在Golang日志中记录请求ID

在Golang中，为了在日志中记录请求ID，你可以使用log包结合自定义的日志格式和上下文（context）来实现。以下是一个简单的示例：

1. 首先，创建一个用于生成和存储请求ID的函数：

package main

import (
	"context"
	"crypto/rand"
	"encoding/hex"
)

const requestIDKey = "requestID"

func generateRequestID() string {
	bytes := make([]byte, 8)
	_, err := rand.Read(bytes)
	if err != nil {
		panic(err)
	}
	return hex.EncodeToString(bytes)
}

func contextWithRequestID(parent context.Context) context.Context {
	requestID := generateRequestID()
	return context.WithValue(parent, requestIDKey, requestID)
}

2. 然后，创建一个自定义的日志记录器，它可以从上下文中提取请求ID并将其添加到日志消息中：

package main

import (
	"context"
	"log"
)

type loggerWithRequestID struct {
	logger *log.Logger
}

func (l *loggerWithRequestID) log(ctx context.Context, level int, msg string, args ...interface{}) {
	requestID, ok := ctx.Value(requestIDKey).(string)
	if !ok {
		requestID = "unknown"
	}
	l.logger.Printf("[%s] [%s] %s", requestID, level, msg, args...)
}

func newLoggerWithRequestID(logger *log.Logger) *loggerWithRequestID {
	return &loggerWithRequestID{logger: logger}
}

3. 最后，在你的应用程序中使用这个自定义的日志记录器：

package main

import (
	"context"
	"net/http"
)

func main() {
	logger := log.Default()
	loggerWithRequestID := newLoggerWithRequestID(logger)

	http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
		ctx := contextWithRequestID(r.Context())
		loggerWithRequestID.log(ctx, "info", "Handling request")

		// Your handler logic here

		loggerWithRequestID.log(ctx, "info", "Finished handling request")
	})

	http.ListenAndServe(":8080", nil)
}

现在，每当你的应用程序处理一个HTTP请求时，它都会在日志中记录一个带有请求ID的消息。这样，你可以轻松地跟踪和调试特定的请求。