-
Notifications
You must be signed in to change notification settings - Fork 85
/
logger.go
107 lines (90 loc) · 2.77 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
package logger
import (
"bytes"
"fmt"
"io"
"os"
"strconv"
"strings"
"github.com/rs/zerolog"
"github.com/rs/zerolog/log"
)
type JobEvent struct {
Type string `json:"type"`
Data interface{} `json:"data"`
Node string `json:"node"`
Job string `json:"job"`
}
var Stdout = struct{ io.Writer }{os.Stdout}
var Stderr = struct{ io.Writer }{os.Stderr}
func init() { //nolint:gochecknoinits // init with zerolog is idiomatic
zerolog.TimeFieldFormat = zerolog.TimeFormatUnix
logLevelString := strings.ToLower(os.Getenv("LOG_LEVEL"))
logTypeString := strings.ToLower(os.Getenv("LOG_TYPE"))
switch {
case logLevelString == "trace":
zerolog.SetGlobalLevel(zerolog.TraceLevel)
case logLevelString == "debug":
zerolog.SetGlobalLevel(zerolog.DebugLevel)
case logLevelString == "error":
zerolog.SetGlobalLevel(zerolog.ErrorLevel)
case logLevelString == "warn":
zerolog.SetGlobalLevel(zerolog.WarnLevel)
case logLevelString == "fatal":
zerolog.SetGlobalLevel(zerolog.FatalLevel)
default:
zerolog.SetGlobalLevel(zerolog.InfoLevel)
}
textWriter := zerolog.ConsoleWriter{Out: Stderr, TimeFormat: "[0607]", NoColor: false, PartsOrder: []string{
zerolog.TimestampFieldName,
zerolog.LevelFieldName,
zerolog.CallerFieldName,
zerolog.MessageFieldName}}
textWriter.FormatMessage = func(i interface{}) string {
return fmt.Sprintf("%s", i)
}
textWriter.FormatFieldName = func(i interface{}) string {
return fmt.Sprintf("%s:", i)
}
textWriter.FormatFieldValue = func(i interface{}) string {
return strings.ToUpper(fmt.Sprintf("%s", i))
}
zerolog.CallerMarshalFunc = func(file string, line int) string {
short := file
separatorCount := 2
countedSeparators := 0
for i := len(file) - 1; i > 0; i-- {
if file[i] == '/' {
countedSeparators += 1
if countedSeparators >= separatorCount {
short = file[i+1:]
break
}
}
}
file = short
return file + ":" + strconv.Itoa(line)
}
// we default to text output
var useLogWriter io.Writer = textWriter
if logTypeString == "json" {
// we just want json
useLogWriter = os.Stdout
} else if logTypeString == "combined" {
// we just want json and text and events
useLogWriter = zerolog.MultiLevelWriter(textWriter, os.Stdout)
} else if logTypeString == "event" {
// we just want events
useLogWriter = io.Discard
}
log.Logger = zerolog.New(useLogWriter).With().Timestamp().Caller().Logger()
}
func LoggerWithRuntimeInfo(runtimeInfo string) zerolog.Logger {
return log.With().Str("R", runtimeInfo).Logger()
}
func LoggerWithNodeAndJobInfo(nodeID, jobID string) zerolog.Logger {
return log.With().Str("N", nodeID).Str("J", jobID).Logger()
}
func LoggerTestLogger(logBuffer *bytes.Buffer) zerolog.Logger {
return zerolog.New(zerolog.MultiLevelWriter(io.MultiWriter(logBuffer, os.Stdout)))
}