如何在Golang中实现中间件机制_处理请求预处理和日志记录

func loggingMiddleware(next http.Handler) http.Handler {
    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        // ✅ 请求前：记录开始时间、方法、路径
        start := time.Now()
        log.Printf("START %s %s", r.Method, r.URL.Path)

        // ✅ 调用下游 handler（可能是下一个中间件，也可能是最终业务 handler）
        next.ServeHTTP(w, r)

        // ✅ 请求后：记录耗时、状态码
        duration := time.Since(start)
        log.Printf("END %s %s [%d] %.2fms", r.Method, r.URL.Path, w.Header().Get("Status"), float64(duration.Microseconds())/1000)
    })
}

handler := http.HandlerFunc(yourBusinessHandler)
handler = authMiddleware(handler)     // 鉴权（需在 CORS 后？视需求而定）
handler = corsMiddleware(handler)     // 跨域（通常放最外层或紧邻业务层）
handler = loggingMiddleware(handler)  // 日志（建议最外层，覆盖全部）

http.ListenAndServe(":8080", handler)

type ctxKey string
const userCtxKey ctxKey = "user"

func authMiddleware(next http.Handler) http.Handler {
    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        token := r.Header.Get("Authorization")
        userID, err := validateToken(token)
        if err != nil {
            http.Error(w, "Unauthorized", http.StatusUnauthorized)
            return
        }
        // ✅ 将用户信息注入 context
        ctx := context.WithValue(r.Context(), userCtxKey, userID)
        r = r.WithContext(ctx)

        next.ServeHTTP(w, r)
    })
}

// 在业务 handler 中读取：
func yourBusinessHandler(w http.ResponseWriter, r *http.Request) {
    userID := r.Context().Value(userCtxKey).(string) // 类型断言需谨慎（建议封装工具函数）
    fmt.Fprintf(w, "Hello, user %s", userID)
}