lychiyu/bbgo_origin
1
0
Fork 0
mirror of https://github.com/c9s/bbgo.git synced 2024-09-20 08:11:08 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix: add rate limit on telegram api and split messages by unicode with size limitation

Browse Source
This commit is contained in:
zenix 2022-09-20 17:02:02 +09:00
parent 08e6e5efdf
commit 097860af6b
1 changed files with 33 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

36
pkg/interact/telegram.go
View File

@ -5,8 +5,10 @@ import (
"fmt"
"strings"
"time"
"unicode/utf8"
log "github.com/sirupsen/logrus"
"golang.org/x/time/rate"
"gopkg.in/tucnak/telebot.v2"
)
@ -15,6 +17,10 @@ func init() {
_ = Reply(&TelegramReply{})
}
var sendLimiter = rate.NewLimiter(10, 2)
const maxMessageSize int = 3000
type TelegramSessionMap map[int64]*TelegramSession
type TelegramSession struct {
@ -62,7 +68,14 @@ type TelegramReply struct {
}
func (r *TelegramReply) Send(message string) {
checkSendErr(r.bot.Send(r.session.Chat, message))
var left, right int
for left, right = 0, maxMessageSize; right < len(message); left, right = right, right+maxMessageSize {
for !utf8.RuneStart(message[right]) {
right--
}
checkSendErr(r.bot.Send(r.session.Chat, message[left:right]))
}
checkSendErr(r.bot.Send(r.session.Chat, message[left:]))
}
func (r *TelegramReply) Message(message string) {
@ -132,7 +145,7 @@ func (tm *Telegram) SetTextMessageResponder(responder Responder) {
tm.textMessageResponder = responder
}
func (tm *Telegram) Start(context.Context) {
func (tm *Telegram) Start(ctx context.Context) {
tm.Bot.Handle(telebot.OnCallback, func(c *telebot.Callback) {
log.Infof("[telegram] onCallback: %+v", c)
})
@ -158,7 +171,24 @@ func (tm *Telegram) Start(context.Context) {
if reply.set {
reply.build()
if len(reply.message) > 0 || reply.menu != nil {
checkSendErr(tm.Bot.Send(m.Chat, reply.message, reply.menu))
var left, right int
for left, right = 0, maxMessageSize; right < len(reply.message); left, right = right, right+maxMessageSize {
for !utf8.RuneStart(reply.message[right]) {
right--
}
if err := sendLimiter.Wait(ctx); err != nil {
log.WithError(err).Errorf("telegram send limit exceeded")
}
checkSendErr(tm.Bot.Send(m.Chat, reply.message[left:right]))
}
if left < len(reply.message) {
if err := sendLimiter.Wait(ctx); err != nil {
log.WithError(err).Errorf("telegram send limit exceeded")
}
// only set menu on the last message
checkSendErr(tm.Bot.Send(m.Chat, reply.message[left:], reply.menu))
}
}
}
})