iridium/neko
2
2
Fork 0
mirror of https://github.com/m1k1o/neko.git synced 2024-07-24 14:40:50 +12:00
Code Issues Packages Projects Releases Wiki Activity

disconnect socket on system disconnect event.

Browse Source
This commit is contained in:
Miroslav Šedivý 2021-03-25 11:31:49 +01:00
parent 80d229041a
commit 86d9e769f2
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/component/internal/messages.ts
View File

@ -40,6 +40,7 @@ export interface NekoEvents {
}
export class NekoMessages extends EventEmitter<NekoEvents> {
private websocket: NekoWebSocket
private state: NekoState
private _log: Logger
@ -48,7 +49,9 @@ export class NekoMessages extends EventEmitter<NekoEvents> {
this._log = new Logger('messages')
this.state = state
websocket.on('message', async (event: string, payload: any) => {
this.websocket = websocket
this.websocket.on('message', async (event: string, payload: any) => {
// @ts-ignore
if (typeof this[event] === 'function') {
// @ts-ignore
@ -95,6 +98,7 @@ export class NekoMessages extends EventEmitter<NekoEvents> {
protected [EVENT.SYSTEM_DISCONNECT]({ message }: message.SystemDisconnect) {
this._log.debug('EVENT.SYSTEM_DISCONNECT')
this.websocket.disconnect(new Error(message))
this.emit('connection.disconnect', message)
}