feat: 收到停止信号时先断开连接

SaltWood-Studio · Sep 15, 2024 · 83dd6b6 · 83dd6b6
1 parent ee0ef79
commit 83dd6b6
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git 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
@@ -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);
 }