Ensure WebSocket server is closed

coocos · Apr 4, 2021 · c6daaaf · c6daaaf
1 parent da8d542
commit c6daaaf
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/websocket/server.ts b/src/websocket/server.ts
@@ -17,6 +17,9 @@ function createWebSocketServer(httpServer: http.Server): ws.Server {
       socket.destroy();
     }
   });
+  httpServer.on("close", () => {
+    webSocketServer.close();
+  });
   return webSocketServer;
 }