ReVanced/revanced-bots
1
0
Fork 0
mirror of https://github.com/ReVanced/revanced-bots.git synced 2026-01-11 13:56:15 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(bots/discord): hanging process when disconnecting from API too many times

Browse Source
This commit is contained in:
PalmDevs
2024-07-30 21:30:35 +07:00
parent 2efedc47df
commit d31616ebcb
6 changed files with 12 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
packages/api/src/classes/ClientWebSocket.ts
View File

@@ -49,19 +49,13 @@ export class ClientWebSocketManager {
const timeout = setTimeout(() => {
if (!this.ready) {
this.#socket?.close(DisconnectReason.TooSlow)
throw new Error('WebSocket connection was not readied in time')
this._handleDisconnect(DisconnectReason.TooSlow, 'WebSocket connection was not readied in time')
}
}, this.timeout)
const errorBeforeReadyHandler = (err: Error) => {
cleanup()
throw err
}
const closeBeforeReadyHandler = (code: number, reason: Buffer) => {
clearTimeout(timeout)
this._handleDisconnect(code, reason.toString())
throw new Error('WebSocket connection closed before ready')
cleanup()
}
const readyHandler = () => {
@@ -71,15 +65,14 @@ export class ClientWebSocketManager {
rs()
}
const socket = this.#socket
const cleanup = () => {
this.#socket.off('open', readyHandler)
this.#socket.off('close', closeBeforeReadyHandler)
this.#socket.off('error', errorBeforeReadyHandler)
socket.off('open', readyHandler)
socket.off('close', closeBeforeReadyHandler)
clearTimeout(timeout)
}
this.#socket.on('open', readyHandler)
this.#socket.on('error', errorBeforeReadyHandler)
this.#socket.on('close', closeBeforeReadyHandler)
} catch (e) {
rj(e)