迁移日志库
This commit is contained in:
2
logger/README.md
Normal file
2
logger/README.md
Normal file
@ -0,0 +1,2 @@
|
||||
# go-logger
|
||||
基于zap log的包装,方便引入使用
|
395
logger/config.go
Normal file
395
logger/config.go
Normal file
@ -0,0 +1,395 @@
|
||||
// Package logger...
|
||||
//
|
||||
// Description : config 日志配置
|
||||
//
|
||||
// Author : go_developer@163.com<张德满>
|
||||
//
|
||||
// Date : 2021-01-02 3:07 下午
|
||||
package logger
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"os"
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
"go.uber.org/zap/zapcore"
|
||||
)
|
||||
|
||||
// TimeIntervalType 日志时间间隔类型
|
||||
type TimeIntervalType uint
|
||||
|
||||
const (
|
||||
// TimeIntervalTypeMinute 按分钟切割
|
||||
TimeIntervalTypeMinute = TimeIntervalType(0)
|
||||
// TimeIntervalTypeHour 按小时切割
|
||||
TimeIntervalTypeHour = TimeIntervalType(1)
|
||||
// TimeIntervalTypeDay 按天切割
|
||||
TimeIntervalTypeDay = TimeIntervalType(2)
|
||||
// TimeIntervalTypeMonth 按月切割
|
||||
TimeIntervalTypeMonth = TimeIntervalType(3)
|
||||
// TimeIntervalTypeYear 按年切割
|
||||
TimeIntervalTypeYear = TimeIntervalType(4)
|
||||
)
|
||||
|
||||
const (
|
||||
// DefaultDivisionChar 默认的时间格式分隔符
|
||||
DefaultDivisionChar = "-"
|
||||
)
|
||||
|
||||
// RotateLogConfig 日志切割的配置
|
||||
//
|
||||
// Author : go_developer@163.com<张德满>
|
||||
//
|
||||
// Date : 3:08 下午 2021/1/2
|
||||
type RotateLogConfig struct {
|
||||
TimeIntervalType TimeIntervalType // 日志切割的时间间隔类型 0 - 小时 1 - 天 2 - 月 3 - 年
|
||||
TimeInterval time.Duration // 日志切割的时间间隔
|
||||
LogPath string // 存储日志的路径
|
||||
LogFileName string // 日志文件名
|
||||
DivisionChar string // 日志文件拼时间分隔符
|
||||
FullLogFormat string // 完整的日志格式
|
||||
MaxAge time.Duration // 日志最长保存时间
|
||||
}
|
||||
|
||||
// SetRotateLogConfigOption 设置日志切割的选项
|
||||
//
|
||||
// Author : go_developer@163.com<张德满>
|
||||
//
|
||||
// Date : 3:13 下午 2021/1/2
|
||||
type SetRotateLogConfigFunc func(rlc *RotateLogConfig)
|
||||
|
||||
// WithTimeIntervalType 设置日志切割时间间隔
|
||||
//
|
||||
// Author : go_developer@163.com<张德满>
|
||||
//
|
||||
// Date : 3:34 下午 2021/1/2
|
||||
func WithTimeIntervalType(timeIntervalType TimeIntervalType) SetRotateLogConfigFunc {
|
||||
return func(rlc *RotateLogConfig) {
|
||||
rlc.TimeIntervalType = timeIntervalType
|
||||
}
|
||||
}
|
||||
|
||||
// WithDivisionChar 设置分隔符
|
||||
//
|
||||
// Author : go_developer@163.com<张德满>
|
||||
//
|
||||
// Date : 3:49 下午 2021/1/2
|
||||
func WithDivisionChar(divisionChar string) SetRotateLogConfigFunc {
|
||||
return func(rlc *RotateLogConfig) {
|
||||
rlc.DivisionChar = divisionChar
|
||||
}
|
||||
}
|
||||
|
||||
// WithMaxAge 设置日志保存时间
|
||||
//
|
||||
// Author : go_developer@163.com<张德满>
|
||||
//
|
||||
// Date : 5:03 下午 2021/1/2
|
||||
func WithMaxAge(maxAge time.Duration) SetRotateLogConfigFunc {
|
||||
return func(rlc *RotateLogConfig) {
|
||||
rlc.MaxAge = maxAge
|
||||
}
|
||||
}
|
||||
|
||||
// NewRotateLogConfig 生成日志切割的配置
|
||||
//
|
||||
// Author : go_developer@163.com<张德满>
|
||||
//
|
||||
// Date : 3:53 下午 2021/1/2
|
||||
func NewRotateLogConfig(logPath string, logFile string, option ...SetRotateLogConfigFunc) (*RotateLogConfig, error) {
|
||||
if len(logPath) == 0 || len(logFile) == 0 {
|
||||
return nil, LogPathEmptyError()
|
||||
}
|
||||
c := &RotateLogConfig{
|
||||
TimeIntervalType: 0,
|
||||
LogPath: logPath,
|
||||
LogFileName: logFile,
|
||||
DivisionChar: "",
|
||||
}
|
||||
|
||||
for _, o := range option {
|
||||
o(c)
|
||||
}
|
||||
|
||||
if err := formatConfig(c); nil != err {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return c, nil
|
||||
}
|
||||
|
||||
// formatConfig 格式化配置
|
||||
//
|
||||
// Author : go_developer@163.com<张德满>
|
||||
//
|
||||
// Date : 4:23 下午 2021/1/2
|
||||
func formatConfig(c *RotateLogConfig) error {
|
||||
|
||||
if len(c.DivisionChar) == 0 {
|
||||
c.DivisionChar = DefaultDivisionChar
|
||||
}
|
||||
// 格式化路径
|
||||
logPathByte := []byte(c.LogPath)
|
||||
if string(logPathByte[len(logPathByte)-1]) != "/" {
|
||||
c.LogPath = c.LogPath + "/"
|
||||
}
|
||||
// 检测路径是否存在,不存在自动创建
|
||||
if _, err := os.Stat(c.LogPath); nil != err {
|
||||
if !os.IsNotExist(err) {
|
||||
// 异常不是路径不存在,抛异常
|
||||
return DealLogPathError(err, c.LogPath)
|
||||
}
|
||||
if err := os.Mkdir(c.LogPath, os.ModePerm); nil != err {
|
||||
return DealLogPathError(err, "创建日志目录失败")
|
||||
}
|
||||
}
|
||||
|
||||
// 生成格式化日志全路径
|
||||
switch c.TimeIntervalType {
|
||||
case TimeIntervalTypeMinute:
|
||||
c.TimeInterval = time.Minute
|
||||
c.FullLogFormat = c.LogPath + "%Y" + c.DivisionChar + "%m" + c.DivisionChar + "%d" + c.DivisionChar + "%H" + c.DivisionChar + "%M" + c.DivisionChar + c.LogFileName
|
||||
case TimeIntervalTypeHour:
|
||||
c.TimeInterval = time.Hour
|
||||
c.FullLogFormat = c.LogPath + "%Y" + c.DivisionChar + "%m" + c.DivisionChar + "%d" + c.DivisionChar + "%H" + c.DivisionChar + c.LogFileName
|
||||
case TimeIntervalTypeDay:
|
||||
c.TimeInterval = time.Hour * 24
|
||||
c.FullLogFormat = c.LogPath + "%Y" + c.DivisionChar + "%m" + c.DivisionChar + "%d" + c.DivisionChar + c.LogFileName
|
||||
case TimeIntervalTypeMonth:
|
||||
c.TimeInterval = time.Hour * 24 * 30
|
||||
c.FullLogFormat = c.LogPath + "%Y" + c.DivisionChar + "%m" + c.DivisionChar + c.LogFileName
|
||||
case TimeIntervalTypeYear:
|
||||
c.TimeInterval = time.Hour * 24 * 365
|
||||
c.FullLogFormat = c.LogPath + "%Y" + c.DivisionChar + c.LogFileName
|
||||
default:
|
||||
return LogSplitTypeError(c.TimeIntervalType)
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
// ============== 以下为zap相关配置
|
||||
|
||||
const (
|
||||
// defaultMessageKey 默认的message key
|
||||
defaultMessageKey = "message"
|
||||
// defaultLevelKey 默认的level
|
||||
defaultLevelKey = "level"
|
||||
// defaultTimeKey 默认时间key
|
||||
defaultTimeKey = "time"
|
||||
// defaultCallerKey 默认的文件key
|
||||
defaultCallerKey = "file"
|
||||
// defaultUserShortCaller 是否使用短的文件调用格式
|
||||
defaultUseShortCaller = true
|
||||
// defaultUseJsonFormat 日志默认使用json格式
|
||||
defaultUseJsonFormat = true
|
||||
)
|
||||
|
||||
// defaultTimeEncoder 默认的时间处理
|
||||
//
|
||||
// Author : go_developer@163.com<张德满>
|
||||
//
|
||||
// Date : 11:53 下午 2021/1/2
|
||||
func defaultTimeEncoder(t time.Time, enc zapcore.PrimitiveArrayEncoder) {
|
||||
sec := t.UnixNano() / 1e9
|
||||
ms := t.UnixNano() / 1e6 % 1e3
|
||||
ns := t.UnixNano() % 1e6
|
||||
enc.AppendString(time.Unix(sec, ns).Format("2006-01-02 15:04:05") + "." + fmt.Sprintf("%v", ms) + "+" + fmt.Sprintf("%v", ns))
|
||||
}
|
||||
|
||||
// SecondTimeEncoder 秒级时间戳格式化
|
||||
//
|
||||
// Author : go_developer@163.com<张德满>
|
||||
//
|
||||
// Date : 8:34 下午 2021/1/3
|
||||
func SecondTimeEncoder(t time.Time, enc zapcore.PrimitiveArrayEncoder) {
|
||||
enc.AppendString(t.Format("2006-01-02 15:04:05"))
|
||||
}
|
||||
|
||||
// MsTimeEncoder 毫秒时间格式化方法
|
||||
//
|
||||
// Author : go_developer@163.com<张德满>
|
||||
//
|
||||
// Date : 8:35 下午 2021/1/3
|
||||
func MsTimeEncoder(t time.Time, enc zapcore.PrimitiveArrayEncoder) {
|
||||
sec := t.UnixNano() / 1e9
|
||||
ms := t.UnixNano() / 1e6 % 1e3
|
||||
enc.AppendString(time.Unix(sec, 0).Format("2006-01-02 15:04:05") + "." + fmt.Sprintf("%v", ms))
|
||||
}
|
||||
|
||||
// defaultEncodeDuration 默认的原始时间处理
|
||||
//
|
||||
// Author : go_developer@163.com<张德满>
|
||||
//
|
||||
// Date : 11:56 下午 2021/1/2
|
||||
func defaultEncodeDuration(d time.Duration, enc zapcore.PrimitiveArrayEncoder) {
|
||||
enc.AppendInt64(int64(d) / 1000000)
|
||||
}
|
||||
|
||||
// OptionLogger 日志配置的选项
|
||||
//
|
||||
// Author : go_developer@163.com<张德满>
|
||||
//
|
||||
// Date : 11:41 下午 2021/1/2
|
||||
type OptionLogger struct {
|
||||
UseJsonFormat bool // 日志使用json格式
|
||||
MessageKey string // message 字段
|
||||
LevelKey string // level 字段
|
||||
TimeKey string // 时间字段
|
||||
CallerKey string // 记录日志的文件的代码行数
|
||||
UseShortCaller bool // 使用短的调用文件格式
|
||||
TimeEncoder zapcore.TimeEncoder // 格式化时间的函数
|
||||
EncodeDuration zapcore.DurationEncoder // 原始时间信息
|
||||
}
|
||||
|
||||
// 设置日志配置
|
||||
type SetLoggerOptionFunc func(o *OptionLogger)
|
||||
|
||||
// WithUseJsonFormat 日志是否使用json格式数据
|
||||
//
|
||||
// Author : go_developer@163.com<张德满>
|
||||
//
|
||||
// Date : 12:30 上午 2021/1/3
|
||||
func WithUseJsonFormat(isJsonFormat bool) SetLoggerOptionFunc {
|
||||
return func(o *OptionLogger) {
|
||||
o.UseJsonFormat = isJsonFormat
|
||||
}
|
||||
}
|
||||
|
||||
// WithMessageKey 使用message key
|
||||
//
|
||||
// Author : go_developer@163.com<张德满>
|
||||
//
|
||||
// Date : 12:32 上午 2021/1/3
|
||||
func WithMessageKey(messageKey string) SetLoggerOptionFunc {
|
||||
return func(o *OptionLogger) {
|
||||
messageKey = strings.Trim(messageKey, " ")
|
||||
if len(messageKey) == 0 {
|
||||
return
|
||||
}
|
||||
o.MessageKey = messageKey
|
||||
}
|
||||
}
|
||||
|
||||
// WithLevelKey 设置level key
|
||||
//
|
||||
// Author : go_developer@163.com<张德满>
|
||||
//
|
||||
// Date : 12:33 上午 2021/1/3
|
||||
func WithLevelKey(levelKey string) SetLoggerOptionFunc {
|
||||
return func(o *OptionLogger) {
|
||||
levelKey = strings.Trim(levelKey, " ")
|
||||
if len(levelKey) == 0 {
|
||||
return
|
||||
}
|
||||
o.LevelKey = levelKey
|
||||
}
|
||||
}
|
||||
|
||||
// WithTimeKey 设置time key ...
|
||||
//
|
||||
// Author : go_developer@163.com<张德满>
|
||||
//
|
||||
// Date : 12:34 上午 2021/1/3
|
||||
func WithTimeKey(timeKey string) SetLoggerOptionFunc {
|
||||
return func(o *OptionLogger) {
|
||||
timeKey = strings.Trim(timeKey, " ")
|
||||
if len(timeKey) == 0 {
|
||||
return
|
||||
}
|
||||
o.TimeKey = timeKey
|
||||
}
|
||||
}
|
||||
|
||||
// WithCallerKey 设置caller key
|
||||
//
|
||||
// Author : go_developer@163.com<张德满>
|
||||
//
|
||||
// Date : 12:37 上午 2021/1/3
|
||||
func WithCallerKey(callerKey string) SetLoggerOptionFunc {
|
||||
return func(o *OptionLogger) {
|
||||
callerKey = strings.Trim(callerKey, " ")
|
||||
if len(callerKey) == 0 {
|
||||
return
|
||||
}
|
||||
o.CallerKey = callerKey
|
||||
}
|
||||
}
|
||||
|
||||
// WithShortCaller 是否使用短caller格式
|
||||
//
|
||||
// Author : go_developer@163.com<张德满>
|
||||
//
|
||||
// Date : 12:39 上午 2021/1/3
|
||||
func WithShortCaller(useShortCaller bool) SetLoggerOptionFunc {
|
||||
return func(o *OptionLogger) {
|
||||
o.UseShortCaller = useShortCaller
|
||||
}
|
||||
}
|
||||
|
||||
// WithTimeEncoder 设置格式化时间方法
|
||||
//
|
||||
// Author : go_developer@163.com<张德满>
|
||||
//
|
||||
// Date : 12:41 上午 2021/1/3
|
||||
func WithTimeEncoder(encoder zapcore.TimeEncoder) SetLoggerOptionFunc {
|
||||
return func(o *OptionLogger) {
|
||||
if nil == encoder {
|
||||
return
|
||||
}
|
||||
o.TimeEncoder = encoder
|
||||
}
|
||||
}
|
||||
|
||||
// WithEncodeDuration 原始时间
|
||||
//
|
||||
// Author : go_developer@163.com<张德满>
|
||||
//
|
||||
// Date : 12:42 上午 2021/1/3
|
||||
func WithEncodeDuration(encoder zapcore.DurationEncoder) SetLoggerOptionFunc {
|
||||
return func(o *OptionLogger) {
|
||||
if nil == encoder {
|
||||
return
|
||||
}
|
||||
o.EncodeDuration = encoder
|
||||
}
|
||||
}
|
||||
|
||||
// GetEncoder 获取空中台输出的encoder
|
||||
//
|
||||
// Author : go_developer@163.com<张德满>
|
||||
//
|
||||
// Date : 6:24 下午 2021/1/2
|
||||
func GetEncoder(option ...SetLoggerOptionFunc) zapcore.Encoder {
|
||||
ol := &OptionLogger{
|
||||
UseJsonFormat: defaultUseJsonFormat,
|
||||
MessageKey: defaultMessageKey,
|
||||
LevelKey: defaultLevelKey,
|
||||
TimeKey: defaultTimeKey,
|
||||
TimeEncoder: defaultTimeEncoder,
|
||||
CallerKey: defaultCallerKey,
|
||||
EncodeDuration: defaultEncodeDuration,
|
||||
UseShortCaller: defaultUseShortCaller,
|
||||
}
|
||||
for _, o := range option {
|
||||
o(ol)
|
||||
}
|
||||
ec := zapcore.EncoderConfig{
|
||||
MessageKey: ol.MessageKey,
|
||||
LevelKey: ol.LevelKey,
|
||||
EncodeLevel: zapcore.CapitalLevelEncoder,
|
||||
TimeKey: ol.TimeKey,
|
||||
EncodeTime: ol.TimeEncoder,
|
||||
CallerKey: ol.CallerKey,
|
||||
EncodeCaller: zapcore.ShortCallerEncoder,
|
||||
EncodeDuration: ol.EncodeDuration,
|
||||
}
|
||||
if !ol.UseShortCaller {
|
||||
ec.EncodeCaller = zapcore.FullCallerEncoder
|
||||
}
|
||||
if !ol.UseJsonFormat {
|
||||
return zapcore.NewConsoleEncoder(ec)
|
||||
}
|
||||
return zapcore.NewJSONEncoder(ec)
|
||||
}
|
26
logger/debug.go
Normal file
26
logger/debug.go
Normal file
@ -0,0 +1,26 @@
|
||||
// Package logger...
|
||||
//
|
||||
// Description : logger...
|
||||
//
|
||||
// Author : go_developer@163.com<张德满>
|
||||
//
|
||||
// Date : 2021-01-03 12:58 上午
|
||||
package logger
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"encoding/json"
|
||||
)
|
||||
|
||||
// FormatJson 格式化输出json
|
||||
//
|
||||
// Author : go_developer@163.com<张德满>
|
||||
//
|
||||
// Date : 1:06 上午 2021/1/3
|
||||
func FormatJson(src interface{}) string {
|
||||
byteData, _ := json.Marshal(src)
|
||||
|
||||
var str bytes.Buffer
|
||||
_ = json.Indent(&str, byteData, "", " ")
|
||||
return str.String()
|
||||
}
|
64
logger/error.go
Normal file
64
logger/error.go
Normal file
@ -0,0 +1,64 @@
|
||||
// Package logger...
|
||||
//
|
||||
// Description : error 定义日志处理过程中的各种错误
|
||||
//
|
||||
// Author : go_developer@163.com<张德满>
|
||||
//
|
||||
// Date : 2021-01-02 2:44 下午
|
||||
package logger
|
||||
|
||||
import "github.com/pkg/errors"
|
||||
|
||||
// CreateLogFileError 创建日志文件失败
|
||||
//
|
||||
// Author : go_developer@163.com<张德满>
|
||||
//
|
||||
// Date : 2:55 下午 2021/1/2
|
||||
func CreateLogFileError(err error, logFilePath string) error {
|
||||
return errors.Wrapf(err, "创建日志文件失败,日志文件路径 : %s", logFilePath)
|
||||
}
|
||||
|
||||
// LogPathEmptyError 日志路径为空
|
||||
//
|
||||
// Author : go_developer@163.com<张德满>
|
||||
//
|
||||
// Date : 4:03 下午 2021/1/2
|
||||
func LogPathEmptyError() error {
|
||||
return errors.Wrap(errors.New("日志存储路径或者日志文件名为空"), "日志存储路径或者日志文件名为空")
|
||||
}
|
||||
|
||||
// CustomTimeIntervalError 自定义日志切割时间间隔错误
|
||||
//
|
||||
// Author : go_developer@163.com<张德满>
|
||||
//
|
||||
// Date : 4:11 下午 2021/1/2
|
||||
func CustomTimeIntervalError() error {
|
||||
return errors.Wrap(errors.New("自定义时间间隔错误,必须是大于0的值"), "自定义时间间隔错误,必须是大于0的值")
|
||||
}
|
||||
|
||||
// DealLogPathError 日志路径处理异常
|
||||
//
|
||||
// Author : go_developer@163.com<张德满>
|
||||
//
|
||||
// Date : 4:31 下午 2021/1/2
|
||||
func DealLogPathError(err error, logPath string) error {
|
||||
return errors.Wrapf(err, "日志路径检测处理异常, 日志路径 : %s", logPath)
|
||||
}
|
||||
|
||||
// LogSplitTypeError 日志切割类型错误
|
||||
//
|
||||
// Author : go_developer@163.com<张德满>
|
||||
//
|
||||
// Date : 4:50 下午 2021/1/2
|
||||
func LogSplitTypeError(splitType TimeIntervalType) error {
|
||||
return errors.Wrapf(errors.New("日志切割时间类型错误"), "日志切割时间类型错误, 传入类型 : %v", splitType)
|
||||
}
|
||||
|
||||
// CreateIOWriteError 创建日志实例失败
|
||||
//
|
||||
// Author : go_developer@163.com<张德满>
|
||||
//
|
||||
// Date : 5:20 下午 2021/1/2
|
||||
func CreateIOWriteError(err error) error {
|
||||
return errors.Wrapf(err, "创建日志实例失败")
|
||||
}
|
84
logger/logger.go
Normal file
84
logger/logger.go
Normal file
@ -0,0 +1,84 @@
|
||||
// Package logger...
|
||||
//
|
||||
// Description : logger 日志文件
|
||||
//
|
||||
// Author : go_developer@163.com<张德满>
|
||||
//
|
||||
// Date : 2021-01-02 5:04 下午
|
||||
package logger
|
||||
|
||||
import (
|
||||
"io"
|
||||
"os"
|
||||
|
||||
"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, consoleOutput bool, encoder zapcore.Encoder, splitConfig *RotateLogConfig) (*zap.Logger, error) {
|
||||
loggerLevelDeal := zap.LevelEnablerFunc(func(lvl zapcore.Level) bool {
|
||||
return lvl >= loggerLevel
|
||||
})
|
||||
l := &Logger{
|
||||
splitConfig: splitConfig,
|
||||
encoder: encoder,
|
||||
}
|
||||
var (
|
||||
err error
|
||||
loggerWriter io.Writer
|
||||
)
|
||||
// 获取 日志实现
|
||||
if loggerWriter, err = l.getWriter(); nil != err {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
fileHandlerList := []zapcore.Core{
|
||||
zapcore.NewCore(encoder, zapcore.AddSync(loggerWriter), loggerLevelDeal),
|
||||
}
|
||||
|
||||
// 设置控制台输出
|
||||
if consoleOutput {
|
||||
fileHandlerList = append(fileHandlerList, zapcore.NewCore(encoder, zapcore.AddSync(os.Stdout), loggerLevelDeal))
|
||||
}
|
||||
|
||||
// 最后创建具体的Logger
|
||||
core := zapcore.NewTee(fileHandlerList...)
|
||||
|
||||
log := zap.New(core, zap.AddCaller()) // 需要传入 zap.AddCaller() 才会显示打日志点的文件名和行数
|
||||
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
|
||||
}
|
72
logger/logger_test.go
Normal file
72
logger/logger_test.go
Normal file
@ -0,0 +1,72 @@
|
||||
// Package logger...
|
||||
//
|
||||
// Description : logger_test 单元测试
|
||||
//
|
||||
// Author : go_developer@163.com<张德满>
|
||||
//
|
||||
// Date : 2021-01-02 4:59 下午
|
||||
package logger
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"testing"
|
||||
"time"
|
||||
|
||||
"go.uber.org/zap"
|
||||
|
||||
"go.uber.org/zap/zapcore"
|
||||
)
|
||||
|
||||
// Test_Logger ...
|
||||
//
|
||||
// Author : go_developer@163.com<张德满>
|
||||
//
|
||||
// Date : 5:50 下午 2021/1/2
|
||||
func Test_Logger(t *testing.T) {
|
||||
encoder := GetEncoder(WithTimeEncoder(MsTimeEncoder))
|
||||
c, err := NewRotateLogConfig("./logs", "test.log", WithTimeIntervalType(TimeIntervalTypeMinute), WithMaxAge(120*time.Second))
|
||||
if nil != err {
|
||||
panic(err)
|
||||
}
|
||||
l, err := NewLogger(zapcore.InfoLevel, true, encoder, c)
|
||||
if nil != err {
|
||||
panic(err)
|
||||
}
|
||||
|
||||
for {
|
||||
l.Info("这是一条测试日志", zap.Any("lala", "不限制类型"))
|
||||
l.Debug("这是一条测试日志", zap.Any("lala", "不限制类型"))
|
||||
time.Sleep(1 * time.Second)
|
||||
}
|
||||
}
|
||||
|
||||
// Test_FormatJson 测试json格式化输出
|
||||
//
|
||||
// Author : go_developer@163.com<张德满>
|
||||
//
|
||||
// Date : 1:08 上午 2021/1/3
|
||||
func Test_FormatJson(t *testing.T) {
|
||||
data := map[string]interface{}{
|
||||
"name": "zhangdeman",
|
||||
"age": 18,
|
||||
}
|
||||
str := FormatJson(data)
|
||||
fmt.Println(str)
|
||||
|
||||
boolData := true
|
||||
str = FormatJson(boolData)
|
||||
fmt.Println(str)
|
||||
}
|
||||
|
||||
// Test_Time 测试格式化纳秒时间戳
|
||||
//
|
||||
// Author : go_developer@163.com<张德满>
|
||||
//
|
||||
// Date : 8:32 下午 2021/1/3
|
||||
func Test_Time(t *testing.T) {
|
||||
ti := time.Now()
|
||||
sec := ti.UnixNano() / 1e9
|
||||
ms := ti.UnixNano() / 1e6 % 1e3
|
||||
ns := ti.UnixNano() % 1e6
|
||||
fmt.Println(time.Unix(sec, ns).Format("2006-01-02 15:04:05") + "." + fmt.Sprintf("%v", ms) + "+" + fmt.Sprintf("%v", ns))
|
||||
}
|
151
logger/wrapper/http_gin.go
Normal file
151
logger/wrapper/http_gin.go
Normal file
@ -0,0 +1,151 @@
|
||||
// Package wrapper...
|
||||
//
|
||||
// Description : http_gin 使用gin框架时的,记录日志
|
||||
//
|
||||
// Author : go_developer@163.com<张德满>
|
||||
//
|
||||
// Date : 2021-01-03 3:43 下午
|
||||
package wrapper
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
|
||||
"github.com/gin-gonic/gin"
|
||||
"github.com/go-developer/gopkg/logger"
|
||||
"go.uber.org/zap"
|
||||
"go.uber.org/zap/zapcore"
|
||||
)
|
||||
|
||||
// NewGinWrapperLogger 使用gin框架记录日志
|
||||
//
|
||||
// Author : go_developer@163.com<张德满>
|
||||
//
|
||||
// Date : 3:45 下午 2021/1/3
|
||||
func NewGinWrapperLogger(loggerLevel zapcore.Level, consoleOutput bool, encoder zapcore.Encoder, splitConfig *logger.RotateLogConfig, extractFieldList []string) (*GinWrapper, error) {
|
||||
var (
|
||||
err error
|
||||
l *zap.Logger
|
||||
)
|
||||
if l, err = logger.NewLogger(loggerLevel, consoleOutput, encoder, splitConfig); nil != err {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return &GinWrapper{
|
||||
loggerInstance: l,
|
||||
extractFieldList: extractFieldList,
|
||||
}, nil
|
||||
}
|
||||
|
||||
// GinWrapper 包装gin实例
|
||||
//
|
||||
// Author : go_developer@163.com<张德满>
|
||||
//
|
||||
// Date : 3:59 下午 2021/1/3
|
||||
type GinWrapper struct {
|
||||
loggerInstance *zap.Logger // zap 的日志实例
|
||||
extractFieldList []string // 从gin中抽取的字段
|
||||
ginCtx *gin.Context // gin 实例
|
||||
}
|
||||
|
||||
// GetLogger 为每一次请求生成不同的日志实例,包含独立的gin上下文
|
||||
//
|
||||
// Author : go_developer@163.com<张德满>
|
||||
//
|
||||
// Date : 4:02 下午 2021/1/3
|
||||
func (gw *GinWrapper) GetLogger(ginCtx *gin.Context) *GinWrapper {
|
||||
return &GinWrapper{
|
||||
loggerInstance: gw.loggerInstance,
|
||||
extractFieldList: gw.extractFieldList,
|
||||
ginCtx: ginCtx,
|
||||
}
|
||||
}
|
||||
|
||||
// formatFieldList 格式化日志field列表
|
||||
//
|
||||
// Author : go_developer@163.com<张德满>
|
||||
//
|
||||
// Date : 4:13 下午 2021/1/3
|
||||
func (gw *GinWrapper) formatFieldList(inputFieldList []zap.Field) []zap.Field {
|
||||
if nil == inputFieldList {
|
||||
inputFieldList = make([]zap.Field, 0)
|
||||
}
|
||||
if nil != gw.ginCtx {
|
||||
// 自动扩充抽取字段,字段不存在的话,忽略掉
|
||||
for _, extractField := range gw.extractFieldList {
|
||||
if v, exist := gw.ginCtx.Get(extractField); exist {
|
||||
byteData, _ := json.Marshal(v)
|
||||
inputFieldList = append(inputFieldList, zap.String(extractField, string(byteData)))
|
||||
}
|
||||
}
|
||||
}
|
||||
return inputFieldList
|
||||
}
|
||||
|
||||
// Debug 日志
|
||||
//
|
||||
// Author : go_developer@163.com<张德满>
|
||||
//
|
||||
// Date : 4:14 下午 2021/1/3
|
||||
func (gw *GinWrapper) Debug(msg string, field ...zap.Field) {
|
||||
fieldList := gw.formatFieldList(field)
|
||||
gw.loggerInstance.Debug(msg, fieldList...)
|
||||
}
|
||||
|
||||
// Info 日志
|
||||
//
|
||||
// Author : go_developer@163.com<张德满>
|
||||
//
|
||||
// Date : 4:28 下午 2021/1/3
|
||||
func (gw *GinWrapper) Info(msg string, field ...zap.Field) {
|
||||
fieldList := gw.formatFieldList(field)
|
||||
gw.loggerInstance.Info(msg, fieldList...)
|
||||
}
|
||||
|
||||
// Warn 日志
|
||||
//
|
||||
// Author : go_developer@163.com<张德满>
|
||||
//
|
||||
// Date : 4:29 下午 2021/1/3
|
||||
func (gw *GinWrapper) Warn(msg string, field ...zap.Field) {
|
||||
fieldList := gw.formatFieldList(field)
|
||||
gw.loggerInstance.Warn(msg, fieldList...)
|
||||
}
|
||||
|
||||
// Error 日志
|
||||
//
|
||||
// Author : go_developer@163.com<张德满>
|
||||
//
|
||||
// Date : 4:29 下午 2021/1/3
|
||||
func (gw *GinWrapper) Error(msg string, field ...zap.Field) {
|
||||
fieldList := gw.formatFieldList(field)
|
||||
gw.loggerInstance.Error(msg, fieldList...)
|
||||
}
|
||||
|
||||
// Panic 日志
|
||||
//
|
||||
// Author : go_developer@163.com<张德满>
|
||||
//
|
||||
// Date : 4:29 下午 2021/1/3
|
||||
func (gw *GinWrapper) Panic(msg string, field ...zap.Field) {
|
||||
fieldList := gw.formatFieldList(field)
|
||||
gw.loggerInstance.Panic(msg, fieldList...)
|
||||
}
|
||||
|
||||
// DPanic 日志
|
||||
//
|
||||
// Author : go_developer@163.com<张德满>
|
||||
//
|
||||
// Date : 4:30 下午 2021/1/3
|
||||
func (gw *GinWrapper) DPanic(msg string, field ...zap.Field) {
|
||||
fieldList := gw.formatFieldList(field)
|
||||
gw.loggerInstance.DPanic(msg, fieldList...)
|
||||
}
|
||||
|
||||
// GetZapLoggerInstance 获取zap日志实例
|
||||
//
|
||||
// Author : go_developer@163.com<张德满>
|
||||
//
|
||||
// Date : 2021/01/03 22:56:47
|
||||
func (gw *GinWrapper) GetZapLoggerInstance() *zap.Logger {
|
||||
return gw.loggerInstance
|
||||
}
|
Reference in New Issue
Block a user