added more debug logs

Gioppoo · Nov 18, 2020 · 5a21265 · 5a21265
1 parent 6ca7856
commit 5a21265
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/rpc/ws/client.go b/rpc/ws/client.go
@@ -89,12 +89,14 @@ func (c *Client) handleMessage(message []byte) {
 func (c *Client) handleNewSubscriptionMessage(requestID, subID uint64) {
 	c.lock.Lock()
 	defer c.lock.Unlock()
+
 	if traceEnabled {
 		zlog.Debug("received new subscription message",
 			zap.Uint64("message_id", requestID),
 			zap.Uint64("subscription_id", subID),
 		)
 	}
+
 	callBack, found := c.subscriptionByRequestID[requestID]
 	if !found {
 		zlog.Error("cannot find websocket message handler for a new stream.... this should not happen",
@@ -105,6 +107,12 @@ func (c *Client) handleNewSubscriptionMessage(requestID, subID uint64) {
 	}
 	callBack.subID = subID
 	c.subscriptionByWSSubID[subID] = callBack
+
+	zlog.Debug("registered ws subscription",
+		zap.Uint64("subscription_id", subID),
+		zap.Uint64("request_id", requestID),
+		zap.Int("subscription_count", len(c.subscriptionByWSSubID)),
+	)
 	return
 }