feat: initial HSAP platform
Huaxu Sentinel Active Safety Platform with embedded algorithm code, Docker Compose setup, and vendored dataset scaffolds for clone-and-run. Co-authored-by: Cursor <cursoragent@cursor.com>
This commit is contained in:
0
platform/as_platform/redis/__init__.py
Normal file
0
platform/as_platform/redis/__init__.py
Normal file
51
platform/as_platform/redis/bus.py
Normal file
51
platform/as_platform/redis/bus.py
Normal file
@@ -0,0 +1,51 @@
|
||||
"""Redis 连接与 Job 事件总线。"""
|
||||
from __future__ import annotations
|
||||
|
||||
import json
|
||||
from functools import lru_cache
|
||||
from typing import Any
|
||||
|
||||
from as_platform.config import JOB_QUEUE_KEY, REDIS_URL
|
||||
|
||||
try:
|
||||
import redis
|
||||
except ImportError:
|
||||
redis = None # type: ignore
|
||||
|
||||
|
||||
@lru_cache(maxsize=1)
|
||||
def get_redis():
|
||||
if not REDIS_URL or redis is None:
|
||||
return None
|
||||
return redis.from_url(REDIS_URL, decode_responses=True)
|
||||
|
||||
|
||||
def ping_redis() -> bool:
|
||||
try:
|
||||
r = get_redis()
|
||||
return bool(r and r.ping())
|
||||
except Exception:
|
||||
return False
|
||||
|
||||
|
||||
def publish(event: str, payload: dict[str, Any]) -> None:
|
||||
r = get_redis()
|
||||
if not r:
|
||||
return
|
||||
r.publish("as:events", json.dumps({"event": event, **payload}, ensure_ascii=False))
|
||||
|
||||
|
||||
def push_job(job_id: str) -> None:
|
||||
r = get_redis()
|
||||
if not r:
|
||||
raise RuntimeError("Redis 未配置,无法使用 worker 模式")
|
||||
r.lpush(JOB_QUEUE_KEY, job_id)
|
||||
publish("job.queued", {"job_id": job_id})
|
||||
|
||||
|
||||
def pop_job(timeout: int = 5) -> str | None:
|
||||
r = get_redis()
|
||||
if not r:
|
||||
return None
|
||||
item = r.brpop(JOB_QUEUE_KEY, timeout=timeout)
|
||||
return item[1] if item else None
|
||||
Reference in New Issue
Block a user