Revert socket counter

Novage · Jun 23, 2024 · f970f41 · f970f41
1 parent 8d9fa21
commit f970f41
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/lib/uws-tracker.ts b/lib/uws-tracker.ts
@@ -184,6 +184,7 @@ export class UWebSocketsTracker {
       compression: this.settings.websockets.compression,
       maxPayloadLength: this.settings.websockets.maxPayloadLength,
       idleTimeout: this.settings.websockets.idleTimeout,
+      open: this.onOpen,
       upgrade: this.onUpgrade,
       drain: (ws: WebSocket<SocketContext>) => {
         if (debugWebSocketsEnabled) {
@@ -195,6 +196,10 @@ export class UWebSocketsTracker {
     });
   }
 
+  private readonly onOpen = (): void => {
+    this.webSocketsCount++;
+  };
+
   private readonly onUpgrade = (
     response: HttpResponse,
     request: HttpRequest,
@@ -271,8 +276,6 @@ export class UWebSocketsTracker {
       );
     }
 
-    this.webSocketsCount++;
-
     response.upgrade<Omit<SocketContext, "ws">>(
       {
         sendMessage,