mirror of
https://github.com/c9s/bbgo.git
synced 2024-11-24 07:45:15 +00:00
notifier: redirect error, panic, fatal error to telegram
This commit is contained in:
parent
cacc24fb3c
commit
b8e18dd75c
|
@ -1012,6 +1012,8 @@ func (environ *Environment) setupTelegram(userConfig *Config, telegramBotToken s
|
||||||
var notifier = telegramnotifier.New(bot, opts...)
|
var notifier = telegramnotifier.New(bot, opts...)
|
||||||
Notification.AddNotifier(notifier)
|
Notification.AddNotifier(notifier)
|
||||||
|
|
||||||
|
log.AddHook(telegramnotifier.NewLogHook(notifier))
|
||||||
|
|
||||||
// allocate a store, so that we can save the chatID for the owner
|
// allocate a store, so that we can save the chatID for the owner
|
||||||
var messenger = interact.NewTelegram(bot)
|
var messenger = interact.NewTelegram(bot)
|
||||||
|
|
||||||
|
|
39
pkg/notifier/telegramnotifier/logrus_look.go
Normal file
39
pkg/notifier/telegramnotifier/logrus_look.go
Normal file
|
@ -0,0 +1,39 @@
|
||||||
|
package telegramnotifier
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"github.com/sirupsen/logrus"
|
||||||
|
"golang.org/x/time/rate"
|
||||||
|
)
|
||||||
|
|
||||||
|
var limiter = rate.NewLimiter(rate.Every(time.Minute), 3)
|
||||||
|
|
||||||
|
type LogHook struct {
|
||||||
|
notifier *Notifier
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewLogHook(notifier *Notifier) *LogHook {
|
||||||
|
return &LogHook{
|
||||||
|
notifier: notifier,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t *LogHook) Levels() []logrus.Level {
|
||||||
|
return []logrus.Level{
|
||||||
|
logrus.ErrorLevel,
|
||||||
|
logrus.FatalLevel,
|
||||||
|
logrus.PanicLevel,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t *LogHook) Fire(e *logrus.Entry) error {
|
||||||
|
if !limiter.Allow() {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
var message = fmt.Sprintf("[%s] %s", e.Level.String(), e.Message)
|
||||||
|
t.notifier.Notify(message)
|
||||||
|
return nil
|
||||||
|
}
|
|
@ -35,7 +35,7 @@ func UseBroadcast() Option {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// New
|
// New returns a telegram notifier instance
|
||||||
func New(bot *telebot.Bot, options ...Option) *Notifier {
|
func New(bot *telebot.Bot, options ...Option) *Notifier {
|
||||||
notifier := &Notifier{
|
notifier := &Notifier{
|
||||||
bot: bot,
|
bot: bot,
|
||||||
|
|
|
@ -11,7 +11,7 @@ import (
|
||||||
"golang.org/x/time/rate"
|
"golang.org/x/time/rate"
|
||||||
)
|
)
|
||||||
|
|
||||||
var limiter = rate.NewLimiter(rate.Every(time.Minute), 45)
|
var limiter = rate.NewLimiter(rate.Every(time.Minute), 3)
|
||||||
|
|
||||||
type LogHook struct {
|
type LogHook struct {
|
||||||
Slack *slack.Client
|
Slack *slack.Client
|
||||||
|
|
Loading…
Reference in New Issue
Block a user