AskaEth
e6875f0b4b
- 13-service async plugin framework - Textual TUI with CLI fallback - Plugin hot-reload + permission system - Web management panel (aiohttp) - Bridge-based inter-module communication - 10 regression tests Fixes applied: - PBKDF2-SHA256 auth (was plain SHA256) - Auth bypass removed (was allow-all on fail) - Bare excepts replaced with logged errors - CatFramework/DreamSu -> SenSu naming unified - ServiceManager: health checks + startup_order - Env var credentials (SENSU_ADMIN_PASSWORD etc)
80 lines
2.9 KiB
Python
80 lines
2.9 KiB
Python
#!/usr/bin/env python3
|
|
# -*- coding: utf-8 -*-
|
|
|
|
import logging
|
|
from typing import Dict, Any, List, Optional
|
|
import asyncio
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
class ServiceManager:
|
|
"""服务管理器 - 管理所有框架服务的生命周期"""
|
|
|
|
def __init__(self):
|
|
self.services: Dict[str, Any] = {}
|
|
self._startup_order: List[str] = []
|
|
self._health_checks: Dict[str, Any] = {}
|
|
logger.debug("ServiceManager初始化完成")
|
|
|
|
def register_service(self, name: str, service_instance,
|
|
depends_on: Optional[List[str]] = None,
|
|
health_check=None):
|
|
"""注册服务"""
|
|
if name in self.services:
|
|
logger.warning(f"服务 {name} 已存在，将被覆盖")
|
|
self.services[name] = service_instance
|
|
self._startup_order.append(name)
|
|
if health_check:
|
|
self._health_checks[name] = health_check
|
|
logger.debug(f"服务 {name} 注册成功")
|
|
|
|
def get_service(self, name: str):
|
|
"""获取服务"""
|
|
service = self.services.get(name)
|
|
if not service:
|
|
raise ValueError(f"服务 {name} 未找到。可用: {list(self.services.keys())}")
|
|
return service
|
|
|
|
def has_service(self, name: str) -> bool:
|
|
"""检查服务是否已注册"""
|
|
return name in self.services
|
|
|
|
async def check_health(self, name: str = None) -> Dict[str, bool]:
|
|
"""健康检查"""
|
|
results = {}
|
|
names = [name] if name else list(self._health_checks.keys())
|
|
for n in names:
|
|
if n in self._health_checks:
|
|
try:
|
|
r = self._health_checks[n]()
|
|
if asyncio.iscoroutine(r): r = await r
|
|
results[n] = bool(r)
|
|
except Exception as e:
|
|
logger.warning(f"服务 {n} 健康检查失败: {e}")
|
|
results[n] = False
|
|
return results
|
|
|
|
@property
|
|
def startup_order(self) -> List[str]:
|
|
return list(self._startup_order)
|
|
|
|
def shutdown_all(self):
|
|
"""关闭所有服务"""
|
|
logger.info("开始关闭所有服务")
|
|
for name in reversed(self._startup_order):
|
|
service = self.services.get(name)
|
|
if service and hasattr(service, 'shutdown'):
|
|
try:
|
|
if asyncio.iscoroutinefunction(service.shutdown):
|
|
try:
|
|
loop = asyncio.get_running_loop()
|
|
loop.create_task(service.shutdown())
|
|
except RuntimeError:
|
|
service.shutdown()
|
|
else:
|
|
service.shutdown()
|
|
logger.debug(f"服务 {name} 关闭成功")
|
|
except Exception as e:
|
|
logger.error(f"关闭服务 {name} 时出错: {e}")
|
|
logger.info("所有服务关闭完成")
|