Golang实现自定义recovery中间件

package main
import (
    "github.com/gin-gonic/gin"
    "log"
    "net/http"
    "runtime"
)
func Recovery() gin.HandlerFunc {
    return func(c *gin.Context) {
        defer func() {
            if err := recover(); err != nil {
                const size = 64 << 10
                stack := make([]byte, size)
                stack = stack[:runtime.Stack(stack, false)]
                log.Printf("[GinPanic] %s\n", string(stack))
                c.JSON(http.StatusOK, struct {
                    Code int         `json:"code"`
                    Data interface{} `json:"data"`
                    Msg  string      `json:"msg"`
                }{
                    Code: -1,
                    Data: nil,
                    Msg:  "server panic",
                })
                c.Abort()
            }
        }()
        c.Next()
    }
}

package main
import (
    "github.com/gin-gonic/gin"
    "runtime"
)
func main() {
    r := gin.New()
    r.Use(Recovery())
    r.GET("/test", func(c *gin.Context) {
        panic("Oops! Something went wrong.")
    })
    r.Run(":8080")
}

package main
import (
    "fmt"
    "log"
    "net/http"
    "runtime/debug"
)
// 自定义的recovery中间件
func recoveryMiddleware(next http.Handler) http.Handler {
    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        defer func() {
            if err := recover(); err != nil {
                // 打印错误信息
                log.Println("[Recovery] Panic:", err)
                // 打印堆栈跟踪信息
                log.Printf("[Recovery] Stack Trace:\n%s\n", debug.Stack())
                // 返回一个适当的错误响应给客户端
                fmt.Fprintf(w, `{"code":-1,"data":null,"msg":"server panic"}`)
            }
        }()
        // 继续处理下一个中间件或路由处理函数
        next.ServeHTTP(w, r)
    })
}
// 示例的处理函数
func helloHandler(w http.ResponseWriter, r *http.Request) {
    panic("Oops! Something went wrong.") // 模拟一个错误
    w.Write([]byte("Hello, Recovery Middleware!"))
}
func main() {
    // 创建一个多路复用器
    mux := http.NewServeMux()
    // 注册中间件
    mux.Handle("/test", recoveryMiddleware(http.HandlerFunc(helloHandler)))
    // 创建服务器
    server := &http.Server{
        Addr:    ":8080",
        Handler: mux,
    }
    // 启动服务器
    log.Println("Server is running on http://localhost:8080")
    log.Fatal(server.ListenAndServe())
}