如何用Golang实现Web服务的异常处理_Golang Web异常处理方法汇总

func recoverMiddleware(next http.HandlerFunc) http.HandlerFunc {
    return func(w http.ResponseWriter, r *http.Request) {
        defer func() {
            if err := recover(); err != nil {
                log.Printf("Panic: %v\n", err)
                http.Error(w, "Internal Server Error", http.StatusInternalServerError)
            }
        }()
        next(w, r)
    }
}

func handler(w http.ResponseWriter, r *http.Request) {
    panic("something went wrong")
}

type Response struct {
    Code    int         `json:"code"`
    Message string      `json:"message"`
    Data    interface{} `json:"data,omitempty"`
}

func writeError(w http.ResponseWriter, code int, message string) {
    w.Header().Set("Content-Type", "application/json")
    w.WriteHeader(code)
    json.NewEncoder(w).Encode(Response{
        Code:    code,
        Message: message,
    })
}

if user == nil {
    writeError(w, 404, "User not found")
    return
}

func RecoveryHandler() gin.HandlerFunc {
    return func(c *gin.Context) {
        defer func() {
            if err := recover(); err != nil {
                c.JSON(500, gin.H{
                    "code":    500,
                    "message": "Internal server error",
                })
                c.Abort()
            }
        }()
        c.Next()
    }
}

// 使用
r := gin.New()
r.Use(RecoveryHandler())

type AppError struct {
    Code    int
    Message string
}

func (e AppError) Error() string {
    return e.Message
}

if err != nil {
    if appErr, ok := err.(AppError); ok {
        writeError(w, appErr.Code, appErr.Message)
    } else {
        writeError(w, 500, "Internal error")
    }
    return
}

log.Printf("Panic: %v\nStack: %s", err, debug.Stack())