From f65f19249c51e438aac7586b1e0a83a6f0160c07 Mon Sep 17 00:00:00 2001 From: ikun-11451 <334495606@qq.com> Date: Sun, 5 Oct 2025 23:09:40 +0800 Subject: [PATCH] =?UTF-8?q?=E5=90=8E=E7=AB=AF=E5=A2=9E=E5=8A=A0cpu?= =?UTF-8?q?=E5=8D=A0=E7=94=A8=E7=8E=87=E7=AD=89api=E5=96=B5~?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/api/api_router.py | 6 ++++++ src/api/system_router.py | 19 +++++++++++++++++++ src/main.py | 4 +++- 3 files changed, 28 insertions(+), 1 deletion(-) create mode 100644 src/api/api_router.py create mode 100644 src/api/system_router.py diff --git a/src/api/api_router.py b/src/api/api_router.py new file mode 100644 index 000000000..f1b449e97 --- /dev/null +++ b/src/api/api_router.py @@ -0,0 +1,6 @@ +from fastapi import APIRouter +from src.api.system_router import router as system_router + +# 导出统一的 API 路由 +router = APIRouter() +router.include_router(system_router, prefix="/api") diff --git a/src/api/system_router.py b/src/api/system_router.py new file mode 100644 index 000000000..001456f95 --- /dev/null +++ b/src/api/system_router.py @@ -0,0 +1,19 @@ +import psutil +from fastapi import APIRouter, HTTPException + +router = APIRouter() + +@router.get("/system/stats") +async def get_system_stats(): + """ + 获取系统CPU和内存使用率。 + """ + try: + cpu_percent = psutil.cpu_percent(interval=1) + memory_info = psutil.virtual_memory() + return { + "cpu_percent": cpu_percent, + "memory_percent": memory_info.percent + } + except Exception as e: + raise HTTPException(status_code=500, detail=str(e)) \ No newline at end of file diff --git a/src/main.py b/src/main.py index 63faddfbc..e1acbf620 100644 --- a/src/main.py +++ b/src/main.py @@ -389,8 +389,10 @@ MoFox_Bot(第三方修改版) # 注册API路由 try: from src.api.message_router import router as message_router + from src.api.api_router import router as api_router self.server.register_router(message_router, prefix="/api") - logger.info("API路由注册成功") + self.server.register_router(api_router, prefix="/api") + logger.info("API路由注册成功(含 system_router)") except Exception as e: logger.error(f"注册API路由失败: {e}")