拆分logger库

logger.go

@@ -0,0 +1,136 @@
+// Package logger...
+//
+// Description : logger 日志文件
+//
+// Author : go_developer@163.com<白茶清欢>
+//
+// Date : 2021-01-02 5:04 下午
+package logger
+
+import (
+	"io"
+	"os"
+
+	"github.com/pkg/errors"
+
+	"go.uber.org/zap"
+
+	"go.uber.org/zap/zapcore"
+
+	rotatelogs "github.com/lestrrat-go/file-rotatelogs"
+)
+
+// NewLogger 获取日志实例
+//
+// Author : go_developer@163.com<白茶清欢>
+//
+// Date : 5:05 下午 2021/1/2
+func NewLogger(loggerLevel zapcore.Level, splitConfig *RotateLogConfig, optionFunc ...SetLoggerOptionFunc) (*zap.Logger, error) {
+	if nil == splitConfig {
+		return nil, errors.New("未配置日志切割规则")
+	}
+
+	o := &OptionLogger{}
+
+	for _, f := range optionFunc {
+		f(o)
+	}
+
+	if nil == o.Encoder {
+		o.Encoder = GetEncoder()
+	}
+	loggerLevelDeal := zap.LevelEnablerFunc(func(lvl zapcore.Level) bool {
+		return lvl >= loggerLevel
+	})
+	l := &Logger{
+		splitConfig: splitConfig,
+		encoder:     o.Encoder,
+	}
+	var (
+		err          error
+		loggerWriter io.Writer
+	)
+	// 获取 日志实现
+	if loggerWriter, err = l.getWriter(); nil != err {
+		return nil, err
+	}
+
+	fileHandlerList := []zapcore.Core{
+		zapcore.NewCore(o.Encoder, zapcore.AddSync(loggerWriter), loggerLevelDeal),
+	}
+
+	// 设置控制台输出
+	if o.ConsoleOutput {
+		fileHandlerList = append(fileHandlerList, zapcore.NewCore(o.Encoder, zapcore.AddSync(os.Stdout), loggerLevelDeal))
+	}
+
+	// 最后创建具体的Logger
+	core := zapcore.NewTee(fileHandlerList...)
+
+	// 需要传入 zap.AddCaller() 才会显示打日志点的文件名和行数, 跳过一行可以直接显示业务代码行号,否则显示日志包行号
+	logConfList := make([]zap.Option, 0)
+	if o.WithCaller {
+		logConfList = append(logConfList, zap.AddCaller(), zap.AddCallerSkip(o.WithCallerSkip))
+	}
+	log := zap.New(core, logConfList...)
+	return log, nil
+}
+
+// NewConsoleLogger 获取控制台输出的日志实例
+//
+// Author : go_developer@163.com<白茶清欢>
+//
+// Date : 8:22 下午 2021/4/17
+func NewConsoleLogger(loggerLevel zapcore.Level, optionFunc ...SetLoggerOptionFunc) (*zap.Logger, error) {
+	o := &OptionLogger{}
+
+	for _, f := range optionFunc {
+		f(o)
+	}
+	if nil == o.Encoder {
+		o.Encoder = GetEncoder()
+	}
+	loggerLevelDeal := zap.LevelEnablerFunc(func(lvl zapcore.Level) bool {
+		return lvl >= loggerLevel
+	})
+	fileHandlerList := []zapcore.Core{
+		zapcore.NewCore(o.Encoder, zapcore.AddSync(os.Stdout), loggerLevelDeal),
+	}
+	// 最后创建具体的Logger
+	core := zapcore.NewTee(fileHandlerList...)
+
+	// 需要传入 zap.AddCaller() 才会显示打日志点的文件名和行数, 跳过一行可以直接显示业务代码行号,否则显示日志包行号
+	logConfList := make([]zap.Option, 0)
+	if o.WithCaller {
+		logConfList = append(logConfList, zap.AddCaller(), zap.AddCallerSkip(o.WithCallerSkip))
+	}
+	log := zap.New(core, logConfList...)
+	return log, nil
+}
+
+type Logger struct {
+	splitConfig *RotateLogConfig
+	encoder     zapcore.Encoder
+}
+
+// getWriter 获取日志实例
+//
+// Author : go_developer@163.com<白茶清欢>
+//
+// Date : 5:08 下午 2021/1/2
+func (l *Logger) getWriter() (io.Writer, error) {
+	option := make([]rotatelogs.Option, 0)
+	option = append(option, rotatelogs.WithRotationTime(l.splitConfig.TimeInterval))
+	if l.splitConfig.MaxAge > 0 {
+		option = append(option, rotatelogs.WithMaxAge(l.splitConfig.MaxAge))
+	}
+	var (
+		hook *rotatelogs.RotateLogs
+		err  error
+	)
+	if hook, err = rotatelogs.New(l.splitConfig.FullLogFormat, option...); nil != err {
+		return nil, CreateIOWriteError(err)
+	}
+
+	return hook, nil
+}