83 lines
2.3 KiB
Python
83 lines
2.3 KiB
Python
|
"""分析事件桥 — 消费子进程 event_queue，在主进程写库"""
|
|||
|
|
import logging
|
|||
|
|
import queue
|
|||
|
|
import threading
|
|||
|
|
|
|||
|
|
logger = logging.getLogger("analysis.event_bridge")
|
|||
|
|
|
|||
|
|
_BRIDGE = None
|
|||
|
|
_BRIDGE_LOCK = threading.Lock()
|
|||
|
|
|
|||
|
|
|
|||
|
|
class AnalysisEventBridge(object):
|
|||
|
|
def __init__(self):
|
|||
|
|
self._thread = None
|
|||
|
|
self._running = False
|
|||
|
|
self._queues = []
|
|||
|
|
self._lock = threading.Lock()
|
|||
|
|
|
|||
|
|
def register_queue(self, q):
|
|||
|
|
with self._lock:
|
|||
|
|
if q not in self._queues:
|
|||
|
|
self._queues.append(q)
|
|||
|
|
|
|||
|
|
def unregister_queue(self, q):
|
|||
|
|
with self._lock:
|
|||
|
|
if q in self._queues:
|
|||
|
|
self._queues.remove(q)
|
|||
|
|
|
|||
|
|
def start(self):
|
|||
|
|
if self._running:
|
|||
|
|
return
|
|||
|
|
self._running = True
|
|||
|
|
self._thread = threading.Thread(target=self._loop, name="analysis-event-bridge", daemon=True)
|
|||
|
|
self._thread.start()
|
|||
|
|
logger.info("AnalysisEventBridge 已启动")
|
|||
|
|
|
|||
|
|
def _loop(self):
|
|||
|
|
while self._running:
|
|||
|
|
handled = False
|
|||
|
|
with self._lock:
|
|||
|
|
queues = list(self._queues)
|
|||
|
|
for q in queues:
|
|||
|
|
try:
|
|||
|
|
while True:
|
|||
|
|
msg = q.get_nowait()
|
|||
|
|
self._handle(msg)
|
|||
|
|
handled = True
|
|||
|
|
except queue.Empty:
|
|||
|
|
continue
|
|||
|
|
if not handled:
|
|||
|
|
threading.Event().wait(0.05)
|
|||
|
|
|
|||
|
|
def _handle(self, msg):
|
|||
|
|
if not msg or len(msg) < 2:
|
|||
|
|
return
|
|||
|
|
kind, payload = msg[0], msg[1]
|
|||
|
|
try:
|
|||
|
|
if kind == "event":
|
|||
|
|
self._on_event(payload)
|
|||
|
|
elif kind == "touch":
|
|||
|
|
self._on_touch(payload)
|
|||
|
|
except Exception as e:
|
|||
|
|
logger.exception("事件桥处理失败: %s", e)
|
|||
|
|
|
|||
|
|
def _on_event(self, event):
|
|||
|
|
from app.services.alarm_service import write_alarm, ALARM_EVENT_TYPES
|
|||
|
|
etype = event.get("type", "")
|
|||
|
|
if etype in ALARM_EVENT_TYPES:
|
|||
|
|
write_alarm(event)
|
|||
|
|
|
|||
|
|
def _on_touch(self, payload):
|
|||
|
|
# 已停用：不再写追踪快照
|
|||
|
|
pass
|
|||
|
|
|
|||
|
|
|
|||
|
|
def get_event_bridge():
|
|||
|
|
global _BRIDGE
|
|||
|
|
with _BRIDGE_LOCK:
|
|||
|
|
if _BRIDGE is None:
|
|||
|
|
_BRIDGE = AnalysisEventBridge()
|
|||
|
|
_BRIDGE.start()
|
|||
|
|
return _BRIDGE
|