Go语言Web应用错误处理最佳实践：Handler封装与Panic恢复

package main

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

// Error 结构体定义了自定义错误的信息，包括底层错误、HTTP状态码和用户友好的消息。
type Error struct {
    Error   error  // 原始的Go错误对象
    Code    int    // HTTP状态码，如500, 404
    Message string // 用户友好的错误消息
}

// NewError 是一个辅助函数，用于创建Error指针。
func NewError(err error, code int, msg string) *Error {
    return &Error{err, code, msg}
}

// Handler 是一个自定义的HTTP处理函数类型，它返回一个自定义的错误类型*Error。
// 这样，业务逻辑中的错误可以被统一处理。如果返回nil，表示没有错误。
type Handler func(http.ResponseWriter, *http.Request) *Error

// ServeHTTP 实现了 http.Handler 接口。
// 它负责调用实际的业务逻辑处理函数，并捕获其返回的错误或运行时panic。
func (fn Handler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
    // 使用defer和recover()机制捕获运行时panic，防止应用崩溃。
    defer func() {
        if r := recover(); r != nil {
            // 记录panic信息，这对于调试和监控至关重要。
            log.Printf("Panic recovered: %v", r)
            // 在生产环境中，通常返回一个通用的500错误页面，避免暴露敏感的内部错误信息。
            http.Error(w, "服务器内部发生严重错误。", http.StatusInternalServerError)
            // 如果是调试模式，可以选择重新panic以获取完整的堆栈信息。
            // 例如：
            // if os.Getenv("APP_ENV") == "development" {
            //     panic(r)
            // }
        }
    }()

    // 调用实际的业务处理函数，并检查其返回的自定义错误。
    if e := fn(w, r); e != nil {
        // 记录业务逻辑返回的错误信息，便于后期分析。
        log.Printf("业务错误 - Code: %v, Message: \"%s\", Error: %v", e.Code, e.Message, e.Error)

        // 根据错误码进行不同的处理。
        switch e.Code {
        case http.StatusInternalServerError: // 500 内部服务器错误
            http.Error(w, e.Message, e.Code)
            // 可以在此处渲染一个自定义的500错误页面，例如：
            // renderErrorPage(w, "500.html", e.Message)
        case http.StatusNotFound: // 404 页面未找到
            http.NotFound(w, r) // http.NotFound 会设置状态码并写入一个默认消息
            // 可以在此基础上添加自定义的404页面内容，例如：
            // renderErrorPage(w, "404.html", e.Message)
            fmt.Fprint(w, e.Message) // 额外输出自定义消息
        case http.StatusOK: // 200 成功，但可能带有特定消息，例如成功提示
            fmt.Fprint(w, e.Message)
        default: // 处理其他未明确定义的错误码
            http.Error(w, e.Message, e.Code)
            // 默认渲染一个通用的错误页面
        }
    }
}

// 示例业务处理函数：
// myPageHandler 演示了如何根据请求路径返回不同类型的响应，包括正常响应、业务错误、404和panic。
func myPageHandler(w http.ResponseWriter, r *http.Request) *Error {
    // 模拟一个业务逻辑错误
    if r.URL.Path == "/error" {
        return NewError(errors.New("business logic failed"), http.StatusInternalServerError, "业务处理失败，请稍后再试。")
    }
    // 模拟一个404情况，如果路径不匹配预期的页面
    if r.URL.Path != "/" && r.URL.Path != "/home" {
        return NewError(nil, http.StatusNotFound, "请求的页面不存在。")
    }

    // 模拟一个运行时panic
    if r.URL.Path == "/panic" {
        panic("这是一个模拟的运行时panic!")
    }

    // 正常响应
    fmt.Fprint(w, "欢迎来到首页！")
    return nil // 没有错误发生
}

func main() {
    // 将自定义Handler包装器应用于业务处理函数
    // 注意：http.Handle 期望一个 http.Handler 接口，而我们定义的 Handler 类型实现了这个接口。
    http.Handle("/", Handler(myPageHandler))
    http.Handle("/home", Handler(myPageHandler))
    http.Handle("/error", Handler(myPageHandler))
    http.Handle("/panic", Handler(myPageHandler))

    log.Println("服务器启动在 :8080，请访问 /home, /error, /panic 或其他路径")
    log.Fatal(http.ListenAndServe(":8080", nil))
}