-
-
Notifications
You must be signed in to change notification settings - Fork 204
/
log.go
72 lines (56 loc) · 1.5 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
package logger
import (
"io"
"os"
"github.com/sirupsen/logrus"
lWriter "github.com/sirupsen/logrus/hooks/writer"
"github.com/lxc/incus/v6/shared/termios"
)
// Setup a basic empty logger on init.
func init() {
logger := logrus.New()
logger.SetOutput(io.Discard)
Log = newWrapper(logger)
}
// InitLogger intializes a full logging instance.
func InitLogger(filepath string, syslogName string, verbose bool, debug bool, hook logrus.Hook) error {
logger := logrus.New()
logger.Level = logrus.DebugLevel
logger.SetOutput(io.Discard)
// Setup the formatter.
logger.Formatter = &logrus.TextFormatter{PadLevelText: true, FullTimestamp: true, ForceColors: termios.IsTerminal(int(os.Stderr.Fd()))}
// Setup log level.
levels := []logrus.Level{logrus.PanicLevel, logrus.FatalLevel, logrus.ErrorLevel, logrus.WarnLevel}
if debug {
levels = append(levels, logrus.InfoLevel, logrus.DebugLevel)
} else if verbose {
levels = append(levels, logrus.InfoLevel)
}
// Setup writers.
writers := []io.Writer{os.Stderr}
if filepath != "" {
f, err := os.OpenFile(filepath, os.O_WRONLY|os.O_CREATE|os.O_APPEND, 0600)
if err != nil {
return err
}
writers = append(writers, f)
}
logger.AddHook(&lWriter.Hook{
Writer: io.MultiWriter(writers...),
LogLevels: levels,
})
// Setup syslog.
if syslogName != "" {
err := setupSyslog(logger, syslogName)
if err != nil {
return err
}
}
// Add hooks.
if hook != nil {
logger.AddHook(hook)
}
// Set the logger.
Log = newWrapper(logger)
return nil
}