feat(pkg): add default timeout for GetStorageRoot (#347)

taikoxyz · Aug 3, 2023 · 9a4dee0 · 9a4dee0
1 parent c6454af
commit 9a4dee0
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 2 deletions.
diff --git a/pkg/rpc/methods.go b/pkg/rpc/methods.go
@@ -335,8 +335,17 @@ func (c *Client) GetStorageRoot(
 	contract common.Address,
 	height *big.Int,
 ) (common.Hash, error) {
+	var (
+		ctxWithTimeout = ctx
+		cancel         context.CancelFunc
+	)
+	if _, ok := ctx.Deadline(); !ok {
+		ctxWithTimeout, cancel = context.WithTimeout(ctx, defaultTimeout)
+		defer cancel()
+	}
+
 	proof, err := gethclient.GetProof(
-		ctx,
+		ctxWithTimeout,
 		contract,
 		[]string{"0x0000000000000000000000000000000000000000000000000000000000000000"},
 		height,

diff --git a/version/version.go b/version/version.go
@@ -2,7 +2,7 @@ package version
 
 // Version info.
 var (
-	Version = "0.13.0"
+	Version = "0.14.0"
 	Meta    = "dev"
 )