amirho/nest
1
0
Fork 0
mirror of https://github.com/nestjs/nest.git synced 2026-02-21 23:11:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

refactor: extract handle message from WsAdapter

Browse Source
This commit is contained in:
Kamil Myśliwiec
2018-10-17 21:42:24 +02:00
parent 3e9bfff04e
commit a6b001638f
1 changed files with 13 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
packages/websockets/adapters/ws-adapter.ts
View File

@@ -15,10 +15,10 @@ import { MessageMappingProperties } from '../gateway-metadata-explorer';
let wsPackage: any = {}; let wsPackage: any = {};
enum READY_STATE { enum READY_STATE {
CONNECTING_STATE = 0, CONNECTING_STATE = 0,
OPEN_STATE = 1, OPEN_STATE = 1,
CLOSING_STATE = 2, CLOSING_STATE = 2,
CLOSED_STATE = 3, CLOSED_STATE = 3,
} }
export class WsAdapter implements WebSocketAdapter { export class WsAdapter implements WebSocketAdapter {
@@ -79,11 +79,13 @@ export class WsAdapter implements WebSocketAdapter {
), ),
takeUntil(close$), takeUntil(close$),
); );
source$.subscribe(response => { const handleMessage = response => {
if (client.readyState === READY_STATE.OPEN_STATE) { if (client.readyState !== READY_STATE.OPEN_STATE) {
client.send(JSON.stringify(response)); return;
} }
}); client.send(JSON.stringify(response));
};
source$.subscribe(handleMessage);
} }
public bindMessageHandler( public bindMessageHandler(
@@ -108,7 +110,9 @@ export class WsAdapter implements WebSocketAdapter {
} }
public bindErrorHandler(server) { public bindErrorHandler(server) {
server.on(CONNECTION_EVENT, ws => ws.on(ERROR_EVENT, err => this.logger.error(err))); server.on(CONNECTION_EVENT, ws =>
ws.on(ERROR_EVENT, err => this.logger.error(err)),
);
server.on(ERROR_EVENT, err => this.logger.error(err)); server.on(ERROR_EVENT, err => this.logger.error(err));
return server; return server;
} }