BBIT-Kai
60 lines
1.9 KiB
Python
60 lines
1.9 KiB
Python
import logging
|
|
import os
|
|
|
|
import psutil
|
|
from mcp.server.fastmcp import FastMCP
|
|
|
|
logger = logging.getLogger("system_tools")
|
|
|
|
DINGTALK_WEBHOOK = os.environ.get("DINGTALK_WEBHOOK")
|
|
|
|
|
|
def register_system_tools(mcp: FastMCP):
|
|
@mcp.tool()
|
|
def get_server_status() -> dict:
|
|
"""
|
|
获取服务器状态监控信息。
|
|
返回:
|
|
- 包含CPU、内存、磁盘等使用情况的字典
|
|
"""
|
|
try:
|
|
# CPU信息
|
|
cpu_percent = psutil.cpu_percent(interval=1)
|
|
cpu_count = psutil.cpu_count()
|
|
|
|
# 内存信息
|
|
memory = psutil.virtual_memory()
|
|
memory_total = memory.total / (1024 * 1024 * 1024) # GB
|
|
memory_used = memory.used / (1024 * 1024 * 1024) # GB
|
|
memory_percent = memory.percent
|
|
|
|
# 磁盘信息
|
|
disk = psutil.disk_usage("/")
|
|
disk_total = disk.total / (1024 * 1024 * 1024) # GB
|
|
disk_used = disk.used / (1024 * 1024 * 1024) # GB
|
|
disk_percent = disk.percent
|
|
|
|
# 系统启动时间
|
|
boot_time = psutil.boot_time()
|
|
|
|
return {
|
|
"success": True,
|
|
"result": {
|
|
"cpu": {"usage_percent": cpu_percent, "core_count": cpu_count},
|
|
"memory": {
|
|
"total_gb": round(memory_total, 2),
|
|
"used_gb": round(memory_used, 2),
|
|
"usage_percent": memory_percent,
|
|
},
|
|
"disk": {
|
|
"total_gb": round(disk_total, 2),
|
|
"used_gb": round(disk_used, 2),
|
|
"usage_percent": disk_percent,
|
|
},
|
|
"system": {"boot_time": boot_time},
|
|
},
|
|
}
|
|
except Exception as e:
|
|
logger.error(f"获取服务器状态失败: {str(e)}")
|
|
return {"success": False, "error": str(e)}
|