From 67f11717c20b88b73ffc1d13a4517c6bf1dc04ce Mon Sep 17 00:00:00 2001
From: qiangyanwen <508737091@qq.com>
Date: Tue, 13 Dec 2022 10:10:55 +0800
Subject: [PATCH] add test
---
api/sys_monitor_api.py | 52 +++++++++++++++++++++++++++++++++++++++++-
main.py | 3 ++-
2 files changed, 53 insertions(+), 2 deletions(-)
diff --git a/api/sys_monitor_api.py b/api/sys_monitor_api.py
index f2611e8..c94ef61 100644
--- a/api/sys_monitor_api.py
+++ b/api/sys_monitor_api.py
@@ -3,10 +3,13 @@
# @Email :508737091@qq.com
# @Author :qiangyanwen
# @File :sys_monitor_api.py
+import time
-from fastapi import APIRouter, Body, Depends, Query
+from fastapi import APIRouter, Body, Depends, Query,WebSocket,WebSocketDisconnect
from utils.mem import get_disk_info, get_sys_info, get_cpu_info, ge_mem_info
from utils.response import success_200
+from fastapi.responses import HTMLResponse
+from utils.websocket_manager import manager
router = APIRouter(prefix="/api", tags=["系统模块"])
@@ -29,3 +32,50 @@ async def get_cpu():
@router.get("/sys/memory", summary="获取内存信息", name="获取内存信息")
async def get_mem():
return success_200(data=await ge_mem_info(), message="查询内存信息成功")
+
+
+html = """
+
+
+
+ Chat
+
+
+ WebSocket Chat
+
+
+
+
+
+"""
+@router.get("/")
+async def get():
+ return HTMLResponse(html)
+
+@router.websocket("/items/ws")
+async def websocket_endpoint(
+ websocket: WebSocket
+):
+ await websocket.accept()
+ while True:
+ data = await websocket.receive_text()
+ await websocket.send_text(f"消息是: {data}")
diff --git a/main.py b/main.py
index fbed760..ce3cbd7 100644
--- a/main.py
+++ b/main.py
@@ -35,5 +35,6 @@ async def errors_handling(request: Request, call_next):
})
)
+
if __name__ == '__main__':
- uvicorn.run(app="main:app", host=host, port=settings.PROJECT.PORT, reload=Debug)
+ uvicorn.run(app="main:app", host="127.0.0.1", port=settings.PROJECT.PORT, reload=Debug, loop="asyncio")