Initial commit

backend/app/api/websocket.py

@@ -0,0 +1,59 @@
+import asyncio
+import json
+from fastapi import APIRouter, WebSocket, WebSocketDisconnect
+from app.services.system import SystemService
+
+router = APIRouter()
+
+
+@router.websocket("/ws/system")
+async def websocket_system_stats(websocket: WebSocket):
+    """
+    WebSocket endpoint pour stream les stats système en temps réel
+    """
+    await websocket.accept()
+    
+    # Initialiser le cache psutil
+    import psutil
+    psutil.cpu_percent(interval=0.05, percpu=True)
+    
+    try:
+        while True:
+            # Récupère les stats (très rapide avec cache)
+            stats = SystemService.get_system_stats()
+            
+            # Envoie au client
+            await websocket.send_json({
+                "cpu": {
+                    "percent": stats.cpu.percent,
+                    "average": stats.cpu.average,
+                    "cores": stats.cpu.cores,
+                    "per_cpu": stats.cpu.per_cpu
+                },
+                "memory": {
+                    "percent": stats.memory.percent,
+                    "used": stats.memory.used,
+                    "total": stats.memory.total,
+                    "available": stats.memory.available
+                },
+                "processes": [
+                    {
+                        "pid": p.pid,
+                        "name": p.name,
+                        "status": p.status,
+                        "cpu_percent": p.cpu_percent,
+                        "memory_percent": p.memory_percent,
+                        "username": p.username
+                    }
+                    for p in stats.processes
+                ]
+            })
+            
+            # Attendre 1 seconde avant le prochain envoi (1 fps, comme GNOME Monitor)
+            await asyncio.sleep(1)
+            
+    except WebSocketDisconnect:
+        print(f"Client déconnecté")
+    except Exception as e:
+        print(f"Erreur WebSocket: {e}")
+        await websocket.close(code=1000)