golang记录请求id

在Golang中记录请求ID是非常常见和重要的，它能够让开发者更好地追踪和诊断代码中的问题。本文将从如何生成请求ID的角度出发，探讨如何在Golang中记录请求ID。

生成请求ID

生成请求ID的方式有很多种，下面我们介绍两种常见的方法。

首先，可以使用UUID库来生成请求ID。UUID全称为通用唯一识别码，是一种由16字节组成的标识符，通常表示为32位16进制数。在Golang中，我们可以使用标准库提供的uuid库来生成UUID。以下是一个使用uuid库生成请求ID的示例：

import "github.com/google/uuid"

func GenerateRequestID() string {
    id := uuid.New()
    return id.String()
}

上述代码中，我们引入了uuid库，并在GenerateRequestID函数中调用了uuid.New()方法来生成UUID。最后以字符串形式返回请求ID。

其次，如果不想依赖额外的库，也可以使用时间戳和随机数生成请求ID。以下是一个使用时间戳和随机数生成请求ID的示例：

import (
    "strconv"
    "time"
    "math/rand"
)

func GenerateRequestID() string {
    timestamp := time.Now().UnixNano()
    rand.Seed(time.Now().UnixNano())
    randomNum := rand.Intn(1000)
    requestID := strconv.FormatInt(timestamp, 10) + strconv.Itoa(randomNum)
    return requestID
}

上述代码中，我们首先使用time包中的Now()方法获取当前时间戳，然后使用rand包生成一个随机数。将时间戳和随机数转换为字符串，并拼接在一起作为请求ID返回。

记录请求ID

在Golang中，可以使用log库来记录请求ID。log库提供了方便的函数来打印输出信息，并且可以自定义日志格式。以下是一个使用log库记录请求ID的示例：

import (
    "log"
    "net/http"
)

func HandleRequest(w http.ResponseWriter, r *http.Request) {
    requestID := GenerateRequestID()
    log.Printf("[%s] Received request", requestID)
    // 处理请求的业务逻辑
    // ...
}

上述代码中，我们首先调用GenerateRequestID函数生成请求ID，然后使用log库的Printf函数打印包含请求ID在内的信息。这样每次接收到请求时，日志中都会包含请求ID，方便追踪和分析。

使用context传递请求ID

除了在log中记录请求ID外，还可以使用context包来传递请求ID。context包提供了一种在请求之间传递变量和请求范围的数据的机制。以下是一个使用context传递请求ID的示例：

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

type key int

const (
    requestIDKey key = iota
)

func WithRequestID(ctx context.Context, requestID string) context.Context {
    return context.WithValue(ctx, requestIDKey, requestID)
}

func RequestIDFromContext(ctx context.Context) string {
    requestID, _ := ctx.Value(requestIDKey).(string)
    return requestID
}

func HandleRequest(w http.ResponseWriter, r *http.Request) {
    requestID := GenerateRequestID()
    ctx := context.Background()
    ctx = WithRequestID(ctx, requestID)
    log.Printf("[%s] Received request", RequestIDFromContext(ctx))
    // 处理请求的业务逻辑
    // ...
}

上述代码中，我们定义了一个key类型的常量requestIDKey，用于标识请求ID在context中的键。WithRequestID函数用于将请求ID存储在context中，RequestIDFromContext函数用于从context中读取请求ID。在HandleRequest函数中，我们首先通过GenerateRequestID函数生成请求ID，然后使用context.Background()创建一个空的context。接着使用WithRequestID函数将请求ID存储在context中。最后在日志中打印出请求ID，以及处理其他业务逻辑。