GithubBackup/SaveAny-Bot
1
0
Fork 0
mirror of https://github.com/krau/SaveAny-Bot.git synced 2026-09-07 16:46:37 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: make retry middleware configurable

Browse Source
This commit is contained in:
krau
2025-07-04 13:38:55 +08:00
parent 14ba2afdf8
commit 80146176f0
2 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
client/bot/handlers/utils/shortcut/message.go
+2 -2
View File
@@ -14,7 +14,7 @@ import (
"github.com/krau/SaveAny-Bot/client/bot/handlers/utils/mediautil" "github.com/krau/SaveAny-Bot/client/bot/handlers/utils/mediautil"
"github.com/krau/SaveAny-Bot/client/bot/handlers/utils/msgelem" "github.com/krau/SaveAny-Bot/client/bot/handlers/utils/msgelem"
"github.com/krau/SaveAny-Bot/client/bot/handlers/utils/re" "github.com/krau/SaveAny-Bot/client/bot/handlers/utils/re"
"github.com/krau/SaveAny-Bot/client/user" uc "github.com/krau/SaveAny-Bot/client/user"
"github.com/krau/SaveAny-Bot/common/cache" "github.com/krau/SaveAny-Bot/common/cache"
"github.com/krau/SaveAny-Bot/common/utils/tgutil" "github.com/krau/SaveAny-Bot/common/utils/tgutil"
"github.com/krau/SaveAny-Bot/common/utils/tphutil" "github.com/krau/SaveAny-Bot/common/utils/tphutil"
@@ -103,7 +103,7 @@ func GetFilesFromUpdateLinkMessageWithReplyEdit(ctx *ext.Context, update *ext.Up
tctx := ctx tctx := ctx
if config.Cfg.Telegram.Userbot.Enable { if config.Cfg.Telegram.Userbot.Enable {
tctx = user.GetCtx() tctx = uc.GetCtx()
} }
for _, link := range msgLinks { for _, link := range msgLinks {
client/middleware/default.go
+2 -1
View File
@@ -9,13 +9,14 @@ import (
"github.com/gotd/td/telegram" "github.com/gotd/td/telegram"
"github.com/krau/SaveAny-Bot/client/middleware/recovery" "github.com/krau/SaveAny-Bot/client/middleware/recovery"
"github.com/krau/SaveAny-Bot/client/middleware/retry" "github.com/krau/SaveAny-Bot/client/middleware/retry"
"github.com/krau/SaveAny-Bot/config"
) )
// https://github.com/iyear/tdl/blob/master/core/tclient/tclient.go // https://github.com/iyear/tdl/blob/master/core/tclient/tclient.go
func NewDefaultMiddlewares(ctx context.Context, timeout time.Duration) []telegram.Middleware { func NewDefaultMiddlewares(ctx context.Context, timeout time.Duration) []telegram.Middleware {
return []telegram.Middleware{ return []telegram.Middleware{
recovery.New(ctx, newBackoff(timeout)), recovery.New(ctx, newBackoff(timeout)),
retry.New(5), retry.New(config.Cfg.Telegram.RpcRetry),
floodwait.NewSimpleWaiter(), floodwait.NewSimpleWaiter(),
} }
} }