refactor: remove presence debounce for instant user departures

refactor: simplify websocket connection and error handling in rpc view

CHANGELOG.md

@@ -6,6 +6,14 @@
 
 
 
+
+## Version 1.6.0 - 2025-12-17
+
+Removes presence debounce to make user departures instant. Simplifies websocket connection and error handling in RPC views for improved reliability.
+
+**Changes:** 2 files, 98 lines
+**Languages:** Python (98 lines)
+
 ## Version 1.5.0 - 2025-12-17
 
 remove umami analytics script

pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "Snek"
-version = "1.5.0"
+version = "1.6.0"
 readme = "README.md"
 #license = { file = "LICENSE", content-type="text/markdown" }
 description = "Snek Chat Application by Molodetz"

src/snek/service/socket.py

@@ -8,9 +8,6 @@ from snek.system.service import BaseService
 logger = logging.getLogger(__name__)
 from snek.system.model import now
 
-PRESENCE_DEBOUNCE_SECONDS = 3.0
-
-
 class SocketService(BaseService):
 
     class Socket:
@@ -48,7 +45,6 @@ class SocketService(BaseService):
         self.users = {}
         self.subscriptions = {}
         self.last_update = str(datetime.now())
-        self._departure_tasks = {}
 
     async def user_availability_service(self):
         logger.info("User availability update service started.")
@@ -90,10 +86,6 @@ class SocketService(BaseService):
         logger.info(f"Added socket for user {s.user['username']}")
 
         is_first_connection = False
-        if user_uid in self._departure_tasks:
-            self._departure_tasks[user_uid].cancel()
-            del self._departure_tasks[user_uid]
-
         if not self.users.get(user_uid):
             self.users[user_uid] = set()
             is_first_connection = True
@@ -147,25 +139,7 @@ class SocketService(BaseService):
                 if user_uid in self.users:
                     self.users[user_uid].discard(s)
                     if len(self.users[user_uid]) == 0:
-                        await self._schedule_departure(user_uid, user_nick, user_color)
-
-    async def _schedule_departure(self, user_uid, user_nick, user_color):
-        if user_uid in self._departure_tasks:
-            return
-
-        async def delayed_departure():
-            try:
-                await asyncio.sleep(PRESENCE_DEBOUNCE_SECONDS)
-                if user_uid in self.users and len(self.users[user_uid]) == 0:
                         await self._broadcast_presence("departed", user_uid, user_nick, user_color)
-                if user_uid in self._departure_tasks:
-                    del self._departure_tasks[user_uid]
-            except asyncio.CancelledError:
-                pass
-            except Exception as ex:
-                logger.warning(f"Error in departure broadcast: {ex}")
-
-        self._departure_tasks[user_uid] = asyncio.create_task(delayed_departure())
 
     async def _broadcast_presence(self, event_type, user_uid, user_nick, user_color):
         if not user_uid or not user_nick:

src/snek/view/rpc.py

@@ -627,27 +627,24 @@ class RPCView(BaseView):
 
         ws = web.WebSocketResponse()
         await ws.prepare(self.request)
-        user_uid = self.request.session.get("uid") if self.request.session.get("logged_in") else None
-
-        try:
-            if user_uid:
-                await self.services.socket.add(ws, user_uid)
+        if self.request.session.get("logged_in"):
+            await self.services.socket.add(ws, self.request.session.get("uid"))
             async for subscription in self.services.channel_member.find(
-                    user_uid=user_uid,
+                user_uid=self.request.session.get("uid"),
                 deleted_at=None,
                 is_banned=False,
             ):
                 await self.services.socket.subscribe(
-                        ws, subscription["channel_uid"], user_uid
+                    ws, subscription["channel_uid"], self.request.session.get("uid")
                 )
         if not scheduled and self.request.app.uptime_seconds < 5:
             await schedule(
-                    user_uid,
+                self.request.session.get("uid"),
                 0,
                 {"event": "refresh", "data": {"message": "Finishing deployment"}},
             )
             await schedule(
-                    user_uid,
+                self.request.session.get("uid"),
                 15,
                 {"event": "deployed", "data": {"uptime": self.request.app.uptime}},
             )
@@ -658,13 +655,12 @@ class RPCView(BaseView):
                 try:
                     await rpc(msg.json())
                 except Exception as ex:
+                    print("XXXXXXXXXX Deleting socket", ex, flush=True)
                     logger.exception(ex)
+                    await self.services.socket.delete(ws)
                     break
             elif msg.type == web.WSMsgType.ERROR:
-                    break
+                pass
             elif msg.type == web.WSMsgType.CLOSE:
-                    break
-        finally:
-            await self.services.socket.delete(ws)
-
+                pass
         return ws