Gateway: send reconnect on shutdown

1 files changed, 11 insertions, 0 deletions

diff --git a/src/gateway/events/Connection.ts b/src/gateway/events/Connection.ts

index 5876c3e1..6f67c0c5 100644
--- a/src/gateway/events/Connection.ts
+++ b/src/gateway/events/Connection.ts
@@ -43,6 +43,17 @@ export async function Connection(this: WS.Server, socket: WebSocket, request: In
         if (index !== -1) openConnections.splice(index, 1);
     });
 
+    for (const sig of ["SIGINT", "SIGTERM", "SIGQUIT"] as const) {
+        process.on(sig, async () => {
+            await Send(socket, {
+                op: OPCODES.Reconnect,
+                s: socket.sequence++,
+                d: Math.round(Math.random() * 5000),
+            });
+            socket.close(1000);
+        });
+    }
+
     const forwardedFor = Config.get().security.forwardedFor;
     const ipAddress = forwardedFor ? (request.headers[forwardedFor.toLowerCase()] as string) : request.socket.remoteAddress;