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

refactor: rename admin socket session map

Browse Source
This commit is contained in:
DecDuck
2024-11-16 16:27:38 +11:00
parent 9011cf5c83
commit 44c60280ef
1 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -5,7 +5,7 @@ import taskHandler, { TaskMessage } from "~/server/internal/tasks";
// TODO add web socket sessions for horizontal scaling
// ID to admin
const socketSessions: { [key: string]: boolean } = {};
const adminSocketSessions: { [key: string]: boolean } = {};
export default defineWebSocketHandler({
open(peer) {
@ -22,7 +22,7 @@ export default defineWebSocketHandler({
return;
}
const admin = session.getAdminUser(dummyEvent);
socketSessions[peer.id] = admin !== undefined;
adminSocketSessions[peer.id] = admin !== undefined;
const rtMsg: TaskMessage = {
id: "connect",
@ -36,17 +36,17 @@ export default defineWebSocketHandler({
},
message(peer, message) {
if (!peer.id) return;
if (socketSessions[peer.id] === undefined) return;
if (adminSocketSessions[peer.id] === undefined) return;
const text = message.text();
if (text.startsWith("connect/")) {
const id = text.substring("connect/".length);
taskHandler.connect(peer.id, id, peer, socketSessions[peer.id]);
taskHandler.connect(peer.id, id, peer, adminSocketSessions[peer.id]);
return;
}
},
close(peer, details) {
if (!peer.id) return;
if (socketSessions[peer.id] === undefined) return;
delete socketSessions[peer.id];
if (adminSocketSessions[peer.id] === undefined) return;
delete adminSocketSessions[peer.id];
},
});