117 lines
3.8 KiB
Python
117 lines
3.8 KiB
Python
|
import platform
|
|||
|
|
import socket
|
|||
|
|
import time
|
|||
|
|
from pathlib import Path
|
|||
|
|
|
|||
|
|
import psutil
|
|||
|
|
|
|||
|
|
from app.utils.common_util import bytes2human
|
|||
|
|
|
|||
|
|
from .schema import (
|
|||
|
|
CpuInfoSchema,
|
|||
|
|
DiskInfoSchema,
|
|||
|
|
MemoryInfoSchema,
|
|||
|
|
PyInfoSchema,
|
|||
|
|
ServerMonitorSchema,
|
|||
|
|
SysInfoSchema,
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
|
|||
|
|
class ServerService:
|
|||
|
|
"""服务监控模块服务层"""
|
|||
|
|
|
|||
|
|
@staticmethod
|
|||
|
|
async def get_server_monitor_info() -> ServerMonitorSchema:
|
|||
|
|
return ServerMonitorSchema(
|
|||
|
|
cpu=ServerService._get_cpu_info(),
|
|||
|
|
mem=ServerService._get_memory_info(),
|
|||
|
|
sys=ServerService._get_system_info(),
|
|||
|
|
py=ServerService._get_python_info(),
|
|||
|
|
disks=ServerService._get_disk_info(),
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
@staticmethod
|
|||
|
|
def _get_cpu_info() -> CpuInfoSchema:
|
|||
|
|
cpu_times = psutil.cpu_times_percent()
|
|||
|
|
cpu_num = psutil.cpu_count(logical=True)
|
|||
|
|
if not cpu_num:
|
|||
|
|
cpu_num = 1
|
|||
|
|
return CpuInfoSchema(
|
|||
|
|
cpu_num=cpu_num,
|
|||
|
|
used=cpu_times.user,
|
|||
|
|
sys=cpu_times.system,
|
|||
|
|
free=cpu_times.idle,
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
@staticmethod
|
|||
|
|
def _get_memory_info() -> MemoryInfoSchema:
|
|||
|
|
memory = psutil.virtual_memory()
|
|||
|
|
return MemoryInfoSchema(
|
|||
|
|
total=bytes2human(memory.total),
|
|||
|
|
used=bytes2human(memory.used),
|
|||
|
|
free=bytes2human(memory.free),
|
|||
|
|
usage=memory.percent,
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
@staticmethod
|
|||
|
|
def _get_system_info() -> SysInfoSchema:
|
|||
|
|
hostname = socket.gethostname()
|
|||
|
|
return SysInfoSchema(
|
|||
|
|
computer_ip=socket.gethostbyname(hostname),
|
|||
|
|
computer_name=platform.node(),
|
|||
|
|
os_arch=platform.machine(),
|
|||
|
|
os_name=platform.platform(),
|
|||
|
|
user_dir=str(Path.cwd()),
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
@staticmethod
|
|||
|
|
def _get_python_info() -> PyInfoSchema:
|
|||
|
|
current_process = psutil.Process()
|
|||
|
|
memory = psutil.virtual_memory()
|
|||
|
|
process_memory = current_process.memory_info()
|
|||
|
|
|
|||
|
|
start_time = current_process.create_time()
|
|||
|
|
run_time = ServerService._calculate_run_time(start_time)
|
|||
|
|
|
|||
|
|
return PyInfoSchema(
|
|||
|
|
name=current_process.name(),
|
|||
|
|
version=platform.python_version(),
|
|||
|
|
start_time=time.strftime("%Y-%m-%d %H:%M:%S", time.localtime(start_time)),
|
|||
|
|
run_time=run_time,
|
|||
|
|
home=str(Path(current_process.exe())),
|
|||
|
|
memory_total=bytes2human(memory.available),
|
|||
|
|
memory_used=bytes2human(process_memory.rss),
|
|||
|
|
memory_free=bytes2human(memory.available - process_memory.rss),
|
|||
|
|
memory_usage=round((process_memory.rss / memory.available) * 100, 2),
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
@staticmethod
|
|||
|
|
def _get_disk_info() -> list[DiskInfoSchema]:
|
|||
|
|
disk_info = []
|
|||
|
|
for partition in psutil.disk_partitions():
|
|||
|
|
try:
|
|||
|
|
usage = psutil.disk_usage(partition.mountpoint)
|
|||
|
|
mount_point = str(Path(partition.mountpoint))
|
|||
|
|
disk_info.append(
|
|||
|
|
DiskInfoSchema(
|
|||
|
|
dir_name=mount_point,
|
|||
|
|
sys_type_name=partition.fstype,
|
|||
|
|
type_name=f"本地固定磁盘（{mount_point}）",
|
|||
|
|
total=bytes2human(usage.total),
|
|||
|
|
used=bytes2human(usage.used),
|
|||
|
|
free=bytes2human(usage.free),
|
|||
|
|
usage=usage.percent,
|
|||
|
|
),
|
|||
|
|
)
|
|||
|
|
except (PermissionError, FileNotFoundError):
|
|||
|
|
continue
|
|||
|
|
return disk_info
|
|||
|
|
|
|||
|
|
@staticmethod
|
|||
|
|
def _calculate_run_time(start_time: float) -> str:
|
|||
|
|
difference = time.time() - start_time
|
|||
|
|
days = int(difference // (24 * 60 * 60))
|
|||
|
|
hours = int((difference % (24 * 60 * 60)) // (60 * 60))
|
|||
|
|
minutes = int((difference % (60 * 60)) // 60)
|
|||
|
|
return f"{days}天{hours}小时{minutes}分钟"
|