feat: use custom OnError function

moira-alert · Sep 27, 2024 · d2205b3 · d2205b3
1 parent b8c57f3
commit d2205b3
Showing 1 changed file with 11 additions and 2 deletions.
diff --git a/senders/telegram/init.go b/senders/telegram/init.go
@@ -80,8 +80,9 @@ func (sender *Sender) Init(senderSettings interface{}, logger moira.Logger, loca
 
 	sender.logger = logger
 	sender.bot, err = telebot.NewBot(telebot.Settings{
-		Token:  cfg.APIToken,
-		Poller: &telebot.LongPoller{Timeout: pollerTimeout},
+		Token:   cfg.APIToken,
+		Poller:  &telebot.LongPoller{Timeout: pollerTimeout},
+		OnError: sender.customOnErrorFunc,
 	})
 	if err != nil {
 		return sender.removeTokenFromError(err)
@@ -123,3 +124,11 @@ func (sender *Sender) runTelebot(contactType string) {
 func telegramLockKey(contactType string) string {
 	return telegramLockPrefix + contactType
 }
+
+func (sender *Sender) customOnErrorFunc(err error, _ telebot.Context) {
+	err = sender.removeTokenFromError(err)
+
+	sender.logger.Error().
+		Error(err).
+		Msg("Error inside telebot")
+}