bbgo_origin/bbgo/slack/logrus_look.go
2020-09-05 16:22:46 +08:00

62 lines
1.2 KiB
Go

package slack
import (
"context"
"fmt"
"strings"
"github.com/sirupsen/logrus"
"github.com/slack-go/slack"
)
type LogHook struct {
Slack *slack.Client
ErrorChannel string
}
func (t *LogHook) Levels() []logrus.Level {
return []logrus.Level{
// log.InfoLevel,
logrus.ErrorLevel,
logrus.PanicLevel,
// log.WarnLevel,
}
}
func (t *LogHook) Fire(e *logrus.Entry) error {
var color = "#F0F0F0"
switch e.Level {
case logrus.DebugLevel:
color = "#9B30FF"
case logrus.InfoLevel:
color = "good"
case logrus.ErrorLevel, logrus.FatalLevel, logrus.PanicLevel:
color = "danger"
default:
color = "warning"
}
var slackAttachments []slack.Attachment = nil
// error fields
var fields []slack.AttachmentField
for k, d := range e.Data {
fields = append(fields, slack.AttachmentField{
Title: k, Value: fmt.Sprintf("%v", d),
})
}
slackAttachments = append(slackAttachments, slack.Attachment{
Color: color,
Title: strings.ToUpper(e.Level.String()),
Fields: fields,
})
_, _, err := t.Slack.PostMessageContext(context.Background(), t.ErrorChannel,
slack.MsgOptionText(":balloon: "+e.Message, true),
slack.MsgOptionAttachments(slackAttachments...))
return err
}