zotan/megalodon
1
0
Fork 0
Code Issues Pull requests Packages Projects Releases Wiki Activity

Merge pull request #22 from h3poteto/iss-20

Browse source 
closes #20 Guard reconnection if the close method is already called
This commit is contained in:
AkiraFukushima 2018-12-10 23:56:25 +09:00 committed by GitHub
parent 0a80a51542 eeb45eedd0
commit 9414e56385
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 8 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
src/web_socket.ts
View file

@ -17,6 +17,7 @@ export default class WebSocket extends EventEmitter {
private _reconnectInterval: number
private _reconnectMaxAttempts: number
private _reconnectCurrentAttempts: number
private _connectionClosed: boolean
/**
* @param url Full url of websocket: e.g. https://pleroma.io/api/v1/streaming
@ -33,12 +34,14 @@ export default class WebSocket extends EventEmitter {
this._reconnectInterval = 1000
this._reconnectMaxAttempts = Infinity
this._reconnectCurrentAttempts = 0
this._connectionClosed = false
}
/**
* Start websocket connection.
*/
public start() {
this._connectionClosed = false
this._resetRetryParams()
this._startWebSocketConnection()
}
@ -58,6 +61,7 @@ export default class WebSocket extends EventEmitter {
*/
public stop() {
if (this._socketConnection) {
this._connectionClosed = true
this._socketConnection.close()
}
this._resetRetryParams()
@ -133,7 +137,10 @@ export default class WebSocket extends EventEmitter {
this.emit('close', {})
} else {
console.log(`Closed connection with ${code}`)
this._reconnect(cli)
// If already called close method, it does not retry.
if (!this._connectionClosed) {
this._reconnect(cli)
}
}
})
conn.on('message', (message: IMessage) => {