diff --git a/app/api/endpoints/dashboard.py b/app/api/endpoints/dashboard.py index f5398642..8b883ac9 100644 --- a/app/api/endpoints/dashboard.py +++ b/app/api/endpoints/dashboard.py @@ -124,3 +124,19 @@ def transfer(days: int = 7, db: Session = Depends(get_db), """ transfer_stat = TransferHistory.statistic(db, days) return [stat[1] for stat in transfer_stat] + + +@router.get("/cpu", summary="获取当前CPU使用率", response_model=int) +def cpu(_: schemas.TokenPayload = Depends(verify_token)) -> Any: + """ + 获取当前CPU使用率 + """ + return SystemUtils.cpu_usage() + + +@router.get("/memory", summary="获取当前内存使用率", response_model=int) +def memory(_: schemas.TokenPayload = Depends(verify_token)) -> Any: + """ + 获取当前内存使用率 + """ + return SystemUtils.memory_usage() diff --git a/app/utils/system.py b/app/utils/system.py index cdef1a52..6bcc3086 100644 --- a/app/utils/system.py +++ b/app/utils/system.py @@ -278,3 +278,17 @@ class SystemUtils: if os.path.isdir(vol): vols.append(vol) return vols + + @staticmethod + def cpu_usage(): + """ + 获取CPU使用率 + """ + return psutil.cpu_percent() + + @staticmethod + def memory_usage(): + """ + 获取内存使用大小 + """ + return psutil.virtual_memory().used