284 lines
9.4 KiB
Python
284 lines
9.4 KiB
Python
|
#!/usr/bin/env python
|
|||
|
|
# -*- coding: utf-8 -*-
|
|||
|
|
|
|||
|
|
"""
|
|||
|
|
VWED任务系统服务管理工具
|
|||
|
|
用于管理VWED服务的启动、停止、状态检查等操作
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
import os
|
|||
|
|
import sys
|
|||
|
|
import time
|
|||
|
|
import signal
|
|||
|
|
import psutil
|
|||
|
|
import argparse
|
|||
|
|
import subprocess
|
|||
|
|
from pathlib import Path
|
|||
|
|
|
|||
|
|
|
|||
|
|
class VWEDServiceManager:
|
|||
|
|
"""VWED服务管理器"""
|
|||
|
|
|
|||
|
|
def __init__(self):
|
|||
|
|
self.service_name = "VWED_Task_System.exe"
|
|||
|
|
self.pid_file = Path("vwed_service.pid")
|
|||
|
|
self.log_file = Path("vwed_service.log")
|
|||
|
|
self.config_file = Path("config.ini")
|
|||
|
|
|
|||
|
|
def start_service(self, headless=True, wait_for_start=True):
|
|||
|
|
"""启动服务"""
|
|||
|
|
if self.is_running():
|
|||
|
|
print("服务已在运行中")
|
|||
|
|
return True
|
|||
|
|
|
|||
|
|
print("启动VWED服务...")
|
|||
|
|
|
|||
|
|
# 构建启动命令
|
|||
|
|
cmd = [self.service_name]
|
|||
|
|
if headless:
|
|||
|
|
cmd.append("--headless")
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
# 启动服务进程
|
|||
|
|
if headless:
|
|||
|
|
# 后台模式，重定向输出到日志文件
|
|||
|
|
with open(self.log_file, 'a', encoding='utf-8') as log_f:
|
|||
|
|
process = subprocess.Popen(
|
|||
|
|
cmd,
|
|||
|
|
stdout=log_f,
|
|||
|
|
stderr=subprocess.STDOUT,
|
|||
|
|
cwd=Path.cwd(),
|
|||
|
|
creationflags=subprocess.CREATE_NO_WINDOW if sys.platform == "win32" else 0
|
|||
|
|
)
|
|||
|
|
else:
|
|||
|
|
# 前台模式
|
|||
|
|
process = subprocess.Popen(cmd, cwd=Path.cwd())
|
|||
|
|
|
|||
|
|
# 记录PID
|
|||
|
|
with open(self.pid_file, 'w') as f:
|
|||
|
|
f.write(str(process.pid))
|
|||
|
|
|
|||
|
|
print(f"服务已启动 (PID: {process.pid})")
|
|||
|
|
|
|||
|
|
if wait_for_start:
|
|||
|
|
# 等待服务完全启动
|
|||
|
|
print("等待服务启动完成...")
|
|||
|
|
for i in range(30): # 最多等待30秒
|
|||
|
|
if self._check_service_port():
|
|||
|
|
print("服务启动成功，可以访问 http://localhost:8000")
|
|||
|
|
return True
|
|||
|
|
time.sleep(1)
|
|||
|
|
|
|||
|
|
print("警告: 服务可能未完全启动，请检查日志文件")
|
|||
|
|
|
|||
|
|
return True
|
|||
|
|
|
|||
|
|
except FileNotFoundError:
|
|||
|
|
print(f"错误: 找不到可执行文件 {self.service_name}")
|
|||
|
|
return False
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f"启动服务失败: {e}")
|
|||
|
|
return False
|
|||
|
|
|
|||
|
|
def stop_service(self, force=False):
|
|||
|
|
"""停止服务"""
|
|||
|
|
if not self.is_running():
|
|||
|
|
print("服务未运行")
|
|||
|
|
self._cleanup_pid_file()
|
|||
|
|
return True
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
pid = self._get_pid()
|
|||
|
|
if pid:
|
|||
|
|
print(f"正在停止服务 (PID: {pid})...")
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
process = psutil.Process(pid)
|
|||
|
|
|
|||
|
|
if not force:
|
|||
|
|
# 优雅关闭
|
|||
|
|
process.terminate()
|
|||
|
|
# 等待进程结束
|
|||
|
|
try:
|
|||
|
|
process.wait(timeout=10)
|
|||
|
|
print("服务已优雅停止")
|
|||
|
|
except psutil.TimeoutExpired:
|
|||
|
|
print("优雅停止超时，强制停止服务...")
|
|||
|
|
process.kill()
|
|||
|
|
print("服务已强制停止")
|
|||
|
|
else:
|
|||
|
|
# 强制关闭
|
|||
|
|
process.kill()
|
|||
|
|
print("服务已强制停止")
|
|||
|
|
|
|||
|
|
except psutil.NoSuchProcess:
|
|||
|
|
print("进程不存在，清理PID文件")
|
|||
|
|
|
|||
|
|
self._cleanup_pid_file()
|
|||
|
|
return True
|
|||
|
|
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f"停止服务失败: {e}")
|
|||
|
|
return False
|
|||
|
|
|
|||
|
|
def restart_service(self, headless=True):
|
|||
|
|
"""重启服务"""
|
|||
|
|
print("重启服务...")
|
|||
|
|
if self.is_running():
|
|||
|
|
if not self.stop_service():
|
|||
|
|
return False
|
|||
|
|
|
|||
|
|
# 等待进程完全停止
|
|||
|
|
time.sleep(2)
|
|||
|
|
|
|||
|
|
return self.start_service(headless)
|
|||
|
|
|
|||
|
|
def service_status(self, show_logs=True):
|
|||
|
|
"""检查服务状态"""
|
|||
|
|
print("检查VWED服务状态...")
|
|||
|
|
print("-" * 50)
|
|||
|
|
|
|||
|
|
# 检查进程状态
|
|||
|
|
if self.is_running():
|
|||
|
|
pid = self._get_pid()
|
|||
|
|
try:
|
|||
|
|
process = psutil.Process(pid)
|
|||
|
|
print(f"✓ 服务正在运行")
|
|||
|
|
print(f" PID: {pid}")
|
|||
|
|
print(f" 内存使用: {process.memory_info().rss / 1024 / 1024:.1f} MB")
|
|||
|
|
print(f" CPU使用率: {process.cpu_percent():.1f}%")
|
|||
|
|
print(f" 运行时间: {time.time() - process.create_time():.0f} 秒")
|
|||
|
|
except psutil.NoSuchProcess:
|
|||
|
|
print("✗ 进程不存在，清理PID文件")
|
|||
|
|
self._cleanup_pid_file()
|
|||
|
|
return False
|
|||
|
|
else:
|
|||
|
|
print("✗ 服务未运行")
|
|||
|
|
|
|||
|
|
# 检查端口状态
|
|||
|
|
if self._check_service_port():
|
|||
|
|
print("✓ 服务端口 8000 可访问")
|
|||
|
|
print(" 访问地址: http://localhost:8000")
|
|||
|
|
else:
|
|||
|
|
print("✗ 服务端口 8000 不可访问")
|
|||
|
|
|
|||
|
|
# 检查配置文件
|
|||
|
|
if self.config_file.exists():
|
|||
|
|
print("✓ 配置文件存在")
|
|||
|
|
else:
|
|||
|
|
print("✗ 配置文件不存在")
|
|||
|
|
|
|||
|
|
# 显示最近日志
|
|||
|
|
if show_logs and self.log_file.exists():
|
|||
|
|
print("\n最近的日志内容:")
|
|||
|
|
print("-" * 30)
|
|||
|
|
try:
|
|||
|
|
with open(self.log_file, 'r', encoding='utf-8') as f:
|
|||
|
|
lines = f.readlines()
|
|||
|
|
for line in lines[-10:]: # 显示最后10行
|
|||
|
|
print(line.strip())
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f"读取日志失败: {e}")
|
|||
|
|
|
|||
|
|
return self.is_running()
|
|||
|
|
|
|||
|
|
def is_running(self):
|
|||
|
|
"""检查服务是否运行中"""
|
|||
|
|
pid = self._get_pid()
|
|||
|
|
if not pid:
|
|||
|
|
return False
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
return psutil.pid_exists(pid)
|
|||
|
|
except:
|
|||
|
|
return False
|
|||
|
|
|
|||
|
|
def _get_pid(self):
|
|||
|
|
"""获取服务PID"""
|
|||
|
|
if not self.pid_file.exists():
|
|||
|
|
return None
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
with open(self.pid_file, 'r') as f:
|
|||
|
|
return int(f.read().strip())
|
|||
|
|
except (ValueError, FileNotFoundError):
|
|||
|
|
return None
|
|||
|
|
|
|||
|
|
def _cleanup_pid_file(self):
|
|||
|
|
"""清理PID文件"""
|
|||
|
|
if self.pid_file.exists():
|
|||
|
|
try:
|
|||
|
|
self.pid_file.unlink()
|
|||
|
|
except:
|
|||
|
|
pass
|
|||
|
|
|
|||
|
|
def _check_service_port(self):
|
|||
|
|
"""检查服务端口是否可访问"""
|
|||
|
|
try:
|
|||
|
|
import socket
|
|||
|
|
with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
|
|||
|
|
s.settimeout(1)
|
|||
|
|
result = s.connect_ex(('localhost', 8000))
|
|||
|
|
return result == 0
|
|||
|
|
except:
|
|||
|
|
return False
|
|||
|
|
|
|||
|
|
|
|||
|
|
def main():
|
|||
|
|
"""主函数"""
|
|||
|
|
parser = argparse.ArgumentParser(description='VWED任务系统服务管理工具')
|
|||
|
|
parser.add_argument('action', choices=['start', 'stop', 'restart', 'status', 'logs'],
|
|||
|
|
help='操作: start(启动), stop(停止), restart(重启), status(状态), logs(查看日志)')
|
|||
|
|
parser.add_argument('--headless', action='store_true',
|
|||
|
|
help='后台模式运行（适用于start和restart）')
|
|||
|
|
parser.add_argument('--force', action='store_true',
|
|||
|
|
help='强制停止服务（适用于stop）')
|
|||
|
|
parser.add_argument('--no-wait', action='store_true',
|
|||
|
|
help='启动后不等待服务完全启动（适用于start）')
|
|||
|
|
|
|||
|
|
args = parser.parse_args()
|
|||
|
|
|
|||
|
|
manager = VWEDServiceManager()
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
if args.action == 'start':
|
|||
|
|
success = manager.start_service(
|
|||
|
|
headless=args.headless,
|
|||
|
|
wait_for_start=not args.no_wait
|
|||
|
|
)
|
|||
|
|
sys.exit(0 if success else 1)
|
|||
|
|
|
|||
|
|
elif args.action == 'stop':
|
|||
|
|
success = manager.stop_service(force=args.force)
|
|||
|
|
sys.exit(0 if success else 1)
|
|||
|
|
|
|||
|
|
elif args.action == 'restart':
|
|||
|
|
success = manager.restart_service(headless=args.headless)
|
|||
|
|
sys.exit(0 if success else 1)
|
|||
|
|
|
|||
|
|
elif args.action == 'status':
|
|||
|
|
is_running = manager.service_status()
|
|||
|
|
sys.exit(0 if is_running else 1)
|
|||
|
|
|
|||
|
|
elif args.action == 'logs':
|
|||
|
|
if manager.log_file.exists():
|
|||
|
|
try:
|
|||
|
|
with open(manager.log_file, 'r', encoding='utf-8') as f:
|
|||
|
|
print(f.read())
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f"读取日志失败: {e}")
|
|||
|
|
sys.exit(1)
|
|||
|
|
else:
|
|||
|
|
print("日志文件不存在")
|
|||
|
|
sys.exit(1)
|
|||
|
|
|
|||
|
|
except KeyboardInterrupt:
|
|||
|
|
print("\n操作被用户中断")
|
|||
|
|
sys.exit(1)
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f"操作失败: {e}")
|
|||
|
|
sys.exit(1)
|
|||
|
|
|
|||
|
|
|
|||
|
|
if __name__ == "__main__":
|
|||
|
|
main()
|