add 10s timeout for batch rpc call

CortexFoundation · Jul 21, 2024 · fe99aa1 · fe99aa1
1 parent f089f48
commit fe99aa1
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/rpc.go b/rpc.go
@@ -17,7 +17,7 @@
 package robot
 
 import (
-	//"context"
+	"context"
 	"errors"
 	"fmt"
 	"strconv"
@@ -90,7 +90,9 @@ func (m *Monitor) rpcBatchBlockByNumber(from, to uint64) ([]*types.Block, error)
 		}
 	}
 
-	if err := m.cl.BatchCall(reqs); err != nil {
+	ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
+	defer cancel()
+	if err := m.cl.BatchCallContext(ctx, reqs); err != nil {
 		return nil, err
 	}