golib/logger/log.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
	fields map[string]Field
}

func New(out io.Writer, level Level, opts ...Option) *Logger {
	if out == nil {
		out = os.Stderr
	}

	var al = zap.NewAtomicLevelAt(level)
	var cfg zap.Config
	if level == DebugLevel {
		cfg = zap.NewDevelopmentConfig()
	} else {
		cfg = zap.NewProductionConfig()
	}

	core := zapcore.NewCore(
		zapcore.NewJSONEncoder(cfg.EncoderConfig),
		zapcore.AddSync(out),
		al,
	)
	return &Logger{l: zap.New(core, opts...), al: &al, fields: make(map[string]Field)}
}

// 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 mergeFields(fields []Field) []Field {
	var mfields = []Field{}
	for _, field := range std.fields {
		mfields = append(mfields, field)
	}
	mfields = append(mfields, fields...)
	return mfields
}

func (l *Logger) Debug(msg string, fields ...Field) {
	l.l.Debug(msg, mergeFields(fields)...)
}

func (l *Logger) Info(msg string, fields ...Field) {
	l.l.Info(msg, mergeFields(fields)...)
}

func (l *Logger) Warn(msg string, fields ...Field) {
	l.l.Warn(msg, mergeFields(fields)...)
}

func (l *Logger) Error(msg string, fields ...Field) {
	l.l.Error(msg, mergeFields(fields)...)
}

func (l *Logger) Panic(msg string, fields ...Field) {
	l.l.Panic(msg, mergeFields(fields)...)
}

func (l *Logger) Fatal(msg string, fields ...Field) {
	l.l.Fatal(msg, mergeFields(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(field Field) {
	std.fields[field.Key] = field
}

func WithTarget(target string) {
	With(String("target", target))
}

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() }