diff --git a/alembic/versions/comparison_updated_at.py b/alembic/versions/comparison_updated_at.py new file mode 100644 index 0000000..8e22ea9 --- /dev/null +++ b/alembic/versions/comparison_updated_at.py @@ -0,0 +1,50 @@ +"""comparison_record 加 updated_at 列(报警水位)+ 回填现有行 + 索引 + +新增 updated_at:server_default + onupdate = func.now()(DB 时钟)。比价失败报警 worker 用它做 +单调水位(WHERE updated_at > watermark)。加列后回填现有行 = created_at,避免冷启动 max(updated_at) +为 NULL;再置 NOT NULL + 建索引 ix_comparison_updated(水位查询按它)。batch 模式兼容 SQLite。 + +Revision ID: comparison_updated_at +Revises: deepseek_v4_flash_price +Create Date: 2026-08-04 00:00:00.000000 +""" + +from collections.abc import Sequence + +import sqlalchemy as sa + +from alembic import op + +revision: str = "comparison_updated_at" +down_revision: str | Sequence[str] | None = "deepseek_v4_flash_price" +branch_labels: str | Sequence[str] | None = None +depends_on: str | Sequence[str] | None = None + + +def upgrade() -> None: + # 1) 先加可空列(不带 default,避免各库对 add-column-with-default 的差异) + with op.batch_alter_table("comparison_record", schema=None) as batch_op: + batch_op.add_column(sa.Column("updated_at", sa.DateTime(timezone=True), nullable=True)) + + # 2) 回填现有行 = created_at(全新环境表为空,回填 no-op) + op.get_bind().execute( + sa.text( + "UPDATE comparison_record SET updated_at = created_at WHERE updated_at IS NULL" + ) + ) + + # 3) 置 NOT NULL + server_default + 建索引 + with op.batch_alter_table("comparison_record", schema=None) as batch_op: + batch_op.alter_column( + "updated_at", + existing_type=sa.DateTime(timezone=True), + nullable=False, + server_default=sa.text("CURRENT_TIMESTAMP"), + ) + batch_op.create_index("ix_comparison_updated", ["updated_at"], unique=False) + + +def downgrade() -> None: + with op.batch_alter_table("comparison_record", schema=None) as batch_op: + batch_op.drop_index("ix_comparison_updated") + batch_op.drop_column("updated_at") diff --git a/app/core/compare_alert_worker.py b/app/core/compare_alert_worker.py new file mode 100644 index 0000000..7462c43 --- /dev/null +++ b/app/core/compare_alert_worker.py @@ -0,0 +1,312 @@ +"""比价失败报警后台任务:周期扫 comparison_record 新落定记录 → 规则命中 → 飞书汇总。 + +结构仿 heartbeat_monitor_worker(单实例文件锁 + asyncio 轮询 + 优雅退出);发送与 DB 全同步, +放 asyncio.to_thread。水位存 app_config(key=compare_alert.last_watermark,值=上次处理的最大 +updated_at ISO 串),查询用 updated_at 自身比较、规避时区。见 spec 第 5/6 节。 +""" +from __future__ import annotations + +import asyncio +import contextlib +import logging +import os +import time +from collections.abc import Iterator +from dataclasses import replace as _dc_replace +from datetime import datetime +from pathlib import Path + +from sqlalchemy import func, select +from sqlalchemy.exc import SQLAlchemyError + +from app.core.config import settings +from app.core.rewards import CN_TZ +from app.db.session import SessionLocal +from app.integrations import feishu_notifier +from app.models.app_config import AppConfig +from app.models.comparison import ComparisonRecord +from app.models.user import User +from app.services import trace_stuck +from app.services.compare_alert import ( + AlertHit, + classify_cancelled_fallback, + classify_record, + make_hit, +) +from app.services.compare_alert_format import format_alert_card + +logger = logging.getLogger("shagua.compare_alert") + +WATERMARK_KEY = "compare_alert.last_watermark" +_LOCK_PATH = Path(__file__).resolve().parents[2] / "data" / "compare_alert.lock" + + +def _read_watermark(db) -> datetime | None: + row = db.get(AppConfig, WATERMARK_KEY) + if row is None or not row.value: + return None + try: + return datetime.fromisoformat(row.value) + except (ValueError, TypeError): + return None + + +def _write_watermark(db, value: datetime) -> None: + iso = value.isoformat() + row = db.get(AppConfig, WATERMARK_KEY) + if row is None: + db.add(AppConfig(key=WATERMARK_KEY, value=iso, updated_by_admin_id=None)) + else: + row.value = iso + db.commit() + + +def _send_card(card: dict) -> None: + """发飞书交互卡片(webhook 为空则只打日志、不外发)。失败抛 FeishuNotifyError 由调用方处理。""" + webhook = settings.COMPARE_ALERT_FEISHU_WEBHOOK + if not webhook: + title = card.get("header", {}).get("title", {}).get("content", "") + logger.info("[compare-alert] webhook 未配置,仅打印: title=%s", title) + return + feishu_notifier.send_feishu_card( + webhook, card, timeout=settings.COMPARE_ALERT_FEISHU_TIMEOUT_SEC + ) + + +def _trace_dir(base: Path, trace_url: str | None) -> Path | None: + """URL → trace 目录 Path;trace_url 缺失/无法解析 → None。""" + name = trace_stuck.dir_name_from_trace_url(trace_url) + if not name: + return None + return base / name + + +def _fmt_stuck(sp: trace_stuck.StuckPoint) -> str: + """StuckPoint → 「平台·环节 110帧/32s」;stuck_ms 为 None 时省略时长。""" + s = f"{sp.label()} {sp.frames}帧" + if sp.stuck_ms is not None: + s += f"/{round(sp.stuck_ms / 1000)}s" + return s + + +def build_hits( + records: list, + *, + work_log_dir: str, + stuck_threshold: int, + max_tail: int, + max_trace_reads: int, + cancelled_ms_threshold: int, + cancelled_step_threshold: int, + timeout_keywords: tuple[str, ...], + unrecognized_keywords: tuple[str, ...], + biz_exclude_keywords: tuple[str, ...], +) -> list[AlertHit]: + """编排:cancelled 走 trace 优先(读到确认没卡则不报,读不到回退保底);failed 附卡点。 + + trace 读取限量 max_trace_reads 次/轮;任何 trace 异常都在 trace_stuck 内部降级为 + 「读不到」,cancelled 因而回退保底、failed 不附卡点,绝不影响报警发送。 + """ + base = Path(work_log_dir) if work_log_dir else None + reads = 0 + hits: list = [] + for rec in records: + if rec.status == "cancelled": + res = None + if base is not None and reads < max_trace_reads: + td = _trace_dir(base, rec.trace_url) + if td is not None: + res = trace_stuck.read_stuck_points( + td, threshold=stuck_threshold, max_tail=max_tail + ) + reads += 1 + if res is not None and res.readable: + if res.points: + stuck = "、".join(_fmt_stuck(sp) for sp in res.points) + hit = _dc_replace(make_hit(rec, "T5", "深度放弃"), stuck_point=stuck) + else: + hit = None # 读到且确认没卡 → 不报 + else: + hit = classify_cancelled_fallback( + rec, + cancelled_ms_threshold=cancelled_ms_threshold, + cancelled_step_threshold=cancelled_step_threshold, + ) + if hit is not None: + hits.append(hit) + else: + hit = classify_record( + rec, + cancelled_ms_threshold=cancelled_ms_threshold, + cancelled_step_threshold=cancelled_step_threshold, + timeout_keywords=timeout_keywords, + unrecognized_keywords=unrecognized_keywords, + biz_exclude_keywords=biz_exclude_keywords, + ) + if ( + hit is not None + and hit.alert_type in ("T1", "T2", "T6") + and base is not None + and reads < max_trace_reads + ): + td = _trace_dir(base, rec.trace_url) + if td is not None: + sp = trace_stuck.last_step(td) + reads += 1 + if sp is not None: + # failed 是「末帧停在哪」:last_step 返回的 frames 是该平台总帧数(非"卡住"帧数)、 + # stuck_ms=None,带上帧数/时长会误导,故只用 label() 显示环节 + hit = _dc_replace(hit, stuck_point=sp.label()) + if hit is not None: + hits.append(hit) + return hits + + +def _scan_and_alert() -> None: + """一轮:读水位 → 查有更新记录 → 规则 → 有命中发飞书 → 成功推进水位。同步,放 to_thread 调。""" + with SessionLocal() as db: + watermark = _read_watermark(db) + if watermark is None: + # 冷启动:水位 = 当前 max(updated_at),不回溯历史失败。空表则本轮不建水位、下轮再说—— + # 不用 datetime.now():那是本地时钟,与 SQLite 的 updated_at(UTC CURRENT_TIMESTAMP) + # 不同源、会差 8h,导致新记录永远追不上水位。只用 DB 产出的 updated_at 值。 + max_updated = db.scalar(select(func.max(ComparisonRecord.updated_at))) + if max_updated is None: + logger.info("[compare-alert] 冷启动:表空,待有记录后再建水位") + return + _write_watermark(db, max_updated) + logger.info("[compare-alert] 冷启动,水位初始化=%s", max_updated) + return + + records = list( + db.scalars( + select(ComparisonRecord) + .where(ComparisonRecord.updated_at > watermark) + .order_by(ComparisonRecord.updated_at.asc()) + ) + ) + if not records: + return + batch_max = max(r.updated_at for r in records) + + hits = build_hits( + records, + work_log_dir=settings.COMPARE_ALERT_PRICEBOT_WORK_LOG_DIR, + stuck_threshold=settings.COMPARE_ALERT_STUCK_FRAME_THRESHOLD, + max_tail=settings.COMPARE_ALERT_TRACE_MAX_TAIL_FRAMES, + max_trace_reads=settings.COMPARE_ALERT_TRACE_MAX_RECORDS, + cancelled_ms_threshold=settings.COMPARE_ALERT_CANCELLED_MS_THRESHOLD, + cancelled_step_threshold=settings.COMPARE_ALERT_CANCELLED_STEP_THRESHOLD, + timeout_keywords=settings.compare_alert_timeout_keywords, + unrecognized_keywords=settings.compare_alert_unrecognized_keywords, + biz_exclude_keywords=settings.compare_alert_biz_exclude_keywords, + ) + + if hits or settings.COMPARE_ALERT_SEND_EMPTY: + label = datetime.now(CN_TZ).strftime("%Y-%m-%d %H:%M") + interval_min = max(1, settings.COMPARE_ALERT_SCAN_INTERVAL_SEC // 60) + + if hits: + # join User 取手机号 + uids = {h.user_id for h in hits if h.user_id is not None} + if uids: + users = db.scalars(select(User).where(User.id.in_(uids))) + phone_map = {u.id: u.phone for u in users} + else: + phone_map = {} + else: + phone_map = {} + + card = format_alert_card( + hits, + window_label=label, + phone_map=phone_map, + interval_min=interval_min, + max_detail_per_type=settings.COMPARE_ALERT_MAX_DETAIL_PER_TYPE, + max_total=settings.COMPARE_ALERT_MAX_TOTAL, + ) + + try: + _send_card(card) + except feishu_notifier.FeishuNotifyError: + logger.warning("[compare-alert] 发送失败,水位不推进、下轮补发", exc_info=True) + return # 不推进水位 + + _write_watermark(db, batch_max) + if hits: + logger.info("[compare-alert] 本轮命中 %d 条,水位推进到 %s", len(hits), batch_max) + + +# ---- 单实例锁 + 轮询循环(结构同 heartbeat_monitor_worker)--------------------- +def _touch_lock() -> None: + with contextlib.suppress(FileNotFoundError): + os.utime(_LOCK_PATH, None) + + +@contextlib.contextmanager +def _single_instance_lock(stale_after_sec: int) -> Iterator[bool]: + _LOCK_PATH.parent.mkdir(parents=True, exist_ok=True) + fd: int | None = None + try: + try: + fd = os.open(str(_LOCK_PATH), os.O_CREAT | os.O_EXCL | os.O_WRONLY) + except FileExistsError: + try: + age = time.time() - _LOCK_PATH.stat().st_mtime + except FileNotFoundError: + age = stale_after_sec + 1 + if age > stale_after_sec: + with contextlib.suppress(FileNotFoundError): + _LOCK_PATH.unlink() + try: + fd = os.open(str(_LOCK_PATH), os.O_CREAT | os.O_EXCL | os.O_WRONLY) + except FileExistsError: + fd = None + if fd is None: + yield False + return + os.write(fd, f"pid={os.getpid()} started_at={int(time.time())}\n".encode("ascii")) + yield True + finally: + if fd is not None: + os.close(fd) + with contextlib.suppress(FileNotFoundError): + _LOCK_PATH.unlink() + + +async def _run_loop() -> None: + interval = max(60, int(settings.COMPARE_ALERT_SCAN_INTERVAL_SEC)) + lock_stale_after = max(interval * 3, 600) + with _single_instance_lock(lock_stale_after) as lock_acquired: + if not lock_acquired: + logger.warning("compare-alert skipped: another worker owns lock") + return + logger.info("compare-alert worker started interval=%ss", interval) + try: + while True: + try: + _touch_lock() + await asyncio.to_thread(_scan_and_alert) + except SQLAlchemyError: + logger.exception("compare-alert db error") + except Exception: # noqa: BLE001 - 后台任务不因单次异常退出 + logger.exception("compare-alert unexpected error") + await asyncio.sleep(interval) + except asyncio.CancelledError: + logger.info("compare-alert worker stopped") + raise + + +def start_compare_alert_worker() -> asyncio.Task | None: + if not settings.COMPARE_ALERT_ENABLED: + logger.info("compare-alert worker disabled") + return None + return asyncio.create_task(_run_loop(), name="compare-alert-worker") + + +async def stop_compare_alert_worker(task: asyncio.Task | None) -> None: + if task is None: + return + task.cancel() + with contextlib.suppress(asyncio.CancelledError): + await task diff --git a/app/core/config.py b/app/core/config.py index 3a79762..c72d76e 100644 --- a/app/core/config.py +++ b/app/core/config.py @@ -132,6 +132,26 @@ class Settings(BaseSettings): HEARTBEAT_TIMEOUT_MINUTES: int = 60 # 多久没心跳算掉线(1 小时,避免短暂离线误判被杀) HEARTBEAT_SCAN_INTERVAL_SEC: int = 60 # 扫描周期 + # ===== 比价失败报警(常驻 worker 周期扫 comparison_record → 飞书汇总)===== + COMPARE_ALERT_ENABLED: bool = False # 总开关(默认关;启用用 .env COMPARE_ALERT_ENABLED=true 覆盖,别改这默认值);关时 worker 不启动 + COMPARE_ALERT_SCAN_INTERVAL_SEC: int = 900 # 扫描间隔(默认 15min,可配 1800=30min) + COMPARE_ALERT_FEISHU_WEBHOOK: str = "" # 群机器人 webhook(敏感,放 .env 别硬编码进代码);空则 worker 仅打日志不外发 + COMPARE_ALERT_FEISHU_TIMEOUT_SEC: float = 10.0 # 飞书 POST 读/连超时 + COMPARE_ALERT_CANCELLED_MS_THRESHOLD: int = 90000 # T5 耗时阈值(ms) + COMPARE_ALERT_CANCELLED_STEP_THRESHOLD: int = 30 # T5 步数阈值 + COMPARE_ALERT_TIMEOUT_KEYWORDS: str = "超时,启动,加载" # T2 关键词(逗号分隔) + COMPARE_ALERT_UNRECOGNIZED_KEYWORDS: str = "未识别" # T6 关键词 + COMPARE_ALERT_BIZ_EXCLUDE_KEYWORDS: str = "未找到,打烊,起送,门店,店内,不配送,这些菜,未入驻,休息" # T1 information 业务词排除 + COMPARE_ALERT_MAX_DETAIL_PER_TYPE: int = 20 # 单类型明细截断 + COMPARE_ALERT_MAX_TOTAL: int = 50 # 本期总命中截断(超则只给计数) + COMPARE_ALERT_SEND_EMPTY: bool = False # 无命中是否发「本期无异常」简讯 + + # ===== 卡死定位(读 pricebot trace 末段判原地打转)===== + COMPARE_ALERT_PRICEBOT_WORK_LOG_DIR: str = "" # pricebot work_logs 绝对路径(敏感,放 .env);空=跳过 trace、cancelled 全走保底 + COMPARE_ALERT_STUCK_FRAME_THRESHOLD: int = 15 # 末段连续同环节达此帧数判卡死 + COMPARE_ALERT_TRACE_MAX_TAIL_FRAMES: int = 40 # 每平台最多往前读多少帧 + COMPARE_ALERT_TRACE_MAX_RECORDS: int = 30 # 每轮最多对多少条命中记录读 trace(限量) + # ===== 短信 ===== SMS_MOCK: bool = True SMS_CODE_TTL_SEC: int = 300 @@ -217,6 +237,22 @@ class Settings(BaseSettings): phones.add(self.test_account_phone) return frozenset(phones) + def _csv(self, raw: str) -> tuple[str, ...]: + """逗号分隔字符串 → 去空白非空元组(报警关键词解析共用)。""" + return tuple(w.strip() for w in raw.split(",") if w.strip()) + + @property + def compare_alert_timeout_keywords(self) -> tuple[str, ...]: + return self._csv(self.COMPARE_ALERT_TIMEOUT_KEYWORDS) + + @property + def compare_alert_unrecognized_keywords(self) -> tuple[str, ...]: + return self._csv(self.COMPARE_ALERT_UNRECOGNIZED_KEYWORDS) + + @property + def compare_alert_biz_exclude_keywords(self) -> tuple[str, ...]: + return self._csv(self.COMPARE_ALERT_BIZ_EXCLUDE_KEYWORDS) + # ===== 美团联盟 CPS ===== # 未配置时所有 /api/v1/meituan/* 接口 200 返空(优雅降级),不影响登录/领券等其他业务。 MT_CPS_APP_KEY: str = "" diff --git a/app/integrations/feishu_notifier.py b/app/integrations/feishu_notifier.py new file mode 100644 index 0000000..c24d8d1 --- /dev/null +++ b/app/integrations/feishu_notifier.py @@ -0,0 +1,48 @@ +"""飞书群自定义机器人发送(text / post 消息)。 + +自定义机器人「关键词」验证:消息 content.text 必须含机器人配置的关键词,否则飞书返回 code!=0 +(如 19024 Key Words Not Found)——本项目消息由 compare_alert_format 生成,标题已含「比价失败报警」。 +不需要签名(sign)/IP 白名单。文档:https://open.feishu.cn/document/client-docs/bot-v3/add-custom-bot +""" +from __future__ import annotations + +import httpx + + +class FeishuNotifyError(Exception): + """飞书发送失败(网络错误 / 非 2xx / 业务 code!=0,含关键词不匹配)。""" + + +def _post_feishu(webhook_url: str, payload: dict, timeout: float) -> None: + """内部 helper:POST payload 到飞书 webhook 并校验响应。失败抛 FeishuNotifyError。""" + try: + resp = httpx.post(webhook_url, json=payload, timeout=timeout) + except httpx.HTTPError as e: + raise FeishuNotifyError(f"feishu request failed: {e}") from e + if resp.status_code >= 300: + raise FeishuNotifyError(f"feishu http {resp.status_code}: {resp.text[:200]}") + try: + data = resp.json() + except ValueError as e: + raise FeishuNotifyError(f"feishu bad json: {resp.text[:200]}") from e + code = data.get("code", data.get("StatusCode", 0)) + if code not in (0, None): + raise FeishuNotifyError(f"feishu code={code} msg={data.get('msg') or data.get('StatusMessage')}") + + +def send_feishu_text(webhook_url: str, text: str, *, timeout: float = 10.0) -> None: + """POST 一条 text 消息到飞书群机器人 webhook。失败(网络/HTTP/业务 code)抛 FeishuNotifyError。""" + payload = {"msg_type": "text", "content": {"text": text}} + _post_feishu(webhook_url, payload, timeout) + + +def send_feishu_post(webhook_url: str, title: str, content: list, *, timeout: float = 10.0) -> None: + """发飞书富文本(post)。content 是段落数组,每段是元素数组[{tag:text/a,...}]。失败抛 FeishuNotifyError。""" + payload = {"msg_type": "post", "content": {"post": {"zh_cn": {"title": title, "content": content}}}} + _post_feishu(webhook_url, payload, timeout) + + +def send_feishu_card(webhook_url: str, card: dict, *, timeout: float = 10.0) -> None: + """发飞书交互卡片(interactive)。card 为 schema 2.0 卡片 dict。失败抛 FeishuNotifyError。""" + payload = {"msg_type": "interactive", "card": card} + _post_feishu(webhook_url, payload, timeout) diff --git a/app/main.py b/app/main.py index 7d2a0cc..4920940 100644 --- a/app/main.py +++ b/app/main.py @@ -45,6 +45,10 @@ from app.api.v1.user import router as user_router from app.api.v1.wallet import router as wallet_router from app.api.v1.wxpay import router as wxpay_router from app.core import media +from app.core.compare_alert_worker import ( + start_compare_alert_worker, + stop_compare_alert_worker, +) from app.core.config import settings from app.core.cps_reconcile_worker import ( start_cps_reconcile_worker, @@ -123,6 +127,7 @@ async def lifespan(_: FastAPI) -> AsyncIterator[None]: observe_task = start_observe_worker() inactivity_task = start_inactivity_reset_worker() llm_cost_backfill_task = start_llm_cost_backfill_worker() + compare_alert_task = start_compare_alert_worker() try: yield finally: @@ -133,6 +138,7 @@ async def lifespan(_: FastAPI) -> AsyncIterator[None]: await stop_observe_worker(observe_task) await stop_inactivity_reset_worker(inactivity_task) await stop_llm_cost_backfill_worker(llm_cost_backfill_task) + await stop_compare_alert_worker(compare_alert_task) await aclose_pricebot_client() mt_meituan.close_client() logger.info("shutting down") diff --git a/app/models/comparison.py b/app/models/comparison.py index 51956d4..3086af2 100644 --- a/app/models/comparison.py +++ b/app/models/comparison.py @@ -49,6 +49,9 @@ class ComparisonRecord(Base): # 单列 user_id 索引只能过滤,排序仍要把该用户全部记录取出来排一遍;这条复合索引的**反向扫** # 恰好等于 (created_at DESC, id DESC),PG 直接取前 n 条、免排序。列序不能动。 Index("ix_comparison_user_created", "user_id", "created_at", "id"), + # 比价报警 worker 的水位查询 WHERE updated_at > watermark 走它。显式命名(不用列上 + # index=True 的自动名 ix_comparison_record_updated_at)以与迁移 create_index 同名、免 autogenerate 漂移。 + Index("ix_comparison_updated", "updated_at"), ) id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True) @@ -160,6 +163,15 @@ class ComparisonRecord(Base): created_at: Mapped[datetime] = mapped_column( DateTime(timezone=True), server_default=func.now(), index=True, nullable=False ) + # 记录任一次更新(建 running 行 / done / abort 落终态)的 DB 时钟时间。比价报警 worker 的 + # 水位列:按 updated_at 单调推进扫描,任何记录落定/更新都刷新它 > 水位、必被下轮扫到, + # 根治 created_at 水位漏掉「慢失败」(落定延迟 p99 达 7-10min)。onupdate 在 ORM UPDATE 时自动刷新。 + updated_at: Mapped[datetime] = mapped_column( + DateTime(timezone=True), + server_default=func.now(), + onupdate=func.now(), + nullable=False, + ) def __repr__(self) -> str: # pragma: no cover return ( diff --git a/app/services/compare_alert.py b/app/services/compare_alert.py new file mode 100644 index 0000000..7cc114d --- /dev/null +++ b/app/services/compare_alert.py @@ -0,0 +1,98 @@ +"""比价失败报警规则:一条记录 → 命中的 AlertHit(或 None)。 + +纯函数、不碰 DB(阈值/关键词由调用方从 config 传入),便于单测与调阈值。判定顺序保证四类互斥: + failed → fail_reason 空(且 information 非业务)=T1 / 含未识别=T6 / 含超时词=T2 / 其余业务不报; + cancelled 且耗时或步数超阈值=T5(深度放弃);success/running 不报。 +口径依据见 docs/superpowers/specs/2026-08-04-compare-fail-alert-design.md 第 3 节。 +""" +from __future__ import annotations + +from dataclasses import dataclass +from datetime import datetime +from typing import Any + +ALERT_TYPE_LABELS: dict[str, str] = { + "T1": "系统技术失败", + "T6": "商品识别失败", + "T2": "启动/超时失败", + "T5": "深度放弃(cancelled)", +} + + +@dataclass(frozen=True) +class AlertHit: + trace_id: str + alert_type: str + reason: str + app_version: str | None + created_at: datetime | None + trace_url: str | None + user_id: int | None + total_ms: int | None = None + step_count: int | None = None + stuck_point: str | None = None + + +def make_hit(rec: Any, alert_type: str, reason: str) -> AlertHit: + return AlertHit( + trace_id=rec.trace_id, + alert_type=alert_type, + reason=reason, + app_version=getattr(rec, "app_version", None), + created_at=getattr(rec, "created_at", None), + trace_url=getattr(rec, "trace_url", None), + user_id=getattr(rec, "user_id", None), + total_ms=getattr(rec, "total_ms", None), + step_count=getattr(rec, "step_count", None), + ) + + +def classify_cancelled_fallback( + rec: Any, + *, + cancelled_ms_threshold: int, + cancelled_step_threshold: int, +) -> AlertHit | None: + """cancelled 保底判定(读不到 trace 时用):超耗时或步数阈值 → T5 深度放弃,否则 None。纯函数。""" + ms = rec.total_ms + step = rec.step_count + deep = (ms is not None and ms > cancelled_ms_threshold) or ( + step is not None and step > cancelled_step_threshold + ) + if deep: + return make_hit(rec, "T5", "深度放弃") + return None + + +def classify_record( + rec: Any, + *, + cancelled_ms_threshold: int, + cancelled_step_threshold: int, + timeout_keywords: tuple[str, ...], + unrecognized_keywords: tuple[str, ...], + biz_exclude_keywords: tuple[str, ...], +) -> AlertHit | None: + """判定单条记录是否触发报警。rec 需有 status/fail_reason/information/total_ms/step_count/ + trace_id/app_version 属性(ComparisonRecord 或等价对象)。""" + status = rec.status + if status == "failed": + fail_reason = rec.fail_reason + if fail_reason is None: + info = (rec.information or "").strip() + if info and any(w in info for w in biz_exclude_keywords): + return None + return make_hit(rec, "T1", f"技术失败·{info[:80] or '比价过程出错'}") + if any(w in fail_reason for w in unrecognized_keywords): + return make_hit(rec, "T6", f"识别失败·{fail_reason[:80]}") + if any(w in fail_reason for w in timeout_keywords): + return make_hit(rec, "T2", fail_reason[:80]) + return None + if status == "cancelled": + return classify_cancelled_fallback( + rec, + cancelled_ms_threshold=cancelled_ms_threshold, + cancelled_step_threshold=cancelled_step_threshold, + ) + return None + diff --git a/app/services/compare_alert_format.py b/app/services/compare_alert_format.py new file mode 100644 index 0000000..5542a14 --- /dev/null +++ b/app/services/compare_alert_format.py @@ -0,0 +1,243 @@ +"""AlertHit[] → 飞书群机器人消息。 + +提供三个格式化函数: +- format_alert_message: 纯文本(保留,已有集成测试依赖)。 +- format_alert_post: 富文本 post(行式明细:时间｜手机｜版本｜原因｜trace 超链接)。 +- format_alert_card: schema 2.0 卡片 + table 组件(正式发送格式)。 + +按触发类型分组,每类给计数 + 明细(trace/版本/原因)。两级截断防报警风暴:单类型超 +max_detail_per_type 只列前 N + 「另有 M 条」;本期总量超 max_total 只给各类型计数、提示去分析库查。 +标题含关键词「比价失败报警」——飞书自定义机器人用关键词验证,消息必须含它,否则被拒收。 +""" +from __future__ import annotations + +from app.services.compare_alert import ALERT_TYPE_LABELS, AlertHit + +ALERT_KEYWORD = "比价失败报警" + +_TYPE_ORDER = ("T1", "T6", "T2", "T5") + + +def _detail_line(h: AlertHit) -> str: + ver = h.app_version or "?" + return f" - trace {h.trace_id} | {ver} | {h.reason}" + + +def format_alert_message( + hits: list[AlertHit], + *, + window_label: str, + max_detail_per_type: int, + max_total: int, +) -> str: + total = len(hits) + header = f"🚨 {ALERT_KEYWORD} · {window_label} · 本期触发 {total} 条" + + grouped: dict[str, list[AlertHit]] = {t: [] for t in _TYPE_ORDER} + for h in hits: + grouped.setdefault(h.alert_type, []).append(h) + + lines = [header] + counts_only = total > max_total + for t in _TYPE_ORDER: + bucket = grouped.get(t) or [] + if not bucket: + continue + lines.append(f"• {ALERT_TYPE_LABELS[t]} {len(bucket)} 条") + if counts_only: + continue + shown = bucket[:max_detail_per_type] + lines.extend(_detail_line(h) for h in shown) + if len(bucket) > max_detail_per_type: + lines.append(f" …另有 {len(bucket) - max_detail_per_type} 条") + if counts_only: + lines.append(f"(本期命中超 {max_total} 条,仅列计数,明细见分析库 comparison_record)") + return "\n".join(lines) + + +def format_alert_post( + hits: list[AlertHit], + *, + window_label: str, + phone_map: dict[int, str], + max_detail_per_type: int, + max_total: int, +) -> tuple[str, list]: + """行式富文本:返回 (title, content)。title 含 ALERT_KEYWORD(飞书关键词验证)。 + + content: 摘要段(各类型计数) + 表头段 + 明细行(每条「时间｜手机｜版本｜原因｜」+ trace 超链接 a 元素)。 + phone_map: {user_id: phone};明细手机号取 phone_map.get(hit.user_id) or "-"。 + 截断规则:总命中 > max_total → 只出摘要+各类型计数(不列明细); + 否则明细最多列前 max_detail_per_type 条,超出加「…另有 N 条」。 + """ + total = len(hits) + title = f"🚨 {ALERT_KEYWORD} · {window_label}" + + grouped: dict[str, list[AlertHit]] = {t: [] for t in _TYPE_ORDER} + for h in hits: + grouped.setdefault(h.alert_type, []).append(h) + + # 摘要段:各类型计数 + summary_parts = [f"{ALERT_TYPE_LABELS[t]} {len(grouped[t])}" for t in _TYPE_ORDER if grouped.get(t)] + summary_text = f"合计 {total} 条：" + " ｜ ".join(summary_parts) + content: list[list[dict]] = [ + [{"tag": "text", "text": summary_text}], + ] + + counts_only = total > max_total + if counts_only: + content.append([{"tag": "text", "text": f"(本期命中超 {max_total} 条,仅列计数,明细见分析库 comparison_record)"}]) + return title, content + + # 表头段 + content.append([{"tag": "text", "text": "时间 ｜ 手机号 ｜ 版本 ｜ 失败原因 ｜ trace"}]) + + # 明细行(按类型顺序展开,每条一段) + shown_count = 0 + for t in _TYPE_ORDER: + bucket = grouped.get(t) or [] + if not bucket: + continue + for h in bucket[:max_detail_per_type]: + time_str = h.created_at.strftime("%m-%d %H:%M") if h.created_at else "-" + phone = phone_map.get(h.user_id) if h.user_id is not None else None + phone = phone or "-" + ver = h.app_version or "-" + row: list[dict] = [{"tag": "text", "text": f"{time_str} ｜ {phone} ｜ {ver} ｜ {h.reason} ｜ "}] + if h.trace_url: + row.append({"tag": "a", "text": "trace", "href": h.trace_url}) + else: + row.append({"tag": "text", "text": h.trace_id[:16]}) + content.append(row) + shown_count += 1 + if len(bucket) > max_detail_per_type: + content.append([{"tag": "text", "text": f"…另有 {len(bucket) - max_detail_per_type} 条"}]) + + return title, content + + +# ---- format_alert_card (schema 2.0 卡片 + table 组件) ---- + +def _cost_cell(total_ms: int | None, step_count: int | None) -> str: + """组合「用时」列值。有 ms → '{N}s'，有 step_count → '{M}步'，两者用 ' / ' 连；都无 → '-'。""" + parts = [] + if total_ms is not None: + parts.append(f"{round(total_ms / 1000)}s") + if step_count is not None: + parts.append(f"{step_count}步") + return " / ".join(parts) or "-" + + +def _build_card(title_text: str, elements: list[dict]) -> dict: + """组装 schema 2.0 红色 header 卡片;三条路径只需决定 elements。""" + return { + "schema": "2.0", + "header": { + "title": {"tag": "plain_text", "content": title_text}, + "template": "red", + }, + "body": {"elements": elements}, + } + + +def _build_table_rows( + hits: list[AlertHit], + *, + phone_map: dict[int, str], + max_detail_per_type: int, +) -> list[dict]: + """按 _TYPE_ORDER 顺序展开，每类型最多 max_detail_per_type 条。""" + grouped: dict[str, list[AlertHit]] = {t: [] for t in _TYPE_ORDER} + for h in hits: + if h.alert_type in grouped: + grouped[h.alert_type].append(h) + # 非 _TYPE_ORDER 类型静默跳过(与既有行为一致) + + rows = [] + for t in _TYPE_ORDER: + bucket = grouped.get(t) or [] + for h in bucket[:max_detail_per_type]: + time_str = h.created_at.strftime("%m-%d %H:%M") if h.created_at else "-" + phone = (phone_map.get(h.user_id) if h.user_id is not None else None) or "-" + trace = ( + f"[链接]({h.trace_url})" if h.trace_url + else (h.trace_id or "")[:12] + ) + rows.append({ + "time": time_str, + "phone": phone, + "cost": _cost_cell(h.total_ms, h.step_count), + "reason": h.reason, + "stuck": h.stuck_point or "-", + "ver": h.app_version or "-", + "trace": trace, + }) + return rows + + +_TABLE_COLUMNS = [ + {"name": "time", "display_name": "时间", "data_type": "text"}, + {"name": "phone", "display_name": "手机号", "data_type": "text"}, + {"name": "cost", "display_name": "用时", "data_type": "text"}, + {"name": "reason", "display_name": "失败原因", "data_type": "text"}, + {"name": "stuck", "display_name": "卡点", "data_type": "text"}, + {"name": "ver", "display_name": "版本", "data_type": "text"}, + {"name": "trace", "display_name": "trace", "data_type": "lark_md"}, +] + + +def format_alert_card( + hits: list[AlertHit], + *, + window_label: str, + phone_map: dict[int, str], + interval_min: int, + max_detail_per_type: int, + max_total: int, +) -> dict: + """返回飞书 schema 2.0 卡片 dict（配合 send_feishu_card 发送）。 + + - header: template=red，title 含 ALERT_KEYWORD（飞书关键词验证必须）。 + - body 第一个元素: markdown 摘要（数据范围 + 合计 + 各类型计数）。 + - 空 hits: 只有摘要「本期无异常」。 + - total > max_total: 只有摘要（提示去 comparison_record 查），不加 table。 + - 否则: 第二个元素为 table（列序 time/phone/cost/reason/stuck/ver/trace）。 + """ + total = len(hits) + title_text = f"🚨 {ALERT_KEYWORD} · {window_label}" + + # ---------- 空 hits ---------- + if total == 0: + md_content = f"数据范围：近 {interval_min} 分钟\n本期无异常" + return _build_card(title_text, [{"tag": "markdown", "content": md_content}]) + + # ---------- 摘要 ---------- + grouped_count: dict[str, int] = {} + for h in hits: + grouped_count[h.alert_type] = grouped_count.get(h.alert_type, 0) + 1 + count_parts = [ + f"{ALERT_TYPE_LABELS[t]} {grouped_count[t]}" + for t in _TYPE_ORDER + if grouped_count.get(t) + ] + md_content = ( + f"数据范围：近 {interval_min} 分钟\n" + f"**合计 {total} 条**：" + " ｜ ".join(count_parts) + ) + + # ---------- 截断：超 max_total 只出摘要 ---------- + if total > max_total: + md_content += f"\n超 {max_total} 条仅列计数，明细见分析库 comparison_record" + return _build_card(title_text, [{"tag": "markdown", "content": md_content}]) + + # ---------- 常规：摘要 + table ---------- + rows = _build_table_rows(hits, phone_map=phone_map, max_detail_per_type=max_detail_per_type) + table_element = { + "tag": "table", + "page_size": 10, + "row_height": "low", + "header_style": {"background_style": "grey", "bold": True}, + "columns": _TABLE_COLUMNS, + "rows": rows, + } + return _build_card(title_text, [{"tag": "markdown", "content": md_content}, table_element]) diff --git a/app/services/trace_stuck.py b/app/services/trace_stuck.py new file mode 100644 index 0000000..5f982cc --- /dev/null +++ b/app/services/trace_stuck.py @@ -0,0 +1,159 @@ +"""比价卡死定位:读 pricebot trace 末段,判某平台是否原地打转(卡死)。 + +同机直读 {WORK_LOG_DIR}/{dir_name}/{platform}/step_*.json,只取头部字段 +(pipeline_step/detected_page),不解析后面的无障碍树(windows,占单帧 99% 体积)。 +判据与降级见 docs/superpowers/specs/2026-08-05-compare-stuck-detection-design.md。 +""" +from __future__ import annotations + +import re +from dataclasses import dataclass +from datetime import datetime +from pathlib import Path + +# pipeline_step / 平台名 → 中文(映射不到原样显示英文,不阻断)。按 pricebot 实际枚举补全。 +PIPELINE_STEP_LABELS: dict[str, str] = { + "set_address": "定位", + "enter_store": "进店", + "add_one_dish": "加菜", + "match_dish": "找菜", + "checkout": "结算", +} +PLATFORM_LABELS: dict[str, str] = { + "meituan": "美团", + "eleme": "饿了么", + "jd_waimai": "京东外卖", +} + +_PIPE_RE = re.compile(r'"pipeline_step":\s*"([^"]*)"') +_PAGE_RE = re.compile(r'"detected_page":\s*"([^"]*)"') +_TS_RE = re.compile(r'"timestamp":\s*"([^"]*)"') +_STEP_NUM_RE = re.compile(r"step_(\d+)") + + +def _parse_ts(s: str | None) -> datetime | None: + if not s: + return None + try: + return datetime.fromisoformat(s) + except (ValueError, TypeError): + return None + + +@dataclass(frozen=True) +class StuckPoint: + platform: str + pipeline_step: str + frames: int # 末段连续困住的帧数(上限 max_tail) + stuck_ms: int | None = None # 末段连续卡住的时长(ms);无 timestamp 时 None + + def label(self) -> str: + p = PLATFORM_LABELS.get(self.platform, self.platform) + s = PIPELINE_STEP_LABELS.get(self.pipeline_step, self.pipeline_step) + return f"{p}·{s}" + + +@dataclass(frozen=True) +class StuckResult: + readable: bool # trace 是否读到(区分「读不到」与「读到但没卡」) + points: list[StuckPoint] # 卡死平台列表;readable=True 且空 = 确认没卡死 + + +def dir_name_from_trace_url(trace_url: str | None) -> str | None: + """.../traces/{dir_name}/ → dir_name;空/异常 → None。""" + if not trace_url: + return None + name = trace_url.rstrip("/").rsplit("/", 1)[-1] + return name or None + + +def _step_num(path: Path) -> int: + m = _STEP_NUM_RE.search(path.name) + return int(m.group(1)) if m else -1 + + +def _read_head(path: Path, nbytes: int = 4096) -> tuple[str | None, str | None, str | None]: + """只读文件头部,抠 (pipeline_step, detected_page, timestamp)。它们在 json 最前面。""" + try: + with open(path, encoding="utf-8", errors="replace") as f: + head = f.read(nbytes) + except OSError: + return None, None, None + ps = _PIPE_RE.search(head) + pg = _PAGE_RE.search(head) + ts = _TS_RE.search(head) + return (ps.group(1) if ps else None, pg.group(1) if pg else None, ts.group(1) if ts else None) + + +def _platform_stuck( + platform: str, step_files: list[Path], threshold: int, max_tail: int +) -> StuckPoint | None: + """末帧往前数连续同 (pipeline_step, detected_page) 的帧数 ≥threshold → 卡死。""" + tail = step_files[-max_tail:] + heads = [_read_head(p) for p in tail] # [(ps, pg, ts), ...] + last_ps, last_pg, _ = heads[-1] + if last_ps is None: + return None + seg_ts: list[str | None] = [] # 连续段的 timestamp(逆序:末帧在前) + for ps, pg, ts in reversed(heads): + if ps == last_ps and pg == last_pg: + seg_ts.append(ts) + else: + break + count = len(seg_ts) + if count < threshold: + return None + # seg_ts[0]=末帧, seg_ts[-1]=段首帧;两端都能解析才算时长 + t_last, t_first = _parse_ts(seg_ts[0]), _parse_ts(seg_ts[-1]) + stuck_ms = round((t_last - t_first).total_seconds() * 1000) if t_last and t_first else None + if stuck_ms is not None and stuck_ms < 0: + stuck_ms = None # 时钟不单调(帧 timestamp 回退)→ 降级为不显示时长 + return StuckPoint(platform, last_ps, count, stuck_ms) + + +def read_stuck_points(trace_dir: Path, *, threshold: int, max_tail: int) -> StuckResult: + """逐平台判卡死。目录不存在/异常/无平台帧 → StuckResult(readable=False, points=[])。""" + try: + if not trace_dir.is_dir(): + return StuckResult(readable=False, points=[]) + points: list[StuckPoint] = [] + any_frames = False + for pdir in sorted(trace_dir.iterdir()): + if not pdir.is_dir(): + continue + step_files = sorted(pdir.glob("step_*.json"), key=_step_num) + if not step_files: + continue + any_frames = True + sp = _platform_stuck(pdir.name, step_files, threshold, max_tail) + if sp is not None: + points.append(sp) + if not any_frames: + return StuckResult(readable=False, points=[]) + return StuckResult(readable=True, points=points) + except OSError: + return StuckResult(readable=False, points=[]) + + +def last_step(trace_dir: Path) -> StuckPoint | None: + """failed 用:取帧数最多平台的末帧环节(不要求原地打转)。读不到 → None。""" + try: + if not trace_dir.is_dir(): + return None + best: tuple[int, str, list[Path]] | None = None + for pdir in sorted(trace_dir.iterdir()): + if not pdir.is_dir(): + continue + step_files = sorted(pdir.glob("step_*.json"), key=_step_num) + # 平局(同帧数)时取字典序第一个平台(sorted 保证稳定) + if step_files and (best is None or len(step_files) > best[0]): + best = (len(step_files), pdir.name, step_files) + if best is None: + return None + _, platform, step_files = best + ps, _pg, _ts = _read_head(step_files[-1]) + if ps is None: + return None + return StuckPoint(platform, ps, len(step_files)) # stuck_ms=None(failed 不算时长) + except OSError: + return None diff --git a/docs/superpowers/plans/2026-08-05-compare-stuck-detection.md b/docs/superpowers/plans/2026-08-05-compare-stuck-detection.md new file mode 100644 index 0000000..6ad1df7 --- /dev/null +++ b/docs/superpowers/plans/2026-08-05-compare-stuck-detection.md @@ -0,0 +1,735 @@ +# 比价卡死定位报警增强 Implementation Plan + +> **For agentic workers:** REQUIRED SUB-SKILL: Use superpowers:subagent-driven-development (recommended) or superpowers:executing-plans to implement this plan task-by-task. Steps use checkbox (`- [ ]`) syntax for tracking. + +**Goal:** 让 cancelled 报警用 trace 末段「原地打转」判卡死并定位卡在哪个环节，读不到 trace 回退耗时/帧数保底；failed 类附卡点。 + +**Architecture:** 判定保持纯函数（`compare_alert.py`），trace 读取单独成层（`trace_stuck.py`，同机直读 pricebot work_logs、只读帧头部字段），worker 编排（cancelled trace 优先 + 保底、failed 附卡点）。卡点拼进 `AlertHit.reason`，复用现有 `format_alert_post` 展示，不依赖卡片 table 固化。 + +**Tech Stack:** Python 3.11+、FastAPI、SQLAlchemy、pytest。无新依赖（仅标准库 `re`/`json`/`pathlib`/`dataclasses`）。 + +参考 spec：`docs/superpowers/specs/2026-08-05-compare-stuck-detection-design.md` + +--- + +## File Structure + +- **Create** `app/services/trace_stuck.py` — 卡死判据 + 薄 IO。`StuckPoint`/`StuckResult`、`read_stuck_points`、`last_step`、`dir_name_from_trace_url`、文案表。 +- **Create** `tests/test_trace_stuck.py` — trace_stuck 单测。 +- **Modify** `app/core/config.py` — 加 4 个配置项。 +- **Modify** `app/services/compare_alert.py` — `_hit` 改公开 `make_hit`；新增纯函数 `classify_cancelled_fallback`；`classify_record` 的 cancelled 分支改调它（行为不变）。 +- **Create** `tests/test_compare_alert_fallback.py` — `classify_cancelled_fallback`/`make_hit` 单测。 +- **Modify** `app/core/compare_alert_worker.py` — 新增 `build_hits`/`_trace_dir` 编排；`_scan_and_alert` 用 `build_hits` 替换 `classify_batch`。 +- **Create** `tests/test_compare_alert_stuck_worker.py` — `build_hits` 集成测。 + +--- + +## Task 1: 配置项 + +**Files:** +- Modify: `app/core/config.py:147`（在 `COMPARE_ALERT_SEND_EMPTY` 行后追加） + +- [ ] **Step 1: 加 4 个配置字段** + +在 `app/core/config.py` 第 147 行 `COMPARE_ALERT_SEND_EMPTY: bool = False ...` 之后，紧接着追加： + +```python + # ===== 卡死定位(读 pricebot trace 末段判原地打转)===== + COMPARE_ALERT_PRICEBOT_WORK_LOG_DIR: str = "" # pricebot work_logs 绝对路径(敏感,放 .env);空=跳过 trace、cancelled 全走保底 + COMPARE_ALERT_STUCK_FRAME_THRESHOLD: int = 15 # 末段连续同环节达此帧数判卡死 + COMPARE_ALERT_TRACE_MAX_TAIL_FRAMES: int = 40 # 每平台最多往前读多少帧 + COMPARE_ALERT_TRACE_MAX_RECORDS: int = 30 # 每轮最多对多少条命中记录读 trace(限量) +``` + +- [ ] **Step 2: 跑现有测试确认不破** + +Run: `pytest tests/ -q -k "config or defaults"` +Expected: PASS（新增字段都有默认值，不影响 `test_defaults`） + +- [ ] **Step 3: Commit** + +```bash +git add app/core/config.py +git commit -m "feat(compare-alert): 卡死定位 4 个配置项 + +Co-Authored-By: Claude Opus 4.8 " +``` + +--- + +## Task 2: trace_stuck 模块 + +**Files:** +- Create: `app/services/trace_stuck.py` +- Test: `tests/test_trace_stuck.py` + +- [ ] **Step 1: 写失败测试** + +创建 `tests/test_trace_stuck.py`： + +```python +"""trace_stuck 单测:用 tmp 造 step_*.json(只含头部字段)验证卡死判据。""" +import json +from pathlib import Path + +from app.services.trace_stuck import ( + StuckPoint, + dir_name_from_trace_url, + last_step, + read_stuck_points, +) + + +def _frame(pdir: Path, idx: int, step: str, page: str) -> None: + """造一帧 step json:头部放 pipeline_step/detected_page,尾部塞大 windows 模拟真实。""" + pdir.mkdir(parents=True, exist_ok=True) + body = { + "trace_id": "t", "step": idx, "platform": pdir.name, + "pipeline_step": step, "detected_page": page, + "windows": [{"nodes": ["x" * 200]}], + } + (pdir / f"step_{idx:03d}.json").write_text( + json.dumps(body, ensure_ascii=False), encoding="utf-8" + ) + + +def test_stuck_when_tail_repeats_same_step(tmp_path): + pdir = tmp_path / "meituan" + for i in range(20): + _frame(pdir, i, "add_one_dish", "meal_detail_popup") + res = read_stuck_points(tmp_path, threshold=15, max_tail=40) + assert res.readable is True + assert res.points == [StuckPoint("meituan", "add_one_dish", 20)] + + +def test_not_stuck_when_progressing(tmp_path): + pdir = tmp_path / "eleme" + _frame(pdir, 0, "set_address", "home") + for i in range(1, 8): + _frame(pdir, i, "enter_store", "store") + res = read_stuck_points(tmp_path, threshold=15, max_tail=40) + assert res.readable is True + assert res.points == [] + + +def test_adding_many_dishes_not_stuck_when_page_changes(tmp_path): + # add_one_dish 重复但 detected_page 在跳(换菜)=推进,不判卡死 + pdir = tmp_path / "meituan" + for i in range(20): + _frame(pdir, i, "add_one_dish", "menu" if i % 2 == 0 else "dish_popup") + res = read_stuck_points(tmp_path, threshold=15, max_tail=40) + assert res.points == [] + + +def test_below_threshold_not_stuck(tmp_path): + pdir = tmp_path / "meituan" + for i in range(10): # < 15 + _frame(pdir, i, "add_one_dish", "meal_detail_popup") + res = read_stuck_points(tmp_path, threshold=15, max_tail=40) + assert res.points == [] + + +def test_missing_dir_not_readable(tmp_path): + res = read_stuck_points(tmp_path / "nope", threshold=15, max_tail=40) + assert res.readable is False + assert res.points == [] + + +def test_empty_dir_no_platform_frames_not_readable(tmp_path): + (tmp_path / "emptysub").mkdir() + res = read_stuck_points(tmp_path, threshold=15, max_tail=40) + assert res.readable is False + + +def test_per_platform_one_stuck_one_normal(tmp_path): + m = tmp_path / "meituan" + for i in range(18): + _frame(m, i, "add_one_dish", "meal_detail_popup") + e = tmp_path / "eleme" + _frame(e, 0, "set_address", "home") + for i in range(1, 6): + _frame(e, i, "enter_store", "store") + res = read_stuck_points(tmp_path, threshold=15, max_tail=40) + assert res.readable is True + assert res.points == [StuckPoint("meituan", "add_one_dish", 18)] + + +def test_last_step_returns_busiest_platform_last_env(tmp_path): + m = tmp_path / "meituan" + for i in range(20): + _frame(m, i, "add_one_dish", "meal_detail_popup") + e = tmp_path / "eleme" + for i in range(3): + _frame(e, i, "enter_store", "store") + sp = last_step(tmp_path) + assert sp == StuckPoint("meituan", "add_one_dish", 20) + + +def test_dir_name_from_trace_url(): + assert dir_name_from_trace_url("https://x/traces/20260804_1_abc/") == "20260804_1_abc" + assert dir_name_from_trace_url("https://x/traces/20260804_1_abc") == "20260804_1_abc" + assert dir_name_from_trace_url("") is None + assert dir_name_from_trace_url(None) is None +``` + +- [ ] **Step 2: 跑测试确认失败** + +Run: `pytest tests/test_trace_stuck.py -q` +Expected: FAIL（`ModuleNotFoundError: app.services.trace_stuck`） + +- [ ] **Step 3: 实现 trace_stuck.py** + +创建 `app/services/trace_stuck.py`： + +```python +"""比价卡死定位:读 pricebot trace 末段,判某平台是否原地打转(卡死)。 + +同机直读 {WORK_LOG_DIR}/{dir_name}/{platform}/step_*.json,只取头部字段 +(pipeline_step/detected_page),不解析后面的无障碍树(windows,占单帧 99% 体积)。 +判据与降级见 docs/superpowers/specs/2026-08-05-compare-stuck-detection-design.md。 +""" +from __future__ import annotations + +import re +from dataclasses import dataclass +from pathlib import Path + +# pipeline_step / 平台名 → 中文(映射不到原样显示英文,不阻断)。按 pricebot 实际枚举补全。 +PIPELINE_STEP_LABELS: dict[str, str] = { + "set_address": "定位", + "enter_store": "进店", + "add_one_dish": "加菜", + "match_dish": "找菜", + "checkout": "结算", +} +PLATFORM_LABELS: dict[str, str] = { + "meituan": "美团", + "eleme": "饿了么", + "jd_waimai": "京东外卖", +} + +_PIPE_RE = re.compile(r'"pipeline_step":\s*"([^"]*)"') +_PAGE_RE = re.compile(r'"detected_page":\s*"([^"]*)"') +_STEP_NUM_RE = re.compile(r"step_(\d+)") + + +@dataclass(frozen=True) +class StuckPoint: + platform: str + pipeline_step: str + frames: int # 末段连续困住的帧数(上限 max_tail) + + def label(self) -> str: + p = PLATFORM_LABELS.get(self.platform, self.platform) + s = PIPELINE_STEP_LABELS.get(self.pipeline_step, self.pipeline_step) + return f"{p}·{s}" + + +@dataclass(frozen=True) +class StuckResult: + readable: bool # trace 是否读到(区分「读不到」与「读到但没卡」) + points: list[StuckPoint] # 卡死平台列表;readable=True 且空 = 确认没卡死 + + +def dir_name_from_trace_url(trace_url: str | None) -> str | None: + """.../traces/{dir_name}/ → dir_name;空/异常 → None。""" + if not trace_url: + return None + name = trace_url.rstrip("/").rsplit("/", 1)[-1] + return name or None + + +def _step_num(path: Path) -> int: + m = _STEP_NUM_RE.search(path.name) + return int(m.group(1)) if m else -1 + + +def _read_head(path: Path, nbytes: int = 4096) -> tuple[str | None, str | None]: + """只读文件头部,抠 (pipeline_step, detected_page)。它们在 json 最前面。""" + with open(path, "r", encoding="utf-8") as f: + head = f.read(nbytes) + ps = _PIPE_RE.search(head) + pg = _PAGE_RE.search(head) + return (ps.group(1) if ps else None, pg.group(1) if pg else None) + + +def _platform_stuck( + platform: str, step_files: list[Path], threshold: int, max_tail: int +) -> StuckPoint | None: + """末帧往前数连续同 (pipeline_step, detected_page) 的帧数 ≥threshold → 卡死。""" + tail = step_files[-max_tail:] + heads = [_read_head(p) for p in tail] + last_ps, last_pg = heads[-1] + if last_ps is None: + return None + count = 0 + for ps, pg in reversed(heads): + if ps == last_ps and pg == last_pg: + count += 1 + else: + break + if count >= threshold: + return StuckPoint(platform, last_ps, count) + return None + + +def read_stuck_points(trace_dir: Path, *, threshold: int, max_tail: int) -> StuckResult: + """逐平台判卡死。目录不存在/异常/无平台帧 → StuckResult(readable=False, points=[])。""" + try: + if not trace_dir.is_dir(): + return StuckResult(readable=False, points=[]) + points: list[StuckPoint] = [] + any_frames = False + for pdir in sorted(trace_dir.iterdir()): + if not pdir.is_dir(): + continue + step_files = sorted(pdir.glob("step_*.json"), key=_step_num) + if not step_files: + continue + any_frames = True + sp = _platform_stuck(pdir.name, step_files, threshold, max_tail) + if sp is not None: + points.append(sp) + if not any_frames: + return StuckResult(readable=False, points=[]) + return StuckResult(readable=True, points=points) + except OSError: + return StuckResult(readable=False, points=[]) + + +def last_step(trace_dir: Path) -> StuckPoint | None: + """failed 用:取帧数最多平台的末帧环节(不要求原地打转)。读不到 → None。""" + try: + if not trace_dir.is_dir(): + return None + best: tuple[int, str, list[Path]] | None = None + for pdir in sorted(trace_dir.iterdir()): + if not pdir.is_dir(): + continue + step_files = sorted(pdir.glob("step_*.json"), key=_step_num) + if step_files and (best is None or len(step_files) > best[0]): + best = (len(step_files), pdir.name, step_files) + if best is None: + return None + _, platform, step_files = best + ps, _pg = _read_head(step_files[-1]) + if ps is None: + return None + return StuckPoint(platform, ps, len(step_files)) + except OSError: + return None +``` + +- [ ] **Step 4: 跑测试确认通过** + +Run: `pytest tests/test_trace_stuck.py -q` +Expected: PASS（9 passed） + +- [ ] **Step 5: Commit** + +```bash +git add app/services/trace_stuck.py tests/test_trace_stuck.py +git commit -m "feat(compare-alert): trace_stuck 卡死判据(末段原地打转) + +Co-Authored-By: Claude Opus 4.8 " +``` + +--- + +## Task 3: compare_alert 抽出 fallback + 公开 make_hit + +**Files:** +- Modify: `app/services/compare_alert.py`（`_hit`→`make_hit`；新增 `classify_cancelled_fallback`；cancelled 分支改调它） +- Test: `tests/test_compare_alert_fallback.py` + +- [ ] **Step 1: 写失败测试** + +创建 `tests/test_compare_alert_fallback.py`： + +```python +"""classify_cancelled_fallback / make_hit 单测。""" +from app.services.compare_alert import classify_cancelled_fallback, make_hit + + +class _Rec: + def __init__(self, **kw): + self.trace_id = kw.get("trace_id", "t") + self.status = kw.get("status", "cancelled") + self.total_ms = kw.get("total_ms") + self.step_count = kw.get("step_count") + self.fail_reason = kw.get("fail_reason") + self.information = kw.get("information") + self.app_version = kw.get("app_version") + self.created_at = kw.get("created_at") + self.trace_url = kw.get("trace_url") + self.user_id = kw.get("user_id") + + +def test_fallback_deep_by_ms(): + hit = classify_cancelled_fallback( + _Rec(total_ms=95000, step_count=5), + cancelled_ms_threshold=90000, cancelled_step_threshold=30, + ) + assert hit is not None and hit.alert_type == "T5" and "深度放弃" in hit.reason + + +def test_fallback_deep_by_step(): + hit = classify_cancelled_fallback( + _Rec(total_ms=1000, step_count=35), + cancelled_ms_threshold=90000, cancelled_step_threshold=30, + ) + assert hit is not None and hit.alert_type == "T5" + + +def test_fallback_shallow_none(): + hit = classify_cancelled_fallback( + _Rec(total_ms=5000, step_count=3), + cancelled_ms_threshold=90000, cancelled_step_threshold=30, + ) + assert hit is None + + +def test_make_hit_carries_fields(): + hit = make_hit(_Rec(trace_id="tx", app_version="0.6.0"), "T5", "卡在 美团·加菜") + assert hit.trace_id == "tx" + assert hit.reason == "卡在 美团·加菜" + assert hit.app_version == "0.6.0" +``` + +- [ ] **Step 2: 跑测试确认失败** + +Run: `pytest tests/test_compare_alert_fallback.py -q` +Expected: FAIL（`ImportError: cannot import name 'classify_cancelled_fallback'`） + +- [ ] **Step 3: 改 compare_alert.py** + +在 `app/services/compare_alert.py`： + +(a) 把 `def _hit(` 改名为 `def make_hit(`（第 33 行），并把 `classify_record` 内 4 处 `_hit(` 调用改成 `make_hit(`（原 T1/T6/T2/T5 分支）。 + +(b) 在 `make_hit` 之后、`classify_record` 之前，新增： + +```python +def classify_cancelled_fallback( + rec: Any, + *, + cancelled_ms_threshold: int, + cancelled_step_threshold: int, +) -> AlertHit | None: + """cancelled 保底判定(读不到 trace 时用):超耗时或步数阈值 → T5 深度放弃,否则 None。纯函数。""" + ms = rec.total_ms + step = rec.step_count + deep = (ms is not None and ms > cancelled_ms_threshold) or ( + step is not None and step > cancelled_step_threshold + ) + if deep: + return make_hit( + rec, "T5", + f"深度放弃·等待 {round((ms or 0) / 1000)}s / {step or 0} 步后退出", + ) + return None +``` + +(c) 把 `classify_record` 里的 cancelled 分支（原 `if status == "cancelled":` 那整段）替换为： + +```python + if status == "cancelled": + return classify_cancelled_fallback( + rec, + cancelled_ms_threshold=cancelled_ms_threshold, + cancelled_step_threshold=cancelled_step_threshold, + ) + return None +``` + +（`classify_record` 行为不变，只是把 cancelled 逻辑抽到 `classify_cancelled_fallback`。） + +- [ ] **Step 4: 跑测试确认通过（含现有 rules 测试不回归）** + +Run: `pytest tests/test_compare_alert_fallback.py tests/test_compare_alert_rules.py -q` +Expected: PASS（新测试 4 passed，现有 rules 测试仍全 PASS） + +- [ ] **Step 5: Commit** + +```bash +git add app/services/compare_alert.py tests/test_compare_alert_fallback.py +git commit -m "feat(compare-alert): 抽出 classify_cancelled_fallback + 公开 make_hit + +Co-Authored-By: Claude Opus 4.8 " +``` + +--- + +## Task 4: worker 编排 build_hits + +**Files:** +- Modify: `app/core/compare_alert_worker.py`（加 imports、`_trace_dir`、`build_hits`；`_scan_and_alert` 改用 `build_hits`） +- Test: `tests/test_compare_alert_stuck_worker.py` + +- [ ] **Step 1: 写失败测试** + +创建 `tests/test_compare_alert_stuck_worker.py`： + +```python +"""build_hits 集成测:cancelled trace 优先/保底切换、failed 附卡点、限量。""" +import json +from pathlib import Path + +from app.core.compare_alert_worker import build_hits + + +class _Rec: + def __init__(self, **kw): + self.trace_id = kw.get("trace_id", "t") + self.status = kw.get("status", "cancelled") + self.total_ms = kw.get("total_ms") + self.step_count = kw.get("step_count") + self.fail_reason = kw.get("fail_reason") + self.information = kw.get("information") + self.app_version = kw.get("app_version") + self.created_at = kw.get("created_at") + self.trace_url = kw.get("trace_url") + self.user_id = kw.get("user_id") + + +def _frame(pdir: Path, idx: int, step: str, page: str) -> None: + pdir.mkdir(parents=True, exist_ok=True) + body = {"pipeline_step": step, "detected_page": page, "windows": [{"n": ["x" * 200]}]} + (pdir / f"step_{idx:03d}.json").write_text( + json.dumps(body, ensure_ascii=False), encoding="utf-8" + ) + + +_KW = dict( + stuck_threshold=15, max_tail=40, max_trace_reads=30, + cancelled_ms_threshold=90000, cancelled_step_threshold=30, + timeout_keywords=("超时",), unrecognized_keywords=("未识别",), biz_exclude_keywords=(), +) + + +def test_cancelled_stuck_reports_via_trace(tmp_path): + for i in range(18): + _frame(tmp_path / "20260804_x" / "meituan", i, "add_one_dish", "meal_detail_popup") + rec = _Rec(status="cancelled", trace_url="https://x/traces/20260804_x/", + total_ms=5000, step_count=3) # 保底不会中,靠 trace 判卡死 + hits = build_hits([rec], work_log_dir=str(tmp_path), **_KW) + assert len(hits) == 1 + assert hits[0].alert_type == "T5" + assert "卡在" in hits[0].reason and "美团·加菜" in hits[0].reason + + +def test_cancelled_readable_not_stuck_no_report(tmp_path): + # trace 确认没卡(在推进);即便 total_ms/step 超阈值也不报(信 trace,不回退保底) + p = tmp_path / "20260804_y" / "eleme" + _frame(p, 0, "set_address", "home") + for i in range(1, 6): + _frame(p, i, "enter_store", "store") + rec = _Rec(status="cancelled", trace_url="https://x/traces/20260804_y/", + total_ms=95000, step_count=40) + hits = build_hits([rec], work_log_dir=str(tmp_path), **_KW) + assert hits == [] + + +def test_cancelled_unreadable_falls_back(tmp_path): + rec = _Rec(status="cancelled", trace_url="https://x/traces/nope/", + total_ms=95000, step_count=3) + hits = build_hits([rec], work_log_dir=str(tmp_path), **_KW) + assert len(hits) == 1 + assert hits[0].alert_type == "T5" and "深度放弃" in hits[0].reason + + +def test_no_work_log_dir_uses_fallback(tmp_path): + rec = _Rec(status="cancelled", trace_url="https://x/traces/y/", + total_ms=95000, step_count=3) + hits = build_hits([rec], work_log_dir="", **_KW) + assert len(hits) == 1 and "深度放弃" in hits[0].reason + + +def test_failed_gets_stuck_point_appended(tmp_path): + for i in range(20): + _frame(tmp_path / "20260804_f" / "meituan", i, "add_one_dish", "meal_detail_popup") + rec = _Rec(status="failed", fail_reason="启动超时", + trace_url="https://x/traces/20260804_f/") + hits = build_hits([rec], work_log_dir=str(tmp_path), **_KW) + assert len(hits) == 1 + assert hits[0].alert_type == "T2" + assert "卡在 美团·加菜" in hits[0].reason + + +def test_max_trace_reads_zero_skips_trace(tmp_path): + for i in range(18): + _frame(tmp_path / "20260804_z" / "meituan", i, "add_one_dish", "meal_detail_popup") + rec = _Rec(status="cancelled", trace_url="https://x/traces/20260804_z/", + total_ms=95000, step_count=3) + kw = {**_KW, "max_trace_reads": 0} + hits = build_hits([rec], work_log_dir=str(tmp_path), **kw) + # 没读 trace → 回退保底 → deep(95s) → 深度放弃 + assert len(hits) == 1 and "深度放弃" in hits[0].reason +``` + +- [ ] **Step 2: 跑测试确认失败** + +Run: `pytest tests/test_compare_alert_stuck_worker.py -q` +Expected: FAIL（`ImportError: cannot import name 'build_hits'`） + +- [ ] **Step 3: 改 compare_alert_worker.py** + +(a) 顶部 imports 段，把 +```python +from app.services.compare_alert import classify_batch +from app.services.compare_alert_format import ALERT_KEYWORD, format_alert_post +``` +改为 +```python +from dataclasses import replace as _dc_replace + +from app.services import trace_stuck +from app.services.compare_alert import ( + classify_cancelled_fallback, + classify_record, + make_hit, +) +from app.services.compare_alert_format import ALERT_KEYWORD, format_alert_post +``` + +(b) 在 `_scan_and_alert` 之前新增两个函数： + +```python +def _trace_dir(base: Path, trace_url: str | None) -> Path | None: + name = trace_stuck.dir_name_from_trace_url(trace_url) + if not name: + return None + return base / name + + +def build_hits( + records: list, + *, + work_log_dir: str, + stuck_threshold: int, + max_tail: int, + max_trace_reads: int, + cancelled_ms_threshold: int, + cancelled_step_threshold: int, + timeout_keywords: tuple[str, ...], + unrecognized_keywords: tuple[str, ...], + biz_exclude_keywords: tuple[str, ...], +) -> list: + """编排:cancelled 走 trace 优先(读到确认没卡则不报,读不到回退保底);failed 附卡点。 + + trace 读取限量 max_trace_reads 次/轮;任何 trace 异常都在 trace_stuck 内部降级为 + 「读不到」,cancelled 因而回退保底、failed 不附卡点,绝不影响报警发送。 + """ + base = Path(work_log_dir) if work_log_dir else None + reads = 0 + hits: list = [] + for rec in records: + if rec.status == "cancelled": + res = None + if base is not None and reads < max_trace_reads: + td = _trace_dir(base, rec.trace_url) + if td is not None: + res = trace_stuck.read_stuck_points( + td, threshold=stuck_threshold, max_tail=max_tail + ) + reads += 1 + if res is not None and res.readable: + if res.points: + reason = "卡在 " + "、".join(sp.label() for sp in res.points) + hit = make_hit(rec, "T5", reason) + else: + hit = None # 读到且确认没卡 → 不报 + else: + hit = classify_cancelled_fallback( + rec, + cancelled_ms_threshold=cancelled_ms_threshold, + cancelled_step_threshold=cancelled_step_threshold, + ) + if hit is not None: + hits.append(hit) + else: + hit = classify_record( + rec, + cancelled_ms_threshold=cancelled_ms_threshold, + cancelled_step_threshold=cancelled_step_threshold, + timeout_keywords=timeout_keywords, + unrecognized_keywords=unrecognized_keywords, + biz_exclude_keywords=biz_exclude_keywords, + ) + if ( + hit is not None + and hit.alert_type in ("T1", "T2", "T6") + and base is not None + and reads < max_trace_reads + ): + td = _trace_dir(base, rec.trace_url) + if td is not None: + sp = trace_stuck.last_step(td) + reads += 1 + if sp is not None: + hit = _dc_replace(hit, reason=f"{hit.reason}｜卡在 {sp.label()}") + if hit is not None: + hits.append(hit) + return hits +``` + +(c) 在 `_scan_and_alert` 里，把 +```python + hits = classify_batch( + records, + cancelled_ms_threshold=settings.COMPARE_ALERT_CANCELLED_MS_THRESHOLD, + cancelled_step_threshold=settings.COMPARE_ALERT_CANCELLED_STEP_THRESHOLD, + timeout_keywords=settings.compare_alert_timeout_keywords, + unrecognized_keywords=settings.compare_alert_unrecognized_keywords, + biz_exclude_keywords=settings.compare_alert_biz_exclude_keywords, + ) +``` +替换为 +```python + hits = build_hits( + records, + work_log_dir=settings.COMPARE_ALERT_PRICEBOT_WORK_LOG_DIR, + stuck_threshold=settings.COMPARE_ALERT_STUCK_FRAME_THRESHOLD, + max_tail=settings.COMPARE_ALERT_TRACE_MAX_TAIL_FRAMES, + max_trace_reads=settings.COMPARE_ALERT_TRACE_MAX_RECORDS, + cancelled_ms_threshold=settings.COMPARE_ALERT_CANCELLED_MS_THRESHOLD, + cancelled_step_threshold=settings.COMPARE_ALERT_CANCELLED_STEP_THRESHOLD, + timeout_keywords=settings.compare_alert_timeout_keywords, + unrecognized_keywords=settings.compare_alert_unrecognized_keywords, + biz_exclude_keywords=settings.compare_alert_biz_exclude_keywords, + ) +``` + +- [ ] **Step 4: 跑测试确认通过** + +Run: `pytest tests/test_compare_alert_stuck_worker.py -q` +Expected: PASS（6 passed） + +- [ ] **Step 5: 跑报警相关全量测试确认不回归** + +Run: `pytest tests/ -q -k "compare_alert or trace_stuck"` +Expected: PASS（全绿） + +- [ ] **Step 6: Commit** + +```bash +git add app/core/compare_alert_worker.py tests/test_compare_alert_stuck_worker.py +git commit -m "feat(compare-alert): worker 编排 build_hits(cancelled trace 优先+保底、failed 附卡点) + +Co-Authored-By: Claude Opus 4.8 " +``` + +--- + +## 收尾 + +- [ ] **全量测试**：`pytest -q`（对齐 preexisting 失败基线，不新增失败） +- [ ] **lint**：`ruff check app/services/trace_stuck.py app/core/compare_alert_worker.py app/services/compare_alert.py` +- [ ] **本地联调（可选）**：把 `.env` 的 `COMPARE_ALERT_PRICEBOT_WORK_LOG_DIR` 指向本地 `E:\project\pricebot-backend\data\work_logs`，用真实 cancelled trace 目录验证卡点文案。 +- [ ] **清理临时脚本**：`git rm --cached` 无关，直接删 `scripts/_probe_trace_timing.py`（若确认不再用，另行确认 `scripts/_test_alert_card.py`）。 + +## 不在本 plan（后续单独排） + +- 卡片 schema 2.0 table 组件固化（`format_alert_card` + `send_feishu_card`，当前仍在 `scripts/_test_alert_card.py`）——卡点已随 `reason` 在现有 `format_alert_post` 展示，不阻塞本功能。 +- pricebot 侧改动（本方案零改 pricebot）。 +- `PIPELINE_STEP_LABELS` 全枚举补全（映射不到原样英文，可随线上观察增量补）。 diff --git a/docs/superpowers/specs/2026-08-04-compare-fail-alert-design.md b/docs/superpowers/specs/2026-08-04-compare-fail-alert-design.md new file mode 100644 index 0000000..8052660 --- /dev/null +++ b/docs/superpowers/specs/2026-08-04-compare-fail-alert-design.md @@ -0,0 +1,197 @@ +# 比价失败报警机制 · 设计文档 + +- **日期**：2026-08-04 +- **状态**：设计待评审（v2，含数据复审修订） +- **范围**：app-server（`shaguabijia-app-server`） +- **数据源**：`comparison_record` 单表（线上快照已导入本地 `cr_analysis` 分析库，3867 行，覆盖 2026-06-09 ~ 08-04） + +## 复审修订记录（2026-08-04, v2） + +结合 `cr_analysis` 实测数据复审后的改动： +1. **[必修] 水位改用 `updated_at`**：原 `created_at` 水位会系统性漏报慢失败（落定延迟 p99 达 7.5–10min）。改为给 `comparison_record` 加 `updated_at` 列、水位按 `updated_at` 单调推进（§5/§6）。 +2. **新增规则 T6「商品识别失败」**：「未识别到商品」96 条纳入报警，作为识别能力信号（§3）。 +3. **T1 加业务词排除**：清掉 `fail_reason IS NULL` 里 7 条 `information` 实为业务的误报（§3）。 +4. 补充：NULL 语义、时区口径、`business_type` 复核、单窗口截断阈值（§7/§9）。 +5. 数据证伪、未采纳的改动：T2 关键词已完备（35 条技术词全被「超时/启动/加载」覆盖），不扩。 + +--- + +## 1. 背景与目标 + +比价（外卖 `business_type=food`）由客户端无障碍自动化 + pricebot 多平台 LLM 驱动，链路长、失败形态多。目前**没有任何主动发现失败的手段**——只能人工翻库或等用户反馈。 + +**目标**：新增一个**近实时、记录级**的比价失败报警机制。每隔 15–30 分钟扫描新落定的比价记录，逐条按预定规则判定「是否属于需要关注的失败」，把命中的记录**汇总成一条飞书消息**发到告警群，并**报出每条触发的原因**。 + +**关键设计取向**（均由数据分析与评审确认）： +- **粒度是「记录」不是「失败率」**：逐条判定，不算比率、不设样本量门槛、不做基线对比。日均比价量小（完成约 45 条/天），比率方案在小窗口会剧烈抖动；记录级方案规避了这个问题。 +- **只报「技术性失败」「识别失败」与「深度放弃」**，不报正常业务结局。 +- **有触发才发，无触发静默**：不刷屏。 + +## 2. 数据分析依据（基线） + +全量 3867 条记录级 `status` 分布（详见附录 A）： + +| status | 数量 | 占比 | 说明 | +|---|---|---|---| +| success | 1501 | 38.8% | 成功（含 `below_minimum` 未满起送，被归一为 success） | +| cancelled | 1394 | 36.0% | 用户中途退出 | +| failed | 957 | 24.7% | 失败（T1 技术 386 + T2 超时 35 + T6 识别 96 + 业务 440） | +| running | 15 | 0.4% | 悬挂未收尾 | + +支撑规则设计的关键事实： +- **`failed` 是混合桶**：`fail_reason IS NULL` 的 393 条是纯系统技术失败（其中「比价过程出错，请稍后重试」占 294，是 `_GENERIC_INFO` 兜底黑话）；`fail_reason` 非空的多为业务结局，但夹杂「启动淘宝超时」等技术问题 35 条、「未识别到商品」96 条。 +- **业务失败不该报**：打烊、无此店、无此菜、未起送、单点不配送是正常结局。 +- **落定延迟很长**：`total_ms`（≈ 记录从建行到落定的时长）p99 = failed 449s、cancelled 602s，max 16min。**这是水位必须用 `updated_at` 而非 `created_at` 的直接依据**。 +- **cancelled 缺退出上下文**：99.3% 终止原因就一句「用户终止比价」，且 100% 没有 `platforms`/结果数据（都在 `running` 阶段被中止）。唯一可用信号是退出时机（`total_ms`/`step_count`）。cancelled 的 `step_count` 中位 5、p90 31；`total_ms` 中位 24s、p90 124s。参照系：一次成功比价中位 113s / 38 步。 + +## 3. 报警规则（v1） + +worker 每轮查询「上次水位之后有更新」的记录，对每条按下表判定；命中任一即计入本期汇总。四个规则互斥（一条记录最多归一类）。 + +| 类型 | 判定条件（SQL 语义） | 触发原因文案 | 历史量(2月) | +|---|---|---|---| +| **T1 系统技术失败** | `status='failed' AND fail_reason IS NULL AND (information IS NULL OR information !~ 业务词)` | `技术失败·{information 去空白截断; 空则"比价过程出错"}` | ≈386 | +| **T6 商品识别失败** | `status='failed' AND fail_reason ~ '未识别'` | `识别失败·未识别到商品` | ≈96 | +| **T2 超时/启动失败** | `status='failed' AND fail_reason IS NOT NULL AND fail_reason ~ 超时关键词` | `{fail_reason}`（如「启动淘宝超时」） | ≈35 | +| **T5 cancelled 深度放弃** | `status='cancelled' AND (total_ms > 90000 OR step_count > 30)` | `深度放弃·等待 {total_ms/1000 取整}s / {step_count} 步后退出` | ≈250 | + +**判定顺序（保证互斥）**： +1. `status='failed'`： + - `fail_reason IS NULL` → 若 `information` 命中**业务词**（`未找到|打烊|起送|门店|店内|不配送|这些菜|未入驻|休息`）则**不报**（业务失败漏派生 fail_reason，约 7 条）；否则 **T1**。 + - `fail_reason` 含「未识别」→ **T6**。 + - `fail_reason` 含超时关键词（`超时|启动|加载`）→ **T2**。 + - 其余（干净业务原因）→ **不报**。 +2. `status='cancelled'` 且（`total_ms>90000` 或 `step_count>30`）→ **T5**；否则不报。 +3. `status IN ('success','running')` → 不报。 + +**阈值/关键词（可配初值）**：T5 的 `90000ms`/`30步` 取自 cancelled 分布约 p90（评审选定「B 中档」）。超时关键词 `超时,启动,加载`、识别关键词 `未识别`、业务排除词均可配。 + +每条命中记录在汇总里附带：`trace_id`、`app_version`、`business_type`、触发原因文案、`created_at`。 + +## 4. 非目标与暂缓项 + +| 项 | 处理 | 原因 | +|---|---|---| +| 业务失败（打烊/无店/无菜/未起送/单点不配送） | **不报** | 正常业务结局 | +| success、早退 cancelled（≤90s 且 ≤30 步） | **不报** | 无报警价值 | +| **T3 running 悬挂** | **本期暂缓** | 评审决定先不报；但见下方 🔴 | +| **T4 单平台适配失效**（`platforms[].status='failed'`） | 暂不纳入（未来增强） | 量大、与整体失败重叠、噪音高 | +| 失败率 / cancelled 率等**比率型**指标 | 不做 | 本设计是记录级 | + +> 🔴 **待独立排查的回归线索**（非本报警范围，留档）：`running` 悬挂 15 条**全部集中在 2026-07-28 之后**，此前两个月几乎为 0。强烈提示某次发版后 `harvest_done`/`harvest_abort` 收尾链路（`app/repositories/comparison.py`）回归，建议单独开 issue。排查确认后可在 v2 把 T3 作为独立高频告警加回。 + +## 5. 架构与组件 + +沿用项目现有**常驻 asyncio worker** 范式（与 `heartbeat_monitor_worker` 等一致）。各组件单一职责、可独立测试： + +| 组件 | 路径 | 职责 | +|---|---|---| +| **数据模型改动** | `app/models/comparison.py` + alembic 迁移 | `comparison_record` 新增 `updated_at`（`server_default=func.now()`, `onupdate=func.now()`）+ 索引 `ix_comparison_updated`。为水位提供单调递增的落定时间。 | +| **规则模块** | `app/services/compare_alert.py` | 纯函数：输入一批 ORM 记录 → 输出 `[(记录, 触发类型, 原因文案)]`。判定逻辑与阈值全在此，无 I/O，易测易调。 | +| **扫描 worker** | `app/core/compare_alert_worker.py` | 仿 `heartbeat_monitor_worker`：单实例文件锁 + `asyncio` 轮询 + 优雅退出。每轮：读水位 → 查有更新记录 → 调规则 → 有命中则格式化并发飞书 → 推进水位。 | +| **飞书通知器** | `app/integrations/feishu_notifier.py` | 实现群机器人 webhook 发送。发送失败抛异常由 worker 处理。 | +| **水位存储** | 复用 `app_config` 表 | key=`compare_alert.last_watermark`，value=上次处理的最大 `updated_at`。 | +| **启停挂载** | `app/main.py` lifespan | `start_compare_alert_worker()` / `stop_compare_alert_worker()`，与现有 worker 同处注册。 | + +> **`onupdate` 生效前提**：现有 `harvest_done`/`harvest_abort`/`upsert_record` 均走 ORM `setattr`+`commit` 更新，`onupdate=func.now()` 会自动刷新 `updated_at`，无需改写路径。 + +## 6. 数据流与水位管理（updated_at 方案） + +``` +每 interval 秒： + 读 app_config['compare_alert.last_watermark'] → watermark + （空 → 冷启动：watermark = 当前 max(updated_at)，只报之后新落定的，不回溯历史） + 查 comparison_record + WHERE updated_at > watermark + ORDER BY updated_at ASC + 逐条套 T1/T6/T2/T5 规则 → 命中集合（按类型分组） + 若命中集合非空： + 格式化飞书消息 → feishu_notifier.send() + 成功 → 水位 = 本批 max(updated_at) + 失败 → 不更新水位（log），下一轮重扫补发 + 若命中集合为空： + 水位 = 本批 max(updated_at)（无记录则不动；可选 SEND_EMPTY 发简讯） +``` + +- **零漏报**：任何记录落定/更新时 `updated_at` 刷新为当前 DB 时钟 > 水位，必被下一轮扫到——无论 `created_at` 多早、落定多慢（根治了 `created_at` 水位漏掉慢失败的问题）。 +- **规避时区**：水位存的是 DB 产出的 `updated_at` 值，查询用 `updated_at` 自身比较，**不依赖 worker 本地时钟与 DB 时钟对齐**（`created_at` 存 naive 北京、`func.now()` 为 DB 时钟，二者口径不同，但本方案只用 `updated_at` 自比较，不受影响）。 +- **发送失败不推进水位**：保证不漏；恢复后一次补发。 +- **一条记录可能被扫多次**（running 更新→落定更新，`updated_at` 变两次）：但只有落定后 `status` 才命中规则，running 阶段扫到不命中，无副作用；不会重复报。 + +## 7. 飞书消息格式 + +群机器人消息（文本或富文本 `post`），按类型分组： + +``` +🚨 比价失败报警 · 2026-08-04 08:00–08:30 · 本期触发 7 条 +• 系统技术失败 3 条 + - trace abc123 | v0.3.4 | 比价过程出错 +• 商品识别失败 2 条 + - trace abc200 | v0.5.1 | 未识别到商品 +• 启动/超时失败 1 条 + - trace def456 | v0.3.4 | 启动淘宝超时 +• 深度放弃(cancelled) 1 条 + - trace ghi789 | v0.6.3 | 等待 98s / 26 步后退出 +``` + +- **截断阈值**：单类型明细超 `MAX_DETAIL_PER_TYPE`（默认 20）条时，只列前 20 条 + 「另有 N 条」；本期总命中超 `MAX_TOTAL`（默认 50）条时降级为只给各类型计数，提示去分析库查（防报警风暴，如 07-14 那种高失败日）。 + +## 8. 配置项 + +`app/core/config.py`（`pydantic-settings`）： + +| 配置 | 默认 | 说明 | +|---|---|---| +| `COMPARE_ALERT_ENABLED` | `False` | 总开关；关时 worker 不启动 | +| `COMPARE_ALERT_SCAN_INTERVAL_SEC` | `1800` | 扫描间隔，可配 900（15min） | +| `COMPARE_ALERT_FEISHU_WEBHOOK` | `""` | 群机器人 webhook；空则 worker 仅打日志不外发 | +| `COMPARE_ALERT_CANCELLED_MS_THRESHOLD` | `90000` | T5 耗时阈值（ms） | +| `COMPARE_ALERT_CANCELLED_STEP_THRESHOLD` | `30` | T5 步数阈值 | +| `COMPARE_ALERT_TIMEOUT_KEYWORDS` | `"超时,启动,加载"` | T2 关键词 | +| `COMPARE_ALERT_UNRECOGNIZED_KEYWORDS` | `"未识别"` | T6 关键词 | +| `COMPARE_ALERT_BIZ_EXCLUDE_KEYWORDS` | `"未找到,打烊,起送,门店,店内,不配送,这些菜,未入驻,休息"` | T1 的 information 业务词排除 | +| `COMPARE_ALERT_MAX_DETAIL_PER_TYPE` | `20` | 单类型明细截断 | +| `COMPARE_ALERT_MAX_TOTAL` | `50` | 本期总命中截断（超则只给计数） | +| `COMPARE_ALERT_SEND_EMPTY` | `False` | 无命中是否发「本期无异常」简讯 | + +> **`business_type` 复核**：当前数据全为 `food`，规则未按 `business_type` 限定。接入 `ecom`/`coupon` 时需复核各规则（尤其 T2/T6 关键词与 T5 阈值是否仍适用）。 + +## 9. 错误处理与边界 + +- **worker 单轮异常吞掉不退出**（`except Exception: logger.exception`），仿 heartbeat。 +- **DB / 飞书发送异常**：log，本轮不推进水位，下轮重试补发。 +- **单实例锁**：文件锁 `data/compare_alert.lock`（O_CREAT|O_EXCL + stale 检测）。 +- **NULL 语义**：T5 中 `total_ms`/`step_count` 为 NULL 的 cancelled，SQL 比较 `NULL>90000` 为 false → 不命中（无数据不报，符合预期）。T1 中 `information IS NULL` 时业务词排除不触发（视为非业务）→ 仍属 T1，原因文案兜底「比价过程出错」。 +- **冷启动不回溯历史**：首次启动水位=当前 `max(updated_at)`，避免把历史失败一次性全报。 + +## 10. 测试策略 + +仿现有 `tests/` 风格（`TestClient` + monkeypatch 外部依赖，SQLite 临时库）： + +- `test_compare_alert_rules.py`：喂各类记录（T1/T6/T2/T5 命中样本 + 业务失败/success/早退 cancelled/running 反例 + T1 业务词误入反例），断言分类与原因文案；覆盖阈值边界（`total_ms=90000` 不命中、`90001` 命中）与 NULL 语义。 +- `test_compare_alert_worker.py`：monkeypatch notifier 与 `SessionLocal`，验证 `updated_at` 水位推进、发送失败不推进、冷启动=max、命中汇总、截断逻辑。 +- `test_feishu_notifier.py`：monkeypatch HTTP，验证消息体格式与发送失败抛异常。 +- 迁移测试：`updated_at` 列 + 索引存在，`onupdate` 在 ORM 更新时刷新。 + +## 11. 未来增强 + +1. **T3 running 悬挂告警**：待第 4 节 🔴 回归排查后，作为独立高频告警加回。 +2. **T4 单平台适配失效**：`platforms[].status='failed'` 逐平台维度。 +3. **cancelled 退出上下文埋点**：客户端终止时上报退出阶段、已比出平台数、是否已看到中间结果——让 cancelled 从「只有时机」升级为「可归因」。 +4. **分维度统计**：汇总附带按 `app_version`/`source_platform` 的命中分布，辅助定位回归版本/平台。 +5. **趋势型报警**：记录级之上叠加比率/环比（需另设样本量保护）。 + +## 附录 A：分析数据来源与复现 + +- **来源**：线上 PostgreSQL 16 `pg_dump` 单表 `comparison_record`（plain SQL，187MB）。 +- **本地环境**：Docker 容器 `shaguabijia-pg`（postgres:16-alpine），独立分析库 `cr_analysis`（用户 `shaguabijia_app`）。导入：`docker cp` dump 进容器后 `psql -f`（末尾外键引用 `public.user` 报错属预期，单表 dump 无 user 表，不影响数据与索引）。 +- **样本**：3867 行，2026-06-09 ~ 08-04。 +- **关键分布**（供实现期回归对照）： + +| 指标 | success | failed | cancelled | +|---|---|---|---| +| 数量 | 1501 | 957 | 1394 | +| step_count 中位 / p90 | 38 / 61 | 22 / 47 | 5 / 31 | +| total_ms 中位 / p90 / p99 | 113s / 200s / 391s | 77s / 168s / 449s | 24s / 124s / 602s | + +- **failed 细分**（合计 957）：T1 系统技术 386（`fail_reason IS NULL` 393 − 业务误入 7）、T2 超时/启动 35、T6 识别失败 96、业务失败 440（含误入的 7 条）。 diff --git a/docs/superpowers/specs/2026-08-05-compare-stuck-detection-design.md b/docs/superpowers/specs/2026-08-05-compare-stuck-detection-design.md new file mode 100644 index 0000000..04ccf26 --- /dev/null +++ b/docs/superpowers/specs/2026-08-05-compare-stuck-detection-design.md @@ -0,0 +1,163 @@ +# 比价「卡死定位」报警增强设计 + +- 日期：2026-08-05 +- 分支：feat-compare-fail-alert（延续一期） +- 关联：`docs/superpowers/specs/2026-08-04-compare-fail-alert-design.md`（一期报警） + +## 1. 背景与问题 + +一期报警对 **cancelled（中途退出）** 的判据（`app/services/compare_alert.py` 的 `classify_record`）： + +``` +cancelled 且 (total_ms > 90s 或 step_count > 30步) → T5 深度放弃 +``` + +这个判据量的是「投入多少」，不是「卡没卡」，两头都错： + +- **漏报**：一进平台就卡在登录墙 / 加载失败，5 秒 2 步就退 → 判「不深度」→ 不报。但这是真卡死。 +- **误报**：用户正常挑了 100 秒、点了 40 步，比完价不满意退了 → `>90s` → 报「深度放弃」。但根本没卡。 + +根因：`total_ms`/`step_count` 是**整场**的量，把「卡在一步反复失败」和「正常深度使用」混为一谈。 + +### 1.1 数据佐证（真实 trace） + +- **卡死例**（`20260804_114703` meituan）：`pipeline_step` 从 `set_address`(step 0-1) → `enter_store`(3-8) → **`add_one_dish`(9 一路到 120+，110+ 帧全困在这一个环节)**。且 `timing.json` 里根本没有 meituan——`step_profiler` 只在平台 `is_done` 时落 timing，卡死平台永不 done。 +- **正常例**（`20260803_165239` eleme）：`set_address`(0) → `enter_store`(1-7) → done，每个环节 ≤7 帧就推进走了。 + +**卡死的结构特征**：某 `pipeline_step` 连续几十上百帧不变（原地打转）；正常则是逐环节推进、单环节 ≤7 帧。两者空档极大（7 vs 110+），可用一个帧数阈值干净区分，且**不需要大量数据归纳环节语义**。 + +## 2. 目标 + +- cancelled 判据：从「整场耗时/帧数阈值」→「trace 末段原地打转」，抓真卡死（含短时卡死）、不误报正常深度使用。 +- **判定与展示一体**：直接报「卡在 平台·环节」。 +- failed 类（T1/T2/T6）：判定不变，best-effort 补卡点定位。 +- 稳：读不到 trace 回退原耗时/帧数保底，**绝不阻断报警发送**。 + +## 3. 取数：同机直读（不改 pricebot） + +app-server 与 pricebot **同机**。trace 落盘在 `{WORK_LOG_DIR}/{dir_name}/`： + +- **dir_name 从 `comparison_record.trace_url` 尾段抠**：`trace_url = {base}/traces/{dir_name}/`，尾段就是磁盘目录名，新老格式都对得上，规避从 `trace_id` 反推老格式「首帧时刻」的难题。 +- 只读末段帧的**头部字段**（`pipeline_step` / `detected_page`），不解析后面的无障碍树（`windows`，占单帧 99% 体积）。 +- 不改 pricebot、不需要 `INTERNAL_API_SECRET`、不走网络。 + +> 备选途径（已否决）：pricebot 加内部接口（要改两仓 + secret）、公网 `trace_url` GET timing.json（本地 SSL 大面积超时 + timing.json 缺卡死平台）。同机直读最优。 + +## 4. 架构分层 + +保持判定纯函数、IO 单独成层： + +| 模块 | 职责 | 性质 | +|---|---|---| +| `services/compare_alert.py`（微调） | failed 判定不变；cancelled 只保留**回退保底**判定（`>90s`/`>30步`） | 纯函数 | +| `services/trace_stuck.py`（新） | 给定 trace 目录 → 逐平台读末段 → 判「原地打转」→ 返回卡点列表 | 薄 IO + 纯逻辑 | +| `core/compare_alert_worker.py`（编排） | 先跑纯 `classify_batch` 出候选，再对候选调 `trace_stuck` 增强 | 编排 | + +## 5. trace_stuck 模块 + +### 5.1 卡死判据（逐平台） + +对某平台的 `step_*.json` 序列，从**末帧往前**数，连续 `(pipeline_step, detected_page)` 都相同的帧数 ≥ N → 判该平台卡死，卡点 = 该 `pipeline_step`。 + +- `N = COMPARE_ALERT_STUCK_FRAME_THRESHOLD`（默认 **15**；正常环节 ≤7 帧、卡死 110+ 帧，空档极大）。 +- **「无推进」= `(pipeline_step, detected_page)` 双不变**（页面没跳转、环节没变）。这样能区分： + - 「加多菜」：`pipeline_step` 相同但 `detected_page` 在跳（换菜/回菜单）= 推进 → 不判卡死； + - 「卡在一步」：两者都不变 = 原地打转 → 卡死。 +- 从末帧往前最多读 `COMPARE_ALERT_TRACE_MAX_TAIL_FRAMES`（默认 **40**）帧，够判 ≥N 即停，防超长 trace 全读。 + +### 5.2 逐平台聚合（B 方案：不漏） + +一条 trace **逐平台**判，所有卡死平台都收集——不只「帧数最多」的那个。因为「帧数最多」会在**卡死平台帧数不是最多**时漏报（如另一平台正常加了 8 道菜跑了 30 帧、卡死平台一进就卡登录 5 帧退），而那恰是短时卡死。多个卡死平台都列进 reason。 + +### 5.3 接口 + +```python +@dataclass(frozen=True) +class StuckPoint: + platform: str + pipeline_step: str + frames: int # 末段连续困住的帧数 + +@dataclass(frozen=True) +class StuckResult: + readable: bool # trace 是否读到（区分「读不到」与「读到但没卡」） + points: list[StuckPoint] # 卡死平台列表；readable=True 且空 = 确认没卡死 + +def read_stuck_points(trace_dir: Path, *, threshold: int, max_tail: int) -> StuckResult: + """逐平台判卡死。目录不存在/异常/无平台帧 → StuckResult(readable=False, points=[])。""" + +def last_step(trace_dir: Path) -> StuckPoint | None: + """failed 用：取帧数最多平台的末帧环节（不要求原地打转）。读不到 → None。""" +``` + +## 6. 判定流 + +### 6.1 cancelled（trace 优先 → 保底） + +``` +worker 对 cancelled 候选: + res = read_stuck_points(dir) + if not res.readable: # 读不到 trace（目录被清/生产一时读不到）→ 回退保底 + >90s或>30步 → T5「深度放弃·等待Xs/Y步」; 否则不报 + elif res.points: # 读到且有卡死平台 → 报卡死 + 报 T5, reason = "卡在 " + "、".join(f"{平台}·{环节}" for res.points) + else: # 读到且没卡死（末段在推进 = 正常深度使用后退出）→ 不报 + 不报 +``` + +### 6.2 failed（T1/T2/T6，判定不变 + 附卡点） + +``` +worker 对 failed 命中: + sp = last_step(dir) # 读不到 → None + if sp: reason += f"｜卡在 {平台}·{环节}" +``` + +`failed` 只取「末帧停在哪」，不要求原地打转（它已失败、末帧即失败点）。intent 阶段就失败（无平台目录，典型 T6）→ 不附，reason 原样。 + +## 7. 卡点文案映射 + +`PIPELINE_STEP_LABELS`（小映射表，映射不到原样显示英文、不阻断）： + +| pipeline_step | 中文 | +|---|---| +| `set_address` | 定位 | +| `enter_store` | 进店 | +| `add_one_dish` | 加菜 | +| …（实现时按 pricebot 实际枚举补全） | | + +平台名同样映射（`meituan`→美团、`eleme`→饿了么、`jd_waimai`→京东外卖）。 + +## 8. 配置（`app/core/config.py`；路径敏感项放 `.env`） + +| 配置 | 默认 | 说明 | +|---|---|---| +| `COMPARE_ALERT_PRICEBOT_WORK_LOG_DIR` | `""` | pricebot work_logs 绝对路径；**空 = 跳过 trace、全走保底**（行为等同一期） | +| `COMPARE_ALERT_STUCK_FRAME_THRESHOLD` | `15` | N：末段连续同环节达此帧数判卡死 | +| `COMPARE_ALERT_TRACE_MAX_TAIL_FRAMES` | `40` | 每平台最多往前读多少帧 | +| `COMPARE_ALERT_TRACE_MAX_RECORDS` | `30` | 每轮最多对多少条命中记录读 trace（限量） | + +## 9. 展示 + +卡片「失败原因」列下附一行卡点小字，**不新增列**。cancelled 卡死时卡点即 reason 本身；failed 的卡点附在原因后。 + +> 依赖：本期展示复用一期卡片的「失败原因」列。若一期卡片（schema 2.0 table 组件，当前仍在临时脚本 `scripts/_test_alert_card.py`）尚未固化为正式 `format_alert_card` + `send_feishu_card`，本期实现时一并固化。 + +## 10. 降级与成本 + +- 只对命中记录读、限量 `MAX_RECORDS`、每平台只读末段头部字段、单文件读加超时。 +- **任何异常降级**：cancelled 回退保底、failed 不附卡点，绝不阻断报警。 +- `work_log_dir` 未配 → 整个 trace 增强跳过，行为等同一期（纯保底）。 + +## 11. 测试 + +- **trace_stuck 单测**：卡死正例（meituan 目录 → 判出 `add_one_dish`）、正常负例（eleme → 不判卡死）、加多菜不误判（`detected_page` 在变）、末段不足 N 帧、读不到目录降级。 +- **worker 集成**：trace 优先命中 vs 读不到回退保底切换；failed 附卡点；限量 `MAX_RECORDS` 生效。 +- fixture 用 tmp 造 `step_*.json`，**只含头部字段**（trace_id/step/platform/pipeline_step/detected_page）即可，不需无障碍树。 + +## 12. 不做（YAGNI） + +- 不改 pricebot（不加内部接口）。 +- 不落库（不加 `comparison_record` 列、不做迁移）。 +- 不做每帧耗时（`timing.json` 缺卡死平台，且报警用不上逐帧耗时）。 +- 不做环节黑白名单 / 语义分类（结构判据已够，且需大数据）。 diff --git a/tests/conftest.py b/tests/conftest.py index 581aa36..4ea1521 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -31,6 +31,7 @@ os.environ.setdefault("WXPAY_MCH_ID", "test-mch") os.environ.setdefault("WXPAY_MCH_SERIAL_NO", "test-serial") os.environ.setdefault("WXPAY_PUBLIC_KEY_ID", "test-pubkey-id") os.environ.setdefault("RATE_LIMIT_ENABLED", "false") # 限流内存计数会跨用例累加,测试关掉 +os.environ.setdefault("COMPARE_ALERT_ENABLED", "false") # 报警 worker 测试不启动(避免 .env 的 true 干扰 test_defaults) # 穿山甲发奖回调:测试里开启 + 给个 mock 验签密钥,test 内自签自验闭环 os.environ.setdefault("PANGLE_CALLBACK_ENABLED", "true") os.environ.setdefault("PANGLE_REWARD_SECRET", "test-pangle-secret-only-for-pytest") diff --git a/tests/test_compare_alert_config.py b/tests/test_compare_alert_config.py new file mode 100644 index 0000000..352d88b --- /dev/null +++ b/tests/test_compare_alert_config.py @@ -0,0 +1,17 @@ +"""比价报警配置默认值与关键词解析。""" +from __future__ import annotations + +from app.core.config import settings + + +def test_defaults() -> None: + assert settings.COMPARE_ALERT_ENABLED is False + assert settings.COMPARE_ALERT_SCAN_INTERVAL_SEC == 900 + assert settings.COMPARE_ALERT_CANCELLED_MS_THRESHOLD == 90000 + assert settings.COMPARE_ALERT_CANCELLED_STEP_THRESHOLD == 30 + + +def test_keyword_parsing() -> None: + assert settings.compare_alert_timeout_keywords == ("超时", "启动", "加载") + assert settings.compare_alert_unrecognized_keywords == ("未识别",) + assert "打烊" in settings.compare_alert_biz_exclude_keywords diff --git a/tests/test_compare_alert_fallback.py b/tests/test_compare_alert_fallback.py new file mode 100644 index 0000000..1e85e43 --- /dev/null +++ b/tests/test_compare_alert_fallback.py @@ -0,0 +1,62 @@ +"""classify_cancelled_fallback / make_hit 单测。""" +from app.services.compare_alert import classify_cancelled_fallback, make_hit + + +class _Rec: + def __init__(self, **kw): + self.trace_id = kw.get("trace_id", "t") + self.status = kw.get("status", "cancelled") + self.total_ms = kw.get("total_ms") + self.step_count = kw.get("step_count") + self.fail_reason = kw.get("fail_reason") + self.information = kw.get("information") + self.app_version = kw.get("app_version") + self.created_at = kw.get("created_at") + self.trace_url = kw.get("trace_url") + self.user_id = kw.get("user_id") + + +def test_fallback_deep_by_ms(): + hit = classify_cancelled_fallback( + _Rec(total_ms=95000, step_count=5), + cancelled_ms_threshold=90000, cancelled_step_threshold=30, + ) + assert hit is not None and hit.alert_type == "T5" and "深度放弃" in hit.reason + + +def test_fallback_deep_by_step(): + hit = classify_cancelled_fallback( + _Rec(total_ms=1000, step_count=35), + cancelled_ms_threshold=90000, cancelled_step_threshold=30, + ) + assert hit is not None and hit.alert_type == "T5" + + +def test_fallback_shallow_none(): + hit = classify_cancelled_fallback( + _Rec(total_ms=5000, step_count=3), + cancelled_ms_threshold=90000, cancelled_step_threshold=30, + ) + assert hit is None + + +def test_make_hit_carries_fields(): + hit = make_hit(_Rec(trace_id="tx", app_version="0.6.0"), "T5", "卡在 美团·加菜") + assert hit.trace_id == "tx" + assert hit.reason == "卡在 美团·加菜" + assert hit.app_version == "0.6.0" + + +def test_make_hit_carries_total_ms_and_step(): + hit = make_hit(_Rec(trace_id="tx", total_ms=602000, step_count=157), "T5", "深度放弃") + assert hit.total_ms == 602000 + assert hit.step_count == 157 + + +def test_make_hit_total_ms_step_default_none(): + # rec 没有这两个属性时安全降级为 None(不报错) + class _Bare: + trace_id = "b"; status = "cancelled"; reason = None + app_version = None; created_at = None; trace_url = None; user_id = None + hit = make_hit(_Bare(), "T1", "技术失败") + assert hit.total_ms is None and hit.step_count is None diff --git a/tests/test_compare_alert_format.py b/tests/test_compare_alert_format.py new file mode 100644 index 0000000..23f0dba --- /dev/null +++ b/tests/test_compare_alert_format.py @@ -0,0 +1,572 @@ +"""AlertHit[] → 飞书消息:分组 / 截断 / 含关键词(text + post 两种格式)。""" +from __future__ import annotations + +from datetime import datetime + +from app.services.compare_alert import AlertHit +from app.services.compare_alert_format import ALERT_KEYWORD, format_alert_message, format_alert_post + + +def _hits(n, alert_type="T1"): + return [ + AlertHit( + trace_id=f"t{i}", + alert_type=alert_type, + reason="比价过程出错", + app_version="v0.3.4", + created_at=None, + trace_url=None, + user_id=None, + ) + for i in range(n) + ] + + +# ---- format_alert_message(纯文本,保留原有测试) ---- + +def test_contains_keyword_and_count(): + msg = format_alert_message( + _hits(2), window_label="2026-08-04 08:00–08:30", + max_detail_per_type=20, max_total=50, + ) + assert ALERT_KEYWORD in msg + assert "本期触发 2 条" in msg + assert "系统技术失败 2 条" in msg + assert "t0" in msg and "v0.3.4" in msg + + +def test_group_by_type(): + hits = _hits(1, "T1") + _hits(1, "T6") + _hits(1, "T5") + msg = format_alert_message(hits, window_label="w", max_detail_per_type=20, max_total=50) + assert "系统技术失败 1 条" in msg + assert "商品识别失败 1 条" in msg + assert "深度放弃(cancelled) 1 条" in msg + + +def test_per_type_truncation(): + msg = format_alert_message(_hits(25), window_label="w", max_detail_per_type=20, max_total=50) + assert msg.count("t0") == 1 + assert "另有 5 条" in msg + + +def test_total_truncation_counts_only(): + msg = format_alert_message(_hits(60), window_label="w", max_detail_per_type=20, max_total=50) + assert "系统技术失败 60 条" in msg + assert "t0" not in msg + assert "分析库" in msg + + +# ---- format_alert_post(富文本 post) ---- + +def _hits_with_meta(n, alert_type="T1", *, user_id=None, trace_url=None, created_at=None): + return [ + AlertHit( + trace_id=f"tr{i}", + alert_type=alert_type, + reason="比价过程出错", + app_version="v1.2.3", + created_at=created_at or datetime(2026, 8, 4, 10, 30), + trace_url=trace_url, + user_id=user_id, + ) + for i in range(n) + ] + + +def test_post_title_contains_keyword(): + hits = _hits_with_meta(1) + title, content = format_alert_post( + hits, window_label="2026-08-04 10:00", phone_map={}, max_detail_per_type=20, max_total=50, + ) + assert ALERT_KEYWORD in title + + +def test_post_content_is_list(): + hits = _hits_with_meta(2) + title, content = format_alert_post( + hits, window_label="w", phone_map={}, max_detail_per_type=20, max_total=50, + ) + assert isinstance(content, list) + assert len(content) >= 1 + # 每个段落是 list[dict] + for para in content: + assert isinstance(para, list) + for elem in para: + assert "tag" in elem + + +def test_post_summary_count(): + hits = _hits_with_meta(3, "T1") + _hits_with_meta(2, "T6") + title, content = format_alert_post( + hits, window_label="w", phone_map={}, max_detail_per_type=20, max_total=50, + ) + # 摘要第一段含合计数和类型计数 + first_para_text = "".join(e.get("text", "") for e in content[0]) + assert "合计 5 条" in first_para_text + assert "系统技术失败 3" in first_para_text + assert "商品识别失败 2" in first_para_text + + +def test_post_phone_map_applied(): + hits = _hits_with_meta(1, user_id=42) + title, content = format_alert_post( + hits, window_label="w", phone_map={42: "13800138000"}, max_detail_per_type=20, max_total=50, + ) + # 找明细行(非摘要非表头)中含手机号 + all_text = " ".join( + e.get("text", "") for para in content for e in para + ) + assert "13800138000" in all_text + + +def test_post_no_user_id_shows_dash(): + hits = _hits_with_meta(1, user_id=None) + title, content = format_alert_post( + hits, window_label="w", phone_map={}, max_detail_per_type=20, max_total=50, + ) + all_text = " ".join(e.get("text", "") for para in content for e in para) + assert "｜ - ｜" in all_text + + +def test_post_trace_url_becomes_a_element(): + hits = _hits_with_meta(1, trace_url="https://trace.example.com/tr0") + title, content = format_alert_post( + hits, window_label="w", phone_map={}, max_detail_per_type=20, max_total=50, + ) + # 找 tag=a 的元素 + a_elements = [e for para in content for e in para if e.get("tag") == "a"] + assert len(a_elements) == 1 + assert a_elements[0]["href"] == "https://trace.example.com/tr0" + assert a_elements[0]["text"] == "trace" + + +def test_post_no_trace_url_shows_trace_id_prefix(): + hits = _hits_with_meta(1, trace_url=None) + title, content = format_alert_post( + hits, window_label="w", phone_map={}, max_detail_per_type=20, max_total=50, + ) + # 无 trace_url 时:tag=text, text=trace_id[:16] + detail_texts = [e.get("text", "") for para in content for e in para if e.get("tag") == "text"] + # trace_id 是 tr0,截 16 位 + assert any("tr0" in t for t in detail_texts) + + +def test_post_total_truncation_no_detail(): + hits = _hits_with_meta(60) + title, content = format_alert_post( + hits, window_label="w", phone_map={}, max_detail_per_type=20, max_total=50, + ) + # 超 max_total:只有摘要+截断提示,无表头、无明细 + all_text = " ".join(e.get("text", "") for para in content for e in para) + assert "合计 60 条" in all_text + assert "时间 ｜ 手机号" not in all_text + assert "tr0" not in all_text + assert "分析库" in all_text + + +def test_post_per_type_truncation(): + hits = _hits_with_meta(25) + title, content = format_alert_post( + hits, window_label="w", phone_map={}, max_detail_per_type=20, max_total=50, + ) + all_text = " ".join(e.get("text", "") for para in content for e in para) + assert "另有 5 条" in all_text + + +def test_post_created_at_formatting(): + hits = [ + AlertHit( + trace_id="tx1", + alert_type="T1", + reason="测试", + app_version="v2.0", + created_at=datetime(2026, 8, 4, 10, 30), + trace_url=None, + user_id=None, + ) + ] + title, content = format_alert_post( + hits, window_label="w", phone_map={}, max_detail_per_type=20, max_total=50, + ) + all_text = " ".join(e.get("text", "") for para in content for e in para) + assert "08-04 10:30" in all_text + + +def test_post_no_created_at_shows_dash(): + hits = [ + AlertHit( + trace_id="tx2", + alert_type="T1", + reason="测试", + app_version=None, + created_at=None, + trace_url=None, + user_id=None, + ) + ] + title, content = format_alert_post( + hits, window_label="w", phone_map={}, max_detail_per_type=20, max_total=50, + ) + all_text = " ".join(e.get("text", "") for para in content for e in para) + # 无 created_at 时间显示 "-" + assert "- ｜" in all_text + + +# ---- format_alert_card(schema 2.0 卡片) ---- + +from app.services.compare_alert_format import format_alert_card # noqa: E402 + + +def _card_hits(n, alert_type="T1", *, total_ms=None, step_count=None, user_id=None, + trace_url=None, created_at=None): + return [ + AlertHit( + trace_id=f"card{i}", + alert_type=alert_type, + reason="比价过程出错", + app_version="v1.5.0", + created_at=created_at or datetime(2026, 8, 4, 10, 30), + trace_url=trace_url, + user_id=user_id, + total_ms=total_ms, + step_count=step_count, + ) + for i in range(n) + ] + + +def test_card_schema_and_header(): + card = format_alert_card( + _card_hits(1), + window_label="2026-08-04 10:00", + phone_map={}, + interval_min=15, + max_detail_per_type=20, + max_total=50, + ) + assert card["schema"] == "2.0" + assert card["header"]["template"] == "red" + assert ALERT_KEYWORD in card["header"]["title"]["content"] + + +def test_card_body_markdown_summary(): + card = format_alert_card( + _card_hits(3), + window_label="2026-08-04 10:00", + phone_map={}, + interval_min=15, + max_detail_per_type=20, + max_total=50, + ) + elements = card["body"]["elements"] + md = elements[0] + assert md["tag"] == "markdown" + assert "近 15 分钟" in md["content"] + assert "合计 3 条" in md["content"] + + +def test_card_has_table_seven_columns(): + card = format_alert_card( + _card_hits(2), + window_label="w", + phone_map={}, + interval_min=10, + max_detail_per_type=20, + max_total=50, + ) + elements = card["body"]["elements"] + # 第二个元素是 table + assert len(elements) >= 2 + table = elements[1] + assert table["tag"] == "table" + cols = table["columns"] + assert len(cols) == 7 + display_names = [c["display_name"] for c in cols] + assert display_names == ["时间", "手机号", "用时", "失败原因", "卡点", "版本", "trace"] + # 「卡点」列在「失败原因」后、「版本」前 + names = [c["name"] for c in cols] + reason_idx = names.index("reason") + stuck_idx = names.index("stuck") + ver_idx = names.index("ver") + assert reason_idx < stuck_idx < ver_idx + # trace 列用 lark_md + trace_col = next(c for c in cols if c["name"] == "trace") + assert trace_col["data_type"] == "lark_md" + # 其余列 data_type 均为 text + for c in cols: + if c["name"] != "trace": + assert c["data_type"] == "text" + + +def test_card_cost_cell_format(): + """cost 格式: total_ms 和 step_count 均有值时 '{Ns} / {M步}'""" + hits = [ + AlertHit( + trace_id="t1", + alert_type="T1", + reason="r", + app_version="v1", + created_at=datetime(2026, 8, 4, 10, 0), + trace_url=None, + user_id=None, + total_ms=602_000, + step_count=157, + ) + ] + card = format_alert_card( + hits, + window_label="w", + phone_map={}, + interval_min=10, + max_detail_per_type=20, + max_total=50, + ) + table = card["body"]["elements"][1] + row = table["rows"][0] + assert row["cost"] == "602s / 157步" + + +def test_card_cost_only_ms(): + hits = [ + AlertHit( + trace_id="t2", + alert_type="T1", + reason="r", + app_version="v1", + created_at=None, + trace_url=None, + user_id=None, + total_ms=30_000, + step_count=None, + ) + ] + card = format_alert_card( + hits, + window_label="w", + phone_map={}, + interval_min=10, + max_detail_per_type=20, + max_total=50, + ) + table = card["body"]["elements"][1] + row = table["rows"][0] + assert row["cost"] == "30s" + + +def test_card_cost_only_steps(): + hits = [ + AlertHit( + trace_id="t3", + alert_type="T1", + reason="r", + app_version="v1", + created_at=None, + trace_url=None, + user_id=None, + total_ms=None, + step_count=42, + ) + ] + card = format_alert_card( + hits, + window_label="w", + phone_map={}, + interval_min=10, + max_detail_per_type=20, + max_total=50, + ) + table = card["body"]["elements"][1] + row = table["rows"][0] + assert row["cost"] == "42步" + + +def test_card_cost_none_when_both_missing(): + hits = _card_hits(1, total_ms=None, step_count=None) + card = format_alert_card( + hits, + window_label="w", + phone_map={}, + interval_min=10, + max_detail_per_type=20, + max_total=50, + ) + table = card["body"]["elements"][1] + row = table["rows"][0] + assert row["cost"] == "-" + + +def test_card_total_truncation_no_table(): + """超 max_total 时只有 markdown 摘要，没有 table 元素""" + card = format_alert_card( + _card_hits(60), + window_label="w", + phone_map={}, + interval_min=10, + max_detail_per_type=20, + max_total=50, + ) + elements = card["body"]["elements"] + assert len(elements) == 1 + assert elements[0]["tag"] == "markdown" + assert "comparison_record" in elements[0]["content"] + + +def test_card_empty_hits(): + """空 hits 返回含「本期无异常」的卡片（无 table）""" + card = format_alert_card( + [], + window_label="w", + phone_map={}, + interval_min=10, + max_detail_per_type=20, + max_total=50, + ) + assert card["schema"] == "2.0" + elements = card["body"]["elements"] + assert len(elements) == 1 + assert elements[0]["tag"] == "markdown" + assert "本期无异常" in elements[0]["content"] + + +def test_card_rows_count_respects_per_type_limit(): + """每类型最多 max_detail_per_type 条""" + hits = _card_hits(25, "T1") + card = format_alert_card( + hits, + window_label="w", + phone_map={}, + interval_min=10, + max_detail_per_type=20, + max_total=50, + ) + table = card["body"]["elements"][1] + assert len(table["rows"]) == 20 + + +def test_card_trace_url_becomes_markdown_link(): + hits = _card_hits(1, trace_url="https://trace.example.com/t0") + card = format_alert_card( + hits, + window_label="w", + phone_map={}, + interval_min=10, + max_detail_per_type=20, + max_total=50, + ) + table = card["body"]["elements"][1] + row = table["rows"][0] + assert row["trace"] == "[链接](https://trace.example.com/t0)" + + +def test_card_no_trace_url_shows_trace_id_prefix(): + hits = _card_hits(1, trace_url=None) + card = format_alert_card( + hits, + window_label="w", + phone_map={}, + interval_min=10, + max_detail_per_type=20, + max_total=50, + ) + table = card["body"]["elements"][1] + row = table["rows"][0] + # trace_id = "card0"[:12] + assert row["trace"] == "card0" + + +def test_card_phone_from_map(): + hits = _card_hits(1, user_id=7) + card = format_alert_card( + hits, + window_label="w", + phone_map={7: "13912345678"}, + interval_min=10, + max_detail_per_type=20, + max_total=50, + ) + table = card["body"]["elements"][1] + row = table["rows"][0] + assert row["phone"] == "13912345678" + + +def test_card_version_not_truncated(): + """版本完整显示，不缩写""" + hits = [ + AlertHit( + trace_id="tv1", + alert_type="T1", + reason="r", + app_version="v2.15.3-release", + created_at=None, + trace_url=None, + user_id=None, + ) + ] + card = format_alert_card( + hits, + window_label="w", + phone_map={}, + interval_min=10, + max_detail_per_type=20, + max_total=50, + ) + table = card["body"]["elements"][1] + row = table["rows"][0] + assert row["ver"] == "v2.15.3-release" + + +def test_card_table_has_page_size_and_header_style(): + card = format_alert_card( + _card_hits(1), + window_label="w", + phone_map={}, + interval_min=10, + max_detail_per_type=20, + max_total=50, + ) + table = card["body"]["elements"][1] + assert "page_size" in table + assert "header_style" in table + assert table["header_style"].get("background_style") == "grey" + assert table["header_style"].get("bold") is True + + +def test_card_stuck_point_shown_in_row(): + """stuck_point 有值时行中 stuck 列正确显示；无值时显示「-」。""" + import dataclasses + + from app.services.compare_alert import make_hit + + class _Rec: + trace_id = "sp1" + status = "cancelled" + total_ms = 90000 + step_count = 20 + fail_reason = None + information = None + app_version = "v1.0" + created_at = datetime(2026, 8, 5, 10, 0) + trace_url = None + user_id = None + + hit_with = dataclasses.replace( + make_hit(_Rec(), "T5", "深度放弃"), + stuck_point="美团·加菜 110帧/32s", + ) + hit_without = make_hit(_Rec(), "T5", "深度放弃") # stuck_point=None + + card_with = format_alert_card( + [hit_with], + window_label="w", phone_map={}, interval_min=5, + max_detail_per_type=20, max_total=50, + ) + row_with = card_with["body"]["elements"][1]["rows"][0] + assert row_with["stuck"] == "美团·加菜 110帧/32s" + + card_without = format_alert_card( + [hit_without], + window_label="w", phone_map={}, interval_min=5, + max_detail_per_type=20, max_total=50, + ) + row_without = card_without["body"]["elements"][1]["rows"][0] + assert row_without["stuck"] == "-" diff --git a/tests/test_compare_alert_migration.py b/tests/test_compare_alert_migration.py new file mode 100644 index 0000000..2541a9b --- /dev/null +++ b/tests/test_compare_alert_migration.py @@ -0,0 +1,46 @@ +"""updated_at 列存在、回填 = created_at、onupdate 在 ORM 更新时刷新。""" +from __future__ import annotations + +import time + +from sqlalchemy import inspect + +from app.db.session import SessionLocal, engine +from app.models.comparison import ComparisonRecord + + +def test_updated_at_column_and_index_exist() -> None: + insp = inspect(engine) + cols = {c["name"] for c in insp.get_columns("comparison_record")} + assert "updated_at" in cols + idx_names = {i["name"] for i in insp.get_indexes("comparison_record")} + # 迁移环境手动建索引名为 ix_comparison_updated; + # 测试环境 create_all() 按 SQLAlchemy 命名约定生成 ix_comparison_record_updated_at。 + # 两种环境都验通过即可。 + assert ( + "ix_comparison_updated" in idx_names + or "ix_comparison_record_updated_at" in idx_names + ), f"updated_at index not found; available indexes: {idx_names}" + + +def test_onupdate_refreshes_updated_at() -> None: + db = SessionLocal() + try: + rec = ComparisonRecord(trace_id="alert-updated-at-onupdate", status="running") + db.add(rec) + db.commit() + db.refresh(rec) + first = rec.updated_at + assert first is not None + time.sleep(1.1) # SQLite CURRENT_TIMESTAMP 秒级精度,睡过 1 秒才看得出变化 + rec.status = "failed" + db.commit() + db.refresh(rec) + assert rec.updated_at > first + finally: + db.rollback() + db.query(ComparisonRecord).filter( + ComparisonRecord.trace_id == "alert-updated-at-onupdate" + ).delete() + db.commit() + db.close() diff --git a/tests/test_compare_alert_rules.py b/tests/test_compare_alert_rules.py new file mode 100644 index 0000000..712645c --- /dev/null +++ b/tests/test_compare_alert_rules.py @@ -0,0 +1,61 @@ +"""比价报警规则分类:记录 → AlertHit | None(纯函数,不碰 DB)。""" +from __future__ import annotations + +from types import SimpleNamespace + +import pytest + +from app.services.compare_alert import classify_record + +KW = dict( + cancelled_ms_threshold=90000, + cancelled_step_threshold=30, + timeout_keywords=("超时", "启动", "加载"), + unrecognized_keywords=("未识别",), + biz_exclude_keywords=("未找到", "打烊", "起送", "门店", "店内", "不配送", "这些菜", "未入驻", "休息"), +) + + +def _rec(**kw): + base = dict( + status="failed", fail_reason=None, information=None, + total_ms=None, step_count=None, trace_id="t", app_version=None, business_type="food", + ) + base.update(kw) + return SimpleNamespace(**base) + + +@pytest.mark.parametrize( + ("rec", "expected_type"), + [ + (_rec(status="failed", fail_reason=None, information="比价过程出错，请稍后重试"), "T1"), + (_rec(status="failed", fail_reason=None, information=None), "T1"), + (_rec(status="failed", fail_reason=None, information="美团外卖门店已打烊，无法比价"), None), + (_rec(status="failed", fail_reason="未识别到商品"), "T6"), + (_rec(status="failed", fail_reason="启动淘宝超时, 请稍后重试"), "T2"), + (_rec(status="failed", fail_reason="淘宝闪购店内未找到这些菜品"), None), + (_rec(status="cancelled", total_ms=98000, step_count=10), "T5"), + (_rec(status="cancelled", total_ms=20000, step_count=31), "T5"), + (_rec(status="cancelled", total_ms=20000, step_count=5), None), + (_rec(status="cancelled", total_ms=None, step_count=None), None), + (_rec(status="success"), None), + (_rec(status="running"), None), + ], +) +def test_classify_record_type(rec, expected_type): + hit = classify_record(rec, **KW) + assert (hit.alert_type if hit else None) == expected_type + + +def test_t5_boundary_exclusive(): + assert classify_record(_rec(status="cancelled", total_ms=90000, step_count=30), **KW) is None + assert classify_record(_rec(status="cancelled", total_ms=90001, step_count=30), **KW).alert_type == "T5" + + +def test_reason_texts(): + t1 = classify_record(_rec(status="failed", fail_reason=None, information=" 比价过程出错 "), **KW) + assert t1.reason == "技术失败·比价过程出错" + t1_empty = classify_record(_rec(status="failed", fail_reason=None, information=None), **KW) + assert t1_empty.reason == "技术失败·比价过程出错" + t5 = classify_record(_rec(status="cancelled", total_ms=98000, step_count=26), **KW) + assert t5.reason == "深度放弃" diff --git a/tests/test_compare_alert_stuck_worker.py b/tests/test_compare_alert_stuck_worker.py new file mode 100644 index 0000000..fe9cb61 --- /dev/null +++ b/tests/test_compare_alert_stuck_worker.py @@ -0,0 +1,129 @@ +"""build_hits 集成测:cancelled trace 优先/保底切换、failed 附卡点、限量。""" +import json +from pathlib import Path + +from app.core.compare_alert_worker import _fmt_stuck, build_hits +from app.services.trace_stuck import StuckPoint + + +class _Rec: + def __init__(self, **kw): + self.trace_id = kw.get("trace_id", "t") + self.status = kw.get("status", "cancelled") + self.total_ms = kw.get("total_ms") + self.step_count = kw.get("step_count") + self.fail_reason = kw.get("fail_reason") + self.information = kw.get("information") + self.app_version = kw.get("app_version") + self.created_at = kw.get("created_at") + self.trace_url = kw.get("trace_url") + self.user_id = kw.get("user_id") + + +def _frame(pdir: Path, idx: int, step: str, page: str) -> None: + pdir.mkdir(parents=True, exist_ok=True) + body = {"pipeline_step": step, "detected_page": page, "windows": [{"n": ["x" * 200]}]} + (pdir / f"step_{idx:03d}.json").write_text( + json.dumps(body, ensure_ascii=False), encoding="utf-8" + ) + + +_KW = dict( + stuck_threshold=15, max_tail=40, max_trace_reads=30, + cancelled_ms_threshold=90000, cancelled_step_threshold=30, + timeout_keywords=("超时",), unrecognized_keywords=("未识别",), biz_exclude_keywords=(), +) + + +def test_cancelled_stuck_reports_via_trace(tmp_path): + for i in range(18): + _frame(tmp_path / "20260804_x" / "meituan", i, "add_one_dish", "meal_detail_popup") + rec = _Rec(status="cancelled", trace_url="https://x/traces/20260804_x/", + total_ms=5000, step_count=3) # 保底不会中,靠 trace 判卡死 + hits = build_hits([rec], work_log_dir=str(tmp_path), **_KW) + assert len(hits) == 1 + assert hits[0].alert_type == "T5" + assert hits[0].reason == "深度放弃" + assert "美团·加菜" in hits[0].stuck_point + assert "帧" in hits[0].stuck_point + + +def test_cancelled_readable_not_stuck_no_report(tmp_path): + # trace 确认没卡(在推进);即便 total_ms/step 超阈值也不报(信 trace,不回退保底) + p = tmp_path / "20260804_y" / "eleme" + _frame(p, 0, "set_address", "home") + for i in range(1, 6): + _frame(p, i, "enter_store", "store") + rec = _Rec(status="cancelled", trace_url="https://x/traces/20260804_y/", + total_ms=95000, step_count=40) + hits = build_hits([rec], work_log_dir=str(tmp_path), **_KW) + assert hits == [] + + +def test_cancelled_unreadable_falls_back(tmp_path): + rec = _Rec(status="cancelled", trace_url="https://x/traces/nope/", + total_ms=95000, step_count=3) + hits = build_hits([rec], work_log_dir=str(tmp_path), **_KW) + assert len(hits) == 1 + assert hits[0].alert_type == "T5" and "深度放弃" in hits[0].reason + + +def test_no_work_log_dir_uses_fallback(tmp_path): + rec = _Rec(status="cancelled", trace_url="https://x/traces/y/", + total_ms=95000, step_count=3) + hits = build_hits([rec], work_log_dir="", **_KW) + assert len(hits) == 1 and "深度放弃" in hits[0].reason + + +def test_failed_gets_stuck_point_appended(tmp_path): + for i in range(20): + _frame(tmp_path / "20260804_f" / "meituan", i, "add_one_dish", "meal_detail_popup") + rec = _Rec(status="failed", fail_reason="启动超时", + trace_url="https://x/traces/20260804_f/") + hits = build_hits([rec], work_log_dir=str(tmp_path), **_KW) + assert len(hits) == 1 + assert hits[0].alert_type == "T2" + assert "美团·加菜" in hits[0].stuck_point + assert "卡在" not in hits[0].reason + + +def test_max_trace_reads_zero_skips_trace(tmp_path): + for i in range(18): + _frame(tmp_path / "20260804_z" / "meituan", i, "add_one_dish", "meal_detail_popup") + rec = _Rec(status="cancelled", trace_url="https://x/traces/20260804_z/", + total_ms=95000, step_count=3) + kw = {**_KW, "max_trace_reads": 0} + hits = build_hits([rec], work_log_dir=str(tmp_path), **kw) + # 没读 trace → 回退保底 → deep(95s) → 深度放弃 + assert len(hits) == 1 and "深度放弃" in hits[0].reason + + +def test_shared_reads_budget_across_branches(tmp_path): + # cancelled 和 failed 共用 max_trace_reads 预算;预算=1 时 cancelled 先消耗,failed 拿不到卡点 + for i in range(18): + _frame(tmp_path / "20260804_c" / "meituan", i, "add_one_dish", "meal_detail_popup") + for i in range(20): + _frame(tmp_path / "20260804_d" / "meituan", i, "add_one_dish", "meal_detail_popup") + cancelled = _Rec(status="cancelled", trace_url="https://x/traces/20260804_c/", + total_ms=5000, step_count=3) + failed = _Rec(status="failed", fail_reason="启动超时", + trace_url="https://x/traces/20260804_d/") + kw = {**_KW, "max_trace_reads": 1} + hits = build_hits([cancelled, failed], work_log_dir=str(tmp_path), **kw) + assert len(hits) == 2 + # cancelled 消耗了唯一预算 → 报卡死，卡点在 stuck_point + assert hits[0].alert_type == "T5" and hits[0].stuck_point is not None + # failed 超预算 → 仍报 T2,但 stuck_point 为 None + assert hits[1].alert_type == "T2" and hits[1].stuck_point is None + + +# ---- _fmt_stuck 单测 ---- + +def test_fmt_stuck_with_ms(): + sp = StuckPoint(platform="meituan", pipeline_step="add_one_dish", frames=110, stuck_ms=32000) + assert _fmt_stuck(sp) == "美团·加菜 110帧/32s" + + +def test_fmt_stuck_without_ms(): + sp = StuckPoint(platform="meituan", pipeline_step="add_one_dish", frames=110, stuck_ms=None) + assert _fmt_stuck(sp) == "美团·加菜 110帧" diff --git a/tests/test_compare_alert_worker.py b/tests/test_compare_alert_worker.py new file mode 100644 index 0000000..ab2c880 --- /dev/null +++ b/tests/test_compare_alert_worker.py @@ -0,0 +1,93 @@ +"""扫描 worker 的同步核心 _scan_and_alert:水位读写 / 冷启动 / 发送成功推进、失败不推进。 + +monkeypatch 掉真正的飞书发送;用真实 SQLite(SessionLocal)插入几条记录。 +""" +from __future__ import annotations + +import time + +import pytest + +from app.core import compare_alert_worker as w +from app.db.session import SessionLocal +from app.models.app_config import AppConfig +from app.models.comparison import ComparisonRecord + +WM_KEY = w.WATERMARK_KEY + + +@pytest.fixture() +def clean_db(): + db = SessionLocal() + db.query(ComparisonRecord).delete() + db.query(AppConfig).filter(AppConfig.key == WM_KEY).delete() + db.commit() + yield db + db.query(ComparisonRecord).delete() + db.query(AppConfig).filter(AppConfig.key == WM_KEY).delete() + db.commit() + db.close() + + +def _add(db, trace, status, **kw): + rec = ComparisonRecord(trace_id=trace, status=status, **kw) + db.add(rec) + db.commit() + db.refresh(rec) + return rec + + +def _card_title(card: dict) -> str: + return card.get("header", {}).get("title", {}).get("content", "") + + +def _card_md_content(card: dict) -> str: + elements = card.get("body", {}).get("elements", []) + return elements[0].get("content", "") if elements else "" + + +def test_cold_start_sets_watermark_no_alert(clean_db, monkeypatch): + # 冷启动:表非空 → 水位=当前 max(updated_at),不回溯已有历史失败、不报 + _add(clean_db, "old-fail", "failed", fail_reason=None, information="比价过程出错") + sent = [] + monkeypatch.setattr(w, "_send_card", lambda card: sent.append(card)) + w._scan_and_alert() + assert sent == [] # 冷启动不报历史 + row = clean_db.get(AppConfig, WM_KEY) + assert row is not None # 水位已初始化 + + +def test_alerts_on_new_failed_and_advances(clean_db, monkeypatch): + # seed 一条 + 冷启动建水位;sleep 1.1s 拉开时间(SQLite 秒级精度,否则新记录同秒、追不上水位) + _add(clean_db, "seed", "running") + monkeypatch.setattr(w, "_send_card", lambda card: None) + w._scan_and_alert() # 冷启动,水位=seed.updated_at + time.sleep(1.1) + sent = [] + monkeypatch.setattr(w, "_send_card", lambda card: sent.append(card)) + _add(clean_db, "new-fail", "failed", fail_reason=None, information="比价过程出错") + w._scan_and_alert() + assert len(sent) == 1 + card = sent[0] + # title 含关键词 + assert "比价失败报警" in _card_title(card) + # markdown 摘要含"系统技术失败" + assert "系统技术失败 1" in _card_md_content(card) + + +def test_send_failure_does_not_advance_watermark(clean_db, monkeypatch): + _add(clean_db, "seed2", "running") + monkeypatch.setattr(w, "_send_card", lambda card: None) + w._scan_and_alert() # 冷启动建水位 + wm_before = clean_db.get(AppConfig, WM_KEY).value + time.sleep(1.1) + _add(clean_db, "fail-send", "failed", fail_reason=None, information="比价过程出错") + + def boom(card): + raise w.feishu_notifier.FeishuNotifyError("down") + + monkeypatch.setattr(w, "_send_card", boom) + w._scan_and_alert() # 发送失败 + clean_db.expire_all() + wm_after = clean_db.get(AppConfig, WM_KEY).value + assert wm_after == wm_before # 未推进,下轮补发 diff --git a/tests/test_feishu_notifier.py b/tests/test_feishu_notifier.py new file mode 100644 index 0000000..12b0ced --- /dev/null +++ b/tests/test_feishu_notifier.py @@ -0,0 +1,169 @@ +"""飞书群机器人发送:text/post 消息体格式 / 关键词失败 / 网络失败,均 monkeypatch httpx 不真发。""" +from __future__ import annotations + +import httpx +import pytest + +from app.integrations import feishu_notifier + +# ---- send_feishu_text ---- + +def test_send_posts_text_payload(monkeypatch): + captured = {} + + def fake_post(url, json, timeout): + captured["url"] = url + captured["json"] = json + return httpx.Response(200, json={"code": 0, "msg": "success"}) + + monkeypatch.setattr(feishu_notifier.httpx, "post", fake_post) + feishu_notifier.send_feishu_text("https://open.feishu.cn/hook/xxx", "比价失败报警 · test", timeout=5.0) + assert captured["url"] == "https://open.feishu.cn/hook/xxx" + assert captured["json"] == {"msg_type": "text", "content": {"text": "比价失败报警 · test"}} + + +def test_send_raises_on_keyword_rejection(monkeypatch): + def fake_post(url, json, timeout): + return httpx.Response(200, json={"code": 19024, "msg": "Key Words Not Found"}) + + monkeypatch.setattr(feishu_notifier.httpx, "post", fake_post) + with pytest.raises(feishu_notifier.FeishuNotifyError): + feishu_notifier.send_feishu_text("https://hook", "无关键词文本", timeout=5.0) + + +def test_send_raises_on_http_error(monkeypatch): + def fake_post(url, json, timeout): + return httpx.Response(500, text="boom") + + monkeypatch.setattr(feishu_notifier.httpx, "post", fake_post) + with pytest.raises(feishu_notifier.FeishuNotifyError): + feishu_notifier.send_feishu_text("https://hook", "比价失败报警", timeout=5.0) + + +# ---- send_feishu_post ---- + +def test_send_post_payload_structure(monkeypatch): + """send_feishu_post 发出 msg_type=post 的 payload,结构符合飞书 post 格式。""" + captured = {} + + def fake_post(url, json, timeout): + captured["url"] = url + captured["json"] = json + return httpx.Response(200, json={"code": 0, "msg": "success"}) + + monkeypatch.setattr(feishu_notifier.httpx, "post", fake_post) + content = [[{"tag": "text", "text": "合计 3 条"}]] + feishu_notifier.send_feishu_post( + "https://open.feishu.cn/hook/yyy", + "🚨 比价失败报警 · 2026-08-04", + content, + timeout=5.0, + ) + assert captured["url"] == "https://open.feishu.cn/hook/yyy" + payload = captured["json"] + assert payload["msg_type"] == "post" + zh_cn = payload["content"]["post"]["zh_cn"] + assert zh_cn["title"] == "🚨 比价失败报警 · 2026-08-04" + assert zh_cn["content"] == content + + +def test_send_post_raises_on_code_nonzero(monkeypatch): + """send_feishu_post code!=0 时抛 FeishuNotifyError。""" + def fake_post(url, json, timeout): + return httpx.Response(200, json={"code": 19024, "msg": "Key Words Not Found"}) + + monkeypatch.setattr(feishu_notifier.httpx, "post", fake_post) + with pytest.raises(feishu_notifier.FeishuNotifyError): + feishu_notifier.send_feishu_post( + "https://hook", "🚨 比价失败报警", [[{"tag": "text", "text": "test"}]], timeout=5.0 + ) + + +def test_send_post_raises_on_http_error(monkeypatch): + """send_feishu_post 非 2xx 抛 FeishuNotifyError。""" + def fake_post(url, json, timeout): + return httpx.Response(500, text="internal server error") + + monkeypatch.setattr(feishu_notifier.httpx, "post", fake_post) + with pytest.raises(feishu_notifier.FeishuNotifyError): + feishu_notifier.send_feishu_post( + "https://hook", "🚨 比价失败报警", [[{"tag": "text", "text": "test"}]], timeout=5.0 + ) + + +def test_send_post_raises_on_network_error(monkeypatch): + """send_feishu_post 网络异常抛 FeishuNotifyError。""" + def fake_post(url, json, timeout): + raise httpx.ConnectError("connection refused") + + monkeypatch.setattr(feishu_notifier.httpx, "post", fake_post) + with pytest.raises(feishu_notifier.FeishuNotifyError): + feishu_notifier.send_feishu_post( + "https://hook", "🚨 比价失败报警", [[{"tag": "text", "text": "test"}]], timeout=5.0 + ) + + +# ---- send_feishu_card ---- + +def test_send_card_payload_structure(monkeypatch): + """send_feishu_card 发出 msg_type=interactive + card 字段的 payload。""" + captured = {} + + def fake_post(url, json, timeout): + captured["url"] = url + captured["json"] = json + return httpx.Response(200, json={"code": 0, "msg": "success"}) + + monkeypatch.setattr(feishu_notifier.httpx, "post", fake_post) + card = { + "schema": "2.0", + "header": {"title": {"tag": "plain_text", "content": "🚨 比价失败报警 · test"}, "template": "red"}, + "body": {"elements": [{"tag": "markdown", "content": "近 15 分钟"}]}, + } + feishu_notifier.send_feishu_card("https://open.feishu.cn/hook/zzz", card, timeout=5.0) + assert captured["url"] == "https://open.feishu.cn/hook/zzz" + assert captured["json"]["msg_type"] == "interactive" + assert captured["json"]["card"] is card + + +def test_send_card_uses_default_timeout(monkeypatch): + """send_feishu_card 默认 timeout=10.0。""" + captured = {} + + def fake_post(url, json, timeout): + captured["timeout"] = timeout + return httpx.Response(200, json={"code": 0, "msg": "success"}) + + monkeypatch.setattr(feishu_notifier.httpx, "post", fake_post) + feishu_notifier.send_feishu_card("https://hook", {"schema": "2.0"}) + assert captured["timeout"] == 10.0 + + +def test_send_card_raises_on_code_nonzero(monkeypatch): + """send_feishu_card code!=0 时抛 FeishuNotifyError。""" + def fake_post(url, json, timeout): + return httpx.Response(200, json={"code": 19024, "msg": "Key Words Not Found"}) + + monkeypatch.setattr(feishu_notifier.httpx, "post", fake_post) + with pytest.raises(feishu_notifier.FeishuNotifyError): + feishu_notifier.send_feishu_card("https://hook", {"schema": "2.0"}) + + +def test_send_card_raises_on_http_error(monkeypatch): + """send_feishu_card 非 2xx 抛 FeishuNotifyError。""" + def fake_post(url, json, timeout): + return httpx.Response(500, text="boom") + + monkeypatch.setattr(feishu_notifier.httpx, "post", fake_post) + with pytest.raises(feishu_notifier.FeishuNotifyError): + feishu_notifier.send_feishu_card("https://hook", {"schema": "2.0"}) + + +def test_send_card_raises_on_network_error(monkeypatch): + """send_feishu_card 网络异常抛 FeishuNotifyError。""" + def fake_post(url, json, timeout): + raise httpx.ConnectError("connection refused") + + monkeypatch.setattr(feishu_notifier.httpx, "post", fake_post) + with pytest.raises(feishu_notifier.FeishuNotifyError): + feishu_notifier.send_feishu_card("https://hook", {"schema": "2.0"}) diff --git a/tests/test_trace_stuck.py b/tests/test_trace_stuck.py new file mode 100644 index 0000000..72aa367 --- /dev/null +++ b/tests/test_trace_stuck.py @@ -0,0 +1,154 @@ +"""trace_stuck 单测:用 tmp 造 step_*.json(只含头部字段)验证卡死判据。""" +import json +from pathlib import Path + +from app.services.trace_stuck import ( + StuckPoint, + dir_name_from_trace_url, + last_step, + read_stuck_points, +) + + +def _frame(pdir: Path, idx: int, step: str, page: str) -> None: + """造一帧 step json:头部放 pipeline_step/detected_page,尾部塞大 windows 模拟真实。""" + pdir.mkdir(parents=True, exist_ok=True) + body = { + "trace_id": "t", "step": idx, "platform": pdir.name, + "pipeline_step": step, "detected_page": page, + "windows": [{"nodes": ["x" * 200]}], + } + (pdir / f"step_{idx:03d}.json").write_text( + json.dumps(body, ensure_ascii=False), encoding="utf-8" + ) + + +def test_stuck_when_tail_repeats_same_step(tmp_path): + pdir = tmp_path / "meituan" + for i in range(20): + _frame(pdir, i, "add_one_dish", "meal_detail_popup") + res = read_stuck_points(tmp_path, threshold=15, max_tail=40) + assert res.readable is True + assert res.points == [StuckPoint("meituan", "add_one_dish", 20)] + + +def test_not_stuck_when_progressing(tmp_path): + pdir = tmp_path / "eleme" + _frame(pdir, 0, "set_address", "home") + for i in range(1, 8): + _frame(pdir, i, "enter_store", "store") + res = read_stuck_points(tmp_path, threshold=15, max_tail=40) + assert res.readable is True + assert res.points == [] + + +def test_adding_many_dishes_not_stuck_when_page_changes(tmp_path): + # add_one_dish 重复但 detected_page 在跳(换菜)=推进,不判卡死 + pdir = tmp_path / "meituan" + for i in range(20): + _frame(pdir, i, "add_one_dish", "menu" if i % 2 == 0 else "dish_popup") + res = read_stuck_points(tmp_path, threshold=15, max_tail=40) + assert res.points == [] + + +def test_below_threshold_not_stuck(tmp_path): + pdir = tmp_path / "meituan" + for i in range(10): # < 15 + _frame(pdir, i, "add_one_dish", "meal_detail_popup") + res = read_stuck_points(tmp_path, threshold=15, max_tail=40) + assert res.points == [] + + +def test_missing_dir_not_readable(tmp_path): + res = read_stuck_points(tmp_path / "nope", threshold=15, max_tail=40) + assert res.readable is False + assert res.points == [] + + +def test_empty_dir_no_platform_frames_not_readable(tmp_path): + (tmp_path / "emptysub").mkdir() + res = read_stuck_points(tmp_path, threshold=15, max_tail=40) + assert res.readable is False + + +def test_per_platform_one_stuck_one_normal(tmp_path): + m = tmp_path / "meituan" + for i in range(18): + _frame(m, i, "add_one_dish", "meal_detail_popup") + e = tmp_path / "eleme" + _frame(e, 0, "set_address", "home") + for i in range(1, 6): + _frame(e, i, "enter_store", "store") + res = read_stuck_points(tmp_path, threshold=15, max_tail=40) + assert res.readable is True + assert res.points == [StuckPoint("meituan", "add_one_dish", 18)] + + +def test_last_step_returns_busiest_platform_last_env(tmp_path): + m = tmp_path / "meituan" + for i in range(20): + _frame(m, i, "add_one_dish", "meal_detail_popup") + e = tmp_path / "eleme" + for i in range(3): + _frame(e, i, "enter_store", "store") + sp = last_step(tmp_path) + assert sp == StuckPoint("meituan", "add_one_dish", 20) + + +def test_dir_name_from_trace_url(): + assert dir_name_from_trace_url("https://x/traces/20260804_1_abc/") == "20260804_1_abc" + assert dir_name_from_trace_url("https://x/traces/20260804_1_abc") == "20260804_1_abc" + assert dir_name_from_trace_url("") is None + assert dir_name_from_trace_url(None) is None + + +def test_corrupt_frame_does_not_crash(tmp_path): + # 截断的 UTF-8(pricebot 被 SIGTERM 打断的末帧)不应抛 UnicodeDecodeError + pdir = tmp_path / "meituan" + pdir.mkdir() + (pdir / "step_000.json").write_bytes(b'{"pipeline_step": "add\xff') + res = read_stuck_points(tmp_path, threshold=1, max_tail=40) + assert res.points == [] # 抠不出字段 → 不判卡死;关键是没崩溃 + + +def _frame_ts(pdir: Path, idx: int, step: str, page: str, ts: str) -> None: + pdir.mkdir(parents=True, exist_ok=True) + body = {"pipeline_step": step, "detected_page": page, "timestamp": ts, + "windows": [{"n": ["x" * 200]}]} + (pdir / f"step_{idx:03d}.json").write_text( + json.dumps(body, ensure_ascii=False), encoding="utf-8" + ) + + +def test_stuck_ms_computed_from_timestamps(tmp_path): + # 末段 16 帧都卡在 add_one_dish,timestamp 从 :00 到 :30(每帧+2s) → 卡住 30s + pdir = tmp_path / "meituan" + for i in range(16): + _frame_ts(pdir, i, "add_one_dish", "meal_detail_popup", + f"2026-08-04T12:00:{i * 2:02d}.000000") + res = read_stuck_points(tmp_path, threshold=15, max_tail=40) + assert len(res.points) == 1 + sp = res.points[0] + assert sp.frames == 16 + assert sp.stuck_ms == 30000 # (第15帧:30 - 第0帧:00) = 30s + + +def test_stuck_ms_none_when_timestamp_missing(tmp_path): + # 帧无 timestamp 字段 → stuck_ms 为 None(不报错) + pdir = tmp_path / "meituan" + for i in range(16): + _frame(pdir, i, "add_one_dish", "meal_detail_popup") # 现有 _frame,无 timestamp + res = read_stuck_points(tmp_path, threshold=15, max_tail=40) + assert len(res.points) == 1 + assert res.points[0].stuck_ms is None + + +def test_stuck_ms_none_when_clock_goes_backwards(tmp_path): + # 帧 timestamp 非单调(时钟回退):step_000=:30 ... step_015=:00 → 末帧早于段首 → 负时长 → 降级 None + pdir = tmp_path / "meituan" + for i in range(16): + _frame_ts(pdir, i, "add_one_dish", "meal_detail_popup", + f"2026-08-04T12:00:{(30 - i * 2):02d}.000000") + res = read_stuck_points(tmp_path, threshold=15, max_tail=40) + assert len(res.points) == 1 + assert res.points[0].stuck_ms is None # 负时长降级为 None