Fix WebSockets counter

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