mirror of
https://github.com/nestjs/nest.git
synced 2026-02-21 23:11:44 +00:00
refactor: extract handle message from WsAdapter
This commit is contained in:
@@ -15,10 +15,10 @@ import { MessageMappingProperties } from '../gateway-metadata-explorer';
|
||||
let wsPackage: any = {};
|
||||
|
||||
enum READY_STATE {
|
||||
CONNECTING_STATE = 0,
|
||||
OPEN_STATE = 1,
|
||||
CLOSING_STATE = 2,
|
||||
CLOSED_STATE = 3,
|
||||
CONNECTING_STATE = 0,
|
||||
OPEN_STATE = 1,
|
||||
CLOSING_STATE = 2,
|
||||
CLOSED_STATE = 3,
|
||||
}
|
||||
|
||||
export class WsAdapter implements WebSocketAdapter {
|
||||
@@ -79,11 +79,13 @@ export class WsAdapter implements WebSocketAdapter {
|
||||
),
|
||||
takeUntil(close$),
|
||||
);
|
||||
source$.subscribe(response => {
|
||||
if (client.readyState === READY_STATE.OPEN_STATE) {
|
||||
client.send(JSON.stringify(response));
|
||||
const handleMessage = response => {
|
||||
if (client.readyState !== READY_STATE.OPEN_STATE) {
|
||||
return;
|
||||
}
|
||||
});
|
||||
client.send(JSON.stringify(response));
|
||||
};
|
||||
source$.subscribe(handleMessage);
|
||||
}
|
||||
|
||||
public bindMessageHandler(
|
||||
@@ -108,7 +110,9 @@ export class WsAdapter implements WebSocketAdapter {
|
||||
}
|
||||
|
||||
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));
|
||||
return server;
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user