Add custom timeout in transaction result sdk

flashbots · Jan 11, 2024 · b494114 · b494114
1 parent 3c93c40
commit b494114
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/suave/sdk/sdk.go b/suave/sdk/sdk.go
@@ -104,11 +104,15 @@ type TransactionResult struct {
 }
 
 func (t *TransactionResult) Wait() (*types.Receipt, error) {
+	return t.WaitWithTimeout(10 * time.Second)
+}
+
+func (t *TransactionResult) WaitWithTimeout(timeout time.Duration) (*types.Receipt, error) {
 	if t.receipt != nil {
 		return t.receipt, nil
 	}
 
-	timer := time.NewTimer(10 * time.Second)
+	timer := time.NewTimer(timeout)
 
 	var receipt *types.Receipt
 	var err error