32 lines
628 B
Go
32 lines
628 B
Go
package middleware
|
|
|
|
import (
|
|
"fmt"
|
|
"net/http"
|
|
"runtime/debug"
|
|
|
|
"hfb_jysh/backend/pkg/response"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"go.uber.org/zap"
|
|
)
|
|
|
|
const contextPanicStack = "panic_stack"
|
|
|
|
func Recovery(_ *zap.Logger) gin.HandlerFunc {
|
|
return func(c *gin.Context) {
|
|
defer func() {
|
|
if recovered := recover(); recovered != nil {
|
|
err := fmt.Errorf("panic: %v", recovered)
|
|
_ = c.Error(err)
|
|
c.Set(contextPanicStack, string(debug.Stack()))
|
|
if !c.Writer.Written() {
|
|
response.Error(c, http.StatusInternalServerError, "internal_error", "服务暂时不可用")
|
|
}
|
|
c.Abort()
|
|
}
|
|
}()
|
|
c.Next()
|
|
}
|
|
}
|