/
logger.go
85 lines (72 loc) · 1.75 KB
/
logger.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
package gruutbot
import (
"strings"
"github.com/sirupsen/logrus"
)
type Logger interface {
Trace(v ...interface{})
Debug(v ...interface{})
Info(v ...interface{})
Warn(v ...interface{})
Error(v ...interface{})
Fatal(v ...interface{})
Panic(v ...interface{})
Tracef(format string, v ...interface{})
Debugf(format string, v ...interface{})
Infof(format string, v ...interface{})
Warnf(format string, v ...interface{})
Errorf(format string, v ...interface{})
Fatalf(format string, v ...interface{})
Panicf(format string, v ...interface{})
Traceln(v ...interface{})
Debugln(v ...interface{})
Infoln(v ...interface{})
Warnln(v ...interface{})
Errorln(v ...interface{})
Fatalln(v ...interface{})
Panicln(v ...interface{})
}
type LogLevel string
const (
TraceLevel LogLevel = "trace"
DebugLevel LogLevel = "debug"
InfoLevel LogLevel = "info"
WarnLevel LogLevel = "warn"
ErrorLevel LogLevel = "error"
FatalLevel LogLevel = "fatal"
logLevelKey = "log_level"
)
func logrusLogger(level LogLevel) (lf *logrus.Entry) {
l := logrus.New()
if gviper.IsSet(logLevelKey) {
level = LogLevel(gviper.GetString(logLevelKey))
}
l.SetLevel(logrusLevel(level))
lf = l.WithField("lib", "gruutbot")
lf.Infoln("Log level set to", l.Level)
return
}
func logrusLevel(level LogLevel) logrus.Level {
level = levelToLower(level)
switch level {
case TraceLevel:
return logrus.TraceLevel
case DebugLevel:
return logrus.DebugLevel
case InfoLevel:
return logrus.InfoLevel
case WarnLevel:
return logrus.WarnLevel
case ErrorLevel:
return logrus.ErrorLevel
case FatalLevel:
return logrus.FatalLevel
default:
return logrus.InfoLevel
}
}
func levelToLower(l LogLevel) LogLevel {
s := string(l)
s = strings.ToLower(s)
return LogLevel(s)
}