Fixed websocket bug

backend/app/api/routes/websocket.py

@@ -1,16 +1,19 @@
+import logging
 from fastapi import APIRouter, WebSocket, WebSocketDisconnect
 
 from backend.app.core.websocket import ws_manager
 from backend.app.services.printer_manager import printer_manager, printer_state_to_dict
 
-
+logger = logging.getLogger(__name__)
 router = APIRouter()
 
 
 @router.websocket("/ws")
 async def websocket_endpoint(websocket: WebSocket):
     """WebSocket endpoint for real-time updates."""
+    logger.info("WebSocket client connecting...")
     await ws_manager.connect(websocket)
+    logger.info("WebSocket client connected")
 
     try:
         # Send initial status of all printers
@@ -21,6 +24,7 @@ async def websocket_endpoint(websocket: WebSocket):
                 "printer_id": printer_id,
                 "data": printer_state_to_dict(state),
             })
+        logger.info(f"Sent initial status for {len(statuses)} printers")
 
         # Keep connection alive and handle incoming messages
         while True:
@@ -43,6 +47,8 @@ async def websocket_endpoint(websocket: WebSocket):
                         })
 
     except WebSocketDisconnect:
+        logger.info("WebSocket client disconnected normally")
         await ws_manager.disconnect(websocket)
-    except Exception:
+    except Exception as e:
+        logger.error(f"WebSocket error: {e}", exc_info=True)
         await ws_manager.disconnect(websocket)

backend/app/main.py

@@ -81,8 +81,16 @@ def register_expected_print(printer_id: int, filename: str, archive_id: int):
     )
 
 
+_last_status_broadcast: dict[int, str] = {}
+
 async def on_printer_status_change(printer_id: int, state: PrinterState):
     """Handle printer status changes - broadcast via WebSocket."""
+    # Only broadcast if something meaningful changed (reduce WebSocket spam)
+    status_key = f"{state.connected}:{state.state}:{state.progress}:{state.layer_num}"
+    if _last_status_broadcast.get(printer_id) == status_key:
+        return  # No change, skip broadcast
+    _last_status_broadcast[printer_id] = status_key
+
     await ws_manager.send_printer_status(
         printer_id,
         printer_state_to_dict(state, printer_id),