refactor: rename admin socket session map

This commit is contained in:
DecDuck
2024-11-16 16:27:38 +11:00
parent 9011cf5c83
commit 44c60280ef

View File

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