/
logformat.go
39 lines (34 loc) · 1 KB
/
logformat.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
package cmd
import (
"bytes"
"github.com/fatih/color"
log "github.com/sirupsen/logrus"
)
type logFormatter struct{}
var logLevelColor = map[log.Level]struct{ key, value *color.Color }{
log.PanicLevel: {color.New(color.FgRed), color.New(color.FgRed)},
log.FatalLevel: {color.New(color.FgRed), color.New(color.FgRed)},
log.ErrorLevel: {color.New(color.FgRed), color.New(color.FgRed)},
log.WarnLevel: {color.New(color.FgYellow), nil},
log.InfoLevel: {color.New(color.FgCyan), nil},
log.DebugLevel: {color.New(color.FgWhite), nil},
}
func (l *logFormatter) Format(e *log.Entry) ([]byte, error) {
var buf bytes.Buffer
c := logLevelColor[e.Level]
buf.WriteString(c.key.Sprint(e.Level.String()))
buf.WriteString(": ")
if c.value != nil {
buf.WriteString(c.value.Sprint(e.Message))
} else {
buf.WriteString(e.Message)
}
if stack, ok := e.Data["stack"]; ok {
if s, ok := stack.(string); ok {
buf.WriteString(" - ")
buf.WriteString(formatName(s))
}
}
buf.WriteByte('\n')
return buf.Bytes(), nil
}