97ced1b5e6
Key features implemented: - Updated package metadata and dependencies in PKG-INFO, setup files - Added main.py entry point for backward compatibility with README launch method - Enhanced CLI with config options, system info command, and proper signal handling - Implemented minimal PluginManager loading only plugin-loader core plugin - Refactored PluginLoader to follow minimal core design, removed sandbox/isolation complexity - Updated auto-dependency plugin with safer PL injection mechanism and disabled pl_injection - Removed legacy plugin files (firewall, frp_proxy, ftp_server, multi_lang_deploy, ops_toolbox, security_gateway) as functionality moved to core plugin system - Improved gitignore with comprehensive ignore patterns The changes implement a minimal core framework design where only the plugin-loader is directly loaded by the core, with all other plugins managed through the PL injection mechanism, significantly simplifying the architecture.
65 lines
2.0 KiB
Python
65 lines
2.0 KiB
Python
"""熔断器实现"""
|
|
import time
|
|
from typing import Callable, Any
|
|
from .state import CircuitState
|
|
|
|
|
|
class CircuitBreaker:
|
|
"""熔断器"""
|
|
|
|
def __init__(self, failure_threshold: int = 3, recovery_timeout: int = 60, half_open_requests: int = 1):
|
|
self.failure_threshold = failure_threshold
|
|
self.recovery_timeout = recovery_timeout
|
|
self.half_open_requests = half_open_requests
|
|
|
|
self.state = CircuitState.CLOSED
|
|
self.failure_count = 0
|
|
self.success_count = 0
|
|
self.last_failure_time = 0
|
|
self.half_open_calls = 0
|
|
|
|
def call(self, func: Callable, *args, **kwargs) -> Any:
|
|
"""执行调用"""
|
|
if self.state == CircuitState.OPEN:
|
|
if time.time() - self.last_failure_time >= self.recovery_timeout:
|
|
self.state = CircuitState.HALF_OPEN
|
|
self.half_open_calls = 0
|
|
else:
|
|
raise Exception("熔断器已打开，调用被拒绝")
|
|
|
|
try:
|
|
result = func(*args, **kwargs)
|
|
self._on_success()
|
|
return result
|
|
except Exception as e:
|
|
self._on_failure()
|
|
raise
|
|
|
|
def _on_success(self):
|
|
"""成功回调"""
|
|
self.failure_count = 0
|
|
if self.state == CircuitState.HALF_OPEN:
|
|
self.half_open_calls += 1
|
|
if self.half_open_calls >= self.half_open_requests:
|
|
self.state = CircuitState.CLOSED
|
|
self.half_open_calls = 0
|
|
|
|
def _on_failure(self):
|
|
"""失败回调"""
|
|
self.failure_count += 1
|
|
self.last_failure_time = time.time()
|
|
|
|
if self.state == CircuitState.HALF_OPEN:
|
|
self.state = CircuitState.OPEN
|
|
elif self.failure_count >= self.failure_threshold:
|
|
self.state = CircuitState.OPEN
|
|
|
|
def reset(self):
|
|
"""重置熔断器"""
|
|
self.state = CircuitState.CLOSED
|
|
self.failure_count = 0
|
|
self.half_open_calls = 0
|
|
|
|
def get_state(self) -> str:
|
|
return self.state
|