/
nsqd.go
88 lines (68 loc) · 1.43 KB
/
nsqd.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
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
package log
import (
"strings"
"go.uber.org/zap/zapcore"
)
type NSQLogger struct {
logger Logger
}
func NewNSQLogger(logger Logger) NSQLogger {
return NSQLogger{logger: logger}
}
func (nl NSQLogger) Output(calldepth int, s string) error {
output := strings.SplitN(s, " ", 2) // nolint: gomnd
if len(output) <= 1 {
nl.defaultOutput(s)
return nil
}
logLvl := parseNSQLogLvl(output[0])
if logLvl == -2 {
nl.defaultOutput(s)
return nil
}
nl.logger.ForLogLevel(logLvl).Msg(strings.TrimSpace(output[1]))
return nil
}
func (nl NSQLogger) defaultOutput(s string) {
nl.logger.Info().Msg(s)
}
const unknownErrLevel = -2
func parseNSQLogLvl(s string) zapcore.Level {
var lvl zapcore.Level
switch s {
case "TRC", "DBG":
lvl = zapcore.DebugLevel
case "INF":
lvl = zapcore.InfoLevel
case "WRN":
lvl = zapcore.WarnLevel
case "ERR":
lvl = zapcore.ErrorLevel
case "FTL":
lvl = zapcore.FatalLevel
case "PNC":
lvl = zapcore.PanicLevel
default:
lvl = unknownErrLevel
}
return lvl
}
func (nl NSQLogger) LogLevel() int {
return nsqLogLvlFromZapLogLvl(nl.logger.Level)
}
func nsqLogLvlFromZapLogLvl(level zapcore.Level) int {
var nsqlLL int
switch level {
case zapcore.DebugLevel:
nsqlLL = 0
case zapcore.InfoLevel:
nsqlLL = 1
case zapcore.WarnLevel:
nsqlLL = 2
case zapcore.ErrorLevel, zapcore.FatalLevel, zapcore.PanicLevel, zapcore.DPanicLevel:
nsqlLL = 3
case -2:
nsqlLL = 4
}
return nsqlLL
}