-
Notifications
You must be signed in to change notification settings - Fork 102
/
log.go
108 lines (89 loc) · 2.61 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
96
97
98
99
100
101
102
103
104
105
106
107
108
package log
import (
"fmt"
"log"
"log/syslog"
"os"
"regexp"
"strings"
kitlog "github.com/go-kit/kit/log"
)
// Level is a log level such a Debug or Error
type Level int
const (
syslogFlags = log.Llongfile
normalFlags = log.LUTC | log.Ldate | log.Ltime | log.Llongfile
// LevelDebug enables debug logging
LevelDebug Level = iota
// LevelError enables error logging
LevelError Level = iota
)
var (
debuglog = log.New(os.Stdout, "DEBUG: ", normalFlags)
errlog = log.New(os.Stderr, "ERROR: ", normalFlags)
kitlogLogger = kitlog.NewNopLogger()
kitlogTrimmer = regexp.MustCompile("level=[a-zA-Z]+")
kitlogDebug = kitlog.NewLogfmtLogger(CollectorWriter{debuglog, kitlogTrimmer})
level = LevelError
)
// CollectorWriter implements io.Writer.
type CollectorWriter struct {
stdlog *log.Logger
trimmerRegex *regexp.Regexp
}
// Write implements io.Writer.
func (w CollectorWriter) Write(p []byte) (int, error) {
out := strings.TrimSpace(string(w.trimmerRegex.ReplaceAll(p, []byte{})))
// Adjust the calldepth to account for the kitlog frames
err := w.stdlog.Output(5, out)
if err != nil {
fmt.Fprintf(os.Stderr, "ERROR writing log output: %+v", err)
}
return len(p), nil
}
// SetLevel sets the log level
func SetLevel(l Level) {
level = l
}
// InitSyslog initializes logging to syslog
func InitSyslog() (err error) {
dl, err := syslog.NewLogger(syslog.LOG_NOTICE|syslog.LOG_SYSLOG, syslogFlags)
if err != nil {
return fmt.Errorf("InitSyslog failed to initialize debug logger: %+v", err)
}
debuglog = dl
el, err := syslog.NewLogger(syslog.LOG_ERR|syslog.LOG_SYSLOG, syslogFlags)
if err != nil {
return fmt.Errorf("InitSyslog failed to initialize error logger: %+v", err)
}
errlog = el
return nil
}
// Debug prints a debug message. If syslog is enabled then LOG_NOTICE is used
func Debug(msg string, params ...interface{}) {
if level > LevelDebug {
return
}
if err := debuglog.Output(2, fmt.Sprintf(msg, params...)); err != nil {
fmt.Fprintf(os.Stderr, "ERROR writing log output: %+v", err)
}
}
// Error prints an error message. If syslog is enabled then LOG_ERR is used
func Error(msg string, params ...interface{}) {
if err := errlog.Output(2, fmt.Sprintf(msg, params...)); err != nil {
fmt.Fprintf(os.Stderr, "ERROR writing log output: %+v", err)
}
}
// Fatal logs Error and exits 1
func Fatal(msg string, params ...interface{}) {
if err := errlog.Output(2, fmt.Sprintf(msg, params...)); err != nil {
fmt.Fprintf(os.Stderr, "ERROR writing log output: %+v", err)
}
os.Exit(1)
}
func GetCollectorLogger() kitlog.Logger {
if level > LevelDebug {
return kitlogLogger
}
return kitlogDebug
}