perf👌: 完善系统管理api

This commit is contained in:
HibiKier
2024-01-15 01:19:37 +08:00
parent 45acf4a094
commit b656f89c8e
9 changed files with 68 additions and 280 deletions
+2 -2
View File
@@ -246,14 +246,14 @@ async def _(date_type: Optional[QueryDateType] = None) -> Result:
@ws_router.websocket("/system_status")
async def system_logs_realtime(websocket: WebSocket):
async def system_logs_realtime(websocket: WebSocket, sleep: Optional[int] = 5):
await websocket.accept()
logger.debug("ws system_status is connect")
try:
while websocket.client_state == WebSocketState.CONNECTED:
system_status = await get_system_status()
await websocket.send_text(system_status.json())
await asyncio.sleep(5)
await asyncio.sleep(sleep)
except (WebSocketDisconnect, ConnectionClosedError, ConnectionClosedOK):
pass
return