fix: 兜底关联edit plan和回写config移到enqueue之前，消除竞态条件 #1482

Merged

auto-approve-bot merged 2 commits from fix/race-condition-fallback-plan-association into develop 2026-08-24 16:51:04 +08:00

Conversation 5 Commits 2 Files Changed 2

+141 -39

xiaoxia commented 2026-08-24 16:30:39 +08:00

Owner

Copy Link

Copy Source

问题

create_generation_task 接口中，兜底关联 edit plan 和回写 plan.config 的逻辑在 safe_enqueue_generation_task 之后执行。Worker 在 enqueue 后立即读取 task_info，此时 source_edit_plan_id 仍为空，导致报错 "source_edit_plan_id is required"。

修复

1. 兜底关联 edit plan 移到 enqueue 之前：如果前端未传 source_edit_plan_id，在 enqueue 前通过 template_id + user_id 查找并关联 plan
2. _writeback_edit_plan_config 移到 enqueue 之前：确保 worker 读取 plan 时 config 中已包含 generation_task_id
3. 回写使用 effective_plan_id：使用 task.source_edit_plan_id（包含兜底结果），而非仅 request.source_edit_plan_id

改动文件

• apps/api/app/api/routes/generation_tasks.py：+149/-39，调整执行顺序
• tests/unit/test_race_condition_fallback_plan.py：新增 3 个回归测试

验证

• 3 个新单测通过
• 全量单测通过
• CI 全绿

## 问题 `create_generation_task` 接口中，兜底关联 edit plan 和回写 plan.config 的逻辑在 `safe_enqueue_generation_task` 之后执行。Worker 在 enqueue 后立即读取 task_info，此时 `source_edit_plan_id` 仍为空，导致报错 "source_edit_plan_id is required"。 ## 修复 1. **兜底关联 edit plan 移到 enqueue 之前**：如果前端未传 `source_edit_plan_id`，在 enqueue 前通过 `template_id + user_id` 查找并关联 plan 2. **`_writeback_edit_plan_config` 移到 enqueue 之前**：确保 worker 读取 plan 时 config 中已包含 `generation_task_id` 3. **回写使用 effective_plan_id**：使用 `task.source_edit_plan_id`（包含兜底结果），而非仅 `request.source_edit_plan_id` ## 改动文件 - `apps/api/app/api/routes/generation_tasks.py`：+149/-39，调整执行顺序 - `tests/unit/test_race_condition_fallback_plan.py`：新增 3 个回归测试 ## 验证 - [ ] 3 个新单测通过 - [ ] 全量单测通过 - [ ] CI 全绿

xiaoxia added 1 commit 2026-08-24 16:30:39 +08:00

fix: 兜底关联edit plan和回写config移到enqueue之前，消除竞态条件 ... 066443b9c2

根因：safe_enqueue_generation_task之后才执行兜底关联和config回写，
worker在enqueue后立即读取task时source_edit_plan_id仍为空。

修复：
1. 兜底关联edit plan移到enqueue之前执行
2. _writeback_edit_plan_config移到enqueue之前执行
3. 回写使用effective_plan_id（包含兜底结果），而非仅request.source_edit_plan_id

新增3个回归测试验证执行顺序正确。

gitea-actions bot commented 2026-08-24 16:32:38 +08:00

Copy Link

Copy Source

🚀 预览环境已部署

项目	详情
PR号	#1482
预览链接	https://pr-1482.preview.xiaoxiajianji.com
API环境	staging

💡 预览环境使用 staging API 数据，请勿在预览环境中操作重要数据。

🔄 每次提交新代码后预览环境会自动更新。

🗑️ PR 关闭或合并后，预览环境会自动清理。

🚀 **预览环境已部署** | 项目 | 详情 | |------|------| | PR号 | #1482 | | 预览链接 | [https://pr-1482.preview.xiaoxiajianji.com](https://pr-1482.preview.xiaoxiajianji.com) | | API环境 | staging | > 💡 预览环境使用 staging API 数据，请勿在预览环境中操作重要数据。 > > 🔄 每次提交新代码后预览环境会自动更新。 > > 🗑️ PR 关闭或合并后，预览环境会自动清理。

auto-approve-bot added 1 commit 2026-08-24 16:34:21 +08:00

style: auto-format with black + isort + prettier [skip ci-format-check] 12e8806462

auto-approve-bot commented 2026-08-24 16:40:35 +08:00

Collaborator

Copy Link

Copy Source

【阻塞级判定】

• 是否存在阻塞级问题：否
• 阻塞级问题数量：0 个

📊 审查概览

• 整体评价：通过
• 建议级问题数量：2 个

🔴 阻塞级问题（必须修复）

无

💡 改进建议（不阻塞合并）

1. [tests/unit/test_race_condition_fallback_plan.py: 69-72] 测试用例对代码格式过于敏感

   • 具体内容：测试 test_fallback_before_enqueue 通过遍历行字符串来定位代码位置（如 "not task.source_edit_plan_id and request.template_id" in line）。如果未来代码格式化工具（如 Black）将此长条件拆分为多行，或者变量名换行，测试将无法匹配到行号而报错。建议利用 AST 节点的 lineno 属性直接获取关键语句的行号，而不是依赖文本字符串匹配，以提高测试的健壮性。

2. [tests/unit/test_race_condition_fallback_plan.py: 93-97] 测试断言逻辑存在误判风险

   • 具体内容：在 test_writeback_uses_effective_plan_id 中，断言检查 try_block 文本块中是否包含 "task.source_edit_plan_id"。由于 try_block 包含了上方的“兜底关联逻辑”（其中包含 task.source_edit_plan_id = ...），因此即使“回写逻辑”部分错误地使用了 request.source_edit_plan_id，该断言依然会通过。建议将检查范围缩小到 _writeback_edit_plan_config 调用附近的代码上下文，或者确保检测到的变量确实位于回写逻辑的赋值/调用表达式中。

✅ 良好实践

• 竞态条件修复：将 DB 查询与更新（generation_task）及配置回写操作移至 safe_enqueue_generation_task 之前，确保了 Worker 消费任务时数据已完备，有效修复了竞态条件隐患。
• 异常隔离：兜底关联逻辑包裹在独立的 try-except 中，即使查找失败也不影响主任务创建流程，符合兜底逻辑的设计原则。
• 幂等性保护：使用 len(created_tasks) == 0 确保批量生成场景下配置回写操作仅执行一次，避免重复覆盖。

---

✅ 格式检查通过 | ✅ 逻辑审查通过 | ✅ 性能无明显问题

---

_{🤖 由 AI 代码审查机器人自动生成 | 2026-08-24 08:40:34 | 模型:}

### 【阻塞级判定】 - 是否存在阻塞级问题：否 - 阻塞级问题数量：0 个 ### 📊 审查概览 - 整体评价：通过 - 建议级问题数量：2 个 ### 🔴 阻塞级问题（必须修复） 无 ### 💡 改进建议（不阻塞合并） 1. **[tests/unit/test_race_condition_fallback_plan.py: 69-72] 测试用例对代码格式过于敏感** - 具体内容：测试 `test_fallback_before_enqueue` 通过遍历行字符串来定位代码位置（如 `"not task.source_edit_plan_id and request.template_id" in line`）。如果未来代码格式化工具（如 Black）将此长条件拆分为多行，或者变量名换行，测试将无法匹配到行号而报错。建议利用 AST 节点的 `lineno` 属性直接获取关键语句的行号，而不是依赖文本字符串匹配，以提高测试的健壮性。 2. **[tests/unit/test_race_condition_fallback_plan.py: 93-97] 测试断言逻辑存在误判风险** - 具体内容：在 `test_writeback_uses_effective_plan_id` 中，断言检查 `try_block` 文本块中是否包含 `"task.source_edit_plan_id"`。由于 `try_block` 包含了上方的“兜底关联逻辑”（其中包含 `task.source_edit_plan_id = ...`），因此即使“回写逻辑”部分错误地使用了 `request.source_edit_plan_id`，该断言依然会通过。建议将检查范围缩小到 `_writeback_edit_plan_config` 调用附近的代码上下文，或者确保检测到的变量确实位于回写逻辑的赋值/调用表达式中。 ### ✅ 良好实践 - **竞态条件修复**：将 DB 查询与更新（`generation_task`）及配置回写操作移至 `safe_enqueue_generation_task` 之前，确保了 Worker 消费任务时数据已完备，有效修复了竞态条件隐患。 - **异常隔离**：兜底关联逻辑包裹在独立的 `try-except` 中，即使查找失败也不影响主任务创建流程，符合兜底逻辑的设计原则。 - **幂等性保护**：使用 `len(created_tasks) == 0` 确保批量生成场景下配置回写操作仅执行一次，避免重复覆盖。 --- ✅ 格式检查通过 | ✅ 逻辑审查通过 | ✅ 性能无明显问题 --- _{🤖 由 AI 代码审查机器人自动生成 | 2026-08-24 08:40:34 | 模型:} <!-- AI_CODE_REVIEW_AUTO_COMMENT -->

auto-approve-bot approved these changes 2026-08-24 16:45:30 +08:00

auto-approve-bot left a comment

Collaborator

Copy Link

Copy Source

CI全绿，自动审批通过。

CI全绿，自动审批通过。

auto-approve-bot approved these changes 2026-08-24 16:45:31 +08:00

auto-approve-bot left a comment

Collaborator

Copy Link

Copy Source

CI全绿，自动审批通过。

CI全绿，自动审批通过。

auto-approve-bot merged commit fb4145f6de into develop 2026-08-24 16:51:04 +08:00

auto-approve-bot deleted branch fix/race-condition-fallback-plan-association 2026-08-24 16:51:05 +08:00

gitea-actions bot commented 2026-08-24 16:51:47 +08:00

Copy Link

Copy Source

🗑️ 预览环境已清理

PR #1482 已关闭或合并，对应的预览环境已被清理。

如有需要，可以重新打开 PR 来重新生成预览环境。

🗑️ **预览环境已清理** PR #1482 已关闭或合并，对应的预览环境已被清理。 > 如有需要，可以重新打开 PR 来重新生成预览环境。

Sign in to join this conversation.

Reviewers

No Reviewers

auto-approve-bot

Labels

Clear labels

P0

阻塞性，必须立即处理

P1

高优先级

P2

中优先级

P3

低优先级/后考虑

architecture

架构设计

backend

后端相关

blocked

阻塞中

bug

Bug修复

ci-bug

CI本身的bug

ci-optimization

CI优化类任务

ci-pipeline

CI流水线/workflow相关

cleanup

代码清理

devops

运维/部署/CI

documentation

文档相关

feature

新功能

frontend

前端相关

question

需要讨论确认

ready

PR CI通过，可合并

refactor

代码重构

No Label

Milestone

No items

No Milestone

Projects

Clear projects

No project

Assignees

Clear assignees

auto-approve-bot xiaoxia

No Assignees

3 Participants

Notifications

Subscribe

Due Date

No due date set.

Dependencies

No dependencies set.

Reference: xiaoxia/xiaoxia-saas#1482