shmup/colabbd
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Fix client identity tracking to prevent memory leak

Browse source
This commit is contained in:
Jared Miller 2026-01-27 16:16:14 -05:00
parent 3cf16586aa
commit 925c7a3c0d
Signed by: shmup
GPG key ID: 22B5C6D66A38B06C
1 changed files with 10 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
src/index.ts
View file

@ -8,7 +8,7 @@ Bun.serve({
fetch(req, server) { fetch(req, server) {
const url = new URL(req.url); const url = new URL(req.url);
if (url.pathname === "/ws") { if (url.pathname === "/ws") {
const upgraded = server.upgrade(req, { data: { room: null } }); const upgraded = server.upgrade(req, { data: { room: null, client: null } });
if (!upgraded) { if (!upgraded) {
return new Response("websocket upgrade failed", { status: 400 }); return new Response("websocket upgrade failed", { status: 400 });
} }
@ -17,14 +17,19 @@ Bun.serve({
return new Response("collabd running"); return new Response("collabd running");
}, },
websocket: { websocket: {
open() { open(ws) {
// create client object once and store in ws.data
const client: Client = { ws };
ws.data.client = client;
console.debug("client connected"); console.debug("client connected");
}, },
message(ws, raw) { message(ws, raw) {
const msg = decode(raw.toString()); const msg = decode(raw.toString());
if (!msg) return; if (!msg) return;
const client: Client = { ws }; // reuse the client object from ws.data
const client = ws.data.client;
if (!client) return;
switch (msg.type) { switch (msg.type) {
case "join": { case "join": {
@ -52,9 +57,9 @@ Bun.serve({
} }
}, },
close(ws) { close(ws) {
if (ws.data.room) { if (ws.data.room && ws.data.client) {
const session = getSession(ws.data.room); const session = getSession(ws.data.room);
session?.leave({ ws }); session?.leave(ws.data.client);
} }
console.debug("client disconnected"); console.debug("client disconnected");
}, },