TriviumDB 支持特性详解
September 1, 2026 · View on GitHub
深入剖析 TriviumDB 的架构设计、核心能力与技术实现细节。本文以 v0.8.4 当前源码和正式公共 API 为准。
当前能力快照
- 持久化索引:QuIVer ANN、Hash、Ordered ART、Composite ART、Roaring Bitmap、AC+BM25 TextIndex、业务图块索引。
- 自由 DIY 混合查询:TQL 不是几条固定的“混合搜索模板”,而是一套可编排执行管线。开发者可将向量召回、属性过滤、图扩展、图算法、路径、集合代数、迭代、聚合和重排按业务语义自由串联,并由 Cascades 在预算内选择物理计划。
- 统一查询底座:TQL Parser/AST、WITH 管线、Cascades 优化、Prepared TQL、Path 与一等值结果共同支撑上述自由组合。
- 系统保证:确定性执行、查询/遍历/内存/并行预算、ReadOnly/Immutable 零写、generation 原子发布。
- 跨语言能力:Rust、Python、Node 同步提供四类属性索引管理、Prepared TQL、一等查询值和存储格式观测。
- 严格升级策略:历史静默 API 已移除;误用返回迁移错误。主文件保留有限读取窗口,可重建 sidecar 独立版本化。
目录
架构总览
TriviumDB 采用分层架构,各层职责明确:
flowchart TD
classDef layer fill:#fafafa,stroke:#e0e0e0,stroke-width:2px,color:#333;
classDef module fill:#e3f2fd,stroke:#2196f3,stroke-width:1px,color:#000;
classDef math fill:#f3e5f5,stroke:#9c27b0,stroke-width:1px,color:#000;
classDef storage fill:#e8f5e9,stroke:#4caf50,stroke-width:1px,color:#000;
subgraph Layer1 ["🌐 用户 API 层"]
direction LR
API1[Python binding]:::module
API2[Node.js binding]:::module
API3[Rust pub API]:::module
API4["CLI / TUI (tdb)"]:::module
end
Layer1:::layer
subgraph Layer2 ["⚙️ 数据库核心层 (Database)"]
direction LR
C1[事务控制 Dry-Run]:::module
C2[WAL 编排]:::module
C3[内存预算 & Compaction调度]:::module
end
Layer2:::layer
subgraph Layer3 ["🚀 引擎与执行层"]
direction LR
E1["向量索引\n(BruteForce / QuIVer)"]:::module
E2["图谱遍历\n(Spreading Activation)"]:::module
E3["TQL 查询引擎\n(Parser / Cascades / Pipeline)"]:::module
end
Layer3:::layer
subgraph Layer4 ["🧠 认知管线层 (cognitive.rs)"]
direction LR
M1[FISTA 残差寻隐]:::math
M2[DPP 多样性采样]:::math
M3[NMF 语义矩阵分解]:::math
end
Layer4:::layer
subgraph Layer5 ["🗂️ 内存工作区 (MemTable)"]
direction LR
M_VEC["SoA 向量池\n(基础层 mmap + 增量层 Vec)"]:::module
M_PAY["HashMap\n(Payload 元数据)"]:::module
M_EDGE["图谱邻接表 + 派生目录\n(出边 / 入边 / Label)"]:::module
end
Layer5:::layer
subgraph Layer6 ["💾 持久化层 (Storage)"]
direction LR
S1[".tdb 聚合数据 / 元数据"]:::storage
S2[".vec 分离 mmap 向量文件"]:::storage
S3["WAL + .pidx/.gidx/.text/.quiver\n版本化 sidecar"]:::storage
end
Layer6:::layer
Layer1 ---> Layer2
Layer2 ---> Layer3
Layer3 ---> Layer4
Layer4 ---> Layer5
Layer5 ---> Layer6
三位一体数据模型
每个节点在内部同时持有三种数据,共享全局唯一的 u64 主键:
| 数据层 | 存储位置 | 内容 | 用途 |
|---|---|---|---|
| 向量层 (Vector) | 连续 Vec<T> 数组 (SoA) | f32 × dim 浮点数组 | 语义相似度检索 (稠密召回) |
| 稀疏层 (Sparse Text) | 内存倒排 / AC自动机 | BM25 词频统计 / 匹配树表 | 精确词汇与长文本全文检索 |
| 元数据层 (Payload) | HashMap<u64, JSON> | 任意 JSON Key-Value | 条件过滤、业务数据 |
| 图谱层 (Graph) | HashMap<u64, Vec<Edge>> | 有向带权边邻接表 | 关系遍历、扩散激活 |
为什么选择 SoA 而不是 AoS?
AoS (Array of Structures):每个节点的 {vector, payload, edges} 紧挨存放。
- ❌ 向量检索时 CPU 缓存被无用的 payload 数据污染
- ❌ 无法对向量数组做 SIMD 批量计算
SoA (Structure of Arrays):所有向量连续存入一个大数组,payload 和 edges 各自独立存储。
- ✅ 向量检索时 CPU L1/L2 缓存命中率极高
- ✅ rayon 并行 + SIMD 友好
- ✅ mmap 映射时可直接 OS 层分页 zero-copy 加载向量块
存储引擎与双模式切换
TriviumDB 提供两种互斥的存储模式(StorageMode),且系统支持无缝热切换(只需在打开数据库时更改配置,下一次 flush() 时会自动重组转换结构):
1. Rom 模式(便携单文件优先)
所有数据(向量 + Payload + 边)都被打包进一个致密的 .tdb 二进制文件中,启动时全量装载进内存。
对于几十万节点规模的知识库,它是最理想的格式,只需拷贝一个 .tdb 即可完成库的转移,类似 SQLite。
2. Mmap 模式(大规模零拷贝优先,默认)
启动时,所有大体积的持续增长向量池(Vector Block)将分离为独立的 .vec 文件,而 .tdb 中只记录关系边和 Payload。
- MAP_PRIVATE (COW):通过
memmap2库将数 GB 的向量文件映射到操作系统的虚拟内存中。进程不会真的霸占物理内存,而是由 OS 根据查询压力按需(Page Fault)换入换出。 - 分层向量池(VecPool):内存中维护
基础层(mmap)+ 增量层(Vec)两段结构。新插入的向量只进增量层;delete/update 操作对基础层做 COW 写入,产生进程私有脏页,不改变磁盘文件。直到显式flush()时才统一持久化。
Mmap 的能力边界必须明确:它降低启动和复制成本,但不把 SSD 变成内存。非驻留页首次访问仍需要磁盘 fault-in;当随机访问工作集超过可用 PageCache 时,可能出现持续回收、Major Fault 和尾延迟上升。Config.memory_limit 只约束 TDB 可估算的堆分配,不包含 mmap 映射的驻留页、OS PageCache、pagefile/swap 和其他进程占用。
QuIVer 的 BQ 签名与图拓扑位于匿名堆内存,原始向量位于 mmap 冷层。前者不会作为文件 PageCache 页被直接丢弃,但在启用 swap/pagefile 的系统上仍可能被换出。TDB 不承诺超内存随机工作集下的固定 QPS;应结合查询上下文中的阶段耗时、候选数和进程/系统缺页指标持续观测。
VecPool 混合 Flush 策略
flush() 会根据内部 has_dirty_base 标志自动选择最优写入路径,无需用户干预:
| 场景 | 触发条件 | I/O 代价 | 写入路径 |
|---|---|---|---|
| 纯写入(仅 insert) | has_dirty_base = false | O(Δ),只写新增向量 | 追加路径 |
| 有删改(delete/update 触碰基础层) | has_dirty_base = true | O(N),重写全部数据 | 全量重写路径 |
| 首次 flush(无现有 .vec) | mmap 为空 | O(N),创建新文件 | 全量重写路径 |
追加路径详解(AI 记忆系统、批量导入等纯写场景):
① 将 delta 层字节追加到现有 .vec 文件末尾 → fsync
② 释放旧 mmap(映射窗口固定,感知不到文件扩大后的新区域)
③ 重新以新的总大小 map_copy 整个扩大后的文件
④ 清空 delta 层
``$
以 100 万节点(\text{f32} \times 1536 维,约 6 \text{GB})为例,若单次 \text{flush} 只有 1 万条新增:
- 全量重写:写 6 \text{GB} 数据
- 追加路径:写 60 \text{MB} 数据(仅新增 \text{delta})
**标志位精确追踪**:$has_dirty_base` 只在 `zero_out()` 和 `update()` 操作落入 `index < mmap_count`(基础层区域)时才置 true。对 delta 层内节点的 delete/update **不会**触发全量重写——因为 delta 层本就要在追加时写入,可直接写修改后的值。
### Flush 与 ANN 构建隔离
持久化路径只准备磁盘格式需要的 BQ 元数据和向量块,不会隐式构建 QuIVer。自动 QuIVer 只在查询准备阶段触发,显式构建则由 `build_quiver_index()` 发起。这避免了大规模导入在定期 flush 时突然进入 CPU 密集构图。
QuIVer 快照按 slot 流式读取 FP16/FP32/mmap 向量并直接编码为紧凑 `Bq2Store`,不再保留全库 `Vec<f32>`。构建前还会保守估算签名、双邻接表、ID/slot 映射和 worker bitset 峰值;设置内存预算后,超限构建会在大分配前拒绝。
### 核心容器容量规划
`Config.expected_nodes` 表示本次进程预计的总节点数,`Database::reserve_nodes(additional)` 表示追加预留。预留覆盖向量 delta、Payload/ID 映射、slot 和 fast tag;图边、文本索引、BQ、QuIVer 不在自动预留范围内。配置不写入 `.tdb`,超过预计规模仍可正常增长。
事务和 Python/Node `batch_insert` 在 Dry-Run 后、WAL 前按整批预留。容量预算、算术溢出或 allocator 拒绝时,节点、ID、generation 和 WAL 均不变。
### 压实架构的极致安全取舍(Compaction Trade-off)
由于 TriviumDB 坚持“纯正极简的单文件与单 WAL”架构,没有引入复杂的 LSM-Tree 多段日志(Segmented WAL)机制,为了保证 **100% 的绝对崩溃一致性(Crash Consistency)与 ACID 持久性**,在执行“全量重写路径”时,必须短暂阻塞(Lock)前台并发读写。
- **为什么不采用快照(Snapshot)无锁后台重写?**
如果释放锁在后台缓慢重写 6GB 的向量数据,在此期间前台的新写入将进入 WAL 的末尾。当后台写盘完成并清空旧 WAL(截断)重组新 WAL 时,由于 OS 文件截断与重写的非原子性,**在断电瞬间会导致该时间窗口内的前台数据发生物理级永久丢失(静默丢失)**。
- **作为一款嵌入式 AI 引擎的解法**:
鉴于 99% 的纯插入 AI 记忆场景走的是无感知的“追加路径(Append Path)”,TriviumDB 将掌控权完全交给了开发者。
开发者可以通过 `disable_auto_compaction()` 关闭不可控的后台压实,并在业务低峰期(如凌晨 3 点)主动调用 `compact()` 方法进行手动全量落盘,以此换取系统结构在严苛环境下的绝对健壮与零数据败坏风险。
### 单个 .tdb 底层布局 (Rom 模式 / Mmap 时的元数据底座)
所有数据打包进一个 `.tdb` 二进制文件,内部由四个连续的块组成:
``$
┌────────────────────────┐ \text{offset} 0
│ \text{File} \text{Header} │ 58 字节(当前 \text{v7})
│ \text{MAGIC} + \text{VERSION} + \text{dim} │
│ \text{next\_id} + \text{node\_count} │
│ 各 \text{block} 的 \text{offset} │
├────────────────────────┤ \text{payload\_offset}
│ \text{Payload} \text{Block} │ [\text{node\_id}(8\text{B}) + \text{json\_len}(4\text{B}) + \text{json\_data}] \times \text{N}
├────────────────────────┤ \text{vector\_offset}
│ \text{Vector} \text{Block} │ 连续 \text{f32} 数组(可 \text{mmap} 零拷贝加载,仅 \text{Rom} 模式)
├────────────────────────┤ \text{edge\_offset}
│ \text{Edge} \text{Block} │ [\text{src}(8\text{B}) + \text{dst}(8\text{B}) + \text{label\_len}(2\text{B}) + \text{label} + \text{weight}(4\text{B})] \times \text{M}
├────────────────────────┤ \text{bq\_offset}
│ \text{BQ} \text{Metadata} \text{Block} │ \text{magic} + \text{block\_version} + \text{chunks} + \text{count} + \text{LE} \text{u64}[]
└────────────────────────┘
$``
> **边持久化格式** 从 v7 起保存任意 JSON `metadata`,唯一键为 `(src, dst, label)`,重复 Upsert 原地覆盖权重与元数据。加载器继续兼容 v5/v6,旧边的元数据恢复为 `null`。BQ 元数据块仍采用从 v6 开始的自描述固定小端序布局。
>
> 此外,QuIVer 图索引以独立的 `.tdb.quiver` 文件存储,采用 POD memcpy 极速序列化,重启后零开销恢复。
### 安全写入流程
内存数据 → 写入 .tdb.tmp → fsync 落盘 → 原子 rename 替换 .tdb → 清除 WAL
不管在哪一步崩溃,都不会损坏已有数据:
- 步骤 1-2 崩溃:`.tmp` 残留但旧 `.tdb`/`.vec` 完好 → 重启用旧数据 + WAL 回放
- 步骤 3 崩溃:新文件已就绪 + WAL 仍在 → 重启回放幂等数据(安全冗余)
- 全部完成:清理 WAL,进入干净状态
### 跨平台 I/O 加固(Windows 兼容性)
TriviumDB 的存储层针对 Windows 的强制锁定(Mandatory Locking)语义做了专项加固,消除了在 Linux 上不会出现的幽灵故障:
| 问题 | 根因 | 修复方案 |
|------|------|----------|
| Mmap 模式 flush 100% 失败 | Windows 不允许 rename 覆盖正在被映射的文件 | `flush()` 前强制 `self.mmap = None`,解除内核映射锁 |
| 偶发 rename 失败 | 杀毒软件(Defender/火绒)扫描新文件时短暂独占句柄 | `robust_rename()`:对 `ERROR_ACCESS_DENIED(5)` / `ERROR_SHARING_VIOLATION(32)` 进行指数退避重试(最多 10 次,1→50ms) |
| WAL clear 触发重复扫描 | `remove + create` 使杀毒软件将重建的文件视为新文件再次扫描 | WAL 清空改为 `truncate(true)` 语义,文件句柄不变,不触发新文件扫描 |
> **设计决策**:上述加固无需引入 Manifest/多版本文件系统等重型机制。TriviumDB 是单进程嵌入式数据库(通过 `fs2::try_lock_exclusive` 保证),不存在多进程并发持有同一 mmap 的场景。正确管理单进程内的 mmap 生命周期(先释放再 rename)即可解决根本问题。
### Write-Ahead Log (WAL)
所有写操作(insert / delete / link / unlink / update)在生效前先追加写入 WAL 文件。
- **Append-Only**:仅顺序追加,绝不随机写入,SSD 友好
- **CRC32 校验**:每条记录都附带 CRC32;遇到截断或校验失败时停止回放,绝不跨过损坏继续解释后续字节
- **三种同步模式**:Full(fsync)/ Normal(flush)/ Off(无)
---
## 向量索引策略
TriviumDB 采用**全自动双引擎路由**,无需编译期 Feature 选择,全程运行时自适应:
### BruteForce(热区基础引擎,始终启用)
- **精确度**:100% 精确召回,零误差
- **并行化**:rayon `par_chunks` 多核线性加速
- **原理**:对整个 SoA 向量池做并行余弦相似度扫描
- **激活条件**:< 1 万节点,或 QuIVer 索引尚未构建完成
```rust
// 内部实现伪码
flat_vectors
.par_chunks(dim) // rayon 并行切块
.enumerate()
.map(|(idx, vec)| cosine_sim(query, vec))
.top_k(k) // 取最高分前 K 个
QuIVer ANN 图索引(冷区加速引擎,自动激活)
QuIVer(Quantized Indexed Vector Retrieval)是 TriviumDB 自研的 SOTA 级近似最近邻(ANN)图索引,融合 BQ 二进制量化与 Vamana 图导航,冷热分离架构:
- 精确度:近似搜索,实测 Recall@10 在 20 万规模下达 99%+
- 激活条件:≥ 1 万节点时自动构建
- 搜索流程:
- BQ 签名比对:利用 CPU 原生
Popcount硬件指令,在 Vamana 图导航过程中快速计算 Hamming 距离 - Vamana 图导航:沿着贪心最近邻路径在图中跳转,快速收敛到目标区域
- f32 余弦精排 (Re-rank):仅对候选集从 MemTable 按需读取 f32 原始向量做精准打分
- BQ 签名比对:利用 CPU 原生
核心优势:
| 对比 | BruteForce | QuIVer |
|---|---|---|
| 召回率 | 100% | 97%~99%+ |
| 延迟 | 随节点数线性增长 | 图导航 O(log N),大规模下数量级加速 |
| 增量 Insert | 零开销 | ✅ 实时增量插入,无需重建 |
| 增量 Delete | 零开销 | ✅ Tombstone 软删除,25% 退化自动重建 |
| 增量 Update | 零开销 | ✅ soft_delete + incremental_insert |
| 事务安全 | — | ✅ 分离时间线架构,零回滚开销 |
| 持久化 | — | ✅ .tdb.quiver 独立文件,POD memcpy |
| 内存布局 | 连续 f32 数组 | 冷热分离:BQ 签名(hot) + f32 向量(cold) |
| 激活方式 | 默认 | 自动(1 万节点时构建) |
图谱扩散检索
TriviumDB 的核心创新——Spreading Activation(扩散激活)(受 Anderson, 1983, The Architecture of Cognition 中认知心理学扩散激活理论启发):
高级检索可通过 max_edges_per_node、min_edge_weight 和 edge_direction 精细控制高出度节点:支持出边、入边或双向扩散,弱边在能量归一化前被移除,每节点边上限按确定性的绝对权重顺序选择。默认值保持历史行为:不限边数、阈值为 0、仅沿出边。
图能力分为三条互不混淆的语义路径:
| 路径 | 语义 | 返回内容 | 适用场景 |
|---|---|---|---|
| SA-PPR / Spreading Activation | 边权驱动的软相关性能量传播 | 带 score 的检索命中 | RAG 联想与相关节点补充 |
| Reachability | 按方向、label、深度判断结构可达性 | 确定性最短路径与逐跳 label | 权限链、依赖链、血缘与结构查询 |
| GraphFirst | 图模式先限定合法 anchor,再做集合内精确向量 Top-K | 规范绑定行或 SearchHit | 只允许在结构合法对象中做语义排名 |
Reachability 默认沿出边,可选入边或双向,并通过 max_visited_nodes 防止稠密图失控。GraphFirst 按 anchor NodeId 去重,超过候选预算直接报错;它不会把全库向量近邻混入图约束结果。
工作流程
- 双路锚定 (Hybrid Recall):融合
Aho-Corasick 定点词汇匹配+BM25 倒排相似度+Dense Vector 稠密余弦分数,按alpha权重混合打分,找出最精确的初始锚点,有效解决传统纯向量 RAG 容易在专有名词上“瞎联想”的幻觉缺陷。 - 图谱扩散:从双路召回的锚点池出发,沿邻接表进行 N 跳广度优先遍历
- 热度传播:锚点的相似度得分按边权重衰减传播给邻居节点
- 去重排序:合并锚点和扩散节点,按最终得分排序返回
扩散深度与行为
expand_depth | 行为 |
|---|---|
0 | 纯向量检索,不进行图谱扩散 |
1 | 返回锚点 + 锚点的直接邻居 |
2 | 返回锚点 + 1 跳邻居 + 2 跳邻居 |
N | 返回 N 跳以内的所有关联节点 |
典型应用场景
# AI Agent 记忆系统:用户说了"咖啡"
# 1. 向量检索找到最相似的记忆"昨天去了星巴克"
# 2. 沿图谱扩散,发现关联的人物"小红"和地点"三里屯"
results = db.search(
query_vector=encode("咖啡"),
top_k=3,
expand_depth=2, # 关键!扩散 2 跳
min_score=0.4
)
# 结果:["昨天去了星巴克(0.92)", "小红(0.71)", "三里屯(0.65)"]
边特异性强化(Link Specificity Penalty,本项目自研)
传统入度惩罚使用 1 / (1 + log10(in_degree)),对于入度破千的「黑洞节点」衰减过于缓慢,无法有效阻断能量聚集。
TriviumDB 自研的替代方案——幂函数非线性衰减:
inhibition_factor = 1.0 / in_degree^0.55
| 节点入度 | log10 惩罚系数 | powf(0.55) 惩罚系数 | 效果对比 |
|---|---|---|---|
| 1(叶节点) | 0.500 | 1.000 | 不惩罚 |
| 10 | 0.333 | 0.282 | 更有力 |
| 100 | 0.250 | 0.089 | 显著压制 |
| 1000 | 0.200 | 0.028 | 极强压制 |
这使得「重要但不泛滥」的中层枢纽节点依然能从周围吸收合理的能量,但「全局热点」黑洞被大幅削弱,从而迫使扩散能量向更丰富的亚支路蔓延。
不应期(Refractory Period,疲劳机制,本项目自研)
这是缓解「重复召回」问题的核心机制。命名灵感来源于生物神经元在高频放电后进入不应期、暂时无法再次触发的电生理现象(注:此处为类比性借用,并非精确复现生物神经元行为)。
工作流程:
- 标记(Mark):每次图漫游结束后,排名最高的 Top-15 赢家节点会被打上「疲劳」标记(
fatigue = 1),写入 MemTable 的内部状态映射(RwLock<HashMap<NodeId, u8>>)。 - 抑制(Suppress):下一轮扩散中,若发现目标节点处于疲劳期,该传导路径的能量片段会被直接削减 85%(
fatigue_discount = 0.15)。 - 恢复(Recover):一旦该路径在本轮中被抑制并消耗了疲劳标记,节点的不应期立即解除,不会造成永久封印。
常规场景(无重复访问):
Node A --[energy=0.8]--> Node B → 实际传导 = 0.8
高频重复访问(黑洞热点抑制):
Node A --[energy=0.8]--> Node B(疲劳) → 实际传导 = 0.8 × 0.15 = 0.12
(被节省的 0.68 能量将流向其他未疲劳的邻居节点)
内存开销:疲劳状态存储在独立的 HashMap 中,与向量 SoA 连续内存完全物理隔离,不破坏任何 SIMD / mmap 对齐,零额外计算开销。
关键特性:
- ✅ 仅影响相邻两次搜索——无记忆效应,不影响长期联想
- ✅ 不修改任何边权重——图谱结构本身保持不变
- ✅ 完全运行时状态,不写入 WAL 和 .tdb,无持久化开销
认知检索管线
TriviumDB 内置了一套多层认知检索管线(本项目自研的功能性分层设计,而非业界标准分层模型)。所有数学算子均为纯 Rust 手写,零依赖外部矩阵库。其中借鉴的学术算法包括:
- FISTA: Beck & Teboulle, 2009, "A Fast Iterative Shrinkage-Thresholding Algorithm for Linear Inverse Problems"
- DPP: Kulesza & Taskar, 2012, "Determinantal Point Processes for Machine Learning"
- NMF: Lee & Seung, 1999, "Learning the Parts of Objects by Non-negative Matrix Factorization"
设计哲学
- 可配(Configurable):每个数学参数通过
SearchConfig在运行时控制 - 可关(Runtime Toggleable):每条查询独立决定启用哪些层,不是编译期宏
- 零侵入(Zero-Impact):原有22
search()API 绝对不受影响,认知功能全部收束在search_advanced()入口
管线架构(功能性分层)
| 层级 | 功能 | 实现位置 |
|---|---|---|
| L1/L2 | 意图拆分 + 向量召回 | 外部客户端 + MemTable 向量池 |
| L3 | NMF 语义分解分析 | cognitive.rs · nmf_multiplicative_update |
| L4/L5 | FISTA 稀疏残差 + 影子查询 | cognitive.rs · fista_solve + database.rs 自动触发 |
| L6/L7 | SA-PPR 有限深度扩散 + 边特异性强化 + 不应期抑制 | graph/traversal.rs · 个性化重启 + 出边能量归一化 + 入度惩罚 + 疲劳不应期 |
| L8 | 时间/重要性重排 | 主动向业务侧让权,不侵入底层 |
| L9 | DPP 多样性采样 | cognitive.rs · dpp_greedy + Cholesky 行列式 |
安全拦截层 (Layer 0)
所有进入 search_advanced 的查询会首先经过安全拦截:
- 维度检查:向量维度与库不匹配时立即报错
- NaN / Infinity 毒素检测:向量中包含无效浮点数时扔出清晰错误
- 参数安全钳位:
teleport_alpha、fista_lambda、dpp_quality_weight等全部被强制约束在合法数学范围内
TQL 统一查询语言
TQL (Trivium Query Language) 已发展为完整的三模查询管线,也是 TriviumDB 区别于固定混合检索 API 的关键能力。用户不必接受预设的“向量召回 → 图扩散 → 过滤”顺序,而可以自由决定先按属性缩小集合、再寻路、再与向量候选求交,或先运行图算法产生分数、再过滤和重排。自有 Lexer/Parser/AST、确定性且有界的统计感知 Cascades、NodeSet 物理算子和一等值结果共同保证这种 DIY 能力既自由又可规划、可解释、可预算。
它统一图遍历、文档过滤、向量检索、图算法、聚合与写操作。核心模块包括:
| 模块 | 文件 | 职责 |
|---|---|---|
| 词法分析器 | query/tql_lexer.rs | Token、参数和位置诊断 |
| 语法分析器 | query/tql_parser.rs | 递归下降解析、作用域验证 |
| 抽象语法树 | query/tql_ast.rs | 查询、管线、表达式、聚合和路径结构 |
| Cascades | query/cascades.rs | Memo、成本估算、预算切片与确定性计划选择;优化结果以权威 PhysicalPlan 驱动执行器 lowering(Source/Filter/Expand/Rank 真实物理候选、可序列化物理属性、相邻 Filter 合并与恒等 WITH 消除),优化状态显式为 Complete/Fallback/BudgetExceeded 并通过 EXPLAIN 暴露 |
| 执行器 | query/tql_executor.rs | NodeSet/一等值执行、聚合、路径和图算法 |
| Prepared | query/tql_prepared.rs | 严格参数发现、绑定和重复执行 |
查询入口与可组合管线
| 入口 | 用途 | 示例 |
|---|---|---|
| MATCH | 图谱遍历(沿边跳转) | MATCH (a)-[:knows]->(b) RETURN b |
| FIND | 文档过滤(类 MongoDB) | FIND {type: "event", heat: {$gte: 0.7}} RETURN * |
| SEARCH | 向量检索 | SEARCH VECTOR [...] TOP 10 RETURN * |
| MATCH + RANK | GraphFirst 约束排名 | MATCH (a)-[:rel]->(b) RANK a BY VECTOR [...] TOP 10 RETURN a |
| WITH Pipeline | 跨模组合 | SEARCH ... AS seed WITH seed EXPAND ... RETURN ... |
管线支持 FILTER、RANK、EXPAND、PageRank/WCC/Leiden/SA-PPR、ALL_PATHS、SHORTEST_PATHS、UNION/INTERSECT/EXCEPT、ITERATE。RETURN 支持算术、COALESCE、IS NULL、path()、path_length() 以及 COUNT/SUM/AVG/MIN/MAX/COLLECT。
DML 写操作(v0.6.0 新增)
| 语法 | 功能 | 示例 |
|---|---|---|
| CREATE | 创建节点 | CREATE (a {name: "Alice", age: 30}) |
| SET | 更新属性 | MATCH (a {name: "Alice"}) SET a.age == 31 |
| DELETE | 删除节点 | MATCH (a {name: "Alice"}) DELETE a |
| DETACH DELETE | 删除节点及关联边 | MATCH (a {name: "Alice"}) DETACH DELETE a |
支持的语法元素
| 元素 | 语法 | 示例 |
|---|---|---|
| 节点匹配 | (变量名) | (a) |
| 节点+属性 | (变量名 {key: value}) | (a {id: 42}) |
| 有向边 | -[:标签]-> | -[:knows]-> |
| 通配边 | -[]-> | 匹配任意标签 |
| WHERE 条件 | WHERE 表达式 AND/OR 表达式 | WHERE a.age > 18 |
| RETURN | RETURN 变量名列表 | RETURN a, b |
| 比较运算符 | ==, !=, >, >=, <, <= | b.score >= 0.8 |
执行优化
TQL 的 FIND 入口底层采用三层加速策略:
- 属性二级索引:执行器自动检测是否存在已建索引字段。命中时直接 O(1) 倒排查找,跳过全表扫描。
- Parallel Bit-Tag Array(布隆特征拦截):节点插入时自动展平 JSON 键值对,合成 64 位特征标签
fast_tags。过滤时引擎编译出 Must-have Mask,通过位运算(fast_tags[i] & mask) == mask在几个时钟周期内截断 99% 的不匹配节点,仅少量漏网候选进入完整 JSON 解析。 - JSON 精确验证:对通过前两层的极少数候选节点,执行完整的
$gt/$in/$exists等运算符语义验证。
支持的过滤运算符:$eq / $ne / $gt / $gte / $lt / $lte / $before / $beforeEq / $after / $afterEq / $in / $nin / $exists / $size / $all / $type,以及 $and / $or 逻辑组合。普通范围支持数字和原始字符串字典序且不做跨类型强制转换;时间操作符严格解析 RFC3339 并按绝对时间比较。
持久化属性索引体系
属性索引由 PropertyIndexRegistry 统一管理,并持久化到 .tdb.pidx(当前 v4,可读取 v1–v4)。posting 使用稳定 NodeId/slot 映射,CRUD、slot 复用、重启与 ReadOnly/Immutable 均有专项测试。
数值键编码 v2:Ordered/Composite ART 的有序键使用统一的数值全序编码——i64/u64/f64 共享同一数值顺序,整数范围查询(如 age: {$gte: 30})与浮点阈值不再因键前缀不同而错配为空集。等值索引(Hash/Bitmap)继续保留 JSON 数字的精确类型与值,可区分超过 f64 精度的相邻大整数;Filter 的范围比较也使用精确整数比较。超出 f64 精确整数范围(± 之外)的有序范围边界不参与索引剪枝,自动回退精确扫描。旧 key v1 编码的 .pidx 在打开时有界读取并按索引定义在内存中重建为 v2 编码:ReadOnly/Immutable 保持零写,Writer 在下一次显式 flush 发布 v2 sidecar。
| 类型 | 数据结构 | 主要用途 |
|---|---|---|
| Hash | 类型稳定键 → posting | 等值过滤 |
| Ordered | Safe Rust ART | 范围、前缀、ORDER BY |
| Composite | 多字段 ART | 左前缀、等值 + 末列范围 |
| Bitmap | RoaringTreemap | 低基数、多条件集合运算 |
Planner 可选择单索引、复合索引、Bitmap 或多个 posting 交集;无合适索引时才使用 Fast Tags + 精确 JSON 校验。
三语言 API
db.create_index("kind")
db.create_ordered_index("score")
db.create_composite_index(["tenant", "region"])
db.create_bitmap_index("status")
print(db.index_info())
Node 使用 createIndex/createOrderedIndex/createCompositeIndex/createBitmapIndex/indexInfo;Rust 使用对应 snake_case 方法。删除接口与创建接口一一对应。index_info 返回 kind、完整 fields、entry/distinct/null 计数,复合索引不再压缩成不可解析的单字段字符串。
其他持久化索引
.quiver/.quiver.meta:QuIVer ANN 图与一致性元数据;.text/.text.meta:AC 自动机 + BM25 2-Gram;.gidx:出边块、入边目录、Label 目录与 CRC;.tdb内 BQ block:低成本签名预筛,不等同于完整 QuIVer。
崩溃恢复机制
TriviumDB 的数据安全建立在 WAL + 原子写入的双重保障上:
恢复流程(数据库 open 时自动执行)
1. 校验 WAL 的 `TVWL + v3` 版本头;无头或未知版本在回放前拒绝
2. 逐条读取长度、记录体和 CRC32
3. 完整记录校验通过后回放到 MemTable
4. 遇到截断、非法长度、CRC 不匹配或反序列化失败时停止回放
5. 保留此前已验证记录,不跨过损坏边界继续解释后续字节
崩溃场景矩阵
| 崩溃时机 | .tdb 状态 | .vec 状态 | .flush_ok | 恢复路径 |
|---|---|---|---|---|
| 写 .tdb.tmp 中途 | 旧版本完好 | 旧版本完好 | 有效 | 直接加载旧数据 + WAL 回放 |
| .tdb rename 后、.flush_ok 更新前 | 新版本 | 旧版本(追加路径:新版本) | 失效(大小不符) | .flush_ok 校验失败 → fail-closed 拒绝打开(损坏输入零降级、零伪恢复) |
| 追加写 .vec 后、.tdb 重写前 | 旧版本 | 已追加(比 .flush_ok 记录的大) | 失效 | 同上:fail-closed 拒绝打开 |
| flush 全部完成 | 新版本 | 新版本 | 有效 | 直接加载,无需 WAL |
追加路径的崩溃安全性:
.vec文件追加成功后如果崩溃,.flush_ok中记录的vec_size与实际文件大小不符,下次启动时校验失败。引擎不再降级为"忽略.vec的骨架恢复"——那会依赖 WAL 能完整重建全部基础向量,否则产生"节点与 Payload 正确、向量全零"的伪恢复状态。现在统一 fail-closed:已提交到 WAL 的数据不会丢失,但打开会被拒绝,需由 WAL 完整回放能力或备份恢复后重新进入。发布各阶段的真实强杀矩阵验证只允许"旧完整代际"或"新完整代际"。
WAL 记录类型
WAL 文件以 TVWL + u16 version 版本头开场;历史无头 WAL 不再猜测解析,会返回 UnsupportedWalVersion。带记录的旧版本 WAL 仍要求旧内核恢复并 flush;只有恰好 6 字节、零记录的旧版本 WAL 会在 ReadWrite 持有排他锁时原子升级为当前头,ReadOnly/Immutable 保持零写并拒绝打开。未知未来版本同样在任何记录回放前明确拒绝。图写入支持 (src,dst,label) 唯一边的 weight upsert,以及独立的指定标签 UnlinkLabel 记录。
| 类型 | 内容 |
|---|---|
TxBegin | 事务开始标记(含 tx_id) |
TxCommit | 事务提交封条(含 tx_id,缺失则整个事务丢弃) |
Insert | id + vector + payload |
Delete | id |
Link | src + dst + label + weight |
Unlink | src + dst |
UpdatePayload | id + new_payload |
UpdateVector | id + new_vector |
并发安全与零开销事务
TriviumDB 通过四层机制保障并发安全与数据完整性:
1. 进程与线程锁:
fs2独占文件锁避免多进程同时打开同一数据库写入。Arc<RwLock<MemTable>>允许普通无状态查询共享读;写入、flush、compaction 使用写锁。- WAL 使用独立
Arc<Mutex<Wal>>,固定锁顺序为MemTable write → WAL Mutex。 - poison 通过恢复封装处理;fatigue 查询额外串行,Hook 写重入被拒绝。
2. 验证前置事务与容量原子性:
TriviumDB 的 begin_tx() 提供了一种比传统 MVCC 和 Undo Log 都轻量级得多的验证前置(Dry-Run)架构。
在调用 tx.commit() 后:
- 预检前置:引擎在虚拟映射中验证维度、有限数值、节点存在性、ID 冲突和 Payload 上限。
- 容量门禁:使用 checked arithmetic 估算整批核心容器增量,检查内存预算并调用
try_reserve;失败时不写 WAL、不推进 ID/generation。 - WAL-first:全部验证与预留成功后,一次性追加带事务边界的 WAL。
- Infallible Apply:应用已经验证的操作,再统一同步 QuIVer 增量状态。
普通无状态查询持共享 read guard 完成整条 pipeline;generation 校验保证锁外 QuIVer 构建不会发布过期索引。
Python 绑定架构
多后端动态分发
Python 侧的 TriviumDB 类内部通过 DbBackend 枚举封装三种泛型特化:
enum DbBackend {
F32(Database<f32>),
F16(Database<half::f16>),
U64(Database<u64>),
}
通过 dispatch! 宏实现统一的方法分发,Python 用户无需关心底层类型差异。
dtype 选择指南
| dtype | 单维度字节 | 精度 | 适用场景 |
|---|---|---|---|
f32 | 4 B | 完整精度 | 通用 embedding(推荐默认值) |
f16 | 2 B | 半精度 | 大规模数据集,内存减半,精度损失极小 |
u64 | 8 B | 整数 | SimHash 等二值化/离散化向量 |
数据转换
Python 侧的 dict 与 Rust 侧的 serde_json::Value 通过 pyobject_to_json / json_to_pyobject 双向无损转换。支持的 Python 类型:None / bool / int / float / str / list / dict。
Node.js 绑定架构
Node.js 侧通过 napi-rs 提供原生扩展,自带完整的 TypeScript 类型定义。同样通过 DbBackend 枚举 + dispatch! 宏模式实现多类型动态分发。通过 JsSearchConfig 结构体暂露完整的认知管线配置。
Hook 管理接口 (v0.6.0 新增)
Python 和 Node.js 绑定均新增了以下 Hook 管理接口:
| Python 方法 | Node.js 方法 | 说明 |
|---|---|---|
db.load_ffi_hook(path) | db.loadFfiHook(path) | 加载 C/C++ 动态库插件 |
db.clear_hook() | db.clearHook() | 清除 Hook,恢复 NoopHook |
db.search_with_context(vec, ...) | db.searchWithContext(vec, config?) | 带管线上下文的检索 |
返回的 HookContext / JsHookContext 对象包含:
- timings:各管线阶段耗时(毫秒)
- custom_data:Hook 注入的自定义数据
- aborted:管线是否被 Hook 提前终止
🔌 Hook 扩展系统架构 (v0.6.0)
TriviumDB 的 Hook 系统允许开发者在构建 RAG 系统时,通过 6 个管线关键阶段的注入点来自定义字段、回传数据、内联/外置高性能计算模块。
设计原则
- 零开销可选:默认
NoopHook的所有方法为空实现,编译器内联消除全部调用开销 - 按需覆写:所有方法都有默认空实现,开发者只需覆写感兴趣的阶段
- FFI 友好:
FfiHook支持extern "C"函数签名的 C/C++ 动态库加载
Hook 类型体系
SearchHook (trait)
├── NoopHook — 零开销默认实现(编译器内联消除)
├── CompositeHook — 多 Hook 组合(按注册顺序链式调用)
└── FfiHook — C/C++ 动态库加载(libloading)
6 个管线注入点详解
| Hook 点 | 阶段 | 可修改的数据 | 典型用途 |
|---|---|---|---|
on_pre_search | 查询预处理 | 查询向量、SearchConfig、HookContext | 查询改写、用户上下文注入、条件拦截 |
on_custom_recall | 自定义召回 | 返回 Option<Vec<SearchHit>> | 对接外部 FAISS/ScaNN 索引替代内置召回 |
on_post_recall | 召回后处理 | &mut Vec<SearchHit> | 业务过滤、分数调权、去重 |
on_pre_graph_expand | 图扩散前 | &mut Vec<SearchHit> | 种子集过滤/增强/截断 |
on_rerank | 重排序 | &mut Vec<SearchHit> / 替换 | 外置 Cross-Encoder、ONNX 推理重排 |
on_post_search | 最终后处理 | &mut Vec<SearchHit> | 统计埋点、结果增强、回传自定义数据 |
HookContext 共享上下文
HookContext 在管线各阶段之间传递共享状态:
| 字段 | 类型 | 说明 |
|---|---|---|
custom_data | serde_json::Value | 开发者自定义附加数据(任意 JSON) |
stage_timings | Vec<(String, Duration)> | 管线阶段计时统计(自动填充) |
abort | bool | 设为 true 则跳过后续所有阶段 |
典型使用场景
# 场景 1:加载 C++ 高性能召回插件
db.load_ffi_hook("./libfaiss_hook.so")
# 场景 2:管线性能诊断
hits, ctx = db.search_with_context(query_vec, top_k=10)
for stage, ms in ctx.timings.items():
print(f" {stage}: {ms:.2f}ms")
# 场景 3:业务条件拦截(Rust 侧实现 Hook)
# 在 on_pre_search 中检查用户权限,设置 ctx.abort = true 拒绝查询
模块化架构 (v0.6.0 重构)
原 1815 行的 database.rs 已按职责拆分为 4 个独立模块,提高可维护性和可测试性:
| 模块 | 行数 | 职责 |
|---|---|---|
database/mod.rs | ~560 | Database 结构体、CRUD 操作、生命周期管理 |
database/config.rs | ~110 | StorageMode、Config、SearchConfig 配置类型 |
database/pipeline.rs | ~620 | 检索管线 L0-L9 + 6 个 Hook 注入点(拆为 8 个独立子函数) |
database/transaction.rs | ~460 | 事务系统(TxOp、Transaction)+ WAL 崩溃恢复 |
pipeline.rs 内部子函数
| 函数 | 职责 |
|---|---|
execute_pipeline() | 管线总控 + 6 个 Hook 调用入口 |
recall_text() | L1: AC 自动机 + BM25 文本召回 |
recall_vector() | L2+L3: 向量稠密召回 + 自适应引擎路由 |
quiver_pipeline() | QuIVer ANN 图索引搜索(BQ + Vamana 图导航 + f32 精排) |
brute_force_pipeline() | 暴力全扫管线 |
recall_residual() | L4+L5: FISTA 残差 + 影子查询 |
aggregate_seeds() | seed_map 聚合 + 排序 |
apply_dpp() | L9: DPP 多样性采样 |
公共 API 不承诺保留历史静默入口。已移除的调用会在编译期失败,或返回
ApiMigrationRequired/TDB_API_MIGRATION_REQUIRED并给出替代入口。
跨架构 SIMD 加速
v0.6.0 新增 ARM64 (aarch64) NEON SIMD 支持,与已有的 x86 AVX2 并列,实现跨平台高性能向量计算。
余弦相似度 SIMD 调度
| 架构 | SIMD 路径 | 并行度 | 核心指令 |
|---|---|---|---|
| x86_64 | AVX2 + FMA | 8 × f32 | `_mm256_fmadd_ps$ |
| \text{aarch64} | \text{NEON} | 4 \times \text{f32} | |
| 其他 | 标量回退 | 1 \times \text{f32} | 四路展开循环 |
缓存预取
| 架构 | 预取指令 |
|---|---|
| \text{x86_64} | $_mm_prefetch` (SSE) |
| aarch64 | prfm pldl1keep (inline asm) |
编译时自动选择最优路径,零开销回退,无需用户配置。