logger/logger.go

56 lines
1.4 KiB
Go
Raw Normal View History

2021-12-14 10:53:50 +08:00
package logger
import (
"fmt"
"github.com/caarlos0/env/v6"
2021-12-14 10:53:50 +08:00
"go.uber.org/zap"
"go.uber.org/zap/zapcore"
)
var config LogConfig
2022-05-18 18:53:17 +08:00
var initialized bool
2021-12-14 10:53:50 +08:00
2021-12-14 11:08:57 +08:00
// 避免异常在第一次调用时初始化一个只打印到控制台的logger
func init() {
2022-05-18 18:53:17 +08:00
if !initialized {
// 从环境变量读取配置
var c LogConfig
if err := env.Parse(&c); err != nil {
fmt.Println("日志配置解析错误: " + err.Error())
c = LogConfig{Mode: Dev, LokiEnable: false, FileEnable: false}
}
2022-05-18 11:17:22 +08:00
// 如果值错了直接默认为Prod
if c.Mode != Dev && c.Mode != Prod {
c.Mode = Prod
}
InitLogger(c)
2021-12-15 09:21:08 +08:00
}
2021-12-14 11:08:57 +08:00
}
2021-12-14 10:53:50 +08:00
// InitLogger 初始化日志工具
func InitLogger(c LogConfig) {
config = c
var cores []zapcore.Core
// 生成输出到控制台的Core
consoleCore := initConsoleCore()
cores = append(cores, consoleCore)
// 生成输出到Loki的Core
if config.LokiEnable {
lokiCore := initLokiCore()
cores = append(cores, lokiCore)
}
// 输出到文件的Core
if config.FileEnable {
fileCore := initFileCore()
cores = append(cores, fileCore)
}
// 增加 caller 信息
// AddCallerSkip 输出的文件名和行号是调用封装函数的位置,而不是调用日志函数的位置
2022-05-18 17:43:19 +08:00
logger := zap.New(zapcore.NewTee(cores...), zap.AddCaller(), zap.AddCallerSkip(1))
2022-05-18 18:53:17 +08:00
initialized = true
2022-05-18 18:11:43 +08:00
// 给GORM单独生成一个
gormZap = zap.New(zapcore.NewTee(cores...), zap.AddCaller(), zap.AddCallerSkip(3)).Sugar()
zap.ReplaceGlobals(logger)
2021-12-14 10:53:50 +08:00
}