-
Notifications
You must be signed in to change notification settings - Fork 0
/
handler.go
50 lines (40 loc) · 1.13 KB
/
handler.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
package logrus
import (
"context"
"github.com/go-4devs/log"
lrg "github.com/sirupsen/logrus"
)
// Levels maps
type Levels map[log.Level]lrg.Level
// Option configure levels
type Option func(Levels)
// WithLevel set lorgus level to log level
func WithLevel(level log.Level, loggusLevel lrg.Level) Option {
return func(l Levels) {
l[level] = loggusLevel
}
}
// New create new lorgus handler
func New(logger *lrg.Logger, level log.Level, opts ...Option) log.Handler {
levels := map[log.Level]lrg.Level{
log.LevelEmergency: lrg.PanicLevel,
log.LevelAlert: lrg.FatalLevel,
log.LevelCritical: lrg.ErrorLevel,
log.LevelError: lrg.ErrorLevel,
log.LevelWarning: lrg.WarnLevel,
log.LevelNotice: lrg.InfoLevel,
log.LevelInfo: lrg.InfoLevel,
log.LevelDebug: lrg.DebugLevel,
}
for _, o := range opts {
o(levels)
}
logger.SetLevel(levels[level])
return func(ctx context.Context, level log.Level, msg string, fields log.Fields) {
lrgFields := make(lrg.Fields, len(fields))
for _, field := range fields {
lrgFields[field.Key] = field.Value
}
logger.WithFields(lrgFields).Log(levels[level], msg)
}
}