chore: update socket joinRoom

fastrodev · Oct 31, 2024 · 793d70b · 793d70b
1 parent 6a5b07e
commit 793d70b
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/modules/socket/mod.ts b/modules/socket/mod.ts
@@ -41,12 +41,13 @@ export default function socketModule(s: Fastro) {
     }
 
     function joinRoom(ctx: Context, socket: WebSocket, room: string) {
-        if (!connections.has(room)) {
+        const s = connections.get(room);
+        if (s) {
+            s.add(socket);
+        } else {
             const s = new Set<WebSocket>();
             s.add(socket);
             connections.set(room, s);
-        } else {
-            connections.get(room)?.add(socket);
         }
     }