207 lines
8.8 KiB
Python
207 lines
8.8 KiB
Python
# -*- coding: utf-8 -*-
|
||
"""
|
||
事件数据模型
|
||
─────────────────────────────────────────────────────────────────
|
||
Event —— 核心事件对象，贯穿整个生命周期
|
||
EventResult —— 处理器返回结果
|
||
EventState —— 事件处理状态机
|
||
─────────────────────────────────────────────────────────────────
|
||
"""
|
||
|
||
from __future__ import annotations
|
||
|
||
import time
|
||
import uuid
|
||
from dataclasses import dataclass, field
|
||
from datetime import datetime
|
||
from enum import Enum
|
||
from typing import Any, Dict, List, Optional, Union
|
||
|
||
from uas.event.event_type import EventType, EventPriority, event_registry
|
||
|
||
|
||
# ─────────────────────────────────────────────────────────────────
|
||
# 事件状态
|
||
# ─────────────────────────────────────────────────────────────────
|
||
|
||
class EventState(Enum):
|
||
CREATED = "created" # 已创建，未提交
|
||
PENDING = "pending" # 已提交，等待处理
|
||
PROCESSING = "processing" # 处理中
|
||
HANDLED = "handled" # 已处理完成
|
||
CANCELLED = "cancelled" # 已取消
|
||
FAILED = "failed" # 处理失败
|
||
IGNORED = "ignored" # 无处理器，已忽略
|
||
|
||
|
||
# ─────────────────────────────────────────────────────────────────
|
||
# 事件处理结果
|
||
# ─────────────────────────────────────────────────────────────────
|
||
|
||
@dataclass
|
||
class EventResult:
|
||
"""单个处理器的执行结果"""
|
||
handler_name: str
|
||
success: bool
|
||
data: Any = None
|
||
error: str = ""
|
||
elapsed_ms: float = 0.0
|
||
|
||
def __repr__(self):
|
||
status = "✅" if self.success else "❌"
|
||
return f"EventResult({status} {self.handler_name} {self.elapsed_ms:.1f}ms)"
|
||
|
||
|
||
# ─────────────────────────────────────────────────────────────────
|
||
# 核心事件对象
|
||
# ─────────────────────────────────────────────────────────────────
|
||
|
||
@dataclass
|
||
class Event:
|
||
"""
|
||
核心事件对象
|
||
─────────────────────────────────────────────────────────────
|
||
字段说明：
|
||
event_type —— 事件类型（EventType 枚举 或 自定义字符串）
|
||
source —— 事件来源（设备ID、模块名等）
|
||
params —— 事件携带参数（任意字典）
|
||
priority —— 优先级（默认从注册表读取）
|
||
timestamp —— 发生时间（Unix 时间戳，自动填充）
|
||
event_id —— 唯一标识（UUID，自动生成）
|
||
─────────────────────────────────────────────────────────────
|
||
"""
|
||
|
||
event_type: Union[EventType, str]
|
||
source: str = "unknown"
|
||
params: Dict[str, Any] = field(default_factory=dict)
|
||
priority: Optional[EventPriority] = None
|
||
message: str = ""
|
||
|
||
# 自动填充字段
|
||
event_id: str = field(default_factory=lambda: str(uuid.uuid4()))
|
||
timestamp: float = field(default_factory=time.time)
|
||
state: EventState = field(default=EventState.CREATED)
|
||
|
||
# 运行时字段（不参与构造）
|
||
results: List[EventResult] = field(default_factory=list)
|
||
_cancelled: bool = field(default=False, repr=False)
|
||
_submit_time: Optional[float] = field(default=None, repr=False)
|
||
_done_time: Optional[float] = field(default=None, repr=False)
|
||
|
||
def __post_init__(self):
|
||
# 若未指定优先级，从注册表读取
|
||
if self.priority is None:
|
||
self.priority = event_registry.get_priority(self.type_key)
|
||
|
||
# ── 属性 ──────────────────────────────────────────────────────
|
||
|
||
@property
|
||
def type_key(self) -> str:
|
||
"""统一返回字符串 key"""
|
||
return (self.event_type.value
|
||
if isinstance(self.event_type, EventType)
|
||
else str(self.event_type))
|
||
|
||
@property
|
||
def datetime(self) -> datetime:
|
||
return datetime.fromtimestamp(self.timestamp)
|
||
|
||
@property
|
||
def datetime_str(self) -> str:
|
||
return self.datetime.strftime("%Y-%m-%d %H:%M:%S.%f")[:-3]
|
||
|
||
@property
|
||
def is_cancelled(self) -> bool:
|
||
return self._cancelled
|
||
|
||
@property
|
||
def elapsed_ms(self) -> float:
|
||
"""从提交到完成的总耗时（ms）"""
|
||
if self._submit_time and self._done_time:
|
||
return (self._done_time - self._submit_time) * 1000
|
||
return 0.0
|
||
|
||
@property
|
||
def success_count(self) -> int:
|
||
return sum(1 for r in self.results if r.success)
|
||
|
||
@property
|
||
def failed_count(self) -> int:
|
||
return sum(1 for r in self.results if not r.success)
|
||
|
||
# ── 状态操作 ──────────────────────────────────────────────────
|
||
|
||
def cancel(self) -> bool:
|
||
"""取消事件（仅 cancellable 类型可取消）"""
|
||
if event_registry.is_cancellable(self.type_key):
|
||
self._cancelled = True
|
||
self.state = EventState.CANCELLED
|
||
return True
|
||
return False
|
||
|
||
def mark_pending(self):
|
||
self.state = EventState.PENDING
|
||
self._submit_time = time.time()
|
||
|
||
def mark_processing(self):
|
||
self.state = EventState.PROCESSING
|
||
|
||
def mark_done(self):
|
||
self._done_time = time.time()
|
||
if self._cancelled:
|
||
self.state = EventState.CANCELLED
|
||
elif any(not r.success for r in self.results):
|
||
self.state = EventState.FAILED
|
||
elif not self.results:
|
||
self.state = EventState.IGNORED
|
||
else:
|
||
self.state = EventState.HANDLED
|
||
|
||
def add_result(self, result: EventResult):
|
||
self.results.append(result)
|
||
|
||
# ── 序列化 ────────────────────────────────────────────────────
|
||
|
||
def to_dict(self) -> dict:
|
||
return {
|
||
"event_id": self.event_id,
|
||
"event_type": self.type_key,
|
||
"source": self.source,
|
||
"message": self.message,
|
||
"params": self.params,
|
||
"priority": self.priority.name if self.priority else "NORMAL",
|
||
"timestamp": self.timestamp,
|
||
"datetime": self.datetime_str,
|
||
"state": self.state.value,
|
||
"elapsed_ms": self.elapsed_ms,
|
||
"results": [str(r) for r in self.results],
|
||
}
|
||
|
||
def __repr__(self):
|
||
return (f"Event(id={self.event_id[:8]}, type={self.type_key!r}, "
|
||
f"src={self.source!r}, state={self.state.value}, "
|
||
f"priority={self.priority.name if self.priority else 'N/A'})")
|
||
|
||
|
||
# ─────────────────────────────────────────────────────────────────
|
||
# 便捷工厂函数
|
||
# ─────────────────────────────────────────────────────────────────
|
||
|
||
def make_event(
|
||
event_type: Union[EventType, str],
|
||
source: str = "system",
|
||
message: str = "",
|
||
**params
|
||
) -> Event:
|
||
"""
|
||
快速创建事件
|
||
示例：
|
||
e = make_event(EventType.DEVICE_ERROR, source="sensor-01",
|
||
message="温度传感器异常", code="E001", temp=98.5)
|
||
"""
|
||
return Event(
|
||
event_type = event_type,
|
||
source = source,
|
||
message = message,
|
||
params = params,
|
||
) |