Golang日志中如何追踪请求

在Golang中，我们可以使用context包和第三方日志库（如logrus或zap）来实现请求追踪。下面是一个简单的示例，展示了如何在Golang中使用context和logrus实现请求追踪。

首先，确保已经安装了logrus库：

go get github.com/sirupsen/logrus

然后，创建一个简单的HTTP服务器，使用context传递请求ID，并在日志中记录请求ID：

package main

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

	"github.com/sirupsen/logrus"
)

const requestIDKey = "requestID"

func main() {
	logrus.SetFormatter(&logrus.JSONFormatter{})
	logrus.SetReportCaller(true)

	http.HandleFunc("/", handleRequest)
	logrus.Info("Server started on port 8080")
	http.ListenAndServe(":8080", nil)
}

func handleRequest(w http.ResponseWriter, r *http.Request) {
	requestID := generateRequestID()
	ctx := context.WithValue(r.Context(), requestIDKey, requestID)
	logrus.WithFields(logrus.Fields{
		"requestID": requestID,
	}).Info("Handling request")

	start := time.Now()
	defer func() {
		logrus.WithFields(logrus.Fields{
			"requestID": requestID,
			"duration":  time.Since(start),
		}).Info("Finished handling request")
	}()

	// 模拟处理请求
	time.Sleep(100 * time.Millisecond)
	fmt.Fprintln(w, "Hello, World!")
}

func generateRequestID() string {
	return fmt.Sprintf("%d", time.Now().UnixNano())
}

在这个示例中，我们为每个请求生成一个唯一的请求ID，并将其存储在context中。然后，我们在处理请求的开始和结束时记录日志，包括请求ID和处理持续时间。

要查看请求追踪日志，可以运行此程序并访问http://localhost:8080。日志将显示每个请求的ID和处理时间。

这只是一个简单的示例，实际应用中可能需要更复杂的追踪功能，例如分布式追踪。在这种情况下，可以考虑使用像jaeger-client-go这样的库来实现更高级的追踪功能。