feat: add check timeout

totanvix · Jun 28, 2023 · 246f1de · 246f1de
1 parent 8f61fe4
commit 246f1de
Showing 1 changed file with 10 additions and 4 deletions.
diff --git a/utils/bot/bot.go b/utils/bot/bot.go
@@ -1,9 +1,11 @@
 package bot
 
 import (
+	"errors"
 	"fmt"
 	"log"
 	"strings"
+	"time"
 
 	"zeril-bot/utils/bitcoin"
 	"zeril-bot/utils/kqxs"
@@ -42,13 +44,17 @@ func (b Bot) ResolveHook() error {
 		go b.resolveCallbackCommand()
 	}
 
-	for r := range b.rCh {
-		if r.err != nil {
+	for {
+		select {
+		case r, ok := <-b.rCh:
+			if !ok {
+				return nil
+			}
 			return r.err
+		case <-time.After(10 * time.Second):
+			return errors.New("Timeout")
 		}
 	}
-
-	return nil
 }
 
 func (b Bot) setTypingAction() {