331 lines
15 KiB
Python
331 lines
15 KiB
Python
"""Dashboard 仪表盘插件"""
|
|
import os
|
|
import time
|
|
import json
|
|
import socket
|
|
import subprocess
|
|
import platform
|
|
import psutil
|
|
from collections import deque
|
|
from oss.logger.logger import Log
|
|
from oss.plugin.types import Plugin, Response, register_plugin_type
|
|
|
|
|
|
class DashboardPlugin(Plugin):
|
|
"""仪表盘插件 - 依赖 WebUI 容器"""
|
|
|
|
def __init__(self):
|
|
self.webui = None
|
|
self.views_dir = os.path.join(os.path.dirname(__file__), 'views')
|
|
self._start_time = time.time() # 记录插件启动时间（即项目启动时间）
|
|
self._history_len = 60
|
|
self._cpu_history = deque(maxlen=self._history_len)
|
|
self._ram_history = deque(maxlen=self._history_len)
|
|
self._net_recv_history = deque(maxlen=self._history_len)
|
|
self._net_sent_history = deque(maxlen=self._history_len)
|
|
self._disk_read_history = deque(maxlen=self._history_len)
|
|
self._disk_write_history = deque(maxlen=self._history_len)
|
|
self._net_latency_history = deque(maxlen=self._history_len)
|
|
self._last_net = None
|
|
self._last_disk = None
|
|
|
|
def meta(self):
|
|
from oss.plugin.types import Metadata, PluginConfig, Manifest
|
|
return Manifest(
|
|
metadata=Metadata(
|
|
name="dashboard",
|
|
version="2.0.0",
|
|
author="NebulaShell",
|
|
description="WebUI 仪表盘"
|
|
),
|
|
config=PluginConfig(enabled=True, args={}),
|
|
dependencies=["http-api", "webui"]
|
|
)
|
|
|
|
def set_webui(self, webui):
|
|
self.webui = webui
|
|
|
|
def init(self, deps: dict = None):
|
|
if self.webui:
|
|
Log.info("dashboard", "已获取 WebUI 引用")
|
|
self.webui.register_page(
|
|
path='/dashboard',
|
|
content_provider=self._render_content,
|
|
nav_item={'icon': 'ri-dashboard-line', 'text': '仪表盘'}
|
|
)
|
|
if hasattr(self.webui, 'server') and self.webui.server:
|
|
self.webui.server.router.get("/api/dashboard/stats", self._handle_stats_api)
|
|
self.webui.server.router.get("/api/dashboard/history", self._handle_history_api)
|
|
Log.info("dashboard", "已注册到 WebUI 导航")
|
|
else:
|
|
Log.warn("dashboard", "警告: 未找到 WebUI 依赖")
|
|
|
|
def _get_uptime_str(self):
|
|
"""计算项目运行时间（从插件启动时算起）"""
|
|
elapsed = time.time() - self._start_time
|
|
days = int(elapsed // 86400)
|
|
hours = int((elapsed % 86400) // 3600)
|
|
minutes = int((elapsed % 3600) // 60)
|
|
seconds = int(elapsed % 60)
|
|
if days > 0:
|
|
return f"{days}天{hours}时{minutes}分{seconds}秒"
|
|
elif hours > 0:
|
|
return f"{hours}时{minutes}分{seconds}秒"
|
|
elif minutes > 0:
|
|
return f"{minutes}分{seconds}秒"
|
|
else:
|
|
return f"{seconds}秒"
|
|
|
|
def _get_network_stats(self):
|
|
try:
|
|
net = psutil.net_io_counters()
|
|
now = time.time()
|
|
if self._last_net is None:
|
|
self._last_net = (now, net.bytes_recv, net.bytes_sent)
|
|
return {'recv_rate': 0, 'sent_rate': 0, 'total_recv': net.bytes_recv, 'total_sent': net.bytes_sent}
|
|
elapsed = now - self._last_net[0]
|
|
if elapsed <= 0: elapsed = 1
|
|
recv_rate = (net.bytes_recv - self._last_net[1]) / elapsed
|
|
sent_rate = (net.bytes_sent - self._last_net[2]) / elapsed
|
|
self._last_net = (now, net.bytes_recv, net.bytes_sent)
|
|
return {'recv_rate': round(recv_rate, 1), 'sent_rate': round(sent_rate, 1), 'total_recv': net.bytes_recv, 'total_sent': net.bytes_sent}
|
|
except Exception as e:
|
|
import traceback; print(f"[main.py] 错误:{type(e).__name__}:{e}"); traceback.print_exc()
|
|
return {'recv_rate': 0, 'sent_rate': 0, 'total_recv': 0, 'total_sent': 0}
|
|
|
|
def _get_disk_io_stats(self):
|
|
try:
|
|
disk_io = psutil.disk_io_counters()
|
|
if not disk_io:
|
|
return {'read_rate': 0, 'write_rate': 0}
|
|
now = time.time()
|
|
if self._last_disk is None:
|
|
self._last_disk = (now, disk_io.read_bytes, disk_io.write_bytes)
|
|
return {'read_rate': 0, 'write_rate': 0}
|
|
elapsed = now - self._last_disk[0]
|
|
if elapsed <= 0: elapsed = 1
|
|
read_rate = (disk_io.read_bytes - self._last_disk[1]) / elapsed
|
|
write_rate = (disk_io.write_bytes - self._last_disk[2]) / elapsed
|
|
self._last_disk = (now, disk_io.read_bytes, disk_io.write_bytes)
|
|
return {'read_rate': round(read_rate, 1), 'write_rate': round(write_rate, 1)}
|
|
except Exception as e:
|
|
import traceback; print(f"[main.py] 错误:{type(e).__name__}:{e}"); traceback.print_exc()
|
|
return {'read_rate': 0, 'write_rate': 0}
|
|
|
|
def _get_network_latency(self) -> float:
|
|
"""测量到公共 DNS 8.8.8.8 的 TCP 连接延迟（真实网络波动）"""
|
|
try:
|
|
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
|
|
s.settimeout(2)
|
|
start = time.time()
|
|
s.connect(('8.8.8.8', 53))
|
|
elapsed = (time.time() - start) * 1000 # 毫秒
|
|
s.close()
|
|
return round(elapsed, 1)
|
|
except Exception as e:
|
|
import traceback; print(f"[main.py] 错误:{type(e).__name__}:{e}"); traceback.print_exc()
|
|
return 0.0
|
|
|
|
def _get_network_interfaces(self):
|
|
try:
|
|
interfaces = []
|
|
addrs = psutil.net_if_addrs()
|
|
stats = psutil.net_if_stats()
|
|
for name, addr_list in addrs.items():
|
|
if name == 'lo':
|
|
continue
|
|
info = {'name': name, 'ip': 'N/A', 'mac': 'N/A', 'is_up': False, 'speed': 0}
|
|
for addr in addr_list:
|
|
if addr.family == socket.AF_INET:
|
|
info['ip'] = addr.address
|
|
elif hasattr(psutil, 'AF_LINK') and addr.family == psutil.AF_LINK:
|
|
info['mac'] = addr.address
|
|
if name in stats:
|
|
info['is_up'] = stats[name].isup
|
|
info['speed'] = stats[name].speed
|
|
interfaces.append(info)
|
|
return interfaces
|
|
except Exception as e:
|
|
import traceback; print(f"[main.py] 错误:{type(e).__name__}:{e}"); traceback.print_exc()
|
|
return []
|
|
|
|
def _get_load_info(self):
|
|
try:
|
|
load1, load5, load15 = os.getloadavg()
|
|
return {'load1': round(load1, 2), 'load5': round(load5, 2), 'load15': round(load15, 2)}
|
|
except (OSError, AttributeError):
|
|
return {'load1': 0, 'load5': 0, 'load15': 0}
|
|
|
|
def _handle_stats_api(self, request):
|
|
try:
|
|
cpu_percent = psutil.cpu_percent(interval=0.3)
|
|
mem = psutil.virtual_memory()
|
|
disk = psutil.disk_usage('/')
|
|
net = self._get_network_stats()
|
|
disk_io = self._get_disk_io_stats()
|
|
load = self._get_load_info()
|
|
latency = self._get_network_latency()
|
|
|
|
self._cpu_history.append(round(cpu_percent, 1))
|
|
self._ram_history.append(round(mem.percent, 1))
|
|
self._net_recv_history.append(net['recv_rate'])
|
|
self._net_sent_history.append(net['sent_rate'])
|
|
self._disk_read_history.append(disk_io['read_rate'])
|
|
self._disk_write_history.append(disk_io['write_rate'])
|
|
self._net_latency_history.append(latency)
|
|
|
|
uptime_str = self._get_uptime_str()
|
|
|
|
data = {
|
|
'cpu': {'percent': round(cpu_percent, 1), 'cores': psutil.cpu_count(logical=True)},
|
|
'ram': {'percent': round(mem.percent, 1), 'used': round(mem.used / (1024**3), 1), 'total': round(mem.total / (1024**3), 1)},
|
|
'disk': {'percent': round(disk.percent, 1), 'used': round(disk.used / (1024**3), 1), 'total': round(disk.total / (1024**3), 1)},
|
|
'network': net,
|
|
'disk_io': disk_io,
|
|
'load': load,
|
|
'latency': latency,
|
|
'processes': len(psutil.pids()),
|
|
'uptime': uptime_str
|
|
}
|
|
return Response(status=200, headers={"Content-Type": "application/json"}, body=json.dumps(data))
|
|
except Exception as e:
|
|
return Response(status=500, headers={"Content-Type": "application/json"}, body=json.dumps({'error': str(e)}))
|
|
|
|
def _handle_history_api(self, request):
|
|
try:
|
|
data = {
|
|
'cpu': list(self._cpu_history),
|
|
'ram': list(self._ram_history),
|
|
'net_recv': list(self._net_recv_history),
|
|
'net_sent': list(self._net_sent_history),
|
|
'disk_read': list(self._disk_read_history),
|
|
'disk_write': list(self._disk_write_history),
|
|
'latency': list(self._net_latency_history)
|
|
}
|
|
return Response(status=200, headers={"Content-Type": "application/json"}, body=json.dumps(data))
|
|
except Exception as e:
|
|
return Response(status=500, headers={"Content-Type": "application/json"}, body=json.dumps({'error': str(e)}))
|
|
|
|
def start(self):
|
|
Log.info("dashboard", "仪表盘已启动")
|
|
|
|
def stop(self):
|
|
Log.error("dashboard", "仪表盘已停止")
|
|
|
|
def _render_content(self) -> str:
|
|
"""渲染仪表盘页面 - 纯 HTML/Python 模板"""
|
|
try:
|
|
import psutil
|
|
import platform
|
|
|
|
cpu_percent = psutil.cpu_percent(interval=0.5)
|
|
cpu_cores = psutil.cpu_count(logical=True)
|
|
mem = psutil.virtual_memory()
|
|
ram_percent = round(mem.percent, 1)
|
|
ram_used_gb = round(mem.used / (1024**3), 1)
|
|
ram_total_gb = round(mem.total / (1024**3), 1)
|
|
disk = psutil.disk_usage('/')
|
|
disk_percent = round(disk.percent, 1)
|
|
disk_used_gb = round(disk.used / (1024**3), 1)
|
|
disk_total_gb = round(disk.total / (1024**3), 1)
|
|
|
|
circumference = 2 * 3.14159 * 52
|
|
cpu_dash_offset = round(circumference - (cpu_percent / 100) * circumference, 1)
|
|
ram_dash_offset = round(circumference - (ram_percent / 100) * circumference, 1)
|
|
disk_dash_offset = round(circumference - (disk_percent / 100) * circumference, 1)
|
|
|
|
uptime_str = self._get_uptime_str()
|
|
|
|
disk_color = 'gauge-green' if disk_percent < 50 else ('gauge-orange' if disk_percent < 80 else 'gauge-blue')
|
|
|
|
html = f"""<!DOCTYPE html>
|
|
<html lang="zh-CN">
|
|
<head>
|
|
<meta charset="UTF-8">
|
|
<meta name="viewport" content="width=device-width, initial-scale=1.0">
|
|
<title>系统仪表盘</title>
|
|
<link rel="stylesheet" href="/assets/remixicon.css">
|
|
<style>
|
|
* {{ margin: 0; padding: 0; box-sizing: border-box; }}
|
|
body {{ font-family: -apple-system, BlinkMacSystemFont, 'Segoe UI', Roboto, sans-serif; background: #f5f6fa; padding: 20px; }}
|
|
.container {{ max-width: 1400px; margin: 0 auto; }}
|
|
.card {{ background: white; border-radius: 10px; box-shadow: 0 2px 10px rgba(0,0,0,0.1); padding: 20px; margin-bottom: 20px; }}
|
|
.card-title {{ font-size: 18px; font-weight: 600; color: #2c3e50; margin-bottom: 20px; }}
|
|
.stats-grid {{ display: grid; grid-template-columns: repeat(auto-fit, minmax(250px, 1fr)); gap: 20px; }}
|
|
.stat-card {{ background: #f8f9fa; border-radius: 8px; padding: 20px; text-align: center; }}
|
|
.stat-icon {{ width: 60px; height: 60px; margin: 0 auto 15px; border-radius: 50%; display: flex; align-items: center; justify-content: center; font-size: 28px; color: white; }}
|
|
.stat-icon.cpu {{ background: linear-gradient(135deg, #667eea 0%, #764ba2 100%); }}
|
|
.stat-icon.ram {{ background: linear-gradient(135deg, #f093fb 0%, #f5576c 100%); }}
|
|
.stat-icon.disk {{ background: linear-gradient(135deg, #4facfe 0%, #00f2fe 100%); }}
|
|
.stat-value {{ font-size: 24px; font-weight: 700; color: #2c3e50; margin-bottom: 5px; }}
|
|
.stat-label {{ font-size: 14px; color: #7f8c8d; }}
|
|
.gauge-container {{ position: relative; width: 120px; height: 120px; margin: 0 auto; }}
|
|
.gauge-svg {{ transform: rotate(-90deg); }}
|
|
.gauge-bg {{ fill: none; stroke: #e5e7eb; stroke-width: 8; }}
|
|
.gauge-fill {{ fill: none; stroke: #3498db; stroke-width: 8; stroke-linecap: round; transition: stroke-dashoffset 0.5s; }}
|
|
.gauge-green .gauge-fill {{ stroke: #27ae60; }}
|
|
.gauge-orange .gauge-fill {{ stroke: #f39c12; }}
|
|
.gauge-blue .gauge-fill {{ stroke: #e74c3c; }}
|
|
.gauge-text {{ position: absolute; top: 50%; left: 50%; transform: translate(-50%, -50%); font-size: 18px; font-weight: 600; color: #2c3e50; }}
|
|
.info-grid {{ display: grid; grid-template-columns: repeat(auto-fit, minmax(200px, 1fr)); gap: 15px; margin-top: 20px; }}
|
|
.info-item {{ background: #f8f9fa; padding: 15px; border-radius: 6px; }}
|
|
.info-label {{ font-size: 12px; color: #7f8c8d; margin-bottom: 5px; }}
|
|
.info-value {{ font-size: 14px; color: #2c3e50; font-weight: 600; }}
|
|
</style>
|
|
</head>
|
|
<body>
|
|
<div class="container">
|
|
<div class="card">
|
|
<h2 class="card-title"><i class="ri-dashboard-line"></i> 系统仪表盘</h2>
|
|
<div class="stats-grid">
|
|
<div class="stat-card">
|
|
<div class="stat-icon cpu"><i class="ri-cpu-line"></i></div>
|
|
<div class="stat-value">{cpu_percent}%</div>
|
|
<div class="stat-label">CPU 使用率 ({cpu_cores} 核心)</div>
|
|
</div>
|
|
<div class="stat-card">
|
|
<div class="stat-icon ram"><i class="ri-memory-line"></i></div>
|
|
<div class="stat-value">{ram_percent}%</div>
|
|
<div class="stat-label">内存使用 ({ram_used_gb} GB / {ram_total_gb} GB)</div>
|
|
</div>
|
|
<div class="stat-card">
|
|
<div class="stat-icon disk"><i class="ri-hard-drive-line"></i></div>
|
|
<div class="stat-value">{disk_percent}%</div>
|
|
<div class="stat-label">磁盘使用 ({disk_used_gb} GB / {disk_total_gb} GB)</div>
|
|
</div>
|
|
</div>
|
|
<div class="info-grid">
|
|
<div class="info-item">
|
|
<div class="info-label">系统运行时间</div>
|
|
<div class="info-value">{uptime_str}</div>
|
|
</div>
|
|
<div class="info-item">
|
|
<div class="info-label">操作系统</div>
|
|
<div class="info-value">{platform.system()} {platform.release()}</div>
|
|
</div>
|
|
<div class="info-item">
|
|
<div class="info-label">Python 版本</div>
|
|
<div class="info-value">{platform.python_version()}</div>
|
|
</div>
|
|
<div class="info-item">
|
|
<div class="info-label">主机名</div>
|
|
<div class="info-value">{platform.node()}</div>
|
|
</div>
|
|
</div>
|
|
</div>
|
|
</div>
|
|
<script>
|
|
setTimeout(() => location.reload(), 30000);
|
|
</script>
|
|
</body>
|
|
</html>"""
|
|
return html
|
|
except Exception as e:
|
|
return f"<p>仪表盘渲染出错：{{e}}</p>"
|
|
|
|
register_plugin_type("DashboardPlugin", DashboardPlugin)
|
|
|
|
|
|
def New():
|
|
return DashboardPlugin()
|