init
This commit is contained in:
wangjian
commit
2a9d1efba1
|
|
@ -0,0 +1,14 @@
|
||||||
|
module logging
|
||||||
|
|
||||||
|
go 1.17
|
||||||
|
|
||||||
|
require (
|
||||||
|
go.uber.org/zap v1.21.0
|
||||||
|
gopkg.in/natefinch/lumberjack.v2 v2.0.0
|
||||||
|
)
|
||||||
|
|
||||||
|
require (
|
||||||
|
github.com/BurntSushi/toml v1.1.0 // indirect
|
||||||
|
go.uber.org/atomic v1.7.0 // indirect
|
||||||
|
go.uber.org/multierr v1.6.0 // indirect
|
||||||
|
)
|
||||||
|
|
@ -0,0 +1,266 @@
|
||||||
|
package logging
|
||||||
|
|
||||||
|
import (
|
||||||
|
"os"
|
||||||
|
"path/filepath"
|
||||||
|
"sync"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"go.uber.org/zap"
|
||||||
|
"go.uber.org/zap/zapcore"
|
||||||
|
"gopkg.in/natefinch/lumberjack.v2"
|
||||||
|
)
|
||||||
|
|
||||||
|
type ModOptions func(options *Options)
|
||||||
|
|
||||||
|
// Options 日志文件配置选项
|
||||||
|
type Options struct {
|
||||||
|
Path string // 文件保存地方
|
||||||
|
Prefix string // 日志文件前缀
|
||||||
|
ErrorFileSuffix string // error日志文件后缀
|
||||||
|
WarnFileSuffix string // warn日志文件后缀
|
||||||
|
InfoFileSuffix string // info日志文件后缀
|
||||||
|
DebugFileSuffix string // debug日志文件后缀
|
||||||
|
Level zapcore.Level // 日志等级
|
||||||
|
MaxSize int // 日志文件大小（M）
|
||||||
|
MaxBackups int // 最多存在多少个切片文件
|
||||||
|
MaxAge int // 保存的最大天数
|
||||||
|
Development bool // 是否是开发模式
|
||||||
|
zap.Config
|
||||||
|
}
|
||||||
|
|
||||||
|
var (
|
||||||
|
logger *Logger
|
||||||
|
sp = string(filepath.Separator)
|
||||||
|
errWS, warnWS, infoWS, debugWS zapcore.WriteSyncer // IO输出
|
||||||
|
debugConsoleWS = zapcore.Lock(os.Stdout) // 控制台标准输出
|
||||||
|
errorConsoleWS = zapcore.Lock(os.Stderr)
|
||||||
|
)
|
||||||
|
|
||||||
|
type Logger struct {
|
||||||
|
*zap.Logger
|
||||||
|
sync.RWMutex
|
||||||
|
Opts *Options `json:"opts"`
|
||||||
|
zapConfig zap.Config
|
||||||
|
initialized bool
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewLogger(mod ...ModOptions) *zap.Logger {
|
||||||
|
logger = &Logger{}
|
||||||
|
logger.Lock()
|
||||||
|
defer logger.Unlock()
|
||||||
|
if logger.initialized {
|
||||||
|
logger.Info("[NewLogger] logger initEd")
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
logger.Opts = &Options{
|
||||||
|
Path: "",
|
||||||
|
Prefix: "app",
|
||||||
|
ErrorFileSuffix: "error.log",
|
||||||
|
WarnFileSuffix: "warn.log",
|
||||||
|
InfoFileSuffix: "info.log",
|
||||||
|
DebugFileSuffix: "debug.log",
|
||||||
|
Level: zapcore.DebugLevel,
|
||||||
|
MaxSize: 100,
|
||||||
|
MaxBackups: 60,
|
||||||
|
MaxAge: 30,
|
||||||
|
}
|
||||||
|
|
||||||
|
if logger.Opts.Path == "" {
|
||||||
|
logger.Opts.Path, _ = filepath.Abs(filepath.Dir(filepath.Join(".")))
|
||||||
|
logger.Opts.Path += sp + "logs" + sp
|
||||||
|
}
|
||||||
|
if logger.Opts.Development {
|
||||||
|
logger.zapConfig = zap.NewDevelopmentConfig()
|
||||||
|
logger.zapConfig.EncoderConfig.EncodeTime = timeEncoder
|
||||||
|
} else {
|
||||||
|
logger.zapConfig = zap.NewProductionConfig()
|
||||||
|
logger.zapConfig.EncoderConfig.EncodeTime = timeEncoder
|
||||||
|
}
|
||||||
|
if logger.Opts.OutputPaths == nil || len(logger.Opts.OutputPaths) == 0 {
|
||||||
|
logger.zapConfig.OutputPaths = []string{"stdout"}
|
||||||
|
}
|
||||||
|
if logger.Opts.ErrorOutputPaths == nil || len(logger.Opts.ErrorOutputPaths) == 0 {
|
||||||
|
logger.zapConfig.OutputPaths = []string{"stderr"}
|
||||||
|
}
|
||||||
|
for _, fn := range mod {
|
||||||
|
fn(logger.Opts)
|
||||||
|
}
|
||||||
|
logger.zapConfig.Level.SetLevel(logger.Opts.Level)
|
||||||
|
logger.init()
|
||||||
|
logger.initialized = true
|
||||||
|
return logger.Logger
|
||||||
|
}
|
||||||
|
|
||||||
|
func (logger *Logger) init() {
|
||||||
|
logger.setSyncs()
|
||||||
|
var err error
|
||||||
|
logger.Logger, err = logger.zapConfig.Build(logger.cores())
|
||||||
|
if err != nil {
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
defer logger.Logger.Sync()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (logger *Logger) setSyncs() {
|
||||||
|
f := func(fN string) zapcore.WriteSyncer {
|
||||||
|
return zapcore.AddSync(&lumberjack.Logger{
|
||||||
|
Filename: logger.Opts.Path + sp + logger.Opts.Prefix + "-" + fN,
|
||||||
|
MaxSize: logger.Opts.MaxSize,
|
||||||
|
MaxBackups: logger.Opts.MaxBackups,
|
||||||
|
MaxAge: logger.Opts.MaxAge,
|
||||||
|
Compress: true,
|
||||||
|
LocalTime: true,
|
||||||
|
})
|
||||||
|
}
|
||||||
|
errWS = f(logger.Opts.ErrorFileSuffix)
|
||||||
|
warnWS = f(logger.Opts.WarnFileSuffix)
|
||||||
|
infoWS = f(logger.Opts.InfoFileSuffix)
|
||||||
|
debugWS = f(logger.Opts.DebugFileSuffix)
|
||||||
|
}
|
||||||
|
|
||||||
|
func SetMaxSize(MaxSize int) ModOptions {
|
||||||
|
return func(option *Options) {
|
||||||
|
option.MaxSize = MaxSize
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func SetMaxBackups(MaxBackups int) ModOptions {
|
||||||
|
return func(option *Options) {
|
||||||
|
option.MaxBackups = MaxBackups
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func SetMaxAge(MaxAge int) ModOptions {
|
||||||
|
return func(option *Options) {
|
||||||
|
option.MaxAge = MaxAge
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func SetPath(Path string) ModOptions {
|
||||||
|
return func(option *Options) {
|
||||||
|
option.Path = Path
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func SetPrefix(Prefix string) ModOptions {
|
||||||
|
return func(option *Options) {
|
||||||
|
option.Prefix = Prefix
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func SetLevel(Level zapcore.Level) ModOptions {
|
||||||
|
return func(option *Options) {
|
||||||
|
option.Level = Level
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func SetErrorFileSuffix(ErrorFileSuffix string) ModOptions {
|
||||||
|
return func(option *Options) {
|
||||||
|
option.ErrorFileSuffix = ErrorFileSuffix
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func SetWarnFileSuffix(WarnFileSuffix string) ModOptions {
|
||||||
|
return func(option *Options) {
|
||||||
|
option.WarnFileSuffix = WarnFileSuffix
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func SetInfoFileSuffix(InfoFileSuffix string) ModOptions {
|
||||||
|
return func(option *Options) {
|
||||||
|
option.InfoFileSuffix = InfoFileSuffix
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func SetDebugFileSuffix(DebugFileSuffix string) ModOptions {
|
||||||
|
return func(option *Options) {
|
||||||
|
option.DebugFileSuffix = DebugFileSuffix
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func SetDevelopment(Development bool) ModOptions {
|
||||||
|
return func(option *Options) {
|
||||||
|
option.Development = Development
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// getDefaultLogger returns the default logger.
|
||||||
|
func getDefaultLogger() *Logger {
|
||||||
|
return logger
|
||||||
|
}
|
||||||
|
|
||||||
|
// L returns the default logger.
|
||||||
|
func L() *Logger {
|
||||||
|
return getDefaultLogger()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (logger *Logger) cores() zap.Option {
|
||||||
|
fileEncoder := zapcore.NewJSONEncoder(logger.zapConfig.EncoderConfig)
|
||||||
|
encoderConfig := zap.NewDevelopmentEncoderConfig()
|
||||||
|
encoderConfig.EncodeTime = timeEncoder
|
||||||
|
encoderConfig.EncodeLevel = zapcore.CapitalColorLevelEncoder
|
||||||
|
consoleEncoder := zapcore.NewConsoleEncoder(zapcore.EncoderConfig{
|
||||||
|
TimeKey: "ts",
|
||||||
|
LevelKey: "level",
|
||||||
|
NameKey: "logger",
|
||||||
|
CallerKey: "caller_line",
|
||||||
|
FunctionKey: zapcore.OmitKey,
|
||||||
|
MessageKey: "msg",
|
||||||
|
StacktraceKey: "stacktrace",
|
||||||
|
LineEnding: " ",
|
||||||
|
EncodeLevel: encodeLevel,
|
||||||
|
EncodeTime: encodeTime,
|
||||||
|
EncodeDuration: zapcore.SecondsDurationEncoder,
|
||||||
|
EncodeCaller: encodeCaller,
|
||||||
|
})
|
||||||
|
|
||||||
|
errPriority := zap.LevelEnablerFunc(func(lvl zapcore.Level) bool {
|
||||||
|
return lvl == zapcore.ErrorLevel && zapcore.ErrorLevel-logger.zapConfig.Level.Level() > -1
|
||||||
|
})
|
||||||
|
warnPriority := zap.LevelEnablerFunc(func(lvl zapcore.Level) bool {
|
||||||
|
return lvl == zapcore.WarnLevel && zapcore.WarnLevel-logger.zapConfig.Level.Level() > -1
|
||||||
|
})
|
||||||
|
infoPriority := zap.LevelEnablerFunc(func(lvl zapcore.Level) bool {
|
||||||
|
return lvl == zapcore.InfoLevel && zapcore.InfoLevel-logger.zapConfig.Level.Level() > -1
|
||||||
|
})
|
||||||
|
debugPriority := zap.LevelEnablerFunc(func(lvl zapcore.Level) bool {
|
||||||
|
return lvl == zapcore.DebugLevel && zapcore.DebugLevel-logger.zapConfig.Level.Level() > -1
|
||||||
|
})
|
||||||
|
cores := []zapcore.Core{
|
||||||
|
zapcore.NewCore(fileEncoder, errWS, errPriority),
|
||||||
|
zapcore.NewCore(fileEncoder, warnWS, warnPriority),
|
||||||
|
zapcore.NewCore(fileEncoder, infoWS, infoPriority),
|
||||||
|
zapcore.NewCore(fileEncoder, debugWS, debugPriority),
|
||||||
|
}
|
||||||
|
if logger.Opts.Development {
|
||||||
|
cores = append(cores, []zapcore.Core{
|
||||||
|
zapcore.NewCore(consoleEncoder, errorConsoleWS, errPriority),
|
||||||
|
zapcore.NewCore(consoleEncoder, debugConsoleWS, warnPriority),
|
||||||
|
zapcore.NewCore(consoleEncoder, debugConsoleWS, infoPriority),
|
||||||
|
zapcore.NewCore(consoleEncoder, debugConsoleWS, debugPriority),
|
||||||
|
}...)
|
||||||
|
}
|
||||||
|
return zap.WrapCore(func(c zapcore.Core) zapcore.Core {
|
||||||
|
return zapcore.NewTee(cores...)
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
func timeEncoder(t time.Time, enc zapcore.PrimitiveArrayEncoder) {
|
||||||
|
enc.AppendString(t.Format("2006-01-02 15:04:05.000"))
|
||||||
|
}
|
||||||
|
|
||||||
|
// encodeLevel 自定义日志级别显示
|
||||||
|
func encodeLevel(level zapcore.Level, enc zapcore.PrimitiveArrayEncoder) {
|
||||||
|
enc.AppendString("[" + level.CapitalString() + "]")
|
||||||
|
}
|
||||||
|
|
||||||
|
// encodeTime 自定义时间格式显示
|
||||||
|
func encodeTime(t time.Time, enc zapcore.PrimitiveArrayEncoder) {
|
||||||
|
enc.AppendString("[" + t.Format("2006-01-02 15:04:05.000") + "]")
|
||||||
|
}
|
||||||
|
|
||||||
|
// encodeCaller 自定义行号显示
|
||||||
|
func encodeCaller(caller zapcore.EntryCaller, enc zapcore.PrimitiveArrayEncoder) {
|
||||||
|
enc.AppendString("[" + caller.TrimmedPath() + "]")
|
||||||
|
}
|
||||||
Loading…
Reference in New Issue