Ryan/drop
1
0
Fork 0
mirror of https://github.com/Drop-OSS/drop.git synced 2025-11-15 01:01:20 +10:00
Code Issues Packages Projects Releases Wiki Activity

fix: properly disconnect websockets from task handler

Browse Source
This commit is contained in:
DecDuck
2024-11-16 17:27:27 +11:00
parent bc0c47c487
commit 5358f1f52c
2 changed files with 10 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
server/api/v1/task/index.get.ts
View File

@ -48,5 +48,7 @@ export default defineWebSocketHandler({
if (!peer.id) return; if (!peer.id) return;
if (adminSocketSessions[peer.id] === undefined) return; if (adminSocketSessions[peer.id] === undefined) return;
delete adminSocketSessions[peer.id]; delete adminSocketSessions[peer.id];
taskHandler.disconnectAll(peer.id);
}, },
}); });

8
server/internal/tasks/index.ts
View File

@ -111,6 +111,14 @@ class TaskHandler {
return true; return true;
} }
disconnectAll(id: string) {
for (const [taskId] of Object.keys(this.taskRegistry)) {
delete this.taskRegistry[taskId].clients[id];
}
delete this.clientRegistry[id];
}
disconnect(id: string, taskId: string) { disconnect(id: string, taskId: string) {
if (!this.taskRegistry[taskId]) return false; if (!this.taskRegistry[taskId]) return false;