如何在Golang的HTTP中间件中统一处理请求错误

package main

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

// AppError 定义一个自定义错误类型，包含状态码和错误信息
type AppError struct {
    Code    int    `json:"code"`
    Message string `json:"message"`
}

// Error 实现 error 接口
func (e *AppError) Error() string {
    return e.Message
}

// ErrorHandlerFunc 定义一个返回错误的 handler 函数类型
type ErrorHandlerFunc func(w http.ResponseWriter, r *http.Request) error

// Middleware 错误处理中间件
func Middleware(next ErrorHandlerFunc) http.HandlerFunc {
    return func(w http.ResponseWriter, r *http.Request) {
        // 执行 handler 函数并捕获错误
        err := next(w, r)
        if err != nil {
            // 统一错误处理逻辑
            log.Printf("Error: %v", err)

            // 根据错误类型进行处理
            var appError *AppError
            switch e := err.(type) {
            case *AppError:
                appError = e
            default:
                appError = &AppError{
                    Code:    http.StatusInternalServerError,
                    Message: "Internal Server Error",
                }
            }

            // 设置响应头
            w.Header().Set("Content-Type", "application/json")
            w.WriteHeader(appError.Code)

            // 返回 JSON 格式的错误信息
            fmt.Fprintf(w, `{"error": "%s"}`, appError.Message) // 简化了JSON序列化，实际应用中建议使用json.Marshal
        }
    }
}

// ExampleHandler 示例 handler 函数，可能返回错误
func ExampleHandler(w http.ResponseWriter, r *http.Request) error {
    // 模拟一个错误
    if r.URL.Query().Get("error") == "true" {
        return &AppError{
            Code:    http.StatusBadRequest,
            Message: "Simulated error occurred",
        }
    }

    // 正常处理
    fmt.Fprintln(w, "Hello, World!")
    return nil
}

func main() {
    // 使用中间件包装 handler 函数
    handler := Middleware(ExampleHandler)

    // 注册 handler 函数
    http.HandleFunc("/", handler)

    // 启动服务器
    log.Fatal(http.ListenAndServe(":8080", nil))
}

AppError

AppError

type AppError struct {
    Code    int    `json:"code"`
    Message string `json:"message"`
    Details interface{} `json:"details,omitempty"` // 可选的错误详情
}

AppError

func AnotherExampleHandler(w http.ResponseWriter, r *http.Request) error {
    // 模拟一个错误
    if r.URL.Query().Get("error") == "true" {
        return &AppError{
            Code:    http.StatusBadRequest,
            Message: "Invalid input",
            Details: map[string]interface{}{
                "field":   "username",
                "message": "Username is required",
            },
        }
    }

    // 正常处理
    fmt.Fprintln(w, "Another Hello, World!")
    return nil
}

json.Marshal

AppError

func Middleware(next ErrorHandlerFunc) http.HandlerFunc {
    return func(w http.ResponseWriter, r *http.Request) {
        err := next(w, r)
        if err != nil {
            log.Printf("Error: %v", err)

            var appError *AppError
            switch e := err.(type) {
            case *AppError:
                appError = e
            case *DatabaseError: // 假设你定义了一个 DatabaseError 类型
                appError = &AppError{
                    Code:    http.StatusInternalServerError,
                    Message: "Database error",
                    Details: e.Error(),
                }
            case *APIError: // 假设你定义了一个 APIError 类型
                appError = &AppError{
                    Code:    http.StatusBadGateway,
                    Message: "API error",
                    Details: e.Error(),
                }
            default:
                appError = &AppError{
                    Code:    http.StatusInternalServerError,
                    Message: "Internal Server Error",
                }
            }

            w.Header().Set("Content-Type", "application/json")
            w.WriteHeader(appError.Code)
            // 实际应用中建议使用 json.Marshal
            fmt.Fprintf(w, `{"error": "%s", "details": "%v"}`, appError.Message, appError.Details)
        }
    }
}

DatabaseError

APIError

Error()

http.Request

func Middleware(next ErrorHandlerFunc) http.HandlerFunc {
    return func(w http.ResponseWriter, r *http.Request) {
        // 获取请求信息
        requestURL := r.URL.String()
        requestMethod := r.Method
        // 假设你有一个函数可以从请求中提取用户信息
        userID := GetUserIDFromRequest(r)

        err := next(w, r)
        if err != nil {
            // 记录详细的错误日志
            log.Printf("Error: %v, Request URL: %s, Method: %s, User ID: %d", err, requestURL, requestMethod, userID)

            // ... 错误处理逻辑 ...
        }
    }
}

// 假设的函数，用于从请求中提取用户信息
func GetUserIDFromRequest(r *http.Request) int {
    // 在实际应用中，你需要根据你的认证机制来提取用户信息
    // 例如，从 Cookie 中获取用户 ID
    return 123 // 示例用户 ID
}