-
Notifications
You must be signed in to change notification settings - Fork 32
/
logger.go
145 lines (114 loc) · 2.98 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
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
package logger
import (
log "github.com/sirupsen/logrus"
"os"
)
// Logger methods interface
//
//go:generate mockery --name ILogger
type ILogger interface {
getLevel() log.Level
Debug(args ...interface{})
Debugf(format string, args ...interface{})
Info(args ...interface{})
Infof(format string, args ...interface{})
Warn(args ...interface{})
Warnf(format string, args ...interface{})
Error(args ...interface{})
Errorf(format string, args ...interface{})
Panic(args ...interface{})
Panicf(format string, args ...interface{})
Fatal(args ...interface{})
Fatalf(format string, args ...interface{})
Trace(args ...interface{})
Tracef(format string, args ...interface{})
}
var (
Logger ILogger
)
type LoggerConfig struct {
LogLevel string `mapstructure:"level"`
}
// Application logger
type appLogger struct {
level string
logger *log.Logger
}
// For mapping config logger to email_service logger levels
var loggerLevelMap = map[string]log.Level{
"debug": log.DebugLevel,
"info": log.InfoLevel,
"warn": log.WarnLevel,
"error": log.ErrorLevel,
"panic": log.PanicLevel,
"fatal": log.FatalLevel,
"trace": log.TraceLevel,
}
func (l *appLogger) getLevel() log.Level {
level, exist := loggerLevelMap[l.level]
if !exist {
return log.DebugLevel
}
return level
}
// InitLogger Init logger
func InitLogger(cfg *LoggerConfig) ILogger {
l := &appLogger{level: cfg.LogLevel}
l.logger = log.StandardLogger()
logLevel := l.getLevel()
env := os.Getenv("APP_ENV")
if env == "production" {
log.SetFormatter(&log.JSONFormatter{})
} else {
// The TextFormatter is default, you don't actually have to do this.
log.SetFormatter(&log.TextFormatter{
DisableColors: false,
ForceColors: true,
FullTimestamp: true,
})
}
log.SetLevel(logLevel)
return l
}
func (l *appLogger) Debug(args ...interface{}) {
l.logger.Debug(args...)
}
func (l *appLogger) Debugf(format string, args ...interface{}) {
l.logger.Debugf(format, args...)
}
func (l *appLogger) Info(args ...interface{}) {
l.logger.Info(args...)
}
func (l *appLogger) Infof(format string, args ...interface{}) {
l.logger.Infof(format, args...)
}
func (l *appLogger) Trace(args ...interface{}) {
l.logger.Trace(args...)
}
func (l *appLogger) Tracef(format string, args ...interface{}) {
l.logger.Tracef(format, args...)
}
func (l *appLogger) Error(args ...interface{}) {
l.logger.Error(args...)
}
func (l *appLogger) Errorf(format string, args ...interface{}) {
l.logger.Errorf(format, args...)
}
func (l *appLogger) Warn(args ...interface{}) {
l.logger.Warn(args...)
}
func (l *appLogger) Warnf(format string, args ...interface{}) {
l.logger.Warnf(format, args...)
}
func (l *appLogger) Panic(args ...interface{}) {
l.logger.Panic(args...)
}
func (l *appLogger) Panicf(format string, args ...interface{}) {
l.logger.Panicf(format, args...)
}
func (l *appLogger) Fatal(args ...interface{}) {
l.logger.Fatal(args...)
}
func (l *appLogger) Fatalf(format string, args ...interface{}) {
l.logger.Fatalf(format, args...)
}