-
-
Notifications
You must be signed in to change notification settings - Fork 9
/
logger.go
119 lines (103 loc) · 2.59 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
package config
import (
"fmt"
"io"
"os"
"strings"
"github.com/rs/zerolog"
zlog "github.com/rs/zerolog/log"
)
type Logger struct{}
// Log is exposed on the config as a drop-in replacement for our old logger
var Log *Logger
// These functions are provided to reduce refactoring.
func (l *Logger) Debug(msg string, err ...error) {
if len(err) == 1 {
zlog.Debug().Err(err[0]).Msg(msg)
return
}
zlog.Debug().Msg(msg)
}
func (l *Logger) Debugf(msg string, args ...interface{}) {
zlog.Debug().Msg(fmt.Sprintf(msg, args...))
}
func (l *Logger) Info(msg string, err ...error) {
if len(err) == 1 {
zlog.Info().Err(err[0]).Msg(msg)
return
}
zlog.Info().Msg(msg)
}
func (l *Logger) Infof(msg string, args ...interface{}) {
zlog.Info().Msg(fmt.Sprintf(msg, args...))
}
func (l *Logger) Warn(msg string, err ...error) {
if len(err) == 1 {
zlog.Warn().Err(err[0]).Msg(msg)
return
}
zlog.Warn().Msg(msg)
}
func (l *Logger) Warnf(msg string, args ...interface{}) {
zlog.Warn().Msg(fmt.Sprintf(msg, args...))
}
func (l *Logger) Error(msg string, err ...error) {
if len(err) == 1 {
zlog.Error().Err(err[0]).Msg(msg)
return
}
zlog.Error().Msg(msg)
}
func (l *Logger) Errorf(msg string, args ...interface{}) {
zlog.Error().Msg(fmt.Sprintf(msg, args...))
}
func (l *Logger) Fatal(msg string, err ...error) {
if len(err) == 1 {
zlog.Fatal().Err(err[0]).Msg(msg)
return
}
zlog.Fatal().Msg(msg)
}
func (l *Logger) Fatalf(msg string, args ...interface{}) {
zlog.Fatal().Msg(fmt.Sprintf(msg, args...))
}
func DoConfigureLogger(logPath string, logLevel string, prettyLogging bool) {
writers := io.MultiWriter(os.Stdout)
if len(logPath) > 0 {
if _, err := os.Stat(logPath); os.IsNotExist(err) {
file, err := os.Create(logPath)
if err != nil {
panic(err)
}
writers = io.MultiWriter(os.Stdout, file)
} else {
file, err := os.OpenFile(logPath, os.O_APPEND|os.O_WRONLY, os.ModeAppend)
if err != nil {
panic(err)
}
writers = io.MultiWriter(os.Stdout, file)
}
}
if prettyLogging {
zlog.Logger = zlog.Output(zerolog.ConsoleWriter{Out: writers})
} else {
zlog.Logger = zlog.Output(writers)
}
// Set the log level (default to info)
switch strings.ToLower(logLevel) {
case "debug":
zerolog.SetGlobalLevel(zerolog.DebugLevel)
case "info":
zerolog.SetGlobalLevel(zerolog.InfoLevel)
case "warn":
zerolog.SetGlobalLevel(zerolog.WarnLevel)
case "error":
zerolog.SetGlobalLevel(zerolog.ErrorLevel)
case "fatal":
zerolog.SetGlobalLevel(zerolog.FatalLevel)
case "panic":
zerolog.SetGlobalLevel(zerolog.PanicLevel)
default:
zerolog.SetGlobalLevel(zerolog.InfoLevel)
}
}