back(socket): update last connection on disconnect

t-h2o · Dec 18, 2024 · d43eb45 · d43eb45
1 parent 0d7f309
commit d43eb45
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/backend/matcha/websocket/main_namespace.py b/backend/matcha/websocket/main_namespace.py
@@ -4,6 +4,8 @@
 
 from matcha.utils import flaskprint
 
+from matcha.db.last_connection import db_update_last_connection
+
 from matcha.websocket.socket_manager import SocketManager
 
 
@@ -33,6 +35,7 @@ def on_connect(self, auth):
 
     def on_disconnect(self):
         try:
+            db_update_last_connection(SocketManager.get_user_id(request.sid))
             SocketManager.remove_session(request.sid)
         except Exception as e:
             flaskprint("Error handling message:" + str(e))