51 lines
808 B
Go
51 lines
808 B
Go
|
package event
|
||
|
|
|
||
|
|
import (
|
||
|
|
"os"
|
||
|
|
|
||
|
|
"github.com/leandro-lugaresi/hub"
|
||
|
|
"github.com/sirupsen/logrus"
|
||
|
|
)
|
||
|
|
|
||
|
|
var Log *logrus.Logger
|
||
|
|
|
||
|
|
|
||
|
|
type Hook struct {
|
||
|
|
hub *hub.Hub
|
||
|
|
}
|
||
|
|
|
||
|
|
func NewHook(hub *hub.Hub) *Hook {
|
||
|
|
return &Hook{hub: hub}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (h *Hook) Fire(entry *logrus.Entry) error {
|
||
|
|
h.hub.Publish(Message{
|
||
|
|
Name: "log." + entry.Level.String(),
|
||
|
|
Fields: Data{
|
||
|
|
"time": entry.Time,
|
||
|
|
"level": entry.Level.String(),
|
||
|
|
"msg": entry.Message,
|
||
|
|
},
|
||
|
|
})
|
||
|
|
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (h *Hook) Levels() []logrus.Level {
|
||
|
|
return logrus.AllLevels
|
||
|
|
}
|
||
|
|
|
||
|
|
func init() {
|
||
|
|
hooks := logrus.LevelHooks{}
|
||
|
|
hooks.Add(NewHook(SharedHub()))
|
||
|
|
|
||
|
|
Log = &logrus.Logger{
|
||
|
|
Out: os.Stderr,
|
||
|
|
Formatter: &logrus.TextFormatter{},
|
||
|
|
Hooks: hooks,
|
||
|
|
Level: logrus.DebugLevel,
|
||
|
|
ExitFunc: os.Exit,
|
||
|
|
ReportCaller: false,
|
||
|
|
}
|
||
|
|
}
|