feat(observability): 抓取去掉首页预热,交易补齐链路埋点
两个问题一起处理,都与「出站请求与可观测性」有关。 ## 抓取:正常路径不再多打一次首页 site_session 原先每条通道每 30 分钟打一次 www.rakuten.co.jp/ 做预热,而且预热 返回非 2xx 时 warmed_at 不置位——那种情况下每个请求前都会再打一次首页。 Akamai 的 cookie 随任意页面响应下发,目标页自己就会带回来,专门先打一次首页除了 多一个出站请求(以及多一次被风控计数的机会)之外没有额外收益:首个请求无论打哪个 URL 都是冷的 ~11s,之后都复用 cookie。 改为 cookie 由目标页响应建立(_note_cookies)、超 TTL 主动清空 (_drop_expired_cookies)。首页只保留在失败修复路径上(_rewarm_on_home):目标页 已经吃了挑战页时,拿首页换一套干净 cookie 比继续撞同一个 URL 更安全。happy path 的出站请求数 2 → 1。 _note_cookies 刻意不在每次响应时刷新时刻:TTL 要从「这套 cookie 第一次出现」算起, 每次都刷新会让一套 cookie 被无限续命,反而绕过了 session_ttl_seconds 的本意。 profile_status() 的 warmed 字段名保留(上游健康检查看板在用),语义改为「当前有 可复用的 Akamai cookie」,不再代表「已专门预热过首页」。 ## 交易:此前没有任何有意义的链路数据 根因是 trading 的实际工作两类自动埋点都覆盖不到:站点交互走 Playwright(不经 httpx),worker 主循环是后台 asyncio 任务(没有 HTTP 入口,因此没有根 span)。 于是发给网关的每次 httpx 调用各自成为孤立 trace——观测后台上只剩一堆请求记录。 新增手工埋点: - order.task:一笔下单的根 span,一个 task_id 一条 trace,带 order.route (execute / recovery / already_finished)与终态 order.terminal_status - order.step.*:清车 → 加购 → 校验 → 确认 → 提交 → 付款,每步一个子 span, 带 order.evidence_ref,可从 span 直接定位落盘证据 - site.*:12 个 Playwright 交互方法(用 traced 装饰器而非 with 块——这些方法的 函数体本就很长,再加一层缩进不利于阅读) - account_query:只读查询单的根 span,带 query.outcome 空转的长轮询(30 秒一次、绝大多数返回空)用 suppressed() 屏蔽:量大且没有信息量, 把观测后台刷满的正是它们。领到任务后的网关调用都在任务根 span 底下,不受影响。 闸门 / 风控拦截会被 _execute_with_renewal 吞掉转 needs_human,异常冒不到根 span, 被拦下的单在 trace 里跟成功下单一模一样。加 _execute_recording_errors 一层统一 记录,比每个 except 分支各写一遍省事,也不会漏掉后续新增的分支。 _report_safe 写 span 属性前判断 is_recording():付款后监控是 create_task 起的, asyncio 在创建时就把 context 复制了进去,等它真正跑起来根 span 早已结束—— get_current_span() 拿到的仍是那个已结束的 span(不是 INVALID_SPAN),写属性会打 "Setting attribute on ended span"。当前监控路径不传 terminal_status 走不到那里, 这道判断是防以后。 ## 顺带修掉:instrument_app 从未生效 instrument_app 用 _provider is None 做前置判断,但三个服务都在模块导入时执行 app = create_app(),而 setup_telemetry 要等 lifespan 才跑——那时 _provider 还是 None,照着判断直接 return。**FastAPI 从来没被打桩过,三个服务一条 server span 都没有。** 实测确认两件事:导入期打桩能出 span,lifespan 内打桩出不来(instrument_app 是加 中间件,应用开始服务后加进去不生效);provider 后设也不影响 ProxyTracer 委托到 真实 provider。所以只能在导入期装,判断条件改为 otel_enabled。 app/gateway/main.py 此前完全没接 telemetry,worker 出站请求带过来的 traceparent 没人接上,一条下单链路在网关这里断掉,只看得到 worker 侧那半截。补上 setup_telemetry(service_name="rakuten-gateway") 与 instrument_app / shutdown。 ## 验证 新增 8 个用例:首页零请求、cookie 复用与过期清空、失败后用首页换 cookie、一任务 一 trace 的父子结构、闸门失败标 ERROR、空转不埋点,以及 instrument_app 调用顺序 的回归测试。全量 526 passed。 Playwright 那些 site.* 埋点只做了静态验证(测试用桩替换站点方法),没有跑真实 浏览器下单确认 span 真的落地。 Co-Authored-By: Claude Opus 5 (1M context) <noreply@anthropic.com>
This commit is contained in:
+134
-35
@@ -28,6 +28,9 @@ import contextlib
|
||||
import logging
|
||||
from typing import TYPE_CHECKING
|
||||
|
||||
from opentelemetry import trace
|
||||
from opentelemetry.trace import SpanKind, Status, StatusCode
|
||||
|
||||
from app.shared.errors import (
|
||||
AppError,
|
||||
BrowserDeadError,
|
||||
@@ -35,6 +38,7 @@ from app.shared.errors import (
|
||||
OrderGuardError,
|
||||
)
|
||||
from app.shared.task_state import OrderState, TaskStatus
|
||||
from app.shared.telemetry import record_error, set_attributes, suppressed
|
||||
from app.trading.worker import verify
|
||||
from app.trading.worker.client import GatewayClient
|
||||
from app.trading.worker.evidence import EvidenceStore
|
||||
@@ -47,6 +51,8 @@ if TYPE_CHECKING:
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
tracer = trace.get_tracer(__name__)
|
||||
|
||||
|
||||
def _coerce_state(value: str | None) -> OrderState:
|
||||
"""把网关返回的 state 字符串安全地包成 OrderState;None 或非法值回落到 CREATED"""
|
||||
@@ -115,7 +121,11 @@ class WorkerRunner:
|
||||
logger.info("worker 启动:worker_id=%s", self.worker_id)
|
||||
while self._running:
|
||||
try:
|
||||
task = await self._gateway.lease(self.worker_id, wait=30)
|
||||
# 空转的长轮询不埋点:30 秒一次、绝大多数返回空,每次都会变成一条
|
||||
# 孤立 trace 把观测后台刷满。领到任务后的调用都在 handle() 的根
|
||||
# span 底下,不受这里影响。
|
||||
with suppressed():
|
||||
task = await self._gateway.lease(self.worker_id, wait=30)
|
||||
except AppError as exc:
|
||||
logger.warning("lease 失败:%s (err=%s)", exc.message, exc.err_code)
|
||||
await asyncio.sleep(5)
|
||||
@@ -137,13 +147,42 @@ class WorkerRunner:
|
||||
# ---- 单任务调度 ----
|
||||
|
||||
async def handle(self, task: LeaseTask) -> None:
|
||||
"""单任务调度入口:本地幂等闸门 → 恢复核对 → 执行"""
|
||||
"""单任务调度入口:本地幂等闸门 → 恢复核对 → 执行
|
||||
|
||||
这里开的 span 是**整条下单链路的根**:往下的每一步站点交互、每一次回报
|
||||
网关都挂在它底下,一个 task_id 对应观测后台里的一条 trace。worker 是后台
|
||||
asyncio 任务,没有 HTTP 入口,不开这个根 span 的话下游 httpx 调用会各自
|
||||
散成孤立 trace(这正是「只有请求记录、没有链路」的原因)。
|
||||
"""
|
||||
with tracer.start_as_current_span("order.task", kind=SpanKind.CONSUMER) as span:
|
||||
intent = task.intent or {}
|
||||
set_attributes(
|
||||
span,
|
||||
{
|
||||
"order.task_id": task.task_id,
|
||||
"order.site": task.site,
|
||||
"order.worker_id": self.worker_id,
|
||||
"order.lease_count": task.lease_count,
|
||||
"order.known_state": task.known_state,
|
||||
"order.item_url": intent.get("item_url"),
|
||||
"order.quantity": intent.get("quantity"),
|
||||
},
|
||||
)
|
||||
try:
|
||||
await self._dispatch(task, span)
|
||||
except Exception as exc:
|
||||
record_error(span, exc)
|
||||
raise
|
||||
|
||||
async def _dispatch(self, task: LeaseTask, span: "trace.Span") -> None:
|
||||
"""handle() 的实际分支逻辑,拆出来只为让根 span 的 with 块保持一层缩进"""
|
||||
# 本地幂等闸门:之前已完成过的任务不再执行
|
||||
if await self._db.has_finished(task.task_id):
|
||||
final_state = await self._db.final_state(task.task_id)
|
||||
logger.info(
|
||||
"本地已完成,补报终态:task_id=%s state=%s", task.task_id, final_state
|
||||
)
|
||||
span.set_attribute("order.route", "already_finished")
|
||||
await self._report_safe(
|
||||
task,
|
||||
state=_coerce_state(final_state),
|
||||
@@ -154,10 +193,12 @@ class WorkerRunner:
|
||||
|
||||
# 恢复领取:lease_count > 1 表示 stale → reclaim,必须先核对站点订单
|
||||
if task.lease_count > 1:
|
||||
span.set_attribute("order.route", "recovery")
|
||||
await self._handle_recovery(task)
|
||||
return
|
||||
|
||||
# 常规执行
|
||||
span.set_attribute("order.route", "execute")
|
||||
await self._execute_with_renewal(task)
|
||||
|
||||
async def _handle_recovery(self, task: LeaseTask) -> None:
|
||||
@@ -197,11 +238,24 @@ class WorkerRunner:
|
||||
|
||||
# ---- 常规执行 ----
|
||||
|
||||
async def _execute_recording_errors(self, task: LeaseTask) -> None:
|
||||
"""execute() 外面加一层:异常先记到任务根 span 上,再原样抛出
|
||||
|
||||
下面那一串 except 分支会把异常**吞掉**转成 needs_human / failed 回报,
|
||||
异常冒不到 handle() 的根 span。在这里统一记一次,比每个分支各写一遍省事,
|
||||
也不会漏掉新增的分支。
|
||||
"""
|
||||
try:
|
||||
await self.execute(task)
|
||||
except Exception as exc:
|
||||
record_error(trace.get_current_span(), exc)
|
||||
raise
|
||||
|
||||
async def _execute_with_renewal(self, task: LeaseTask) -> None:
|
||||
"""在租约自动续期的上下文里执行任务"""
|
||||
async with self._renew_lease_every(task, interval=60):
|
||||
try:
|
||||
await self.execute(task)
|
||||
await self._execute_recording_errors(task)
|
||||
except NotImplementedError as exc:
|
||||
# 站点交互未实现(规格 §10):上报 needs_human,不视为 worker 失败
|
||||
logger.warning(
|
||||
@@ -521,38 +575,65 @@ class WorkerRunner:
|
||||
证据载体)时,其 html / screenshot 即本步骤要落盘的页面与整页截图;
|
||||
显式传入的 `html` / `png` 优先级更高,供 step3/step4 等已经单独拿到
|
||||
页面的调用点使用。
|
||||
|
||||
每步一个 span,挂在 handle() 的任务根 span 底下:一条 trace 就是一单的
|
||||
完整流水(清车 → 加购 → 校验 → 确认 → 提交 → 付款),卡在哪一步、每步
|
||||
耗时多少、证据落在哪个 evidence_ref 都能直接读出来。
|
||||
"""
|
||||
result = await action()
|
||||
if isinstance(result, PageSnapshot):
|
||||
if html is None:
|
||||
html = result.html or None
|
||||
if png is None:
|
||||
png = result.screenshot or None
|
||||
elif html is None and isinstance(result, str):
|
||||
# 旧契约兼容:站点方法返回字符串时视为页面 HTML
|
||||
html = result
|
||||
meta = {
|
||||
"step": step_name,
|
||||
"state": state.value,
|
||||
**(evidence_meta or {}),
|
||||
}
|
||||
evidence_ref = self._evidence.write_step(
|
||||
task.task_id, step_no, step_name, html=html, png=png, meta=meta
|
||||
)
|
||||
await self._db.index_evidence(task.task_id, step_no, step_name, evidence_ref)
|
||||
await self._db.record_event(
|
||||
task.task_id, state.value, detail=detail, evidence_ref=evidence_ref
|
||||
)
|
||||
await self._gateway.report(
|
||||
task.task_id,
|
||||
self.worker_id,
|
||||
state=state,
|
||||
payable_yen=payable_yen,
|
||||
pay_deadline=pay_deadline,
|
||||
site_order_id=site_order_id,
|
||||
evidence_ref=evidence_ref,
|
||||
detail=detail,
|
||||
)
|
||||
with tracer.start_as_current_span(f"order.step.{step_name}") as span:
|
||||
set_attributes(
|
||||
span,
|
||||
{
|
||||
"order.task_id": task.task_id,
|
||||
"order.step_no": step_no,
|
||||
"order.step_name": step_name,
|
||||
"order.state": state.value,
|
||||
},
|
||||
)
|
||||
try:
|
||||
result = await action()
|
||||
except Exception as exc:
|
||||
record_error(span, exc)
|
||||
raise
|
||||
|
||||
if isinstance(result, PageSnapshot):
|
||||
if html is None:
|
||||
html = result.html or None
|
||||
if png is None:
|
||||
png = result.screenshot or None
|
||||
elif html is None and isinstance(result, str):
|
||||
# 旧契约兼容:站点方法返回字符串时视为页面 HTML
|
||||
html = result
|
||||
meta = {
|
||||
"step": step_name,
|
||||
"state": state.value,
|
||||
**(evidence_meta or {}),
|
||||
}
|
||||
evidence_ref = self._evidence.write_step(
|
||||
task.task_id, step_no, step_name, html=html, png=png, meta=meta
|
||||
)
|
||||
set_attributes(
|
||||
span,
|
||||
{
|
||||
"order.evidence_ref": evidence_ref,
|
||||
"order.site_order_id": site_order_id,
|
||||
"order.payable_yen": payable_yen,
|
||||
},
|
||||
)
|
||||
await self._db.index_evidence(task.task_id, step_no, step_name, evidence_ref)
|
||||
await self._db.record_event(
|
||||
task.task_id, state.value, detail=detail, evidence_ref=evidence_ref
|
||||
)
|
||||
await self._gateway.report(
|
||||
task.task_id,
|
||||
self.worker_id,
|
||||
state=state,
|
||||
payable_yen=payable_yen,
|
||||
pay_deadline=pay_deadline,
|
||||
site_order_id=site_order_id,
|
||||
evidence_ref=evidence_ref,
|
||||
detail=detail,
|
||||
)
|
||||
|
||||
async def _report_safe(
|
||||
self,
|
||||
@@ -566,7 +647,25 @@ class WorkerRunner:
|
||||
payable_yen: int | None = None,
|
||||
pay_deadline: str | None = None,
|
||||
) -> None:
|
||||
"""回报 gateway,失败只记日志不抛——主循环不能因为回报失败退出"""
|
||||
"""回报 gateway,失败只记日志不抛——主循环不能因为回报失败退出
|
||||
|
||||
顺带把终态标到当前 span 上。这一步是必要的:`_execute_with_renewal` 会把
|
||||
闸门拦截、风控拦截、浏览器掉线等异常**吞掉**转成 needs_human 回报,异常
|
||||
不会冒到 handle() 的根 span,于是一笔被拦下的单在 trace 里看起来跟成功
|
||||
下单一模一样。所有这些分支都汇到这个方法,标在这里最省事也最不容易漏。
|
||||
|
||||
`is_recording()` 那道判断不是多余的:付款后监控(_monitor_order)是
|
||||
`create_task` 起的后台任务,而 asyncio 在创建时就把当时的 context 复制了
|
||||
进去——等它真正跑起来,任务根 span 早已结束,但 `get_current_span()` 拿到
|
||||
的仍是那个**已结束**的 span(不是 INVALID_SPAN)。往上写属性会打
|
||||
"Setting attribute on ended span" 警告。当前监控路径不传 terminal_status
|
||||
走不到这里,但这道判断保证以后传了也不会污染已完成的任务 span。
|
||||
"""
|
||||
span = trace.get_current_span()
|
||||
if terminal_status is not None and span.is_recording():
|
||||
span.set_attribute("order.terminal_status", terminal_status.value)
|
||||
if terminal_status in (TaskStatus.NEEDS_HUMAN, TaskStatus.FAILED):
|
||||
span.set_status(Status(StatusCode.ERROR, detail))
|
||||
try:
|
||||
await self._gateway.report(
|
||||
task.task_id,
|
||||
|
||||
Reference in New Issue
Block a user