Hericium-sys
128 lines
2.9 KiB
Go
128 lines
2.9 KiB
Go
package middlewares
|
|
|
|
import (
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
"time"
|
|
|
|
"skeleton/config"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
rotatelogs "github.com/lestrrat-go/file-rotatelogs"
|
|
"go.uber.org/zap"
|
|
"go.uber.org/zap/zapcore"
|
|
)
|
|
|
|
var Logger *zap.Logger
|
|
|
|
// InitLogger 初始化日志系统
|
|
func InitLogger(cfg *config.LoggerConfig) error {
|
|
// 配置编码器
|
|
var encoderConfig zapcore.EncoderConfig
|
|
if cfg.Format == "json" {
|
|
encoderConfig = zap.NewProductionEncoderConfig()
|
|
} else {
|
|
encoderConfig = zap.NewDevelopmentEncoderConfig()
|
|
encoderConfig.EncodeLevel = zapcore.CapitalColorLevelEncoder
|
|
}
|
|
encoderConfig.TimeKey = "timestamp"
|
|
encoderConfig.EncodeTime = zapcore.ISO8601TimeEncoder
|
|
|
|
// 选择编码器
|
|
var encoder zapcore.Encoder
|
|
if cfg.Format == "json" {
|
|
encoder = zapcore.NewJSONEncoder(encoderConfig)
|
|
} else {
|
|
encoder = zapcore.NewConsoleEncoder(encoderConfig)
|
|
}
|
|
|
|
// 配置日志级别
|
|
level := zap.InfoLevel
|
|
switch strings.ToLower(cfg.Level) {
|
|
case "debug":
|
|
level = zap.DebugLevel
|
|
case "info":
|
|
level = zap.InfoLevel
|
|
case "warn":
|
|
level = zap.WarnLevel
|
|
case "error":
|
|
level = zap.ErrorLevel
|
|
}
|
|
|
|
// 配置输出
|
|
var writeSyncer zapcore.WriteSyncer
|
|
if cfg.Output == "file" {
|
|
// 创建日志目录
|
|
logDir := filepath.Dir(cfg.Filename)
|
|
if err := os.MkdirAll(logDir, 0755); err != nil {
|
|
return err
|
|
}
|
|
|
|
// 构建带日期的文件名模式
|
|
ext := filepath.Ext(cfg.Filename)
|
|
nameWithoutExt := strings.TrimSuffix(cfg.Filename, ext)
|
|
|
|
// 创建按日期轮转的日志写入器
|
|
writer, err := rotatelogs.New(
|
|
nameWithoutExt+".%Y%m%d"+ext,
|
|
rotatelogs.WithLinkName(cfg.Filename),
|
|
rotatelogs.WithRotationTime(24*time.Hour),
|
|
rotatelogs.WithMaxAge(time.Duration(cfg.MaxAge)*24*time.Hour),
|
|
)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
writeSyncer = zapcore.AddSync(writer)
|
|
} else {
|
|
writeSyncer = zapcore.AddSync(os.Stdout)
|
|
}
|
|
|
|
// 创建core
|
|
core := zapcore.NewCore(encoder, writeSyncer, level)
|
|
|
|
// 创建logger
|
|
Logger = zap.New(core, zap.AddCaller(), zap.AddStacktrace(zapcore.ErrorLevel))
|
|
|
|
// 设置为全局logger，这样就可以使用 zap.L() 访问
|
|
zap.ReplaceGlobals(Logger)
|
|
|
|
return nil
|
|
}
|
|
|
|
// GinLogger 返回gin的日志中间件
|
|
func GinLogger() gin.HandlerFunc {
|
|
return gin.LoggerWithWriter(os.Stdout)
|
|
}
|
|
|
|
// GinRecovery 返回gin的恢复中间件
|
|
func GinRecovery() gin.HandlerFunc {
|
|
return gin.RecoveryWithWriter(os.Stdout)
|
|
}
|
|
|
|
// ErrorLogging 错误响应日志中间件
|
|
func ErrorLogging() gin.HandlerFunc {
|
|
return func(c *gin.Context) {
|
|
c.Next()
|
|
|
|
// 记录错误状态码的响应
|
|
if c.Writer.Status() >= 400 {
|
|
Logger.Warn("HTTP错误响应",
|
|
zap.String("method", c.Request.Method),
|
|
zap.String("path", c.Request.URL.Path),
|
|
zap.String("client_ip", c.ClientIP()),
|
|
zap.Int("status_code", c.Writer.Status()),
|
|
zap.String("user_agent", c.GetHeader("User-Agent")),
|
|
)
|
|
}
|
|
}
|
|
}
|
|
|
|
// Sync 同步日志缓冲区
|
|
func Sync() {
|
|
if Logger != nil {
|
|
Logger.Sync()
|
|
}
|
|
}
|