diff options
| author | Bobby <[email protected]> | 2026-03-03 16:54:33 +0530 |
|---|---|---|
| committer | Bobby <[email protected]> | 2026-03-03 16:54:33 +0530 |
| commit | f6e671264734b8d82b48102041ad7f92ec2f1048 (patch) | |
| tree | e0bd44e095f5732c698f7adbfe64680ca424a545 /shrine/utils/logger/logger.go | |
| parent | 8631dd7420c35ce8c8b7da5f19eb4b516abcddc3 (diff) | |
| download | pagoda-f6e671264734b8d82b48102041ad7f92ec2f1048.tar.xz pagoda-f6e671264734b8d82b48102041ad7f92ec2f1048.zip | |
refactor: standardize context variable naming in middleware and meta packages
Diffstat (limited to 'shrine/utils/logger/logger.go')
| -rw-r--r-- | shrine/utils/logger/logger.go | 85 |
1 files changed, 35 insertions, 50 deletions
diff --git a/shrine/utils/logger/logger.go b/shrine/utils/logger/logger.go index 9fc1bc7..176a173 100644 --- a/shrine/utils/logger/logger.go +++ b/shrine/utils/logger/logger.go @@ -4,7 +4,6 @@ import ( "fmt" "os" "strings" - "time" "go.uber.org/zap" "go.uber.org/zap/zapcore" @@ -15,23 +14,20 @@ const prefixWidth = 15 var ( loggerInstance *zap.Logger level zap.AtomicLevel - showTimestamp bool ) -func timeEncoder(t time.Time, enc zapcore.PrimitiveArrayEncoder) { - enc.AppendString(Gray + t.Format(time.RFC3339) + Reset) -} - -func levelEncoder(l zapcore.Level, enc zapcore.PrimitiveArrayEncoder) { +func levelBadge(l logLevel) string { switch l { - case zapcore.DebugLevel: - enc.AppendString(LevelColorDebug) - case zapcore.WarnLevel: - enc.AppendString(LevelColorWarn) - case zapcore.ErrorLevel: - enc.AppendString(LevelColorError) + case levelDebug: + return levelColorDebug + case levelWarn: + return levelColorWarn + case levelError: + return levelColorError + case levelSuccess: + return levelColorSuccess default: - enc.AppendString(LevelColorInfo) + return levelColorInfo } } @@ -45,21 +41,21 @@ func formatPrefix(prefix string) string { padding = strings.Repeat(" ", prefixWidth-len(prefix)) } - return Cyan + "[" + prefix + "]" + Reset + padding + return cyan + "[" + prefix + "]" + reset + padding } -func colorMessage(level LogLevel, msg string) string { - switch level { - case Debug: - return MessageColorDebug + msg + Reset - case Warn: - return MessageColorWarn + msg + Reset - case Error: - return MessageColorError + msg + Reset - case Success: - return MessageColorSuccess + msg + Reset +func colorMessage(l logLevel, msg string) string { + switch l { + case levelDebug: + return messageColorDebug + msg + reset + case levelWarn: + return messageColorWarn + msg + reset + case levelError: + return messageColorError + msg + reset + case levelSuccess: + return messageColorSuccess + msg + reset default: - return MessageColorInfo + msg + Reset + return messageColorInfo + msg + reset } } @@ -67,15 +63,8 @@ func Init() { level = zap.NewAtomicLevelAt(zapcore.InfoLevel) encoderCfg := zapcore.EncoderConfig{ - LevelKey: "level", - MessageKey: "msg", - LineEnding: "\n", - EncodeLevel: levelEncoder, - } - - if showTimestamp { - encoderCfg.TimeKey = "ts" - encoderCfg.EncodeTime = timeEncoder + MessageKey: "msg", + LineEnding: "\n", } encoder := zapcore.NewConsoleEncoder(encoderCfg) @@ -92,11 +81,7 @@ func Init() { })), ) - loggerInstance = zap.New(core, zap.AddCaller()) -} - -func SetTimestamp(enabled bool) { - showTimestamp = enabled + loggerInstance = zap.New(core) } func SetDebug(enabled bool) { @@ -108,38 +93,38 @@ func SetDebug(enabled bool) { } func Debugf(prefix, format string, args ...any) { - log(Debug, zapcore.DebugLevel, prefix, fmt.Sprintf(format, args...)) + log(levelDebug, zapcore.DebugLevel, prefix, fmt.Sprintf(format, args...)) } func Infof(prefix, format string, args ...any) { - log(Info, zapcore.InfoLevel, prefix, fmt.Sprintf(format, args...)) + log(levelInfo, zapcore.InfoLevel, prefix, fmt.Sprintf(format, args...)) } func Successf(prefix, format string, args ...any) { - log(Success, zapcore.InfoLevel, prefix, fmt.Sprintf(format, args...)) + log(levelSuccess, zapcore.InfoLevel, prefix, fmt.Sprintf(format, args...)) } func Warnf(prefix, format string, args ...any) { - log(Warn, zapcore.WarnLevel, prefix, fmt.Sprintf(format, args...)) + log(levelWarn, zapcore.WarnLevel, prefix, fmt.Sprintf(format, args...)) } func Errorf(prefix, format string, args ...any) { - log(Error, zapcore.ErrorLevel, prefix, fmt.Sprintf(format, args...)) + log(levelError, zapcore.ErrorLevel, prefix, fmt.Sprintf(format, args...)) } func Fatalf(prefix, format string, args ...any) { - log(Error, zapcore.ErrorLevel, prefix, fmt.Sprintf(format, args...)) + log(levelError, zapcore.ErrorLevel, prefix, fmt.Sprintf(format, args...)) os.Exit(1) } -func log(levelLabel LogLevel, zapLevel zapcore.Level, prefix string, msg any) { +func log(label logLevel, zapLevel zapcore.Level, prefix string, msg any) { if loggerInstance == nil { panic("logger.Init() was not called") } message := fmt.Sprint(msg) - colored := colorMessage(levelLabel, message) - fullMessage := formatPrefix(prefix) + colored + colored := colorMessage(label, message) + fullMessage := levelBadge(label) + " " + formatPrefix(prefix) + colored loggerInstance.Log(zapLevel, fullMessage) -} +}
\ No newline at end of file |
