104 lines
2.7 KiB
Go
104 lines
2.7 KiB
Go
package middleware
|
||
|
||
import (
|
||
"net/http"
|
||
"runtime/debug"
|
||
"time"
|
||
|
||
"github.com/gin-gonic/gin"
|
||
"go.uber.org/zap"
|
||
)
|
||
|
||
// PanicRecoveryMiddleware Panic恢复中间件
|
||
type PanicRecoveryMiddleware struct {
|
||
logger *zap.Logger
|
||
}
|
||
|
||
// NewPanicRecoveryMiddleware 创建Panic恢复中间件
|
||
func NewPanicRecoveryMiddleware(logger *zap.Logger) *PanicRecoveryMiddleware {
|
||
return &PanicRecoveryMiddleware{
|
||
logger: logger,
|
||
}
|
||
}
|
||
|
||
// GetName 返回中间件名称
|
||
func (m *PanicRecoveryMiddleware) GetName() string {
|
||
return "panic_recovery"
|
||
}
|
||
|
||
// GetPriority 返回中间件优先级
|
||
func (m *PanicRecoveryMiddleware) GetPriority() int {
|
||
return 100 // 最高优先级，第一个执行
|
||
}
|
||
|
||
// Handle 返回中间件处理函数
|
||
func (m *PanicRecoveryMiddleware) Handle() gin.HandlerFunc {
|
||
return gin.RecoveryWithWriter(&panicLogger{logger: m.logger})
|
||
}
|
||
|
||
// IsGlobal 是否为全局中间件
|
||
func (m *PanicRecoveryMiddleware) IsGlobal() bool {
|
||
return true
|
||
}
|
||
|
||
// panicLogger 实现io.Writer接口，用于记录panic信息
|
||
type panicLogger struct {
|
||
logger *zap.Logger
|
||
}
|
||
|
||
// Write 实现io.Writer接口
|
||
func (pl *panicLogger) Write(p []byte) (n int, err error) {
|
||
pl.logger.Error("系统发生严重错误",
|
||
zap.String("error_type", "panic"),
|
||
zap.String("stack_trace", string(p)),
|
||
zap.String("timestamp", time.Now().Format("2006-01-02 15:04:05")),
|
||
)
|
||
return len(p), nil
|
||
}
|
||
|
||
// CustomPanicRecovery 自定义panic恢复中间件
|
||
func (m *PanicRecoveryMiddleware) CustomPanicRecovery() gin.HandlerFunc {
|
||
return func(c *gin.Context) {
|
||
defer func() {
|
||
if err := recover(); err != nil {
|
||
// 获取请求信息
|
||
requestID := c.GetString("request_id")
|
||
traceID := c.GetString("trace_id")
|
||
userID := c.GetString("user_id")
|
||
clientIP := c.ClientIP()
|
||
method := c.Request.Method
|
||
path := c.Request.URL.Path
|
||
userAgent := c.Request.UserAgent()
|
||
|
||
// 记录详细的panic信息
|
||
m.logger.Error("系统发生严重错误",
|
||
zap.Any("panic_error", err),
|
||
zap.String("error_type", "panic"),
|
||
zap.String("request_id", requestID),
|
||
zap.String("trace_id", traceID),
|
||
zap.String("user_id", userID),
|
||
zap.String("client_ip", clientIP),
|
||
zap.String("method", method),
|
||
zap.String("path", path),
|
||
zap.String("user_agent", userAgent),
|
||
zap.String("stack_trace", string(debug.Stack())),
|
||
zap.String("timestamp", time.Now().Format("2006-01-02 15:04:05")),
|
||
)
|
||
|
||
// 返回500错误响应
|
||
c.JSON(http.StatusInternalServerError, gin.H{
|
||
"success": false,
|
||
"message": "服务器内部错误",
|
||
"error_code": "INTERNAL_SERVER_ERROR",
|
||
"request_id": requestID,
|
||
"timestamp": time.Now().Unix(),
|
||
})
|
||
|
||
// 中止请求处理
|
||
c.Abort()
|
||
}
|
||
}()
|
||
|
||
c.Next()
|
||
}
|
||
} |