添加在线统计

app/signalr/hub/metadata.py

@@ -91,8 +91,14 @@ class MetadataHub(Hub[MetadataClientState]):
 
     @override
     async def _clean_state(self, state: MetadataClientState) -> None:
+        user_id = int(state.connection_id)
+        
+        # Remove from online user tracking
+        from app.router.v2.stats import remove_online_user
+        asyncio.create_task(remove_online_user(user_id))
+        
         if state.pushable:
-            await asyncio.gather(*self.broadcast_tasks(int(state.connection_id), None))
+            await asyncio.gather(*self.broadcast_tasks(user_id, None))
         redis = get_redis()
         if await redis.exists(f"metadata:online:{state.connection_id}"):
             await redis.delete(f"metadata:online:{state.connection_id}")
@@ -117,6 +123,10 @@ class MetadataHub(Hub[MetadataClientState]):
         user_id = int(client.connection_id)
         self.get_or_create_state(client)
 
+        # Track online user
+        from app.router.v2.stats import add_online_user
+        asyncio.create_task(add_online_user(user_id))
+
         async with with_db() as session:
             async with session.begin():
                 friends = (