memory leak in ws client (#228)

gagliardetto · Oct 10, 2024 · 05e94f2 · 05e94f2
1 parent d337fef
commit 05e94f2
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/rpc/ws/client.go b/rpc/ws/client.go
@@ -341,6 +341,7 @@ func (c *Client) subscribe(
 	c.conn.SetWriteDeadline(time.Now().Add(writeWait))
 	err = c.conn.WriteMessage(websocket.TextMessage, data)
 	if err != nil {
+		delete(c.subscriptionByRequestID, req.ID)
 		return nil, fmt.Errorf("unable to write request: %w", err)
 	}