chore: add all bots in stream

tgdrive · Aug 10, 2024 · a64eb57 · a64eb57
1 parent c7be0ea
commit a64eb57
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/internal/tgc/workers.go b/internal/tgc/workers.go
@@ -28,6 +28,9 @@ func NewBotWorker() *BotWorker {
 func (w *BotWorker) Set(bots []string, channelID int64) {
 	w.mu.Lock()
 	defer w.mu.Unlock()
+	if _, ok := w.bots[channelID]; ok {
+		return
+	}
 	w.bots[channelID] = bots
 	w.currIdx[channelID] = 0
 }

diff --git a/pkg/services/file.go b/pkg/services/file.go
@@ -750,7 +750,7 @@ func (fs *FileService) GetFileStream(c *gin.Context, download bool) {
 		multiThreads = 0
 
 	} else if fs.cnf.TG.DisableBgBots && len(tokens) > 0 {
-		fs.botWorker.Set(tokens[0:min(len(tokens), fs.cnf.TG.Stream.BotsLimit)], file.ChannelID)
+		fs.botWorker.Set(tokens, file.ChannelID)
 		token, _ = fs.botWorker.Next(file.ChannelID)
 		client, err = tgc.BotClient(c, fs.kv, &fs.cnf.TG, token)
 		if err != nil {