Skip to main content

helix_core/effect/
storage.rs

1/// 存储操作(具体 SQL 由 driver 的 Storage 实现构造)
2#[derive(Debug, Clone)]
3pub enum StorageOp {
4    /// batch upsert:INSERT ... ON CONFLICT(conflict_key) DO UPDATE ...
5    BatchUpsert(UpsertSpec),
6    /// 通用单调 MAX-guard 写:仅当 new value > 当前值才写(CAS / 单调寄存器语义,严格 >)。
7    MonotonicUpsert(MonotonicUpsertSpec),
8    /// batch update:UPDATE table SET patch WHERE key_col IN (key_vals)
9    BatchUpdate(BatchUpdateSpec),
10    /// 守卫式自增:`UPDATE … SET bump_col = bump_col + delta[, set…] WHERE key=? AND ? > guard_col`。
11    /// 通用计数器原语(无 read-modify-write),见 `GuardedBumpSpec`。
12    GuardedBump(GuardedBumpSpec),
13    /// 复合作用域守卫式自增:在 scope + key 精确单行内完成幂等计数与绝对列写。
14    ScopedGuardedBump(ScopedGuardedBumpSpec),
15    /// get:SELECT * FROM table WHERE ...(单行)
16    Get(GetSpec),
17    /// 复合作用域 get:SELECT * FROM table WHERE scope=? AND key=?(单行)
18    ScopedGet(ScopedGetSpec),
19    /// scan:SELECT * FROM table [LIMIT n](多行,C3)
20    Scan(ScanSpec),
21    /// batch delete:`DELETE FROM table WHERE scope_col = ? AND key_col IN (key_vals)`。
22    /// 复合 PK 作用域删除(如 channel_member 成员离场:scope=channel_id + key=user_id)。
23    /// O(k) 单语句;scope 等值约束**绝不跨作用域误删**。见 `BatchDeleteSpec`。
24    BatchDelete(BatchDeleteSpec),
25}
26
27/// 稀疏、非 SQL 存储适配器使用的表身份描述。
28///
29/// 具体表名由业务模块提供;平台驱动只据此建立通用索引,不依赖业务类型或解析 DDL。
30#[derive(Debug, Clone, Copy, PartialEq, Eq)]
31pub struct SparseTableSpec {
32    pub table: &'static str,
33    pub identity_cols: &'static [&'static str],
34    pub scope_col: Option<&'static str>,
35}
36
37/// BatchUpsert 规格
38///
39/// ## MAJ-5 修复
40///
41/// `table` 和 `conflict_key` 改为 `&'static str`,消除热路径的运行时 String 分配。
42/// 表名和冲突键都是编译期已知的常量(示例 "message"、"temporary_id"——core 不解释含义),
43/// `"message".to_string()` 每次消息落库都产生一次堆分配,改为静态引用零成本。
44///
45/// 注意:`conflict_key` 的具体值(如 `"temporary_id"`)由 ACL-1 在 helix-im 侧提供,
46/// helix-core 只传递字符串,不理解其含义。
47#[derive(Debug, Clone)]
48pub struct UpsertSpec {
49    /// 目标表名(编译期常量,如 "message")
50    pub table: &'static str,
51    pub rows: Vec<Row>,
52    /// ON CONFLICT 的列名(如 ACL-1 提供 "temporary_id";core 不知道含义)
53    pub conflict_key: Option<&'static str>,
54    /// 冲突时**不**更新的列(INSERT 仍写其字面值,ON CONFLICT DO UPDATE 排除它们)。
55    ///
56    /// 用途:守卫列 / 本地维护列——新行用提供的字面值,但已存在的行保留既有值
57    /// ("server 缺省 → 回退本地"语义在 upsert 层表达,无需先读后写)。列名编译期常量,
58    /// driver 不解释含义(守 HX-C001)。空 = 旧行为(除 conflict_key 外全列更新)。
59    ///
60    /// 默认空:用 `UpsertSpec::new(...)` 构造可零改动迁移既有调用方。
61    pub exclude_from_update: Vec<&'static str>,
62}
63
64impl UpsertSpec {
65    /// 便捷构造(无排除列,旧默认行为)——既有调用方零改动迁移点。
66    pub fn new(table: &'static str, rows: Vec<Row>, conflict_key: Option<&'static str>) -> Self {
67        Self {
68            table,
69            rows,
70            conflict_key,
71            exclude_from_update: Vec::new(),
72        }
73    }
74}
75
76/// 通用单调 MAX-guard 写规格:仅当 `value > 当前值` 才写入(CAS / 单调寄存器语义)。
77///
78/// 通用原语,core 不含任何业务含义——典型用途:增量同步水位、presence 心跳时间戳、单调计数器。
79/// `scope_key` 是运行时值(由上层提供,如 channel_id.to_string()),core 不解析其含义。
80///
81/// ## 表名 / 列名作数据(HX-C001)
82///
83/// `table` / `key_col` / `value_col` / `touch_col` 都是**编译期常量**(`&'static str`),
84/// 由上层业务模块提供。core 只把它们当字符串透传给 driver 拼 SQL,**绝不**理解其业务含义
85/// ——业务 schema 不进 core(守 HX-C001)。`touch_col` 为 `Some` 时,driver 用自身的
86/// ambient 时钟(driver 层合法)写穿该列为当前毫秒时间戳(非权威镜像列,不参与 MAX guard 判定)。
87#[derive(Debug, Clone)]
88pub struct MonotonicUpsertSpec {
89    /// 目标表名(编译期常量;core 不解释含义)
90    pub table: &'static str,
91    /// PRIMARY KEY 列名(编译期常量)
92    pub key_col: &'static str,
93    /// 单调 MAX-guard 的值列名(编译期常量)
94    pub value_col: &'static str,
95    /// 可选「最近更新时间」列名;`Some` 时 driver 写穿当前时间戳,
96    /// 不参与 MAX guard。`None` = 无此列(最小单调寄存器形态)。
97    pub touch_col: Option<&'static str>,
98    /// PK 运行时值(由上层提供);core 不解析其含义。
99    pub scope_key: String,
100    /// 单调写入值;仅当 `> 当前值` 才前进。
101    pub value: i64,
102}
103
104/// batch_update 规格
105///
106/// MAJ-5:`table` 和 `key_col` 改为 `&'static str`(编译期常量)
107#[derive(Debug, Clone)]
108pub struct BatchUpdateSpec {
109    pub table: &'static str,
110    pub key_col: &'static str,
111    pub key_vals: Vec<SqlValue>,
112    pub patch: Row,
113}
114
115/// BatchDelete 规格:`DELETE FROM table WHERE scope_col = ? AND key_col IN (key_vals)`。
116///
117/// 复合 PK 作用域删除(如 channel_member:scope=channel_id 等值 + key=user_id IN 列表)——
118/// 表名 / 列名编译期常量(core 不解释含义,守 HX-C001),单语句 O(k) 删 k 行。scope 等值约束
119/// **绝不跨作用域误删**(无 scope 会把该 user 从所有 channel 删掉)。`key_vals` 空 → 上层应不产 op。
120#[derive(Debug, Clone)]
121pub struct BatchDeleteSpec {
122    /// 目标表名(编译期常量)
123    pub table: &'static str,
124    /// 作用域列名(等值约束,编译期常量)—— 复合 PK 第一段,如 channel_id。
125    pub scope_col: &'static str,
126    /// 作用域运行时值(定位作用域;core 不解析含义)
127    pub scope_val: SqlValue,
128    /// IN-list 列名(编译期常量)—— 复合 PK 第二段,如 user_id。
129    pub key_col: &'static str,
130    /// 待删行的 key 运行时值列表
131    pub key_vals: Vec<SqlValue>,
132}
133
134/// 守卫式自增规格:单行原子计数器递增 + 可选绝对列写,带前进守卫——**无 read-modify-write**。
135///
136/// SQL 语义(表名 / 列名都是编译期常量,由上层业务模块提供——守 HX-C001,core/driver
137/// 不解释含义;`key_val` / `set_cols` 值 / `bump_delta` / `guard_val` 是运行时数据):
138/// ```sql
139/// UPDATE <table> SET <bump_col> = <bump_col> + <bump_delta> [, <set_cols…> = ?…]
140///   WHERE <key_col> = ? AND ? > <guard_col>
141/// ```
142/// (守卫占位 `?` 绑定 `guard_val`:仅当 `guard_val > 当前 guard_col` 才命中更新。)
143///
144/// ## 为什么需要独立原语(不能用 `BatchUpdate`)
145///
146/// `BatchUpdate` 只能 `SET col = ?`(绝对值)+ `WHERE key IN (…)`——无法表达
147/// `col = col + delta`(自增)也无法表达 `AND ? > guard_col`(前进守卫)。把「读出 +1 写回」
148/// 下沉成单条 SQL `col = col + delta` 是 HX-C005 热路径 O(1) 的硬约束(禁应用层 RMW);
149/// 守卫条件让「同一消息重复投递不重复 +1」「乱序旧消息不污染计数」在 SQL 层幂等成立。
150///
151/// ## 通用性(YAGNI 边界)
152///
153/// 纯计数器语义,core 不含任何业务含义——典型用途:未读计数 +1(guard=last_post_at 防回退)、
154/// 限流桶累加、引用计数。`set_cols` 为空 = 纯自增;`guard_val=i64::MIN`/无意义守卫由上层避免
155/// (上层须保证 guard 语义正确)。`set_cols` 列名同样是 `&'static str` 编译期常量。
156#[derive(Debug, Clone)]
157pub struct GuardedBumpSpec {
158    /// 目标表名(编译期常量;core 不解释含义)
159    pub table: &'static str,
160    /// 主键列名(编译期常量)
161    pub key_col: &'static str,
162    /// 主键运行时值(定位单行;core 不解析含义)
163    pub key_val: SqlValue,
164    /// 被自增的数值列名(编译期常量)
165    pub bump_col: &'static str,
166    /// 自增量(运行时值,通常 +1;负数 = 自减)
167    pub bump_delta: i64,
168    /// 同一前进守卫下的额外计数列自增。
169    ///
170    /// 用于 unread/mention/urgent 等多个派生计数必须在同一条新消息守卫下同时推进的场景。
171    /// 列名仍是编译期常量;delta 使用绑定参数,不参与 SQL 字符串拼接。
172    pub extra_bumps: Vec<(&'static str, i64)>,
173    /// 与自增同事务写入的绝对值列(列名编译期常量,值运行时);空 = 纯自增。
174    pub set_cols: Row,
175    /// 前进守卫列名(编译期常量):仅当 `guard_val > <guard_col> 当前值` 才更新整行。
176    pub guard_col: &'static str,
177    /// 守卫比较值(运行时):严格大于现值才命中(防重复 / 旧值回退)。
178    pub guard_val: i64,
179}
180
181/// 复合作用域守卫式自增规格,避免复合主键热路径退化为应用层 read-modify-write。
182#[derive(Debug, Clone)]
183pub struct ScopedGuardedBumpSpec {
184    /// 目标表名(编译期常量;core 不解释业务 schema)。
185    pub table: &'static str,
186    /// 第一维作用域列名。
187    pub scope_col: &'static str,
188    /// 第一维作用域运行时值。
189    pub scope_val: SqlValue,
190    /// 第二维键列名。
191    pub key_col: &'static str,
192    /// 第二维键运行时值。
193    pub key_val: SqlValue,
194    /// 被自增的数值列。
195    pub bump_col: &'static str,
196    /// 自增量;0 表示只应用同一守卫下的绝对列 patch。
197    pub bump_delta: i64,
198    /// 与计数器同一 SQL 更新的绝对列。
199    pub set_cols: Row,
200    /// 前进守卫列。
201    pub guard_col: &'static str,
202    /// 仅严格大于当前守卫值时更新。
203    pub guard_val: i64,
204}
205
206/// get 规格(单行;MAJ-5:`table`/`key_col` = `&'static str`)。
207#[derive(Debug, Clone)]
208pub struct GetSpec {
209    pub table: &'static str,
210    pub key_col: &'static str,
211    pub key_val: SqlValue,
212}
213
214/// 复合作用域 get 规格,用两个等值键 O(1) 定位复合主键行。
215#[derive(Debug, Clone)]
216pub struct ScopedGetSpec {
217    pub table: &'static str,
218    pub scope_col: &'static str,
219    pub scope_val: SqlValue,
220    pub key_col: &'static str,
221    pub key_val: SqlValue,
222}
223
224/// scan 规格:通用多行扫描读(C3)。on_start 载入单调水位(全表)/ P6 投影查询(等值过滤+排序)。
225/// 过滤/排序**可选 additive**(HX-C004):`filter` 为 None、`order_by` 为空 = 旧全表语义
226/// (零行为改变);列名仍为编译期常量(HX-C001),方向由结构化枚举表达,禁止调用方拼 SQL。
227#[derive(Debug, Clone)]
228pub struct ScanSpec {
229    /// 目标表名(编译期常量)
230    pub table: &'static str,
231    /// 最多返回行数(None = 全表;driver 应设硬上限防御)
232    pub limit: Option<u32>,
233    /// 可选等值过滤 `WHERE <col> = <val>`(列名编译期常量,值运行时)。None = 不过滤(旧全表)。
234    pub filter: Option<(&'static str, SqlValue)>,
235    /// 结构化排序键,按切片顺序形成 `ORDER BY`;空切片 = 不排序(旧 rowid 序)。
236    pub order_by: &'static [ScanOrder],
237}
238
239/// Scan 排序方向。driver 只把枚举映射成固定 SQL 关键字,不接受运行时 SQL 片段。
240#[derive(Debug, Clone, Copy, PartialEq, Eq)]
241pub enum SortDirection {
242    Asc,
243    Desc,
244}
245
246/// 单个 Scan 排序键。列名是编译期常量,业务层可安全组合多列确定序。
247#[derive(Debug, Clone, Copy, PartialEq, Eq)]
248pub struct ScanOrder {
249    pub column: &'static str,
250    pub direction: SortDirection,
251}
252
253impl ScanOrder {
254    pub const fn asc(column: &'static str) -> Self {
255        Self {
256            column,
257            direction: SortDirection::Asc,
258        }
259    }
260
261    pub const fn desc(column: &'static str) -> Self {
262        Self {
263            column,
264            direction: SortDirection::Desc,
265        }
266    }
267}
268
269/// SQL 值类型(不含数据库具体类型)
270#[derive(Debug, Clone)]
271pub enum SqlValue {
272    Text(String),
273    Integer(i64),
274    Real(f64),
275    /// 二进制大对象(M2:无损携带,禁 BLOB→lossy text 静默损坏)
276    Blob(Vec<u8>),
277    Null,
278}
279
280/// 行数据(列名 → SQL 值)
281pub type Row = Vec<(String, SqlValue)>;