完善直播节点监控与边缘代理

backend/main.py

@@ -6,17 +6,63 @@ from fastapi.responses import FileResponse
 from starlette.responses import RedirectResponse
 from starlette.background import BackgroundTask
 import httpx
+import asyncio
+import time
+import psutil
 from database import init_db
 from services.transcoder import transcode_mgr
 from routes import streams_router, nodes_router, auth_router
 from routes.settings_route import router as settings_router
 from config import settings
+from services.load_balancer import lb
+from services.node_store import load_nodes
+
+
+async def monitor_origin():
+    previous = psutil.net_io_counters()
+    previous_time = time.monotonic()
+    while True:
+        await asyncio.sleep(10)
+        current = psutil.net_io_counters()
+        current_time = time.monotonic()
+        elapsed = current_time - previous_time
+        network_in = ((current.bytes_recv - previous.bytes_recv) * 8 / elapsed / 1e6) if elapsed > 0 else None
+        network_out = ((current.bytes_sent - previous.bytes_sent) * 8 / elapsed / 1e6) if elapsed > 0 else None
+        media_available = False
+        viewer_count = None
+        try:
+            async with httpx.AsyncClient(timeout=3) as client:
+                response = await client.get(f"http://127.0.0.1:{settings.media_origin_api_port}/v3/paths/list")
+                response.raise_for_status()
+                payload = response.json()
+                viewer_count = sum(item.get("readers", 0) for item in payload.get("items", []))
+                media_available = True
+        except Exception:
+            pass
+        lb.report_health("default", {
+            "cpu_percent": psutil.cpu_percent(),
+            "memory_percent": psutil.virtual_memory().percent,
+            "bandwidth_mbps": network_in + network_out if network_in is not None and network_out is not None else None,
+            "network_in_mbps": network_in,
+            "network_out_mbps": network_out,
+            "active_connections": len(psutil.net_connections(kind="tcp")),
+            "hls_readers": viewer_count,
+            "viewer_count": viewer_count,
+            "frame_drop_indicators": None,
+            "packet_loss_indicators": None,
+            "media_metrics_available": False,
+        })
+        previous = current
+        previous_time = current_time
 
 
 @asynccontextmanager
 async def lifespan(app: FastAPI):
+    load_nodes()
     await init_db()
+    origin_monitor = asyncio.create_task(monitor_origin())
     yield
+    origin_monitor.cancel()
     transcode_mgr.stop_all()