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
|
package utils
import (
"io"
"os"
"strings"
"syscall"
"time"
"github.com/rs/zerolog"
"github.com/rs/zerolog/log"
"golang.org/x/term"
)
// RunningOnTerminal checks whether stdout is terminal
func RunningOnTerminal() bool {
return term.IsTerminal(syscall.Stdout)
}
type LogWriter struct {
Logger zerolog.Logger
}
func (lw LogWriter) Write(bs []byte) (int, error) {
return lw.Logger.With().Str("level", "info").Logger().Write(bs)
}
func SetupJSONLogger(levelStr string, w io.Writer) {
zerolog.MessageFieldName = "message"
zerolog.LevelFieldName = "level"
var tsHook timestampHook
log.Logger = zerolog.New(w).
Hook(&tsHook).
Level(GetLogLevelOrDebug(levelStr))
}
func SetupDefaultLogger(levelStr string) {
zerolog.MessageFieldName = "message"
zerolog.LevelFieldName = "level"
log.Logger = zerolog.New(zerolog.ConsoleWriter{Out: os.Stderr,
FormatTimestamp: func(_ interface{}) string { return "" },
NoColor: !RunningOnTerminal(),
}).Level(GetLogLevelOrDebug(levelStr))
}
func GetLogLevelOrDebug(levelStr string) zerolog.Level {
levelStr = strings.ToLower(levelStr)
if levelStr == "warning" {
levelStr = "warn"
}
var level zerolog.Level
err := level.UnmarshalText([]byte(levelStr))
if err == nil {
return level
}
log.Warn().Msgf("Unknown log level '%s', defaulting to debug", levelStr)
return zerolog.DebugLevel
}
type timestampHook struct{}
func (h *timestampHook) Run(e *zerolog.Event, _ zerolog.Level, _ string) {
t := time.Now()
ts := t.Format(time.RFC3339)
e.Str("time", ts)
}
|
