refactor(account): S2 读写一次切换 —— CK/健康/重登彻底搬到账号,登录记录退化为历史
S1 加好了列,本提交把**读与写一起切**过去。这是设计里「同一份数据的读与写必须
在同一个提交里切」的要求:先切读或先切写都会出现「写账号、读记录」的中间态,
跑起来不报错,只是默默用错数据。
## 核心改动
**判定「谁有 CK」不再借道登录记录**
- `cookie_holding_filter()` / `cookie_operable_filter()` / `cookie_holding_account_ids_query()`
/ `visible_douyu_task_accounts()` 全部改为作用于 `Account`;
- `cookie_operable_filter()` 的「在途排除」仍查 `login_tasks`(那是**某次尝试**的属性),
用 `NOT EXISTS` 子查询 + 迁移 0042 新增的 `(account_id, attempt_status)` 索引;
- 删除死代码 `latest_cookie_login_task()`;`latest_cookie()` 改为直接读账号。
**CK 写入**
- `_write_login_success()` 写 `Account.cookie / cookie_updated_at / cookie_source`;
- `_mark_attempt_failed()` 改看**账号**有没有 CK(决定是否沿用旧的失败语义);
- 检测模式的 `task.cookie = ""` **不再清空账号的 CK**(见下「行为变更」)。
**自动化作业按账号调度**
- `ck_health`:`_select_tasks` → `_select_accounts`,`db.query(Account)`;新增
`count_remaining()`(刻意不带 LIMIT,否则积压数恒等于批次上限);
- `ck_relogin`:同样按账号选行;结算改为「账号 + 本轮尝试记录」,
退避进度写账号;`settle_fresh_results(batch_id)` 按 `attempt_batch_id` 配对,
`run_ck_relogin` 把 **account_ids** 交给 runner;
- 顺带让**没有任何登录历史的账号**也能被巡检/重登(以前它们永远进不了队列)。
**登录批次**
- `LoginBatchRunner` 去掉 `relogin_task_ids`,重登与登录统一走 `account_ids`;
- 批次行复用逻辑重写:只保证「同一账号不并发两行」,不再关心「哪一行持有 CK」。
失败行复用同时看 `attempt_status`(账号有 CK 时 `_mark_attempt_failed` 刻意不动
`status`,只看 status 会永远命不中 → 每次失败刷一行)。
**删除护栏语义变更(用户可感知)**
- 去掉「持有 CK 的记录不可删」——CK 在账号上,删历史不再影响登录态,历史终于可清理;
- **保留**「在途尝试不可删」(批次线程正拿着那条记录)。
## 行为变更(需评审)
1. **账号检测不再清空 CK**。旧代码在 check 模式写 `task.cookie = ""`,而那时记录的
cookie 就是账号唯一凭证 —— 一次「批量检测」会把账号登录态删掉。解耦后
「检测账号凭据」与「CK 是否有效」本就是两件事(后者归 `ck_health` 判),
继续清空等于凭空制造不可逆的数据丢失(重登当前成功率极低)。
2. **Cookie 页的记录单位从「登录记录」变成「账号」**,所有接口的 id 入参/出参
变成 `account_id`;`DELETE /cookies/{id}` 清的是该账号的 CK,**必写审计**。
3. 重复检测从「同一账号多行持 CK」改为「**同一个 CK 出现在多个账号上**」——
后者才是真风险(两个账号共用一个登录态)。用 sha256 指纹分组:
CK 是 AES-GCM + 随机 nonce 加密,**同一明文每次密文都不同**,比密文永远比不出重复。
## 修掉一个真实并发缺陷
`_write_login_success` 原先被传入 `self.db`(runner 主会话),而它是在**工作线程**里
用 `worker_db` 执行的。`emit()` 会往 `account_events` 插一行,其 `account_id` 外键对
`accounts` 那一行加**共享锁**;这把锁加在另一个会话上,紧接着本会话的
`UPDATE accounts SET cookie=...` 就被自己人挡住 —— 实测 **50 秒后
`Lock wait timeout exceeded`,登录明明成功却写不回 CK**(改造前 UPDATE 打在
`login_tasks` 上所以没撞上)。改为传 `worker_db`:事件本就该和状态变更同事务提交,
顺带修掉跨线程共用一个 Session 的隐患。
## 验证
- 全量 pytest **368 passed / 6 skipped**;迁移链 3 passed;
- 前端 `tsc -b && vite build` 通过;
- `ruff check` / `ruff format --check` 全绿(顺带修掉 main 上遗留的 F821);
- 新增/改写的关键回归:**删掉全部登录历史后账号的 CK 与健康状态仍在**、
**无登录历史的账号也能进巡检队列**、结算不碰 CK、删除护栏改为在途语义。
仍未做(设计里排期的后续阶段):`account_lifecycle` 状态收口(S3)、事件发射点补齐(S4)、
业务事实直发事件(S5)、账号页重构(S6)。注意 `status` 目前只有模型默认值与迁移回填,
所以 `disabled/archived` 还没有任何写入方——S3 落地时要把自动化选行的状态过滤一起补上。
This commit is contained in:
@@ -12,7 +12,7 @@ from typing import Any
|
||||
|
||||
from fastapi import APIRouter, Depends, HTTPException, Query
|
||||
from pydantic import BaseModel, Field
|
||||
from sqlalchemy import and_, or_
|
||||
from sqlalchemy import and_, or_, select
|
||||
from sqlalchemy.orm import Session
|
||||
|
||||
from ..database import get_db
|
||||
@@ -203,27 +203,36 @@ def manual_queue(
|
||||
"""
|
||||
from ..services.automation.jobs.ck_relogin import MANUAL_KINDS
|
||||
|
||||
# 转人工 = 分型属于需人工,或者次数用尽(next_run_at 已清空且失败过)。
|
||||
# 解耦后这一队列的单位是**账号**(2026-10-04):重登进度与 CK 都在 accounts 上。
|
||||
# 「最近尝试时间」仍来自登录记录(那是尝试的属性),用相关子查询取最后一条。
|
||||
latest_attempt_at = (
|
||||
select(LoginTask.attempt_at)
|
||||
.where(LoginTask.account_id == Account.id)
|
||||
.order_by(LoginTask.id.desc())
|
||||
.limit(1)
|
||||
.scalar_subquery()
|
||||
)
|
||||
|
||||
# 转人工 = 分型属于需人工,或者次数用尽(next_at 已清空且失败过)。
|
||||
manual_filter = or_(
|
||||
LoginTask.relogin_failure_kind.in_(tuple(MANUAL_KINDS)),
|
||||
Account.relogin_failure_kind.in_(tuple(sorted(MANUAL_KINDS))),
|
||||
and_(
|
||||
LoginTask.relogin_last_error != "",
|
||||
LoginTask.relogin_next_at.is_(None),
|
||||
LoginTask.relogin_attempts > 0,
|
||||
Account.relogin_last_error != "",
|
||||
Account.relogin_next_at.is_(None),
|
||||
Account.relogin_attempts > 0,
|
||||
),
|
||||
)
|
||||
rows = (
|
||||
db.query(LoginTask, Account.username)
|
||||
.join(Account, LoginTask.account_id == Account.id)
|
||||
.filter(LoginTask.ck_check_status == "invalid")
|
||||
.filter(LoginTask.relogin_failure_kind != "")
|
||||
db.query(Account, latest_attempt_at.label("last_attempt_at"))
|
||||
.filter(Account.ck_check_status == "invalid")
|
||||
.filter(Account.relogin_failure_kind != "")
|
||||
.filter(manual_filter)
|
||||
.order_by(LoginTask.attempt_at.desc(), LoginTask.id.desc())
|
||||
.order_by(latest_attempt_at.desc(), Account.id.desc())
|
||||
.all()
|
||||
)
|
||||
grouped: dict[str, dict] = {}
|
||||
for task, username in rows:
|
||||
kind = task.relogin_failure_kind or ""
|
||||
for account, last_attempt_at in rows:
|
||||
kind = account.relogin_failure_kind or ""
|
||||
# 可重试的类型却已经没有下次排期 = 次数用尽,单独成组。
|
||||
# 否则它会混进「验证码失败」组里,客服看不出「这个已经不会再自动试了」。
|
||||
if kind not in MANUAL_KINDS:
|
||||
@@ -241,12 +250,12 @@ def manual_queue(
|
||||
if len(group["samples"]) < limit:
|
||||
group["samples"].append(
|
||||
{
|
||||
"task_id": task.id,
|
||||
"account_username": username,
|
||||
"relogin_attempts": int(task.relogin_attempts or 0),
|
||||
"last_error": task.relogin_last_error or task.attempt_message or "",
|
||||
"attempt_at": task.attempt_at.isoformat()
|
||||
if task.attempt_at
|
||||
"account_id": account.id,
|
||||
"account_username": account.username,
|
||||
"relogin_attempts": int(account.relogin_attempts or 0),
|
||||
"last_error": account.relogin_last_error or "",
|
||||
"attempt_at": last_attempt_at.isoformat()
|
||||
if last_attempt_at
|
||||
else None,
|
||||
}
|
||||
)
|
||||
|
||||
Reference in New Issue
Block a user