Merge pull request #13531 from bettercalljason/fix-ws-shutdown

fix(ws): close existing connections
nestjs · Jun 3, 2024 · a7c5cae · a7c5cae
2 parents 75fa678 + 6893058
commit a7c5cae
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/packages/platform-ws/adapters/ws-adapter.ts b/packages/platform-ws/adapters/ws-adapter.ts
@@ -158,6 +158,16 @@ export class WsAdapter extends AbstractWsAdapter {
     client.on(CLOSE_EVENT, callback);
   }
 
+  public async close(server: any) {
+    const closeEventSignal = new Promise((resolve, reject) =>
+      server.close(err => (err ? reject(err) : resolve(undefined))),
+    );
+    for (const ws of server.clients) {
+      ws.terminate();
+    }
+    await closeEventSignal;
+  }
+
   public async dispose() {
     const closeEventSignals = Array.from(this.httpServersRegistry)
       .filter(([port]) => port !== UNDERLYING_HTTP_SERVER_PORT)