From 83dd6b6d3290f26de2db17c267bb7c6a1de6f708 Mon Sep 17 00:00:00 2001
From: SALTWOOD <105980161+SALTWOOD@users.noreply.github.com>
Date: Sun, 15 Sep 2024 15:51:07 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20=E6=94=B6=E5=88=B0=E5=81=9C=E6=AD=A2?=
 =?UTF-8?q?=E4=BF=A1=E5=8F=B7=E6=97=B6=E5=85=88=E6=96=AD=E5=BC=80=E8=BF=9E?=
 =?UTF-8?q?=E6=8E=A5?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/Server.ts | 2 +-
 src/index.ts  | 4 ++++
 2 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/src/Server.ts b/src/Server.ts
index 099202c..26afe0d 100644
--- a/src/Server.ts
+++ b/src/Server.ts
@@ -39,7 +39,7 @@ const logAccess = (req: Request, res: Response) => {
 
 export class Server {
     private app;
-    private io: SocketIOServer;
+    public io: SocketIOServer;
     private httpsServer: https.Server;
     public db: SQLiteHelper;
     protected files: File[];
diff --git a/src/index.ts b/src/index.ts
index b1bfc93..e0848f9 100644
--- a/src/index.ts
+++ b/src/index.ts
@@ -6,6 +6,10 @@ import got from "got";
 
 function onStop(signal: string) {
     server.db.close();
+    server.io.sockets.sockets.forEach(socket => {
+        socket.disconnect(true);
+    });
+    server.io.close();
     console.log(`Received ${signal}. Shutting down...`);
     process.exit(0);
 }