-
Notifications
You must be signed in to change notification settings - Fork 0
/
log.go
95 lines (79 loc) · 1.82 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
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
package log
import (
"fmt"
"os"
"path/filepath"
"strings"
"time"
"github.com/rs/zerolog"
"github.com/rs/zerolog/log"
)
type LogErrorWriter struct{}
func (LogErrorWriter) Write(b []byte) (int, error) {
if len(b) <= 0 {
return 0, nil
}
log.Error().CallerSkipFrame(3).Msg(strings.TrimSpace(string(b)))
return len(b), nil
}
type LogInfoWriter struct{}
func (LogInfoWriter) Write(b []byte) (int, error) {
if len(b) <= 0 {
return 0, nil
}
log.Info().CallerSkipFrame(3).Msg(strings.TrimSpace(string(b)))
return len(b), nil
}
func init() {
log.Logger = log.Output(
zerolog.ConsoleWriter{
Out: os.Stderr,
TimeFormat: time.RFC1123Z,
FormatCaller: func(i interface{}) string {
return filepath.Join(
filepath.Base(filepath.Dir(fmt.Sprintf("%s", i))),
filepath.Base(fmt.Sprintf("%s", i)),
)
},
},
).With().Caller().Logger()
}
func Initialize(log_level string) {
switch log_level {
case zerolog.LevelTraceValue:
zerolog.SetGlobalLevel(zerolog.TraceLevel)
case zerolog.LevelDebugValue:
zerolog.SetGlobalLevel(zerolog.DebugLevel)
case zerolog.LevelInfoValue:
zerolog.SetGlobalLevel(zerolog.InfoLevel)
case zerolog.LevelWarnValue:
zerolog.SetGlobalLevel(zerolog.WarnLevel)
case zerolog.LevelErrorValue:
zerolog.SetGlobalLevel(zerolog.ErrorLevel)
case zerolog.LevelFatalValue:
zerolog.SetGlobalLevel(zerolog.FatalLevel)
default:
zerolog.SetGlobalLevel(zerolog.InfoLevel)
}
}
func Trace() *zerolog.Event {
return log.Logger.Trace()
}
func Debug() *zerolog.Event {
return log.Logger.Debug()
}
func Info() *zerolog.Event {
return log.Logger.Info()
}
func Warn() *zerolog.Event {
return log.Logger.Warn()
}
func Error() *zerolog.Event {
return log.Logger.Error()
}
func Fatal() *zerolog.Event {
return log.Logger.Fatal()
}
func Panic() *zerolog.Event {
return log.Logger.Panic()
}