mirror/api
1
0
Fork 0
mirror of https://github.com/unraid/api.git synced 2026-04-30 12:10:15 -05:00
Code Issues Packages Projects Releases Wiki Activity

fix: disconnect relay correctly when mothership kicks the connection

Browse Source
This commit is contained in:
Alexis Tyler
2021-02-12 10:41:12 +10:30
parent 7d8d3fc652
commit f6a4b77380
1 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
app/mothership/sockets/mothership.ts
+4 -3
View File
@@ -45,13 +45,14 @@ export class MothershipSocket extends CustomSocket {
}
onDisconnect() {
return async (code: number, _message: string) => {
const onDisconnect = super.onDisconnect().bind(this as unknown as WebSocketWithHeartBeat);
return async (code: number, message: string) => {
try {
// Close connection to motherships's server's endpoint
await this.disconnectFromMothershipsGraphql();
// Process disconnection
this.onDisconnect();
await onDisconnect(code, message);
} catch (error: unknown) {
if (isNodeError(error, AppError)) {
this.logger.debug('Connection closed with code=%s reason="%s"', code, error.message);
@@ -66,7 +67,7 @@ export class MothershipSocket extends CustomSocket {
return async (data: string) => {
try {
this.logger.silly('Recieved message from mothership\'s relay, forwarding to the internal graphql connection');
await sendMessage(sockets.get('internalGraphql')?.connection, data);
await sendMessage('internalGraphql', data);
this.logger.silly('Message sent to the internal graphql connection successfully.');
} catch (error: unknown) {
if (isNodeError(error, AppError)) {