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)
|
||
}
|