重构: 巨函数拆分 + 清理历史标记注释 + custom_prompt/停止按钮/tunnel 改进

2026-07-31 21:36:12 +08:00
parent 365af554da
commit bd9031d35d
73 changed files with 1421 additions and 1064 deletions
@@ -53,7 +53,7 @@ fn ai_provider_from_row(row: &Row<'_>) -> std::result::Result<AiProviderRecord,
        config: row.get("config")?,
        created_at: row.get("created_at")?,
        updated_at: row.get("updated_at")?,
-        // F-260614-04: enabled/weight 列老库经 v19 迁移补建,DEFAULT 1 / DEFAULT 50。
+        // enabled/weight 列老库经 v19 迁移补建,DEFAULT 1 / DEFAULT 50。
        // from_row 按 i32 取列值兼容(SQLite 无真 BOOLEAN),0→false/非0→true。
        enabled: row.get::<_, i32>("enabled").unwrap_or(1) != 0,
        // weight 读侧 clamp [0,100]:与 insert/update_full 落库的 `.min(100)` 对齐,
@@ -95,7 +95,7 @@ fn ai_tool_execution_from_row(row: &Row<'_>) -> std::result::Result<AiToolExecut
    Ok(AiToolExecutionRecord {
        id: row.get("id")?,
        conversation_id: row.get("conversation_id")?,
-        // F-260619-04:message_id 列老库经 v21 迁移补建。unwrap_or(None) 兜底:
+        // message_id 列老库经 v21 迁移补建。unwrap_or(None) 兜底:
        // 新库空表直接有列;老库行 ALTER 后 NULL;极端情况(迁移未跑/手工删列)防御。
        message_id: row.get("message_id").unwrap_or(None),
        tool_call_id: row.get("tool_call_id")?,
@@ -124,7 +124,7 @@ impl_repo!(
        let is_default = if rec.is_default { 1i32 } else { 0i32 };
        // model_configs:Vec<ModelConfig> → JSON 字符串落 TEXT 列
        let model_configs_json = serde_json::to_string(&rec.model_configs).unwrap_or_else(|_| "[]".into());
-        // F-260614-04: enabled/weight 落库(SQLite 无 BOOLEAN,i32 承载)。
+        // enabled/weight 落库(SQLite 无 BOOLEAN,i32 承载)。
        let enabled_i = if rec.enabled { 1i32 } else { 0i32 };
        let weight_i = rec.weight.min(100) as i32;
        conn.execute(
@@ -420,7 +420,7 @@ mod tests {
    use crate::models::AiProviderRecord;
    use df_ai_core::model::{Capability, IntelligenceTier, Modality, ModelConfig};

-    /// model_configs DB roundtrip + 老库空兼容(F-01 阶段1)
+    /// model_configs DB roundtrip + 老库空兼容
    #[tokio::test]
    async fn ai_provider_model_configs_roundtrip_and_old_db_compat() {
        let db = Database::open_in_memory().await.expect("open_in_memory");
@@ -20,7 +20,7 @@ use super::{now_millis_str, storage_err, validate_column_name};

 /// `knowledges` 表对应 `KnowledgeRecord` 15 个字段的列名(顺序与结构体一致)。
 ///
-/// 多处 `search`/`search_vector` 内联 COLS 串的 DRY 收口(CR-260615-03):集中一处定义,
+/// 多处 `search`/`search_vector` 内联 COLS 串的 DRY 收口:集中一处定义,
 /// 配合下方 `KNOWLEDGE_COL_COUNT` 断言,任一处加列漏改会被测试 `test_knowledge_cols_matches_record`
 /// 立即捕获(`knowledge_from_row` 按 name 取列,SELECT 漏列会运行时 rusqlite 报错,故提前断言)。
 ///
@@ -38,7 +38,7 @@ const KNOWLEDGE_COLS_WITH_EMBEDDING: &str = concat!(

 /// `ideas` 表对应 `IdeaRecord` 14 个字段的列名(顺序与结构体一致)。
 ///
-/// 同 KNOWLEDGE_COLS 的列漂移防护(CR-260615-03):idea 表 INSERT/UPDATE/from_row 三处
+/// 同 KNOWLEDGE_COLS 的列漂移防护:idea 表 INSERT/UPDATE/from_row 三处
 /// 各写一份列名串,加列须三处同步(如 V24 加 related_ids 即三处齐改),漏一处
 /// 只在运行时 rusqlite 报错(INSERT 列数与参数数不匹配 / from_row 取不到列)。集中一处
 /// 定义 + 配合 `IDEA_COL_COUNT` 断言 + 测试 `test_idea_cols_matches_record`,加列漏改即捕获。
@@ -147,7 +147,7 @@ fn knowledge_event_from_row(row: &Row<'_>) -> std::result::Result<KnowledgeEvent
 // ============================================================

 // ============================================================
-// IdeaQuery — 多条件查询入参(F-260621-02 status 下沉 + 关键词 + 排序 + 分页)
+// IdeaQuery — 多条件查询入参(status 下沉 + 关键词 + 排序 + 分页)
 // ============================================================

 /// 灵感多条件查询入参。
@@ -256,7 +256,7 @@ impl_repo!(
 // KnowledgeRepo 的整体更新已由 impl_repo! 宏统一生成的 update_full 提供。

 impl IdeaRepo {
-    /// 多条件查询:动态 WHERE 拼接(status / keyword) + 白名单排序 + 分页(F-260621-02)。
+    /// 多条件查询:动态 WHERE 拼接(status / keyword) + 白名单排序 + 分页。
    ///
    /// 复用 `KnowledgeRepo::search` 的动态 WHERE 模式:if-let 分支按可选条件拼 SQL 片段,
    /// 各分支化参数绑定到 `?N` 占位符。`order_by` 经 `validate_idea_order_by` 白名单校验后
@@ -925,7 +925,7 @@ mod tests {
    use super::*;
    use crate::db::Database;

-    // ---------- COLS 漂移防护(CR-260615-03) ----------
+    // ---------- COLS 漂移防护 ----------

    /// KNOWLEDGE_COLS 列数须等于 KNOWLEDGE_COL_COUNT(任一处漂移:加列漏改 / 串错位 → 立即失败)。
    /// `knowledge_from_row` 按 name 取列,SELECT 漏列会在运行时被 rusqlite 报错;此断言提前到测试期捕获。
@@ -1,4 +1,4 @@
-//! AI 消息 Repo — ai_messages 表(F-260619-03 消息拆分存储)
+//! AI 消息 Repo — ai_messages 表(消息拆分存储)
 //!
 //! 每条 ChatMessage 一行的独立表,替代 `ai_conversations.messages` 整对话 JSON 列存。
 //! 全专用方法(insert_batch / list_by_conversation / delete_range / update_status /
@@ -239,7 +239,6 @@ impl AiMessageRepo {

    /// 全量重写对话的消息(单事务 DELETE + INSERT OR IGNORE,原子)。
    ///
-    /// F-260619-03 批次 B(save_conversation 写路径切 ai_messages)的核心方法:
    /// 全量重写语义——以入参 records 为该对话的**唯一真相**,先删该 conv 全部旧行再批量插。
    /// 单事务保证「删 + 插」原子,无中间空窗(reload 不会读到半删半插的中间态)。
    ///
@@ -472,7 +471,7 @@ mod tests {
        assert_eq!(got[0].content, "替换后的结果", "其他消息不应被改");
    }

-    // ---------- replace_conversation(F-260619-03 批次 B)----------
+    // ---------- replace_conversation ----------

    /// replace_conversation 全量重写:删旧 + 插新原子,list 一致
    #[tokio::test]
@@ -12,7 +12,7 @@
 //! - [`mod@conversation_repo`]:AiProviderRepo/AiConversationRepo/AiToolExecutionRepo
 //! - [`mod@idea_repo`]:IdeaRepo/KnowledgeRepo/KnowledgeEventsRepo + 向量工具
 //! - [`mod@idea_eval_repo`]:IdeaEvalRepo(灵感评估历史追加型审计表 idea_evaluations,V22)
-//! - [`mod@message_repo`]:AiMessageRepo(F-260619-03 消息拆分存储,全专用方法不走宏)
+//! - [`mod@message_repo`]:AiMessageRepo(消息拆分存储,全专用方法不走宏)
 //!
 //! re-export(`pub use ...::*`)保持 `df_storage::crud::XxxRepo` /
 //! `df_storage::crud::is_allowed_column` 路径不变,**调用方零改动**。
@@ -17,7 +17,7 @@ use super::impl_repo;
 use super::{normalize_stored_path, now_millis_str, storage_err, validate_column_name};

 // ============================================================
-// 项目查询入参(F-260621-02 P2/P3 查询维度补全)
+// 项目查询入参(P2/P3 查询维度补全)
 // ============================================================

 /// 项目列表查询条件(可选字段,全 None = 当前 list_active 全量行为)。
@@ -204,7 +204,7 @@ impl ProjectRepo {
        .map_err(storage_err)?
    }

-    /// 按条件查询未删除项目(F-260621-02 P2/P3:关键词搜索 + 排序 + 分页)。
+    /// 按条件查询未删除项目(P2/P3:关键词搜索 + 排序 + 分页)。
    ///
    /// 复用 `KnowledgeRepo::search` 动态 WHERE 拼接模式:按可选字段 if-let 拼 SQL 子句 +
    /// 分支化参数绑定。始终排除 `deleted_at IS NOT NULL`(软删),与 `list_active` 行为对齐。
@@ -119,8 +119,8 @@ impl SettingsRepo {
 pub fn allowed_columns_for(table: &str) -> Option<&'static [&'static str]> {
    Some(match table {
        "ideas" => &[
-            // IDEA-FIX-02: id\created_at 不列入 — 主键与创建时间不可通过通用 update_field 改写
-            // (对标 tasks 白名单 B-260616-16 同款防护,防篡改主键/伪造创建时间)
+            // id\created_at 不列入 — 主键与创建时间不可通过通用 update_field 改写
+            // (对标 tasks 白名单同款防护,防篡改主键/伪造创建时间)
            "title", "description", "status", "priority", "score", "tags", "source",
            "promoted_to", "ai_analysis", "scores", "related_ids", "updated_at",
        ],
@@ -141,15 +141,15 @@ pub fn allowed_columns_for(table: &str) -> Option<&'static [&'static str]> {
            // 否则破坏「review_rounds 唯一写入路径」收口、引入旁路写导致计数错乱。
            "project_id", "title", "description", "priority", "branch_name",
            "assignee",
-            // workflow_def_id / base_branch: 预留字段,阶段4 Git/workflow_defs 集成前无写入路径。
+            // workflow_def_id / base_branch: 预留字段,Git/workflow_defs 集成前无写入路径。
            // 当前推进链用硬编码三模板(task_workflow_templates.rs,不建 workflow_defs 表,
-            // tasks.workflow_def_id 留 None),无任何代码写这两列。白名单列入仅为阶段4 预留 +
+            // tasks.workflow_def_id 留 None),无任何代码写这两列。白名单列入仅为预留 +
            // 允许手动/未来填充,勿判死代码删除。base_branch 同理(code kind 闸门接 git 前预留)。
            "workflow_def_id", "base_branch",
            // output_json:ai_execute 写产出 / ai_self_review 读产出自审 / human_review 展示对象
            // (决策 a:task 中心,产出跟 task 走)。非状态机收口字段,合法可写。
            "output_json",
-            // idea_id(F-260619-01 任务关联灵感,1对1 单向):任务可关联/解关联一条灵感,
+            // idea_id(任务关联灵感,1对1 单向):任务可关联/解关联一条灵感,
            // 非状态机收口字段,合法可写(idea_id 存在性由外键约束 + 上层校验兜底)。
            "idea_id",
            // 知识图谱 Phase 1 V29 三列(对标设计 §2.1):非状态机收口字段,合法可写。
@@ -160,7 +160,7 @@ pub fn allowed_columns_for(table: &str) -> Option<&'static [&'static str]> {
            // 走专用方法 set_status_for_aggregation,不经通用 update_field。
            "queue", "parent_id", "content_json",
            "updated_at",
-            // TODO(B-260616-16): project_id 跨表存在性校验待 commands/task.rs 层补。
+            // TODO: project_id 跨表存在性校验待 commands/task.rs 层补。
            //   通用 CRUD 层(db repo)只懂表/列语义,不持有跨表业务约束(查 projects 表存在性)。
            //   project_id 当前可在白名单内改写,合法目标存在性由上层命令层校验。
        ],
@@ -218,7 +218,7 @@ pub(crate) fn validate_column_name(field: &str, table: &str) -> Result<()> {
    match allowed_columns_for(table) {
        Some(cols) if cols.contains(&field) => Ok(()),
        Some(_) => Err(Error::Storage(format!("表 {} 不允许的字段名: {}", table, field))),
-        // 未登记表保守拒绝(FR-S6: 原放行 Ok,若未来未登记表走通用查询路径,列名直进字符串拼接即 SQL 注入;与 is_allowed_column 的 None=>false 对齐)
+        // 未登记表保守拒绝:原放行 Ok,若未来未登记表走通用查询路径,列名直进字符串拼接即 SQL 注入;与 is_allowed_column 的 None=>false 对齐
        None => Err(Error::Storage(format!("表 {} 未登记列白名单,拒绝防注入", table))),
    }
 }
@@ -47,7 +47,7 @@ fn task_from_row(row: &Row<'_>) -> std::result::Result<TaskRecord, rusqlite::Err
 }

 // ============================================================
-// 任务列表查询入参(F-260621-02 查询维度补全)
+// 任务列表查询入参(查询维度补全)
 // ============================================================

 /// 任务列表动态查询入参。全可选,空 query = 等价当前全量行为(向后兼容)。
@@ -213,7 +213,7 @@ impl TaskRepo {
        .map_err(storage_err)?
    }

-    /// 原子推进任务状态(任务推进链 F-260616-02 唯一 status 写入路径)
+    /// 原子推进任务状态(任务推进链唯一 status 写入路径)
    ///
    /// 下沉 SQL `WHERE id=? AND status=?expected` 做 CAS(Compare-And-Swap)防 TOCTOU:
    /// 并发推进/旁路修改若已改 status,affected_rows==0,本方法返回 None,调用方
@@ -296,7 +296,7 @@ impl TaskRepo {
        .map_err(storage_err)?
    }

-    /// 动态条件列出未删除任务(F-260621-02 查询维度补全)。
+    /// 动态条件列出未删除任务(查询维度补全)。
    ///
    /// 复用 KnowledgeRepo::search 的「动态 WHERE + 参数绑定」模式,但用累积式条件收集
    /// (Vec<String> WHERE 子句 + Vec<rusqlite::Value> 参数)替代 if-let 二分支——
@@ -528,7 +528,7 @@ impl TaskRepo {
    ///
    /// 本方法返回 `Vec<(status, count)>`(SQL GROUP BY 一次查询,数据量小 ~50 无压力),
    /// 聚合规则的具体判定由调用方(commands 层)实现 —— 本层只提供原始计数,不持有
-    /// 业务聚合逻辑(CRUD 层只懂表/列语义,对标 B-260616-16 跨表校验下沉思路)。
+    /// 业务聚合逻辑(CRUD 层只懂表/列语义,对标跨表校验下沉思路)。
    pub async fn count_children_by_status(
        &self,
        parent_id: &str,