121 lines
3.0 KiB
Go
121 lines
3.0 KiB
Go
//
|
||
// log.go
|
||
// Copyright (C) 2022 tiglog <me@tiglog.com>
|
||
//
|
||
// Distributed under terms of the MIT license.
|
||
//
|
||
|
||
package logger
|
||
|
||
import (
|
||
"fmt"
|
||
"io"
|
||
"os"
|
||
|
||
"go.uber.org/zap"
|
||
"go.uber.org/zap/zapcore"
|
||
)
|
||
|
||
type Level = zapcore.Level
|
||
|
||
const (
|
||
DebugLevel = zapcore.DebugLevel
|
||
InfoLevel = zapcore.InfoLevel
|
||
WarnLevel = zapcore.WarnLevel
|
||
ErrorLevel = zapcore.ErrorLevel
|
||
PanicLevel = zapcore.PanicLevel
|
||
FatalLevel = zapcore.FatalLevel
|
||
)
|
||
|
||
type Logger struct {
|
||
l *zap.Logger
|
||
// https://pkg.go.dev/go.uber.org/zap#example-AtomicLevel
|
||
al *zap.AtomicLevel
|
||
}
|
||
|
||
func New(out io.Writer, level Level, opts ...Option) *Logger {
|
||
if out == nil {
|
||
out = os.Stderr
|
||
}
|
||
|
||
al := zap.NewAtomicLevelAt(level)
|
||
cfg := zap.NewProductionEncoderConfig()
|
||
cfg.EncodeTime = zapcore.RFC3339TimeEncoder
|
||
|
||
core := zapcore.NewCore(
|
||
zapcore.NewJSONEncoder(cfg),
|
||
zapcore.AddSync(out),
|
||
al,
|
||
)
|
||
return &Logger{l: zap.New(core, opts...), al: &al}
|
||
}
|
||
|
||
// SetLevel 动态更改日志级别
|
||
// 对于使用 NewTee 创建的 Logger 无效，因为 NewTee 本意是根据不同日志级别
|
||
// 创建的多个 zap.Core，不应该通过 SetLevel 将多个 zap.Core 日志级别统一
|
||
func (l *Logger) SetLevel(level Level) {
|
||
if l.al != nil {
|
||
l.al.SetLevel(level)
|
||
}
|
||
}
|
||
|
||
type Field = zap.Field
|
||
|
||
func (l *Logger) Debug(msg string, fields ...Field) {
|
||
l.l.Debug(msg, fields...)
|
||
}
|
||
|
||
func (l *Logger) Info(msg string, fields ...Field) {
|
||
l.l.Info(msg, fields...)
|
||
}
|
||
|
||
func (l *Logger) Warn(msg string, fields ...Field) {
|
||
l.l.Warn(msg, fields...)
|
||
}
|
||
|
||
func (l *Logger) Error(msg string, fields ...Field) {
|
||
l.l.Error(msg, fields...)
|
||
}
|
||
|
||
func (l *Logger) Panic(msg string, fields ...Field) {
|
||
l.l.Panic(msg, fields...)
|
||
}
|
||
|
||
func (l *Logger) Fatal(msg string, fields ...Field) {
|
||
l.l.Fatal(msg, fields...)
|
||
}
|
||
|
||
func (l *Logger) Sync() error {
|
||
return l.l.Sync()
|
||
}
|
||
|
||
var std = New(os.Stderr, InfoLevel)
|
||
|
||
func Default() *Logger { return std }
|
||
func ReplaceDefault(l *Logger) { std = l }
|
||
|
||
func SetLevel(level Level) { std.SetLevel(level) }
|
||
|
||
func With(fields ...Field) {
|
||
std.l.Core().With(fields)
|
||
}
|
||
func WithRequestId(reqId string) {
|
||
std.l.Core().With([]Field{String("reqId", reqId)})
|
||
}
|
||
|
||
func Debug(msg string, fields ...Field) { std.Debug(msg, fields...) }
|
||
func Info(msg string, fields ...Field) { std.Info(msg, fields...) }
|
||
func Warn(msg string, fields ...Field) { std.Warn(msg, fields...) }
|
||
func Error(msg string, fields ...Field) { std.Error(msg, fields...) }
|
||
func Panic(msg string, fields ...Field) { std.Panic(msg, fields...) }
|
||
func Fatal(msg string, fields ...Field) { std.Fatal(msg, fields...) }
|
||
|
||
func Debugf(format string, args ...any) { std.Debug(fmt.Sprintf(format, args...)) }
|
||
func Infof(format string, args ...any) { std.Info(fmt.Sprintf(format, args...)) }
|
||
func Warnf(format string, args ...any) { std.Warn(fmt.Sprintf(format, args...)) }
|
||
func Errorf(format string, args ...any) { std.Error(fmt.Sprintf(format, args...)) }
|
||
func Panicf(format string, args ...any) { std.Panic(fmt.Sprintf(format, args...)) }
|
||
func Fatalf(format string, args ...any) { std.Fatal(fmt.Sprintf(format, args...)) }
|
||
|
||
func Sync() error { return std.Sync() }
|