From 68ac629f44ede15935ddfd696675415f969ea032 Mon Sep 17 00:00:00 2001
From: xfy <i@rua.plus>
Date: Thu, 23 Jul 2026 11:49:52 +0800
Subject: [PATCH] =?UTF-8?q?fix(sql-console):=20=E5=86=99=E6=93=8D=E4=BD=9C?=
 =?UTF-8?q?=E5=90=8E=E5=85=A8=E9=87=8F=E5=A4=B1=E6=95=88=E7=9B=B8=E5=85=B3?=
 =?UTF-8?q?=E7=BC=93=E5=AD=98?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

SQL 控制台执行 UPDATE/DELETE 等 SQL 直改 DB 时绕过了 server function 的
正常缓存失效路径，导致 moka 单篇缓存（TTL 600s）与 SSR 磁盘缓存继续吐出
陈旧数据（含已删除文章），近 10 分钟的陈旧窗口。

- execute_sql 执行成功后，若含写语句则全量失效文章/搜索/评论缓存 +
  SSR 公开页磁盘缓存 + bump 全局世代号
- cache.rs 新增 invalidate_all_comments（评论列表 + 待审数）
- 抽 writes_affect_cache 纯函数 + 3 个单测覆盖写/只读/混合语句
---
 src/api/database/sql_console.rs | 60 +++++++++++++++++++++++++++++++++
 src/cache.rs                    | 10 ++++++
 2 files changed, 70 insertions(+)

diff --git a/src/api/database/sql_console.rs b/src/api/database/sql_console.rs
index 296f33d..2d60297 100644
--- a/src/api/database/sql_console.rs
+++ b/src/api/database/sql_console.rs
@@ -191,6 +191,16 @@ fn is_read_only(stmt: &sqlparser::ast::Statement) -> bool {
     matches!(stmt, Statement::Query(_) | Statement::Explain { .. })
 }
 
+/// 判断一组语句中是否含有写操作（非只读语句）。
+///
+/// SQL 控制台执行写操作（INSERT/UPDATE/DELETE/TRUNCATE/ALTER/DROP 等）可能直改
+/// posts/comments/tags 等业务表，绕过 server function 的正常缓存失效路径。
+/// 抽成纯函数便于单测「哪些语句集合应触发兜底失效」。
+#[cfg(feature = "server")]
+fn writes_affect_cache(stmts: &[sqlparser::ast::Statement]) -> bool {
+    stmts.iter().any(|s| !is_read_only(s))
+}
+
 /// 把一列的值转成 JSON（按 PG 类型名分发）。
 #[cfg(feature = "server")]
 fn col_to_json(row: &tokio_postgres::Row, idx: usize) -> serde_json::Value {
@@ -305,6 +315,18 @@ pub async fn execute_sql(sql: String, opts: ExecuteSqlOpts) -> Result<SqlResult,
             let stmt_sql = stmt.to_string();
             last_result = execute_one(&client, stmt, &stmt_sql, opts.with_explain, start).await?;
         }
+        // 是否含写语句：抽成纯函数便于单测（见 writes_affect_cache）。
+        let has_write = writes_affect_cache(&asts);
+        // SQL 控制台写操作可能直改 posts/comments/tags，绕过了 server function 的
+        // 正常失效路径（delete_post/create_post 等已在内部失效）。此处全量兜底失效，
+        // 避免最长 10 分钟（TTL_SINGLE_POST=600s）内继续吐出陈旧数据（含已删除文章）。
+        if has_write {
+            crate::cache::invalidate_all_post_caches();
+            crate::cache::invalidate_search_results();
+            crate::cache::invalidate_all_comments();
+            crate::ssr_cache::invalidate_ssr_all_public();
+            crate::ssr_cache::bump_global_generation();
+        }
         Ok(last_result)
     }
     #[cfg(not(feature = "server"))]
@@ -688,4 +710,42 @@ mod tests {
         assert!(!is_read_only(&parse("DELETE FROM t WHERE id = 1")[0]));
         assert!(!is_read_only(&parse("INSERT INTO t (a) VALUES (1)")[0]));
     }
+
+    // ── writes_affect_cache（SQL 控制台兜底失效开关） ─────────────
+    // 任何写语句（非只读）都应触发全量缓存失效，兜底绕过 server function 的直改 DB。
+
+    #[test]
+    fn writes_affect_cache_true_for_write_statements() {
+        for sql in [
+            "UPDATE posts SET deleted_at = NOW() WHERE id = 648",
+            "DELETE FROM posts WHERE id = 648",
+            "INSERT INTO posts (title) VALUES ('x')",
+            "TRUNCATE posts",
+            "ALTER TABLE posts ADD COLUMN x int",
+            "DROP TABLE posts",
+        ] {
+            assert!(
+                writes_affect_cache(&parse(sql)),
+                "写语句应触发兜底失效：{sql:?}"
+            );
+        }
+    }
+
+    #[test]
+    fn writes_affect_cache_false_for_read_only_statements() {
+        for sql in ["SELECT 1", "EXPLAIN SELECT * FROM posts", "SELECT id FROM posts WHERE id = 1"]
+        {
+            assert!(
+                !writes_affect_cache(&parse(sql)),
+                "只读语句不应触发兜底失效：{sql:?}"
+            );
+        }
+    }
+
+    #[test]
+    fn writes_affect_cache_mixed_statements_flagged() {
+        // 多语句中混入任一写语句即应触发（与 allow_multi 语义一致）。
+        let stmts = parse("SELECT 1; UPDATE posts SET deleted_at = NULL WHERE id = 648");
+        assert!(writes_affect_cache(&stmts));
+    }
 }
diff --git a/src/cache.rs b/src/cache.rs
index e605008..3b969f5 100644
--- a/src/cache.rs
+++ b/src/cache.rs
@@ -635,6 +635,16 @@ pub async fn invalidate_pending_count() {
         .await;
 }
 
+/// 全量失效评论缓存（SQL 控制台兜底用：管理员直接改 comments 表时无法定向）。
+///
+/// 正常评论写路径用 [`invalidate_comments_by_post`]（按文章定向）+ [`invalidate_pending_count`]；
+/// 这里全量清空是 SQL 控制台直改 DB 的兜底——无法从任意 SQL 精确解析受影响的 post_id。
+#[cfg(feature = "server")]
+pub fn invalidate_all_comments() {
+    COMMENT_CACHE.invalidate_all();
+    PENDING_COUNT_CACHE.invalidate_all();
+}
+
 #[cfg(all(test, feature = "server"))]
 mod tests {
     use super::*;