feat 初次提交

log/level.go

@@ -0,0 +1,23 @@
+package log
+
+import (
+	"go.uber.org/zap"
+	"go.uber.org/zap/zapcore"
+)
+
+var logLevelMap = map[string]zapcore.Level{
+	"panic": zap.DPanicLevel,
+	"fatal": zap.FatalLevel,
+	"error": zap.ErrorLevel,
+	"warn":  zap.WarnLevel,
+	"info":  zap.InfoLevel,
+	"debug": zap.DebugLevel,
+}
+
+// GetLogLevel get the logLevel from logLevelName
+func GetLogLevel(logLevelName string) zapcore.Level {
+	if v, ok := logLevelMap[logLevelName]; ok {
+		return v
+	}
+	return zap.DebugLevel
+}

log/log.go

@@ -0,0 +1,66 @@
+package log
+
+import "go.uber.org/zap"
+
+var globalLogger *zap.SugaredLogger
+
+// SetLogger 设置日志记录器
+func SetLogger(logger *zap.SugaredLogger) {
+	if logger == nil {
+		return
+	}
+	globalLogger = logger
+}
+
+func GetLogger() *zap.SugaredLogger {
+	return globalLogger
+}
+
+// Debugf 打印调试模板日志
+func Debugf(format string, a ...interface{}) {
+	if globalLogger != nil {
+		globalLogger.Debugf(format, a...)
+	}
+}
+
+// Infof 打印信息模板日志
+func Infof(format string, a ...interface{}) {
+	if globalLogger != nil {
+		globalLogger.Infof(format, a...)
+	}
+}
+
+// Warnf 打印警告模板日志
+func Warnf(format string, a ...interface{}) {
+	if globalLogger != nil {
+		globalLogger.Warnf(format, a...)
+	}
+}
+
+// Errorf 打印错误模板日志
+func Errorf(format string, a ...interface{}) {
+	if globalLogger != nil {
+		globalLogger.Errorf(format, a...)
+	}
+}
+
+// Panicf 打印Panic模板日志
+func Panicf(format string, a ...interface{}) {
+	if globalLogger != nil {
+		globalLogger.Panicf(format, a...)
+	}
+}
+
+// Fatalf 打印致命错误模板日志
+func Fatalf(format string, a ...interface{}) {
+	if globalLogger != nil {
+		globalLogger.Fatalf(format, a...)
+	}
+}
+
+// Close 关闭日志
+func Close() {
+	if globalLogger != nil {
+		_ = globalLogger.Sync()
+	}
+}

log/log_zap.go

@@ -0,0 +1,63 @@
+package log
+
+import (
+	"github.com/natefinch/lumberjack"
+	"go.uber.org/zap"
+	"go.uber.org/zap/zapcore"
+	"os"
+)
+
+func Init(debug bool, maxSize, maxBackups, maxAge int32, level string) {
+	// 格式配置
+	jsonConfig := zapcore.EncoderConfig{
+		MessageKey:     "M",
+		LevelKey:       "L",
+		TimeKey:        "T",
+		NameKey:        "N",
+		CallerKey:      "C",
+		FunctionKey:    zapcore.OmitKey,
+		StacktraceKey:  "S",
+		LineEnding:     zapcore.DefaultLineEnding,
+		EncodeLevel:    zapcore.CapitalLevelEncoder,
+		EncodeTime:     zapcore.TimeEncoderOfLayout("01-02 15:04:05.000"),
+		EncodeDuration: zapcore.SecondsDurationEncoder,
+		EncodeCaller:   zapcore.ShortCallerEncoder,
+		EncodeName: func(loggerName string, enc zapcore.PrimitiveArrayEncoder) {
+			enc.AppendString("[" + loggerName + "]")
+		},
+		ConsoleSeparator: " ",
+	}
+
+	// 日志输出到控制台和文件
+	writeSyncer := []zapcore.WriteSyncer{zapcore.AddSync(os.Stdout)}
+	if !debug {
+		writeSyncer = append(writeSyncer, zapcore.AddSync(&lumberjack.Logger{
+			Filename:   "./logs/log.log", // 日志文件位置
+			MaxSize:    int(maxSize),     // 最大文件大小（MB）
+			MaxBackups: int(maxBackups),  // 保留旧文件的最大个数
+			MaxAge:     int(maxAge),      // 保留旧文件的最大天数
+			Compress:   false,            // 是否压缩/归档旧文件
+			LocalTime:  true,
+		}))
+	}
+
+	var encoder zapcore.Encoder
+	if debug {
+		encoder = zapcore.NewConsoleEncoder(jsonConfig)
+	} else {
+		encoder = zapcore.NewJSONEncoder(jsonConfig)
+	}
+	logger := zap.New(zapcore.NewCore(
+		encoder,
+		zapcore.NewMultiWriteSyncer(writeSyncer...),
+		zap.NewAtomicLevelAt(GetLogLevel(level)),
+	))
+	if debug {
+		logger = logger.WithOptions(
+			zap.AddCaller(),
+			zap.AddCallerSkip(1),
+			zap.AddStacktrace(zapcore.ErrorLevel),
+		)
+	}
+	SetLogger(logger.Sugar())
+}