fix
This commit is contained in:
76
backend-old/pkg/logger/logger.go
Normal file
76
backend-old/pkg/logger/logger.go
Normal file
@ -0,0 +1,76 @@
|
||||
package logger
|
||||
|
||||
import (
|
||||
"os"
|
||||
"go.uber.org/zap"
|
||||
"go.uber.org/zap/zapcore"
|
||||
"gopkg.in/natefinch/lumberjack.v2"
|
||||
"photography-backend/internal/config"
|
||||
)
|
||||
|
||||
// InitLogger 初始化日志记录器
|
||||
func InitLogger(cfg *config.LoggerConfig) (*zap.Logger, error) {
|
||||
// 设置日志级别
|
||||
var level zapcore.Level
|
||||
switch cfg.Level {
|
||||
case "debug":
|
||||
level = zapcore.DebugLevel
|
||||
case "info":
|
||||
level = zapcore.InfoLevel
|
||||
case "warn":
|
||||
level = zapcore.WarnLevel
|
||||
case "error":
|
||||
level = zapcore.ErrorLevel
|
||||
default:
|
||||
level = zapcore.InfoLevel
|
||||
}
|
||||
|
||||
// 创建编码器配置
|
||||
encoderConfig := zap.NewProductionEncoderConfig()
|
||||
encoderConfig.TimeKey = "timestamp"
|
||||
encoderConfig.EncodeTime = zapcore.ISO8601TimeEncoder
|
||||
encoderConfig.EncodeLevel = zapcore.CapitalLevelEncoder
|
||||
|
||||
// 创建编码器
|
||||
var encoder zapcore.Encoder
|
||||
if cfg.Format == "json" {
|
||||
encoder = zapcore.NewJSONEncoder(encoderConfig)
|
||||
} else {
|
||||
encoder = zapcore.NewConsoleEncoder(encoderConfig)
|
||||
}
|
||||
|
||||
// 创建写入器
|
||||
var writers []zapcore.WriteSyncer
|
||||
|
||||
// 控制台输出
|
||||
writers = append(writers, zapcore.AddSync(os.Stdout))
|
||||
|
||||
// 文件输出
|
||||
if cfg.Output == "file" && cfg.Filename != "" {
|
||||
// 确保日志目录存在
|
||||
if err := os.MkdirAll("logs", 0755); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
fileWriter := &lumberjack.Logger{
|
||||
Filename: cfg.Filename,
|
||||
MaxSize: cfg.MaxSize,
|
||||
MaxAge: cfg.MaxAge,
|
||||
MaxBackups: 10,
|
||||
LocalTime: true,
|
||||
Compress: cfg.Compress,
|
||||
}
|
||||
writers = append(writers, zapcore.AddSync(fileWriter))
|
||||
}
|
||||
|
||||
// 合并写入器
|
||||
writer := zapcore.NewMultiWriteSyncer(writers...)
|
||||
|
||||
// 创建核心
|
||||
core := zapcore.NewCore(encoder, writer, level)
|
||||
|
||||
// 创建日志记录器
|
||||
logger := zap.New(core, zap.AddCaller(), zap.AddCallerSkip(1))
|
||||
|
||||
return logger, nil
|
||||
}
|
||||
Reference in New Issue
Block a user