新增简易的8080面板😊

store/@{FutureOSS}/plugin-loader-pro.disabled/core/enhancer.py

@@ -0,0 +1,209 @@
+"""插件加载增强器"""
+from ..circuit.breaker import CircuitBreaker
+from ..recovery.health import HealthChecker
+from ..recovery.auto_fix import AutoRecovery
+from ..utils.logger import ProLogger
+from .config import ProConfig
+
+
+class PluginLoaderEnhancer:
+    """插件加载增强器 - 为现有 plugin-loader 提供高级机制"""
+
+    def __init__(self, plugin_manager, config: ProConfig):
+        self.pm = plugin_manager
+        self.config = config
+        self._breakers = {}
+        self._health_checker = None
+        self._auto_recovery = AutoRecovery(
+            config.auto_recovery.max_attempts,
+            config.auto_recovery.delay
+        )
+        self._enhanced = False
+
+    def enhance(self):
+        """增强 plugin-loader"""
+        if self._enhanced:
+            return
+
+        ProLogger.info("enhancer", "开始增强 plugin-loader...")
+
+        # 1. 为所有插件创建熔断器
+        self._setup_circuit_breakers()
+
+        # 2. 包装启动方法（带重试和容错）
+        self._wrap_start_methods()
+
+        # 3. 启动健康检查
+        self._start_health_check()
+
+        self._enhanced = True
+        ProLogger.info("enhancer", "增强完成，共增强 {} 个插件".format(
+            len(self.pm.plugins)
+        ))
+
+    def _setup_circuit_breakers(self):
+        """为所有插件创建熔断器"""
+        for name, info in self.pm.plugins.items():
+            self._breakers[name] = CircuitBreaker(
+                self.config.circuit_breaker.failure_threshold,
+                self.config.circuit_breaker.recovery_timeout,
+                self.config.circuit_breaker.half_open_requests
+            )
+            ProLogger.debug("enhancer", f"为 {name} 创建熔断器")
+
+    def _wrap_start_methods(self):
+        """包装启动方法"""
+        original_start_all = getattr(self.pm, 'start_all', None)
+        if original_start_all:
+            def wrapped_start_all():
+                self._safe_start_all()
+
+            self.pm.start_all = wrapped_start_all
+            ProLogger.info("enhancer", "已包装 start_all 方法")
+
+        original_init_and_start = getattr(
+            self.pm, 'init_and_start_all', None
+        )
+        if original_init_and_start:
+            def wrapped_init_and_start():
+                self._safe_init_and_start_all()
+
+            self.pm.init_and_start_all = wrapped_init_and_start
+            ProLogger.info("enhancer", "已包装 init_and_start_all 方法")
+
+    def _safe_init_and_start_all(self):
+        """安全的初始化并启动"""
+        ordered = self._get_ordered_plugins()
+
+        # 安全初始化
+        for name in ordered:
+            self._safe_call(name, 'init', '初始化')
+
+        # 安全启动
+        for name in ordered:
+            self._safe_call(name, 'start', '启动')
+
+    def _safe_start_all(self):
+        """安全启动所有"""
+        for name in self.pm.plugins:
+            self._safe_call(name, 'start', '启动')
+
+    def _safe_call(self, name: str, method: str, action: str):
+        """安全调用插件方法（带熔断和重试）"""
+        info = self.pm.plugins.get(name)
+        if not info:
+            return
+
+        instance = info.get("instance")
+        if not instance or not hasattr(instance, method):
+            return
+
+        breaker = self._breakers.get(name)
+        if not breaker:
+            # 没有熔断器，直接调用
+            try:
+                getattr(instance, method)()
+            except Exception as e:
+                ProLogger.error("safe", f"{name} {action}失败: {e}")
+                self._on_plugin_error(name, info, str(e))
+            return
+
+        # 有熔断器，包装调用
+        def do_call():
+            return getattr(instance, method)()
+
+        try:
+            breaker.call(do_call)
+            info["info"].error_count = 0
+            ProLogger.info("safe", f"{name} {action}成功")
+        except Exception as e:
+            ProLogger.error("safe", f"{name} {action}失败: {e}")
+            self._on_plugin_error(name, info, str(e))
+
+    def _on_plugin_error(self, name: str, info: dict, error: str):
+        """插件错误处理"""
+        info["info"].error_count += 1
+        info["info"].last_error = error
+
+        # 自动恢复
+        if self.config.auto_recovery.enabled:
+            plugin_dir = info.get("dir")
+            module = info.get("module")
+
+            if plugin_dir:
+                result = self._auto_recovery.attempt_recovery(
+                    name, plugin_dir, module, info.get("instance")
+                )
+                if result:
+                    info["instance"] = result
+                    info["info"].error_count = 0
+                    ProLogger.info("recovery", f"{name} 自动恢复成功")
+
+    def _start_health_check(self):
+        """启动健康检查"""
+        self._health_checker = HealthChecker(
+            self.config.health_check.interval,
+            self.config.health_check.timeout,
+            self.config.health_check.max_failures
+        )
+
+        for name, info in self.pm.plugins.items():
+            self._health_checker.add_plugin(name, info["instance"])
+
+        self._health_checker.start(
+            on_failure_callback=self._on_health_check_failure
+        )
+        ProLogger.info("enhancer", "健康检查已启动")
+
+    def _on_health_check_failure(self, name: str):
+        """健康检查失败回调"""
+        ProLogger.error("health", f"插件 {name} 健康检查失败")
+
+        info = self.pm.plugins.get(name)
+        if not info:
+            return
+
+        plugin_dir = info.get("dir")
+        module = info.get("module")
+
+        if plugin_dir:
+            result = self._auto_recovery.attempt_recovery(
+                name, plugin_dir, module, info.get("instance")
+            )
+            if result:
+                info["instance"] = result
+                self._health_checker.reset_failure_count(name)
+                ProLogger.info("recovery", f"{name} 健康恢复成功")
+
+    def _get_ordered_plugins(self) -> list[str]:
+        """获取按依赖排序的插件列表"""
+        ordered = []
+        visited = set()
+
+        def visit(name):
+            if name in visited:
+                return
+            visited.add(name)
+
+            info = self.pm.plugins.get(name)
+            if not info:
+                return
+
+            for dep in info["info"].dependencies:
+                clean_dep = dep.rstrip("}")
+                if clean_dep in self.pm.plugins:
+                    visit(clean_dep)
+
+            ordered.append(name)
+
+        for name in self.pm.plugins:
+            visit(name)
+
+        return ordered
+
+    def disable(self):
+        """禁用增强器"""
+        if self._health_checker:
+            self._health_checker.stop()
+        self._enhanced = False
+        ProLogger.info("enhancer", "增强器已禁用")