如何在Go中使用context实现请求幂等性

package main

import (
    "context"
    "fmt"
    "net/http"
    "sync"
)

var idempotentKey = "idempotent_key"

func main() {
    http.HandleFunc("/api", handleRequest)
    http.ListenAndServe(":8080", nil)
}

func handleRequest(w http.ResponseWriter, r *http.Request) {
    ctx := r.Context()

    // 获取幂等性标识符
    idempotentID := ctx.Value(idempotentKey)
    if idempotentID == nil {
        idempotentID = generateID()
        // 将幂等性标识符保存到上下文中
        ctx = context.WithValue(ctx, idempotentKey, idempotentID)
    }

    // 执行幂等操作
    if isIdempotent(ctx, idempotentID) {
        // 处理幂等操作
        fmt.Fprintf(w, "Success")
    } else {
        // 幂等操作已经被执行过了
        fmt.Fprintf(w, "Already Executed")
    }
}

func isIdempotent(ctx context.Context, id string) bool {
    // 这里可以根据业务逻辑判断是否已经执行过幂等操作
    // 例如，可以将幂等性标识符保存到数据库中，判断是否已经存在
    // 这里只是一个简单的示例，直接通过sync.Map来保存幂等性标识符
    idMap := ctx.Value(idempotentKey).(*sync.Map)
    _, ok := idMap.Load(id)
    if ok {
        return false
    }
    idMap.Store(id, struct{}{})
    return true
}

func generateID() string {
    // 这里可以根据业务逻辑生成唯一的幂等性标识符
    // 这里只是一个简单的示例，使用时间戳作为标识符
    return fmt.Sprintf("%d", time.Now().Unix())
}