/
log.go
58 lines (50 loc) · 1.26 KB
/
log.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
package zapLog
import (
"github.com/aaronchen2k/deeptest/internal/pkg/config"
"github.com/aaronchen2k/deeptest/internal/pkg/consts"
myZap "github.com/aaronchen2k/deeptest/pkg/core/zap"
logUtils "github.com/aaronchen2k/deeptest/pkg/lib/log"
"github.com/snowlyg/helper/dir"
"go.uber.org/zap"
"go.uber.org/zap/zapcore"
)
// level 日志级别
var level zapcore.Level
// Init 初始化日志服务
func Init() {
var logger *zap.Logger
logDir := "log"
if consts.RunFrom == consts.FromServer {
logDir = config.CONFIG.Zap.Director
}
if !dir.IsExist(logDir) {
dir.InsureDir(logDir)
}
switch config.CONFIG.Zap.Level { // 初始化配置文件的Level
case "debug":
level = zap.DebugLevel
case "info":
level = zap.InfoLevel
case "warn":
level = zap.WarnLevel
case "error":
level = zap.ErrorLevel
case "dpanic":
level = zap.DPanicLevel
case "panic":
level = zap.PanicLevel
case "fatal":
level = zap.FatalLevel
default:
level = zap.InfoLevel
}
if level == zap.DebugLevel || level == zap.ErrorLevel {
logger = zap.New(myZap.GetEncoderCore(level), zap.AddStacktrace(level))
} else {
logger = zap.New(myZap.GetEncoderCore(level))
}
if config.CONFIG.Zap.ShowLine {
logger = logger.WithOptions(zap.AddCaller())
}
logUtils.Logger = logger
}