From 246f1de3e74291fb916ee8cade1ac64cea438144 Mon Sep 17 00:00:00 2001
From: totanvix <totanvix@gmail.com>
Date: Wed, 28 Jun 2023 13:08:29 +0700
Subject: [PATCH] feat: add check timeout

---
 utils/bot/bot.go | 14 ++++++++++----
 1 file changed, 10 insertions(+), 4 deletions(-)

diff --git a/utils/bot/bot.go b/utils/bot/bot.go
index 422d5eb..75a2214 100644
--- 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() {