[go-每日一库] go使用uber的zap库简单实现日志功能

代码结构：

.
+--- go.mod
+--- go.sum
+--- logs
|   +--- zap.log
+--- main.go
+--- zlog
|   +--- log.go

见代码：

// https://mp.weixin.qq.com/s/Jh2iFY5uGe0qCFdKZWjotA
package zlog

import (
	"github.com/natefinch/lumberjack"
	"go.uber.org/zap"
	"go.uber.org/zap/zapcore"

	"os"
	"path"
	"runtime"
)

var logger *zap.Logger

func init()  {
	encoderConfig := zap.NewProductionEncoderConfig()
	// set time format
	encoderConfig.EncodeTime = zapcore.ISO8601TimeEncoder
	// set log codec -> json
	encoder := zapcore.NewJSONEncoder(encoderConfig)

	file, _ := os.OpenFile("D:\\demo1\\src\\demo\\demo06\\zap-log\\logs\\zap.log", os.O_CREATE|os.O_APPEND|os.O_WRONLY, 644)
	fileWriterSyncer := zapcore.AddSync(file)

	core := zapcore.NewTee(
		// output to console
		zapcore.NewCore(encoder, zapcore.AddSync(os.Stdout), zapcore.DebugLevel),
		// output to file
		zapcore.NewCore(encoder, fileWriterSyncer, zapcore.DebugLevel),
	)

	logger = zap.New(core)
}


// log rotate
func getFileLogWriter() (writeSyncer zapcore.WriteSyncer) {
	// use lumberjack to rotate
	lumberJackLogger := &lumberjack.Logger{
		Filename: "D:\\demo1\\src\\demo\\demo06\\zap-log\\logs\\zap.log",
		MaxSize: 10, // filesize, MB
		MaxBackups: 7, // >7, than rotate
		MaxAge: 1, // frequency, 1 day
		Compress: false,
	}
	return zapcore.AddSync(lumberJackLogger)
}

func Info(message string, fields ...zap.Field) {
	callerFields := getCallerInfoForLog()
	fields = append(fields, callerFields...)
	logger.Info(message, fields...)
}

func Debug(message string, fields ...zap.Field) {
	callerFields := getCallerInfoForLog()
	fields = append(fields, callerFields...)
	logger.Debug(message, fields...)
}

func Error(message string, fields ...zap.Field) {
	callerFields := getCallerInfoForLog()
	fields = append(fields, callerFields...)
	logger.Error(message, fields...)
}

func Warn(message string, fields ...zap.Field) {
	callerFields := getCallerInfoForLog()
	fields = append(fields, callerFields...)
	logger.Warn(message, fields...)
}

func getCallerInfoForLog() (callerFields []zap.Field) {

	pc, file, line, ok := runtime.Caller(2) // 回溯两层，拿到写日志的调用方的函数信息
	if !ok {
		return
	}
	funcName := runtime.FuncForPC(pc).Name()
	funcName = path.Base(funcName) //Base函数返回路径的最后一个元素，只保留函数名

	callerFields = append(callerFields, zap.String("func", funcName), zap.String("file", file), zap.Int("line", line))
	return
}

main.go

package main

import (
	"go.uber.org/zap"
	"zap-log/zlog"
)

type user struct {
	Name string `json:"name"`
}

func main()  {
	user := &user{
		Name: "Kevin",
	}
	zlog.Info("test log", zap.Any("user", user))
}