/
loghooks.go
83 lines (68 loc) · 1.83 KB
/
loghooks.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
package env
import (
"io"
"github.com/sirupsen/logrus"
"gopkg.in/natefinch/lumberjack.v2"
)
// LogFileConfig keeps configuration settings required for lumberjack
type LogFileConfig struct {
Filename string
MaxSize int
Level logrus.Level
Formatter logrus.Formatter
}
// LogFileHook structure implements logfile hook for logrus
type LogFileHook struct {
Config LogFileConfig
logWriter io.Writer
}
// ConsoleWriterHook is a hook that writes logs of specified LogLevels to specified Writer
type ConsoleWriterHook struct {
Writer io.Writer
LogLevels []logrus.Level
Formatter logrus.Formatter
}
// NewLogFileHook instantiates hook and implements Hook interface
func NewLogFileHook(config LogFileConfig) (logrus.Hook, error) {
hook := LogFileHook{
Config: config,
}
hook.logWriter = &lumberjack.Logger{
Filename: config.Filename,
MaxSize: config.MaxSize,
}
return &hook, nil
}
// Levels defines and returns the log levels under which logrus fires the "Fire"
func (hook *LogFileHook) Levels() []logrus.Level {
return logrus.AllLevels[:hook.Config.Level+1]
}
// Fire implements actual write()
func (hook *LogFileHook) Fire(entry *logrus.Entry) (err error) {
b, err := hook.Config.Formatter.Format(entry)
if err != nil {
return err
}
_, err = hook.logWriter.Write(b)
if err != nil {
return err
}
return nil
}
// Fire will be called when some logging function is called with current hook
// It will format log entry to string and write it to appropriate writer
func (hook *ConsoleWriterHook) Fire(entry *logrus.Entry) error {
b, err := hook.Formatter.Format(entry)
if err != nil {
return err
}
_, err = hook.Writer.Write(b)
if err != nil {
return err
}
return err
}
// Levels define on which log levels this hook would trigger
func (hook *ConsoleWriterHook) Levels() []logrus.Level {
return hook.LogLevels
}