photoprism/internal/event/log.go

50 lines
808 B
Go
Raw Normal View History

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,
}
}