56 lines
1.4 KiB
Go
56 lines
1.4 KiB
Go
|
package logger
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"fmt"
|
|||
|
|
"github.com/caarlos0/env/v6"
|
|||
|
|
"go.uber.org/zap"
|
|||
|
|
"go.uber.org/zap/zapcore"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
var config LogConfig
|
|||
|
|
var initialized bool
|
|||
|
|
|
|||
|
|
// 避免异常，在第一次调用时初始化一个只打印到控制台的logger
|
|||
|
|
func init() {
|
|||
|
|
if !initialized {
|
|||
|
|
// 从环境变量读取配置
|
|||
|
|
var c LogConfig
|
|||
|
|
if err := env.Parse(&c); err != nil {
|
|||
|
|
fmt.Println("日志配置解析错误: " + err.Error())
|
|||
|
|
c = LogConfig{Mode: Dev, LokiEnable: false, FileEnable: false}
|
|||
|
|
}
|
|||
|
|
// 如果值错了，直接默认为Prod
|
|||
|
|
if c.Mode != Dev && c.Mode != Prod {
|
|||
|
|
c.Mode = Prod
|
|||
|
|
}
|
|||
|
|
InitLogger(c)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 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 输出的文件名和行号是调用封装函数的位置，而不是调用日志函数的位置
|
|||
|
|
logger := zap.New(zapcore.NewTee(cores...), zap.AddCaller(), zap.AddCallerSkip(1))
|
|||
|
|
initialized = true
|
|||
|
|
// 给GORM单独生成一个
|
|||
|
|
gormZap = zap.New(zapcore.NewTee(cores...), zap.AddCaller(), zap.AddCallerSkip(3)).Sugar()
|
|||
|
|
zap.ReplaceGlobals(logger)
|
|||
|
|
}
|