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}")