Merge branch 'fix-ws-client-closing' of https://github.com/spaiter/nest…

… into spaiter-fix-ws-client-closing
gujieye · Oct 17, 2018 · 3e9bfff · 3e9bfff
2 parents 6dda6c5 + 72a8725
commit 3e9bfff
Showing 1 changed file with 12 additions and 1 deletion.
diff --git a/packages/websockets/adapters/ws-adapter.ts b/packages/websockets/adapters/ws-adapter.ts
@@ -14,6 +14,13 @@ import { MessageMappingProperties } from '../gateway-metadata-explorer';
 
 let wsPackage: any = {};
 
+enum READY_STATE {
+    CONNECTING_STATE = 0,
+    OPEN_STATE = 1,
+    CLOSING_STATE = 2,
+    CLOSED_STATE = 3,
+}
+
 export class WsAdapter implements WebSocketAdapter {
   protected readonly logger = new Logger(WsAdapter.name);
   protected readonly httpServer: Server;
@@ -72,7 +79,11 @@ export class WsAdapter implements WebSocketAdapter {
       ),
       takeUntil(close$),
     );
-    source$.subscribe(response => client.send(JSON.stringify(response)));
+    source$.subscribe(response => {
+      if (client.readyState === READY_STATE.OPEN_STATE) {
+          client.send(JSON.stringify(response));
+      }
+    });
   }
 
   public bindMessageHandler(