68 lines
2.0 KiB
Go
68 lines
2.0 KiB
Go
|
package middleware
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"net/http"
|
|||
|
|
"runtime/debug"
|
|||
|
|
|
|||
|
|
"github.com/gin-gonic/gin"
|
|||
|
|
"github.com/Nanqipro/YunQue-Tech-Projects/ai_english_learning/serve/internal/logger"
|
|||
|
|
"github.com/Nanqipro/YunQue-Tech-Projects/ai_english_learning/serve/internal/common"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// ErrorHandler 全局错误处理中间件
|
|||
|
|
func ErrorHandler() gin.HandlerFunc {
|
|||
|
|
return gin.CustomRecovery(func(c *gin.Context, recovered interface{}) {
|
|||
|
|
if err, ok := recovered.(string); ok {
|
|||
|
|
logger.WithFields(map[string]interface{}{
|
|||
|
|
"error": err,
|
|||
|
|
"method": c.Request.Method,
|
|||
|
|
"path": c.Request.URL.Path,
|
|||
|
|
"ip": c.ClientIP(),
|
|||
|
|
"user_agent": c.Request.UserAgent(),
|
|||
|
|
"stack": string(debug.Stack()),
|
|||
|
|
}).Error("Panic recovered")
|
|||
|
|
}
|
|||
|
|
if err, ok := recovered.(error); ok {
|
|||
|
|
logger.WithFields(map[string]interface{}{
|
|||
|
|
"error": err.Error(),
|
|||
|
|
"method": c.Request.Method,
|
|||
|
|
"path": c.Request.URL.Path,
|
|||
|
|
"ip": c.ClientIP(),
|
|||
|
|
"user_agent": c.Request.UserAgent(),
|
|||
|
|
"stack": string(debug.Stack()),
|
|||
|
|
}).Error("Panic recovered")
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 返回统一的错误响应
|
|||
|
|
common.ErrorResponse(c, http.StatusInternalServerError, "Internal server error")
|
|||
|
|
c.Abort()
|
|||
|
|
})
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// RequestLogger 请求日志中间件
|
|||
|
|
func RequestLogger() gin.HandlerFunc {
|
|||
|
|
return gin.LoggerWithFormatter(func(param gin.LogFormatterParams) string {
|
|||
|
|
logger.WithFields(map[string]interface{}{
|
|||
|
|
"timestamp": param.TimeStamp.Format("2006-01-02 15:04:05"),
|
|||
|
|
"status_code": param.StatusCode,
|
|||
|
|
"latency": param.Latency.String(),
|
|||
|
|
"client_ip": param.ClientIP,
|
|||
|
|
"method": param.Method,
|
|||
|
|
"path": param.Path,
|
|||
|
|
"user_agent": param.Request.UserAgent(),
|
|||
|
|
"error": param.ErrorMessage,
|
|||
|
|
}).Info("HTTP Request")
|
|||
|
|
return ""
|
|||
|
|
})
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
// RateLimiter 简单的速率限制中间件（基于IP）
|
|||
|
|
func RateLimiter() gin.HandlerFunc {
|
|||
|
|
// 这里可以集成更复杂的限流库，如 golang.org/x/time/rate
|
|||
|
|
return func(c *gin.Context) {
|
|||
|
|
// 简单实现，实际项目中应该使用更完善的限流算法
|
|||
|
|
c.Next()
|
|||
|
|
}
|
|||
|
|
}
|