mirror of
https://github.com/c9s/bbgo.git
synced 2024-09-20 08:11:08 +00:00
fix: add rate limit on telegram api and split messages by unicode with size limitation
This commit is contained in:
parent
08e6e5efdf
commit
097860af6b
|
@ -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))
|
||||
}
|
||||
}
|
||||
}
|
||||
})
|
||||
|
|
Loading…
Reference in New Issue
Block a user