@nevuamarkets/poly-websockets 0.2.1 → 0.3.0

package/README.md

@@ -90,6 +90,12 @@ Clears all subscriptions and state:
 - Closes all WebSocket connections
 - Clears the internal order book cache
 
+##### `getStatistics(): { openWebSockets: number; subscribedAssetIds: number }`
+
+Returns statistics about the current state of the subscription manager:
+- `openWebSockets`: The number of websockets that are currently in OPEN state
+- `subscribedAssetIds`: The number of unique asset IDs that are currently subscribed
+
 ### WebSocketHandlers
 
 Interface defining event handlers for different WebSocket events.

package/dist/WSSubscriptionManager.d.ts

@@ -13,6 +13,10 @@ declare class WSSubscriptionManager {
     addSubscriptions(assetIdsToAdd: string[]): Promise<void>;
     removeSubscriptions(assetIdsToRemove: string[]): Promise<void>;
     private reconnectAndCleanupGroups;
+    getStatistics(): {
+        openWebSockets: number;
+        subscribedAssetIds: number;
+    };
     private createWebSocketClient;
 }
 export { WSSubscriptionManager, WebSocketHandlers };

package/dist/WSSubscriptionManager.js

@@ -171,6 +171,16 @@ class WSSubscriptionManager {
             await ((_b = (_a = this.handlers).onError) === null || _b === void 0 ? void 0 : _b.call(_a, err));
         }
     }
+    /*
+        Returns statistics about the current state of the subscription manager.
+
+        Returns an object with:
+        - openWebSockets: The number of websockets that are currently in OPEN state
+        - subscribedAssetIds: The number of unique asset IDs that are currently subscribed
+    */
+    getStatistics() {
+        return this.groupRegistry.getStatistics();
+    }
     async createWebSocketClient(groupId, handlers) {
         var _a, _b;
         const group = this.groupRegistry.findGroupById(groupId);

package/dist/modules/GroupRegistry.d.ts

@@ -36,6 +36,17 @@ export declare class GroupRegistry {
      * Returns the group if found, otherwise undefined.
      */
     findGroupById(groupId: string): WebSocketGroup | undefined;
+    /**
+     * Get statistics about the current state of the registry.
+     *
+     * Returns an object with:
+     * - openWebSockets: The number of websockets that are currently in OPEN state
+     * - subscribedAssetIds: The number of unique asset IDs that are currently subscribed
+     */
+    getStatistics(): {
+        openWebSockets: number;
+        subscribedAssetIds: number;
+    };
     /**
      * Atomically remove **all** groups from the registry and return them so the
      * caller can perform any asynchronous cleanup (closing sockets, etc.)

package/dist/modules/GroupRegistry.js

@@ -7,6 +7,7 @@ exports.GroupRegistry = void 0;
 const async_mutex_1 = require("async-mutex");
 const lodash_1 = __importDefault(require("lodash"));
 const uuid_1 = require("uuid");
+const ws_1 = __importDefault(require("ws"));
 const WebSocketSubscriptions_1 = require("../types/WebSocketSubscriptions");
 const logger_1 = require("../logger");
 /*
@@ -85,6 +86,31 @@ class GroupRegistry {
     findGroupById(groupId) {
         return wsGroups.find(g => g.groupId === groupId);
     }
+    /**
+     * Get statistics about the current state of the registry.
+     *
+     * Returns an object with:
+     * - openWebSockets: The number of websockets that are currently in OPEN state
+     * - subscribedAssetIds: The number of unique asset IDs that are currently subscribed
+     */
+    getStatistics() {
+        let openWebSockets = 0;
+        const uniqueAssetIds = new Set();
+        for (const group of wsGroups) {
+            // Count open websockets
+            if (group.wsClient && group.wsClient.readyState === ws_1.default.OPEN) {
+                openWebSockets++;
+            }
+            // Collect unique asset IDs
+            for (const assetId of group.assetIds) {
+                uniqueAssetIds.add(assetId);
+            }
+        }
+        return {
+            openWebSockets,
+            subscribedAssetIds: uniqueAssetIds.size,
+        };
+    }
     /**
      * Atomically remove **all** groups from the registry and return them so the
      * caller can perform any asynchronous cleanup (closing sockets, etc.)

package/dist/modules/GroupSocket.js

@@ -116,8 +116,9 @@ class GroupSocket {
             var _a, _b, _c;
             try {
                 const messageStr = data.toString();
+                const normalizedMessageStr = messageStr.trim().toUpperCase();
                 // Handle PONG messages that might be sent to message handler during handler reattachment
-                if (messageStr === 'PONG') {
+                if (normalizedMessageStr === 'PONG') {
                     return;
                 }
                 let events = [];

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@nevuamarkets/poly-websockets",
-  "version": "0.2.1",
+  "version": "0.3.0",
   "description": "Plug-and-play Polymarket WebSocket price alerts",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",

package/src/WSSubscriptionManager.ts

@@ -199,6 +199,20 @@ class WSSubscriptionManager {
         }
     }
 
+    /*
+        Returns statistics about the current state of the subscription manager.
+
+        Returns an object with:
+        - openWebSockets: The number of websockets that are currently in OPEN state
+        - subscribedAssetIds: The number of unique asset IDs that are currently subscribed
+    */
+    public getStatistics(): {
+        openWebSockets: number;
+        subscribedAssetIds: number;
+    } {
+        return this.groupRegistry.getStatistics();
+    }
+
     private async createWebSocketClient(groupId: string, handlers: WebSocketHandlers) {
         const group = this.groupRegistry.findGroupById(groupId);
 

package/src/modules/GroupRegistry.ts

@@ -1,6 +1,7 @@
 import { Mutex } from 'async-mutex';
 import _ from 'lodash';
 import { v4 as uuidv4 } from 'uuid';
+import WebSocket from 'ws';
 import { WebSocketGroup, WebSocketStatus } from '../types/WebSocketSubscriptions';
 import { OrderBookCache } from './OrderBookCache';
 import { logger } from '../logger';
@@ -81,6 +82,38 @@ export class GroupRegistry {
         return wsGroups.find(g => g.groupId === groupId);
     }
 
+    /**
+     * Get statistics about the current state of the registry.
+     * 
+     * Returns an object with:
+     * - openWebSockets: The number of websockets that are currently in OPEN state
+     * - subscribedAssetIds: The number of unique asset IDs that are currently subscribed
+     */
+    public getStatistics(): {
+        openWebSockets: number;
+        subscribedAssetIds: number;
+    } {
+        let openWebSockets = 0;
+        const uniqueAssetIds = new Set<string>();
+
+        for (const group of wsGroups) {
+            // Count open websockets
+            if (group.wsClient && group.wsClient.readyState === WebSocket.OPEN) {
+                openWebSockets++;
+            }
+
+            // Collect unique asset IDs
+            for (const assetId of group.assetIds) {
+                uniqueAssetIds.add(assetId);
+            }
+        }
+
+        return {
+            openWebSockets,
+            subscribedAssetIds: uniqueAssetIds.size,
+        };
+    }
+
     /**
      * Atomically remove **all** groups from the registry and return them so the
      * caller can perform any asynchronous cleanup (closing sockets, etc.)

package/src/modules/GroupSocket.ts

@@ -139,9 +139,10 @@ export class GroupSocket {
         const handleMessage = async (data: Buffer) => {
             try {
                 const messageStr = data.toString();
+                const normalizedMessageStr = messageStr.trim().toUpperCase();
 
                 // Handle PONG messages that might be sent to message handler during handler reattachment
-                if (messageStr === 'PONG') {
+                if (normalizedMessageStr === 'PONG') {
                     return;
                 }