From e4ecaf89cf5ee6233094f738c4978020fe63e237 Mon Sep 17 00:00:00 2001
From: Marcus Baldassarre <baldassarremarcus@gmail.com>
Date: Tue, 16 Apr 2024 04:37:18 -0400
Subject: [PATCH] rpc: implement Unwrap() for wsHandshakeError (#29522)

---
 rpc/client.go    | 2 +-
 rpc/websocket.go | 4 ++++
 2 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/rpc/client.go b/rpc/client.go
index eef6ee21cf..05b87ae96c 100644
--- a/rpc/client.go
+++ b/rpc/client.go
@@ -431,7 +431,7 @@ func (c *Client) BatchCallContext(ctx context.Context, b []BatchElem) error {
 	}
 
 	// Wait for all responses to come back.
-	for n := 0; n < len(batchresp) && err == nil; n++ {
+	for n := 0; n < len(batchresp); n++ {
 		resp := batchresp[n]
 		if resp == nil {
 			// Ignore null responses. These can happen for batches sent via HTTP.
diff --git a/rpc/websocket.go b/rpc/websocket.go
index 538e53a31b..9f67caf859 100644
--- a/rpc/websocket.go
+++ b/rpc/websocket.go
@@ -122,6 +122,10 @@ func (e wsHandshakeError) Error() string {
 	return s
 }
 
+func (e wsHandshakeError) Unwrap() error {
+	return e.err
+}
+
 func originIsAllowed(allowedOrigins mapset.Set[string], browserOrigin string) bool {
 	it := allowedOrigins.Iterator()
 	for origin := range it.C {