From adad86e0d60860bd14706d2afe92c0a145424949 Mon Sep 17 00:00:00 2001 From: EclesioMeloJunior Date: Thu, 26 May 2022 15:09:56 -0400 Subject: [PATCH] chore: smooth diffs --- dot/rpc/subscription/websocket.go | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/dot/rpc/subscription/websocket.go b/dot/rpc/subscription/websocket.go index cefa30cdce..688869e312 100644 --- a/dot/rpc/subscription/websocket.go +++ b/dot/rpc/subscription/websocket.go @@ -68,8 +68,6 @@ func (c *WSConn) readWebsocketMessage() (rawBytes []byte, wsMessage *websocketMe // HandleConn handles messages received on websocket connections func (c *WSConn) HandleConn() { - defer c.Wsconn.Close() - for { rawBytes, wsMessage, err := c.readWebsocketMessage() if err != nil { @@ -83,12 +81,14 @@ func (c *WSConn) HandleConn() { } logger.Tracef("websocket message received: %s", string(rawBytes)) + if wsMessage.Method == "" { c.safeSendError(0, big.NewInt(InvalidRequestCode), InvalidRequestMessage) continue } logger.Debugf("ws method %s called with params %v", wsMessage.Method, wsMessage.Params) + if !strings.Contains(wsMessage.Method, "_unsubscribe") && !strings.Contains(wsMessage.Method, "_unwatch") { setupListener := c.getSetupListener(wsMessage.Method)