Go语言Web开发：构建灵活的Per-Handler中间件并传递请求上下文数据

package main

import (
    "fmt"
    "net/http"
)

// checkCSRF 是一个简单的CSRF检查中间件示例
func checkCSRF(next http.HandlerFunc) http.HandlerFunc {
    return func(w http.ResponseWriter, r *http.Request) {
        // 实际的CSRF令牌验证逻辑
        fmt.Println("执行CSRF检查...")
        isValidCSRF := true // 假设验证通过

        if !isValidCSRF {
            http.Error(w, "CSRF token invalid", http.StatusForbidden)
            return
        }
        // 验证通过，调用下一个处理函数
        next.ServeHTTP(w, r)
    }
}

// myHandler 是一个示例的处理函数
func myHandler(w http.ResponseWriter, r *http.Request) {
    fmt.Fprintf(w, "Hello from myHandler!")
}

func main() {
    http.HandleFunc("/", checkCSRF(myHandler))
    fmt.Println("Server started on :8080")
    http.ListenAndServe(":8080", nil)
}

type CSRFHandlerFunc func(w http.ResponseWriter, r *http.Request, csrfToken string)

// csrfCheckMiddleware 尝试通过修改签名传递参数
func csrfCheckMiddleware(next CSRFHandlerFunc) http.HandlerFunc {
    return func(w http.ResponseWriter, r *http.Request) {
        // ... CSRF 验证逻辑 ...
        token := "generated_csrf_token_example" // 假设生成了令牌
        // 验证失败则返回错误
        // ...
        next(w, r, token) // 调用自定义签名的处理函数
    }
}

// myCustomHandler 适配自定义签名
func myCustomHandler(w http.ResponseWriter, r *http.Request, token string) {
    fmt.Fprintf(w, "Hello with CSRF token: %s", token)
}

package main

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

// 定义一个私有的上下文键类型，避免与其他包的键冲突
type contextKey string

// 定义具体的上下文键
const csrfTokenKey contextKey = "csrfToken"
const authUserKey contextKey = "authUser"

// checkCSRFMiddleware 使用 context.Context 传递CSRF令牌
func checkCSRFMiddleware(next http.HandlerFunc) http.HandlerFunc {
    return func(w http.ResponseWriter, r *http.Request) {
        fmt.Println("执行CSRF检查...")
        // 实际的CSRF令牌生成/验证逻辑
        token := "generated_csrf_token_12345" // 假设生成了一个令牌

        // 将令牌添加到请求的上下文中
        ctx := context.WithValue(r.Context(), csrfTokenKey, token)
        // 使用新的上下文调用下一个处理函数
        next.ServeHTTP(w, r.WithContext(ctx))
    }
}

// authenticateMiddleware 使用 context.Context 传递认证用户
func authenticateMiddleware(next http.HandlerFunc) http.HandlerFunc {
    return func(w http.ResponseWriter, r *http.Request) {
        fmt.Println("执行认证检查...")
        // 实际的用户认证逻辑
        userID := "user-123" // 假设认证成功并获取到用户ID

        // 将用户ID添加到请求的上下文中
        ctx := context.WithValue(r.Context(), authUserKey, userID)
        // 使用新的上下文调用下一个处理函数
        next.ServeHTTP(w, r.WithContext(ctx))
    }
}

// secureHandler 是一个需要CSRF令牌和认证用户的处理函数
func secureHandler(w http.ResponseWriter, r *http.Request) {
    // 从上下文中获取CSRF令牌
    csrfToken, ok := r.Context().Value(csrfTokenKey).(string)
    if !ok {
        http.Error(w, "CSRF token not found in context", http.StatusInternalServerError)
        return
    }

    // 从上下文中获取认证用户ID
    userID, ok := r.Context().Value(authUserKey).(string)
    if !ok {
        http.Error(w, "Authenticated user not found in context", http.StatusInternalServerError)
        return
    }

    fmt.Fprintf(w, "Welcome, %s! Your CSRF token is: %s", userID, csrfToken)
}

func main() {
    // 链式调用中间件：从右到左执行
    // secureHandler -> authenticateMiddleware -> checkCSRFMiddleware
    handler := checkCSRFMiddleware(authenticateMiddleware(secureHandler))

    http.HandleFunc("/secure", handler)
    fmt.Println("Server started on :8080")
    http.ListenAndServe(":8080", nil)
}