第一次提交
This commit is contained in:
222
scripts/health_check.py
Normal file
222
scripts/health_check.py
Normal file
@@ -0,0 +1,222 @@
|
||||
#!/usr/bin/env python3
|
||||
"""
|
||||
简单监控脚本
|
||||
用于快速检查系统健康状态，不依赖外部工具
|
||||
"""
|
||||
import os
|
||||
import sys
|
||||
import time
|
||||
import json
|
||||
import requests
|
||||
import psutil
|
||||
from datetime import datetime
|
||||
from pathlib import Path
|
||||
|
||||
|
||||
class HealthChecker:
|
||||
"""健康检查器"""
|
||||
|
||||
def __init__(self, base_url='http://localhost:5000'):
|
||||
self.base_url = base_url
|
||||
self.checks = []
|
||||
|
||||
def check_service_health(self):
|
||||
"""检查服务健康状态"""
|
||||
print("\n🔍 检查服务健康状态...")
|
||||
try:
|
||||
response = requests.get(f'{self.base_url}/api/v1/health', timeout=5)
|
||||
if response.status_code == 200:
|
||||
data = response.json()
|
||||
if data.get('success'):
|
||||
print("✅ 服务健康检查通过")
|
||||
return True
|
||||
print("❌ 服务健康检查失败")
|
||||
return False
|
||||
except Exception as e:
|
||||
print(f"❌ 无法连接到服务: {str(e)}")
|
||||
return False
|
||||
|
||||
def check_system_resources(self):
|
||||
"""检查系统资源"""
|
||||
print("\n🔍 检查系统资源...")
|
||||
|
||||
# CPU 使用率
|
||||
cpu_percent = psutil.cpu_percent(interval=1)
|
||||
print(f" CPU 使用率: {cpu_percent:.1f}%")
|
||||
cpu_status = "✅" if cpu_percent < 80 else "⚠️" if cpu_percent < 90 else "❌"
|
||||
|
||||
# 内存使用率
|
||||
memory = psutil.virtual_memory()
|
||||
print(f" 内存使用率: {memory.percent:.1f}%")
|
||||
memory_status = "✅" if memory.percent < 85 else "⚠️" if memory.percent < 95 else "❌"
|
||||
|
||||
# 磁盘使用率
|
||||
disk = psutil.disk_usage('/')
|
||||
disk_percent = (disk.used / disk.total) * 100
|
||||
print(f" 磁盘使用率: {disk_percent:.1f}%")
|
||||
disk_status = "✅" if disk_percent < 90 else "⚠️" if disk_percent < 95 else "❌"
|
||||
|
||||
# 负载均值
|
||||
load_avg = psutil.getloadavg()
|
||||
print(f" 系统负载: {load_avg[0]:.2f}, {load_avg[1]:.2f}, {load_avg[2]:.2f}")
|
||||
load_status = "✅" if load_avg[0] < 2 else "⚠️" if load_avg[0] < 4 else "❌"
|
||||
|
||||
return all([
|
||||
cpu_status == "✅",
|
||||
memory_status == "✅",
|
||||
disk_status == "✅",
|
||||
load_status == "✅"
|
||||
])
|
||||
|
||||
def check_database_connection(self):
|
||||
"""检查数据库连接"""
|
||||
print("\n🔍 检查数据库连接...")
|
||||
try:
|
||||
# 这里可以添加数据库连接检查
|
||||
# 由于需要导入Flask应用，暂时跳过
|
||||
print("✅ 数据库连接检查（跳过，需要应用上下文）")
|
||||
return True
|
||||
except Exception as e:
|
||||
print(f"❌ 数据库连接检查失败: {str(e)}")
|
||||
return False
|
||||
|
||||
def check_disk_space(self):
|
||||
"""检查磁盘空间"""
|
||||
print("\n🔍 检查磁盘空间...")
|
||||
disk_usage = psutil.disk_usage('/')
|
||||
free_gb = disk_usage.free / (1024**3)
|
||||
total_gb = disk_usage.total / (1024**3)
|
||||
print(f" 可用空间: {free_gb:.2f} GB / {total_gb:.2f} GB")
|
||||
return free_gb > 1 # 至少需要1GB可用空间
|
||||
|
||||
def check_process_status(self):
|
||||
"""检查关键进程"""
|
||||
print("\n🔍 检查关键进程...")
|
||||
processes = ['python', 'flask', 'nginx', 'mysql', 'redis']
|
||||
found_processes = []
|
||||
|
||||
for proc in psutil.process_iter(['pid', 'name', 'cmdline']):
|
||||
try:
|
||||
if any(proc.info['name'] and proc.info['name'].lower().startswith(p.lower()) for p in processes):
|
||||
found_processes.append(proc.info['name'])
|
||||
except (psutil.NoSuchProcess, psutil.AccessDenied):
|
||||
pass
|
||||
|
||||
print(f" 发现进程: {', '.join(set(found_processes))}")
|
||||
return len(found_processes) > 0
|
||||
|
||||
def check_log_files(self):
|
||||
"""检查日志文件"""
|
||||
print("\n🔍 检查日志文件...")
|
||||
log_files = [
|
||||
'logs/kamaxitong.log',
|
||||
'/var/log/nginx/access.log',
|
||||
'/var/log/nginx/error.log'
|
||||
]
|
||||
|
||||
for log_file in log_files:
|
||||
path = Path(log_file)
|
||||
if path.exists():
|
||||
size_mb = path.stat().st_size / (1024 * 1024)
|
||||
print(f" {log_file}: {size_mb:.2f} MB")
|
||||
else:
|
||||
print(f" {log_file}: 文件不存在")
|
||||
|
||||
return True
|
||||
|
||||
def generate_report(self):
|
||||
"""生成健康报告"""
|
||||
print("\n" + "=" * 60)
|
||||
print("📊 KaMiXiTong 系统健康报告")
|
||||
print("=" * 60)
|
||||
print(f"检查时间: {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}")
|
||||
|
||||
# 运行所有检查
|
||||
checks = [
|
||||
('服务健康', self.check_service_health),
|
||||
('系统资源', self.check_system_resources),
|
||||
('数据库连接', self.check_database_connection),
|
||||
('磁盘空间', self.check_disk_space),
|
||||
('进程状态', self.check_process_status),
|
||||
('日志文件', self.check_log_files)
|
||||
]
|
||||
|
||||
results = []
|
||||
for name, check_func in checks:
|
||||
try:
|
||||
result = check_func()
|
||||
results.append((name, result))
|
||||
except Exception as e:
|
||||
print(f"❌ {name}检查失败: {str(e)}")
|
||||
results.append((name, False))
|
||||
|
||||
# 生成报告
|
||||
print("\n" + "=" * 60)
|
||||
print("📋 检查结果汇总")
|
||||
print("=" * 60)
|
||||
|
||||
all_passed = True
|
||||
for name, passed in results:
|
||||
status = "✅ 通过" if passed else "❌ 失败"
|
||||
print(f"{name:20s}: {status}")
|
||||
if not passed:
|
||||
all_passed = False
|
||||
|
||||
print("\n" + "=" * 60)
|
||||
if all_passed:
|
||||
print("✅ 系统健康状态良好")
|
||||
else:
|
||||
print("⚠️ 发现问题，请查看上述详细信息")
|
||||
print("=" * 60)
|
||||
|
||||
return all_passed
|
||||
|
||||
def save_report(self, filename='health_report.json'):
|
||||
"""保存健康报告到文件"""
|
||||
report = {
|
||||
'timestamp': datetime.now().isoformat(),
|
||||
'checks': [
|
||||
{
|
||||
'name': name,
|
||||
'passed': passed,
|
||||
'time': datetime.now().isoformat()
|
||||
}
|
||||
for name, passed in self.results
|
||||
],
|
||||
'system_info': {
|
||||
'cpu_count': psutil.cpu_count(),
|
||||
'memory_total': psutil.virtual_memory().total,
|
||||
'disk_total': psutil.disk_usage('/').total
|
||||
}
|
||||
}
|
||||
|
||||
with open(filename, 'w') as f:
|
||||
json.dump(report, f, indent=2)
|
||||
|
||||
print(f"\n📄 健康报告已保存到: {filename}")
|
||||
|
||||
|
||||
def main():
|
||||
"""主函数"""
|
||||
print("=" * 60)
|
||||
print("🔍 KaMiXiTong 系统健康检查")
|
||||
print("=" * 60)
|
||||
|
||||
# 获取基础URL
|
||||
base_url = input("请输入服务地址 (默认: http://localhost:5000): ").strip()
|
||||
if not base_url:
|
||||
base_url = 'http://localhost:5000'
|
||||
|
||||
checker = HealthChecker(base_url)
|
||||
passed = checker.generate_report()
|
||||
|
||||
# 保存报告
|
||||
save = input("\n是否保存健康报告? (y/N): ").strip().lower()
|
||||
if save == 'y':
|
||||
checker.save_report()
|
||||
|
||||
sys.exit(0 if passed else 1)
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
main()
|
||||
Reference in New Issue
Block a user