fix: handle websocket closed error in servicer client listener

pokt-network · Sep 19, 2023 · 4ce5795 · 4ce5795
1 parent 9705345
commit 4ce5795
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/relayer/client/client.go b/relayer/client/client.go
@@ -137,6 +137,10 @@ func (client *servicerClient) listen(ctx context.Context, newBlocks chan types.B
 
 		_, msg, err := client.wsClient.ReadMessage()
 		if err != nil {
+			if websocket.IsUnexpectedCloseError(err) {
+				// NB: stop this goroutine if the websocket connection is closed
+				return
+			}
 			// TODO: handle error
 			continue
 		}