Skip to main content

spg_storage/
lib.rs

1//! In-memory storage primitives.
2//!
3//! v0.3 is intentionally simple: a flat catalog of tables, each holding rows
4//! as `Vec<Value>` (positional, matching the table's `TableSchema`). No MVCC,
5//! no on-disk format — those land in later milestones.
6#![no_std]
7// v3.3.2 NEON path for l2_distance_sq (aarch64 only). Scoped allow:
8// `unsafe_code = "deny"` at workspace level stays in force for every
9// other crate.
10#![cfg_attr(target_arch = "aarch64", allow(unsafe_code))]
11
12extern crate alloc;
13
14pub mod bloom;
15pub mod halfvec;
16pub mod persistent;
17pub mod persistent_btree;
18pub mod quantize;
19pub mod row_locator;
20pub mod segment;
21pub mod trgm;
22
23pub use self::bloom::{BloomError, BloomFilter};
24pub use self::row_locator::{RowLocator, RowLocatorError};
25pub use self::segment::{
26    BRIN_SIDECAR_MAGIC, BrinSummary, OwnedSegment, SEGMENT_COMPRESS_ALGO_LZSS,
27    SEGMENT_COMPRESS_ALGO_NONE, SEGMENT_MAGIC, SEGMENT_MAGIC_V2, SEGMENT_PAGE_BYTES, SegmentError,
28    SegmentMeta, SegmentReader, derive_brin_summaries, encode_segment, wrap_v2_envelope,
29    wrap_v2_envelope_with_brin,
30};
31
32use alloc::boxed::Box;
33use alloc::collections::{BTreeMap, BTreeSet};
34use alloc::format;
35use alloc::string::{String, ToString};
36use alloc::sync::Arc;
37use alloc::vec::Vec;
38use core::fmt;
39
40use self::persistent::PersistentVec;
41use self::persistent_btree::PersistentBTreeMap;
42
43/// In-cell encoding for `DataType::Vector`. Mirrors
44/// `spg_sql::ast::VecEncoding` — kept here so storage stays
45/// dep-free of `spg-sql`. The engine bridges between the two
46/// at DDL-execution time.
47///
48/// `F32` is the pre-v6 default: each cell holds a raw `Vec<f32>`.
49/// `Sq8` (v6.0.1) stores `Sq8Vector { min, max, bytes: Vec<u8> }`
50/// per cell; 4× compression vs `F32` with recall@10 ≥ 0.95 on
51/// natural embeddings (Gaussian / unit-sphere corpora).
52/// `F16` (v6.0.3, DDL keyword `HALF`) stores each element as
53/// IEEE-754 binary16; 2× compression and bit-exact dequantise.
54#[derive(Debug, Clone, Copy, PartialEq, Eq, Default)]
55pub enum VecEncoding {
56    #[default]
57    F32,
58    Sq8,
59    F16,
60}
61
62impl fmt::Display for VecEncoding {
63    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
64        match self {
65            Self::F32 => f.write_str("F32"),
66            Self::Sq8 => f.write_str("SQ8"),
67            Self::F16 => f.write_str("HALF"),
68        }
69    }
70}
71
72/// Runtime type tags. `Vector { dim, encoding }` / `Varchar(max)` /
73/// `Char(size)` are parameterised; the parameter travels with both
74/// the column schema and the on-wire serialised representation.
75#[derive(Debug, Clone, Copy, PartialEq, Eq)]
76pub enum DataType {
77    /// 16-bit signed. Backed by `Value::SmallInt(i16)`; arithmetic that
78    /// would overflow surfaces as a type error at INSERT time.
79    SmallInt,
80    Int,    // 32-bit signed
81    BigInt, // 64-bit signed
82    Float,  // f64 (PG double precision)
83    Text,
84    /// `VARCHAR(n)` — same byte representation as `Text`, but INSERT
85    /// rejects values longer than `n` Unicode characters.
86    Varchar(u32),
87    /// `CHAR(n)` — same representation as `Text`, but INSERT right-pads
88    /// with U+0020 to exactly `n` Unicode characters (or rejects when
89    /// the input is already longer).
90    Char(u32),
91    Bool,
92    /// pgvector-style fixed-dimension vector. `encoding` selects
93    /// the in-cell representation (`F32` = pre-v6 raw f32 buffer;
94    /// `Sq8` = v6.0.1 8-bit scalar-quantised). The DDL grammar
95    /// surfaces encoding via the optional `USING <encoding>`
96    /// clause: `VECTOR(128) USING SQ8`.
97    Vector {
98        dim: u32,
99        encoding: VecEncoding,
100    },
101    /// `NUMERIC(precision, scale)` — exact fixed-point decimal stored as
102    /// a scaled `i128`. `precision` caps total decimal digits, `scale`
103    /// fixes digits after the decimal point. v1.12 supports up to
104    /// precision 38 (the i128-safe ceiling). `NUMERIC` and `NUMERIC(p)`
105    /// surface as `Numeric { precision: p, scale: 0 }`.
106    Numeric {
107        precision: u8,
108        scale: u8,
109    },
110    /// `DATE` — calendar date with day precision, stored as `i32` days
111    /// since the Unix epoch (1970-01-01).
112    Date,
113    /// `TIMESTAMP` (a.k.a. `MySQL` `DATETIME`) — instant with microsecond
114    /// precision, stored as `i64` microseconds since the Unix epoch.
115    Timestamp,
116    /// v7.9.2 `TIMESTAMPTZ` — bit-identical to `Timestamp` on disk
117    /// (i64 microseconds, UTC by convention). Carried as a distinct
118    /// type tag so the PG-wire layer can advertise OID 1184 (PG's
119    /// `timestamp with time zone`) and `sqlx`/`pgx`/JDBC clients
120    /// decode into their TZ-aware datetime types. The internal
121    /// semantics are unchanged: SPG never stored per-row offsets,
122    /// and neither did PG — `TIMESTAMPTZ` in PG is also UTC i64.
123    Timestamptz,
124    /// `INTERVAL` — calendar-aware span (months + microseconds). v2.11
125    /// supports INTERVAL only as a runtime intermediate (literals,
126    /// arithmetic results); on-disk encoding is rejected so this branch
127    /// can't appear in a `ColumnSchema`.
128    Interval,
129    /// v4.9: `JSON` — text-backed JSON document. We don't parse
130    /// the content (no path operators or jsonb functions yet) —
131    /// the column accepts any TEXT-compatible value and round-trips
132    /// it verbatim. PG OID 114 on the wire.
133    Json,
134    /// v7.9.0: `JSONB` — semantically identical to `Json` on
135    /// the storage side (same `Value::Json` cells, same
136    /// row codec), but advertised as PG OID 3802 on the wire
137    /// so `sqlx`-style clients that bind `jsonb` columns
138    /// decode correctly. mailrs migration blocker #3.
139    Jsonb,
140    /// v7.10.4: `BYTES` / `BYTEA` — variable-length raw binary.
141    /// Backed by `Value::Bytes(Vec<u8>)`. PG wire OID 17. Literal
142    /// forms accepted by parser/engine: PG hex form `'\xDEADBEEF'`
143    /// (case-insensitive hex pairs) and escape form
144    /// `'foo\\000bar'` (the latter decoded at coercion time when
145    /// the target column is BYTEA — TEXT columns leave the
146    /// backslash sequence verbatim).
147    Bytes,
148    /// v7.10.9: `TEXT[]` — single-dimension TEXT array. Elements
149    /// may be NULL (PG semantics). PG wire OID 1009. Literal
150    /// forms: `ARRAY['a', 'b', NULL]` and the PG external form
151    /// `'{a,b,NULL}'::TEXT[]`. Engine implements `= ANY(arr)`,
152    /// `<> ALL(arr)`, and 1-based indexing `arr[i]`. Catalog
153    /// FILE_VERSION 18+; older snapshots reject this DataType
154    /// (forward-only by design — TEXT[] columns aren't readable
155    /// on a pre-v7.10 binary).
156    TextArray,
157    /// v7.11.12: `INT[]` — single-dimension i32 array. PG wire
158    /// OID 1007 (_int4). Same `ARRAY[...]` / `'{1,2,3}'::INT[]`
159    /// literal surface as TEXT[]. Catalog FILE_VERSION 19+.
160    IntArray,
161    /// v7.11.12: `BIGINT[]` — single-dimension i64 array. PG
162    /// wire OID 1016 (_int8). Catalog FILE_VERSION 19+.
163    BigIntArray,
164    /// v7.12.0: PG `tsvector` — ordered, deduplicated set of
165    /// `(lexeme, positions, weight)` tuples. PG wire OID 3614.
166    /// Catalog FILE_VERSION 20+. Storage shape is row-codec
167    /// tag 22; the schema-agnostic `write_value` path emits tag
168    /// 18. Literal: `'foo:1 bar:2,3'::tsvector` (PG external
169    /// form). G-CRIT-3 entry — v7.12.0 only ships the type +
170    /// codec; matching `@@` lands in v7.12.2.
171    TsVector,
172    /// v7.12.0: PG `tsquery` — parse tree of lexemes joined by
173    /// `&` `|` `!` and phrase operators. PG wire OID 3615.
174    /// Catalog FILE_VERSION 20+.
175    TsQuery,
176}
177
178impl fmt::Display for DataType {
179    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
180        match self {
181            Self::SmallInt => f.write_str("SMALLINT"),
182            Self::Int => f.write_str("INT"),
183            Self::BigInt => f.write_str("BIGINT"),
184            Self::Float => f.write_str("FLOAT"),
185            Self::Text => f.write_str("TEXT"),
186            Self::Varchar(n) => write!(f, "VARCHAR({n})"),
187            Self::Char(n) => write!(f, "CHAR({n})"),
188            Self::Bool => f.write_str("BOOL"),
189            Self::Vector { dim, encoding } => match encoding {
190                VecEncoding::F32 => write!(f, "VECTOR({dim})"),
191                VecEncoding::Sq8 => write!(f, "VECTOR({dim}) USING SQ8"),
192                VecEncoding::F16 => write!(f, "VECTOR({dim}) USING HALF"),
193            },
194            Self::Numeric { precision, scale } => {
195                if *scale == 0 {
196                    write!(f, "NUMERIC({precision})")
197                } else {
198                    write!(f, "NUMERIC({precision}, {scale})")
199                }
200            }
201            Self::Date => f.write_str("DATE"),
202            Self::Timestamp => f.write_str("TIMESTAMP"),
203            Self::Timestamptz => f.write_str("TIMESTAMPTZ"),
204            Self::Interval => f.write_str("INTERVAL"),
205            Self::Json => f.write_str("JSON"),
206            Self::Jsonb => f.write_str("JSONB"),
207            Self::Bytes => f.write_str("BYTEA"),
208            Self::TextArray => f.write_str("TEXT[]"),
209            Self::IntArray => f.write_str("INT[]"),
210            Self::BigIntArray => f.write_str("BIGINT[]"),
211            Self::TsVector => f.write_str("TSVECTOR"),
212            Self::TsQuery => f.write_str("TSQUERY"),
213        }
214    }
215}
216
217/// v7.12.0 — one entry in a `Value::TsVector`. The lexeme is the
218/// (already-tokenised + stemmed in v7.12.1+) word; `positions` is
219/// a strictly-ascending list of 1-based positions; `weight` is the
220/// PG weight letter (A=3, B=2, C=1, D=0) — v7.12.0 defaults every
221/// lexeme to D, the v7.12.2 ranking path consumes the weight.
222#[derive(Debug, Clone, PartialEq, Eq)]
223pub struct TsLexeme {
224    pub word: String,
225    pub positions: Vec<u16>,
226    pub weight: u8,
227}
228
229/// v7.12.0 — parse tree for a PG `tsquery`. v7.12.0 ships the
230/// type + codec only; the `to_tsquery` / `plainto_tsquery` lexer
231/// lands in v7.12.1 and the `@@` evaluator in v7.12.2.
232#[derive(Debug, Clone, PartialEq, Eq)]
233pub enum TsQueryAst {
234    /// Single lexeme term. The `weight_mask` is the PG-style
235    /// bitmask of accepted weights (`A=1<<3`, `B=1<<2`, `C=1<<1`,
236    /// `D=1<<0`); `0` = any weight. v7.12.0 always sets it to 0.
237    Term {
238        word: String,
239        weight_mask: u8,
240    },
241    And(Box<TsQueryAst>, Box<TsQueryAst>),
242    Or(Box<TsQueryAst>, Box<TsQueryAst>),
243    Not(Box<TsQueryAst>),
244    /// `phrase <distance> phrase`. v7.12.0 only persists this; the
245    /// match semantics arrive in v7.12.2 alongside `@@`.
246    Phrase {
247        left: Box<TsQueryAst>,
248        right: Box<TsQueryAst>,
249        distance: u16,
250    },
251}
252
253/// A row-cell value, including SQL `NULL`. `Float` uses `f64`; NaN compares
254/// non-equal to itself (PG behaviour) — `PartialEq` is derived so callers
255/// must opt into NaN-aware comparison if they need stronger guarantees.
256#[derive(Debug, Clone, PartialEq)]
257#[non_exhaustive]
258pub enum Value {
259    SmallInt(i16),
260    Int(i32),
261    BigInt(i64),
262    Float(f64),
263    Text(String),
264    Bool(bool),
265    Vector(Vec<f32>),
266    /// v6.0.1: 8-bit scalar-quantised vector cell. Lives in
267    /// columns declared `VECTOR(N) USING SQ8`. Layout per cell:
268    /// `Sq8Vector { min: f32, max: f32, bytes: Vec<u8> }` —
269    /// 4× compression vs `Vector(Vec<f32>)`. The wire layer
270    /// dequantises to `f32` on SELECT; INSERT path quantises
271    /// incoming `Vector(Vec<f32>)` cells into this variant.
272    Sq8Vector(crate::quantize::Sq8Vector),
273    /// v6.0.3: IEEE-754 binary16 vector cell. Lives in columns
274    /// declared `VECTOR(N) USING HALF`. Stores raw u16 LE bits
275    /// (2× compression vs `Vector(Vec<f32>)`). Wire / display
276    /// paths dequantise to f32 bit-exactly; INSERT path converts
277    /// incoming f32 vectors at the engine boundary.
278    HalfVector(crate::halfvec::HalfVector),
279    /// Exact fixed-point decimal. `scaled` holds the value as
280    /// `actual * 10^scale` so the storage type is always integral —
281    /// arithmetic never falls back to floating-point.
282    Numeric {
283        scaled: i128,
284        scale: u8,
285    },
286    /// Days since the Unix epoch (1970-01-01). Negative for earlier dates.
287    Date(i32),
288    /// Microseconds since the Unix epoch (1970-01-01T00:00:00Z).
289    Timestamp(i64),
290    /// Calendar span: `months` (variable-length) + `micros` (fixed-length).
291    /// Runtime-only — cannot appear in a stored row in v2.11.
292    Interval {
293        months: i32,
294        micros: i64,
295    },
296    /// v4.9 `JSON` — raw JSON text. No structural validation
297    /// happens at the storage layer; whatever the parser hands us
298    /// round-trips verbatim. Equality is byte-wise.
299    Json(String),
300    /// v7.10.4 `BYTEA` — raw binary blob. Equality is byte-wise.
301    /// Layout matches `Text`'s length-prefixed shape (`[u32 LE
302    /// len][bytes]`) under tag 18; the engine accepts PG hex
303    /// literals (`'\xDEADBEEF'`) and escape literals at the
304    /// coercion boundary.
305    Bytes(Vec<u8>),
306    /// v7.10.9 `TEXT[]` — single-dimension TEXT array with
307    /// optional NULL elements. Equality is element-wise. PG's
308    /// NULL-element comparison semantics: NULL ≠ NULL inside
309    /// arrays under `=`, so `[NULL] != [NULL]` (the engine
310    /// honours this).
311    TextArray(Vec<Option<String>>),
312    /// v7.11.12 `INT[]` — single-dimension i32 array with optional
313    /// NULL elements. Codec mirrors TextArray with i32 LE per
314    /// element instead of length-prefixed UTF-8.
315    IntArray(Vec<Option<i32>>),
316    /// v7.11.12 `BIGINT[]` — single-dimension i64 array with optional
317    /// NULL elements.
318    BigIntArray(Vec<Option<i64>>),
319    /// v7.12.0 `tsvector` — sorted-by-word, deduped lexeme set with
320    /// positions + weights. The engine enforces sort/dedup on
321    /// construction; consumers can rely on `lexemes.windows(2)`
322    /// being strictly ascending by `word`.
323    TsVector(Vec<TsLexeme>),
324    /// v7.12.0 `tsquery` — boolean / phrase parse tree over
325    /// lexemes. Engine builds via `to_tsquery` family.
326    TsQuery(TsQueryAst),
327    Null,
328}
329
330impl Value {
331    /// Type tag, or `None` for `NULL` (unknown at value level).
332    pub fn data_type(&self) -> Option<DataType> {
333        match self {
334            Self::SmallInt(_) => Some(DataType::SmallInt),
335            Self::Int(_) => Some(DataType::Int),
336            Self::BigInt(_) => Some(DataType::BigInt),
337            Self::Float(_) => Some(DataType::Float),
338            // `Text` covers both unbounded TEXT and bounded VARCHAR/CHAR
339            // — the constraint lives on the column schema, not the value.
340            Self::Text(_) => Some(DataType::Text),
341            Self::Bool(_) => Some(DataType::Bool),
342            Self::Vector(v) => Some(DataType::Vector {
343                dim: u32::try_from(v.len()).expect("vector dim ≤ u32"),
344                encoding: VecEncoding::F32,
345            }),
346            Self::Sq8Vector(q) => Some(DataType::Vector {
347                dim: u32::try_from(q.bytes.len()).expect("vector dim ≤ u32"),
348                encoding: VecEncoding::Sq8,
349            }),
350            Self::HalfVector(h) => Some(DataType::Vector {
351                dim: u32::try_from(h.dim()).expect("vector dim ≤ u32"),
352                encoding: VecEncoding::F16,
353            }),
354            // `Value::Numeric` doesn't carry its precision (the column
355            // schema does); we surface precision=0 as "unknown" and let
356            // the engine reconcile against the column type at coercion
357            // time.
358            Self::Numeric { scale, .. } => Some(DataType::Numeric {
359                precision: 0,
360                scale: *scale,
361            }),
362            Self::Date(_) => Some(DataType::Date),
363            Self::Timestamp(_) => Some(DataType::Timestamp),
364            Self::Interval { .. } => Some(DataType::Interval),
365            Self::Json(_) => Some(DataType::Json),
366            Self::Bytes(_) => Some(DataType::Bytes),
367            Self::TextArray(_) => Some(DataType::TextArray),
368            Self::IntArray(_) => Some(DataType::IntArray),
369            Self::BigIntArray(_) => Some(DataType::BigIntArray),
370            Self::TsVector(_) => Some(DataType::TsVector),
371            Self::TsQuery(_) => Some(DataType::TsQuery),
372            Self::Null => None,
373        }
374    }
375
376    pub const fn is_null(&self) -> bool {
377        matches!(self, Self::Null)
378    }
379}
380
381/// One table row — values are positional and must match
382/// `TableSchema.columns` in length and (modulo NULL) in `DataType`.
383#[derive(Debug, Clone, PartialEq)]
384pub struct Row {
385    pub values: Vec<Value>,
386}
387
388impl Row {
389    pub const fn new(values: Vec<Value>) -> Self {
390        Self { values }
391    }
392
393    pub fn len(&self) -> usize {
394        self.values.len()
395    }
396
397    pub fn is_empty(&self) -> bool {
398        self.values.is_empty()
399    }
400}
401
402#[derive(Debug, Clone, PartialEq)]
403pub struct ColumnSchema {
404    pub name: String,
405    pub ty: DataType,
406    pub nullable: bool,
407    /// Optional `DEFAULT` value, frozen at CREATE TABLE time. `None`
408    /// means "no default" (so omitted columns become NULL, or error
409    /// out when the column is NOT NULL). Literal defaults take this
410    /// path.
411    pub default: Option<Value>,
412    /// v7.9.21 — for DEFAULT expressions that need INSERT-time
413    /// evaluation (e.g. `DEFAULT now()`, `DEFAULT CURRENT_TIMESTAMP`),
414    /// the Display form of the expression. The engine re-parses
415    /// it on each INSERT default-fill, evaluates against an empty
416    /// row context, and coerces to the column type. mailrs G4.
417    /// Persisted in catalog FILE_VERSION 15+; older catalogs
418    /// deserialise with None.
419    pub runtime_default: Option<String>,
420    /// MySQL-style `AUTO_INCREMENT`. When set, an INSERT that leaves
421    /// this column unbound (or sets it to NULL) gets the next integer
422    /// computed from the column's current max + 1.
423    pub auto_increment: bool,
424}
425
426#[derive(Debug, Clone, PartialEq)]
427pub struct TableSchema {
428    pub name: String,
429    pub columns: Vec<ColumnSchema>,
430    /// v6.7.2 — per-table hot-tier byte budget override. `None`
431    /// falls through to the global `SPG_HOT_TIER_BYTES` setting;
432    /// `Some(n)` overrides it for this specific table. Set via
433    /// `ALTER TABLE t SET hot_tier_bytes = X`. Persisted in
434    /// catalog FILE_VERSION 11+.
435    pub hot_tier_bytes: Option<u64>,
436    /// v7.6.1 — FOREIGN KEY constraints declared on this table.
437    /// Engine maintains this in lock-step with `spg-sql`'s parser
438    /// AST; the storage layer carries the on-disk shape so a
439    /// catalog snapshot round-trips without external mapping.
440    /// Persisted in catalog FILE_VERSION 13+. Older catalogs
441    /// deserialise with an empty vec.
442    pub foreign_keys: Vec<ForeignKeyConstraint>,
443    /// v7.9.19 — composite UNIQUE / PRIMARY KEY constraints
444    /// declared at the table level. Each entry's leading column
445    /// has a BTree index (created via the constraint), and INSERT
446    /// path enforces the full-tuple uniqueness via a scan keyed
447    /// by the leading column. Persisted in catalog FILE_VERSION
448    /// 15+. Older catalogs (≤ 14) deserialise with an empty vec.
449    pub uniqueness_constraints: Vec<UniquenessConstraint>,
450    /// v7.13.0 — `CHECK (<expr>)` predicates declared on this
451    /// table. Both column-level inline `CHECK (…)` and
452    /// table-level `CHECK (…)` fold into this list. Each entry
453    /// is the AST Expr's `Display` form, re-parsed on every
454    /// INSERT/UPDATE and evaluated against the candidate row.
455    /// A false / NULL result rejects the mutation (PG semantics).
456    /// Persisted in catalog FILE_VERSION 23+. Older catalogs
457    /// deserialise with an empty vec.
458    pub checks: Vec<String>,
459}
460
461/// v7.9.19 — composite UNIQUE / PRIMARY KEY constraint persisted
462/// on the table schema. The leading column always has a BTree
463/// index (created at CREATE TABLE time); INSERT enforcement
464/// scans that index for collisions on the full column tuple.
465#[derive(Debug, Clone, PartialEq, Eq)]
466pub struct UniquenessConstraint {
467    /// `true` when this constraint was declared as `PRIMARY KEY`
468    /// (vs `UNIQUE`). Semantically PK implies NOT NULL on all
469    /// referenced columns; the engine enforces that at CREATE
470    /// TABLE time.
471    pub is_primary_key: bool,
472    /// Column positions on the parent table. ≥ 1 element. For
473    /// single-column UNIQUE this is exactly one position; the
474    /// BTree index alone enforces it.
475    pub columns: Vec<usize>,
476    /// v7.13.0 — `UNIQUE NULLS NOT DISTINCT` modifier
477    /// (mailrs round-5 G10; PG 15+ surface). When `true`, two
478    /// rows whose constrained columns are all NULL collide on
479    /// the constraint. Default (`false`) is the SQL-standard
480    /// `NULLS DISTINCT` behaviour where any NULL passes.
481    /// Persisted in catalog FILE_VERSION 23+.
482    pub nulls_not_distinct: bool,
483}
484
485/// v7.6.1 — Storage-layer mirror of `spg_sql::ast::ForeignKeyConstraint`.
486/// The engine's CREATE TABLE path translates between the two; keeping
487/// them separate preserves the no-deps boundary between
488/// `spg-storage` and `spg-sql`.
489#[derive(Debug, Clone, PartialEq, Eq)]
490pub struct ForeignKeyConstraint {
491    /// Optional user-supplied constraint name (`CONSTRAINT <name>`
492    /// prefix). Used by `ALTER TABLE DROP CONSTRAINT <name>` in
493    /// v7.6.8; ignored by enforcement.
494    pub name: Option<String>,
495    /// Positions of local columns in this table's column list.
496    /// Same arity as `parent_columns`.
497    pub local_columns: Vec<usize>,
498    /// Referenced parent table name.
499    pub parent_table: String,
500    /// Positions of parent columns in the parent's column list.
501    /// Engine resolves these at CREATE TABLE time (after the parent
502    /// schema is known) so enforcement paths can skip the name
503    /// lookup on every row.
504    pub parent_columns: Vec<usize>,
505    /// Referential action when a parent row is deleted.
506    pub on_delete: FkAction,
507    /// Referential action when a parent row's referenced columns
508    /// are updated.
509    pub on_update: FkAction,
510}
511
512/// v7.6.1 — referential action tag. Mirrors `spg_sql::ast::FkAction`.
513#[derive(Debug, Clone, Copy, PartialEq, Eq)]
514pub enum FkAction {
515    Restrict,
516    Cascade,
517    SetNull,
518    SetDefault,
519    NoAction,
520}
521
522impl FkAction {
523    /// On-disk tag byte (v13 catalog appendix).
524    pub const fn tag(self) -> u8 {
525        match self {
526            Self::Restrict => 0,
527            Self::Cascade => 1,
528            Self::SetNull => 2,
529            Self::SetDefault => 3,
530            Self::NoAction => 4,
531        }
532    }
533    pub const fn from_tag(b: u8) -> Option<Self> {
534        Some(match b {
535            0 => Self::Restrict,
536            1 => Self::Cascade,
537            2 => Self::SetNull,
538            3 => Self::SetDefault,
539            4 => Self::NoAction,
540            _ => return None,
541        })
542    }
543}
544
545impl TableSchema {
546    pub fn column_position(&self, name: &str) -> Option<usize> {
547        self.columns.iter().position(|c| c.name == name)
548    }
549}
550
551/// Key type accepted by secondary indices. Float / NULL / Vector values
552/// can't participate in a B-tree index — `f64` is only `PartialOrd`, NULL
553/// has SQL-three-valued semantics, and Vector belongs to the (future) HNSW
554/// path. Index lookups on those columns fall back to full scan.
555#[derive(Debug, Clone, PartialEq, Eq, PartialOrd, Ord)]
556pub enum IndexKey {
557    Int(i64),
558    Text(String),
559    Bool(bool),
560}
561
562impl IndexKey {
563    pub fn from_value(v: &Value) -> Option<Self> {
564        match v {
565            Value::SmallInt(n) => Some(Self::Int(i64::from(*n))),
566            Value::Int(n) => Some(Self::Int(i64::from(*n))),
567            Value::BigInt(n) => Some(Self::Int(*n)),
568            Value::Text(s) => Some(Self::Text(s.clone())),
569            Value::Bool(b) => Some(Self::Bool(*b)),
570            // Date/Timestamp use their integer storage repr as the
571            // index key — same order semantics, same comparison.
572            Value::Date(d) => Some(Self::Int(i64::from(*d))),
573            Value::Timestamp(t) => Some(Self::Int(*t)),
574            // Numeric isn't (yet) indexable — exact-decimal index keys
575            // would need a stable scale-normalised representation.
576            // Interval isn't index-eligible either (and can't reach this
577            // path through column storage anyway).
578            Value::Null
579            | Value::Float(_)
580            | Value::Vector(_)
581            | Value::Sq8Vector(_)
582            | Value::HalfVector(_)
583            | Value::Numeric { .. }
584            | Value::Interval { .. }
585            | Value::Json(_)
586            | Value::Bytes(_)
587            | Value::TextArray(_)
588            | Value::IntArray(_)
589            | Value::BigIntArray(_)
590            | Value::TsVector(_)
591            | Value::TsQuery(_) => None,
592        }
593    }
594}
595
596/// A single-column secondary index. v2.0 carries either a B-tree map
597/// (the default — used for equality / range lookups on scalar columns)
598/// or a navigable-small-world graph (used for kNN over vector
599/// columns).
600#[derive(Debug, Clone)]
601pub struct Index {
602    pub name: String,
603    pub column_position: usize,
604    pub kind: IndexKind,
605    /// v6.8.0 — column positions of `INCLUDE (col1, col2, …)`
606    /// non-key columns. Carries the planner's "this query is
607    /// covered by the index" signal; lookup paths still resolve
608    /// via the `RowLocator` to fetch the row body, but EXPLAIN
609    /// surfaces the covered-scan annotation so operators can
610    /// confirm the planner sees the coverage.
611    ///
612    /// Empty `Vec` = no `INCLUDE` clause (the legacy shape). v12
613    /// catalog snapshots deserialise with an empty vec.
614    pub included_columns: Vec<usize>,
615    /// v6.8.1 — partial-index predicate stored as its canonical
616    /// Display form (the engine re-parses it on the maintenance
617    /// path). `None` = unconditional index (the legacy shape).
618    /// Persisted as `[u8 has_pred][u16 LE len][bytes]` on the
619    /// catalog snapshot (FILE_VERSION 12, appended after
620    /// `included_columns`).
621    pub partial_predicate: Option<String>,
622    /// v6.8.2 — expression-index key, stored as the expression's
623    /// canonical Display form. `None` = bare column-reference
624    /// index (the legacy shape). Persisted alongside
625    /// `partial_predicate` on the v12 catalog snapshot.
626    pub expression: Option<String>,
627    /// v7.9.29 — `CREATE UNIQUE INDEX …`. When true the engine
628    /// rejects INSERTs whose key already appears in this index
629    /// (combined with `partial_predicate` when present — only
630    /// rows matching the predicate enter the uniqueness check).
631    /// Catalog FILE_VERSION 16+; older snapshots deserialise
632    /// with `false`. mailrs K1.
633    pub is_unique: bool,
634    /// v7.9.29 — extra (non-leading) column positions for
635    /// multi-column indexes (`CREATE INDEX … (a, b, c)`). The
636    /// planner today still only uses the leading
637    /// `column_position` for index seeks, but UNIQUE INDEX
638    /// enforcement walks the full tuple so partial-unique
639    /// invariants like CalDAV `(calendar_id, uid,
640    /// recurrence_id)` are enforced correctly. Catalog
641    /// FILE_VERSION 16+; older snapshots deserialise empty.
642    pub extra_column_positions: Vec<usize>,
643}
644
645/// Default neighbor degree (M) for the NSW graph. Picked at construction
646/// time and persisted with the index.
647pub const NSW_DEFAULT_M: usize = 16;
648
649/// v5.2.2: outcome of a successful [`Catalog::freeze_oldest_to_cold`]
650/// call. The catalog state has already been mutated by the time this
651/// is returned (hot rows dropped + segment registered + Cold locators
652/// flipped). The caller's only remaining concern is `segment_bytes` —
653/// persist them to disk under `<db>.spg/segments/seg_<id>.spg` so a
654/// future restart can reload via the v5.1 `SPG_PRELOAD_COLD_SEGMENT`
655/// path. (v5.3's manifest will subsume this manual step.)
656#[derive(Debug, Clone)]
657pub struct FreezeReport {
658    /// Id allocated by [`Catalog::load_segment_bytes`] for the new
659    /// cold-tier segment. Stable across the call's success path.
660    pub segment_id: u32,
661    /// Number of rows that moved hot → cold. Equals the `max_rows`
662    /// the caller asked for (the API is strict on the count).
663    pub frozen_rows: usize,
664    /// Hot-tier bytes reclaimed by the freeze — the
665    /// [`Table::hot_bytes`] delta before vs after. Useful to feed
666    /// back into the freezer's budget check on the next tick.
667    pub bytes_freed: u64,
668    /// Encoded segment bytes, byte-identical to what
669    /// [`encode_segment`] produced. The catalog already owns a
670    /// copy inside `cold_segments`; this hand-off lets the caller
671    /// persist them without re-encoding.
672    pub segment_bytes: Vec<u8>,
673}
674
675/// v6.7.4 — read-only output of [`Catalog::prepare_freeze_slice`].
676/// Carries every row body + key in a contiguous hot-row range,
677/// already encoded and sorted by PK so the coordinator's merge
678/// step is a k-way merge over already-sorted streams.
679///
680/// `Vec<FreezeSlice>` from N independent workers feeds
681/// [`Catalog::commit_freeze_slices`], which concats + encodes the
682/// merged segment + atomically swaps the catalog state.
683#[derive(Debug, Clone)]
684pub struct FreezeSlice {
685    /// Hot-row index range this slice covered (half-open, in the
686    /// table's `rows: PersistentVec` ordering at call time). The
687    /// commit step uses this to compute the union range that
688    /// gets passed to [`Table::delete_rows`].
689    pub row_range: core::ops::Range<usize>,
690    /// `(pk_u64, encoded_row_body, IndexKey)` triples, sorted
691    /// ascending by `pk_u64`. Per-slice sort happens inside
692    /// `prepare_freeze_slice`; the coordinator does only a
693    /// k-way merge to reach the global PK ordering
694    /// [`encode_segment`] requires.
695    pub rows: Vec<(u64, Vec<u8>, IndexKey)>,
696}
697
698/// v6.7.3 — outcome of a [`Catalog::compact_cold_segments`] call.
699/// The catalog state has already been mutated when this is returned:
700/// the merged segment is loaded into `cold_segments`, the source
701/// segment slots are tombstoned (`None`), and every BTree-index
702/// `RowLocator::Cold` that previously pointed at a source now
703/// points at the merged segment. The caller's remaining job is to
704/// persist `merged_segment_bytes` under
705/// `<db>.spg/segments/seg_<merged_segment_id>.spg` and update the
706/// in-memory `segment_id → path` map (remove the source ids, add
707/// the merged id) so the next CHECKPOINT writes a manifest that
708/// no longer lists the retired sources.
709///
710/// On a no-op (fewer than 2 candidate segments under the threshold),
711/// `merged_segment_id` is `None` and `sources` is empty; the
712/// catalog was not mutated.
713#[derive(Debug, Clone)]
714pub struct CompactReport {
715    /// Source segment ids that were merged + tombstoned.
716    pub sources: Vec<u32>,
717    /// Id allocated for the merged segment. `None` on no-op.
718    pub merged_segment_id: Option<u32>,
719    /// Encoded merged-segment bytes (empty on no-op).
720    pub merged_segment_bytes: Vec<u8>,
721    /// Number of rows that landed in the merged segment.
722    pub merged_rows: usize,
723    /// `Σ source.num_rows − merged_rows`. Rows present in source
724    /// segment payloads but unreferenced by any live BTree
725    /// `Cold` locator — DELETE'd-but-still-frozen rows that
726    /// compaction GC'd during the merge.
727    pub deleted_rows_pruned: usize,
728    /// `Σ source.bytes() − merged.bytes()`. Estimate of on-disk
729    /// space the merge will reclaim once the source segment files
730    /// are GC'd. Saturating subtract — never negative.
731    pub bytes_reclaimed_estimate: u64,
732}
733
734#[derive(Debug, Clone)]
735pub enum IndexKind {
736    /// v4.40: structural-sharing B-tree over `IndexKey`. Replaces the v0.8
737    /// `BTreeMap<IndexKey, Vec<usize>>` — `Index::clone` is now an `Arc`
738    /// bump regardless of index size, so `Catalog::clone` inside the
739    /// v4.34 auto-commit wrap stays O(1) even for tables with secondary
740    /// indices (the case that bottlenecked v4.39 at 1M rows in the
741    /// sweep).
742    ///
743    /// v5.1: value type widened from `Vec<usize>` to `Vec<RowLocator>` so
744    /// a single key can point to a mix of hot-tier rows (`RowLocator::Hot`,
745    /// equivalent to the pre-v5 `usize` row index) and cold-tier rows
746    /// (`RowLocator::Cold { segment_id, page_offset }`) once the v5.2
747    /// freezer starts producing them. Pre-v5.2 only `Hot` entries appear
748    /// — the on-disk encoding stays at `FILE_VERSION` 8 (raw u64 row index)
749    /// because every locator round-trips through `RowLocator::from_legacy_v8_u64`
750    /// without information loss. `FILE_VERSION` 9 with tagged encoding lands
751    /// alongside the first freezer commit (v5.1 step 2b / v5.2).
752    BTree(PersistentBTreeMap<IndexKey, Vec<RowLocator>>),
753    /// Navigable-small-world graph for vector kNN search.
754    Nsw(NswGraph),
755    /// v6.7.1 — BRIN (Block Range INdex). Pure metadata: BRIN
756    /// indexes carry NO in-memory key→locator map. The (min,
757    /// max) summaries live in each cold-tier segment's v2
758    /// envelope sidecar; the BRIN entry in `Table.indices` only
759    /// records THAT a BRIN index exists on this column so the
760    /// segment encoder + planner can opt into the summary path.
761    Brin {
762        /// The cell type at `column_position` at CREATE INDEX time.
763        /// Used by the planner to type-check WHERE-clause range
764        /// predicates against the BRIN-indexed column.
765        column_type: DataType,
766    },
767    /// v7.12.3 — GIN inverted index over a `tsvector` column.
768    ///
769    /// Storage shape: `lexeme word → Vec<RowLocator>`. The posting
770    /// list per word is appended in row-order, so range scans are
771    /// O(matching rows) once the per-word lookup is done. Multi-
772    /// term queries intersect / union posting lists.
773    ///
774    /// `IndexKey::from_value(TsVector)` returns `None` — GIN doesn't
775    /// participate in `try_index_seek` (which is BTree-equality-keyed).
776    /// The engine consults this index through `try_gin_lookup` on
777    /// `WHERE col @@ tsquery` predicates instead.
778    ///
779    /// Backed by a `PersistentBTreeMap` so `Catalog::clone` (the
780    /// per-write snapshot) stays O(1) — same structural-sharing
781    /// invariant as BTree.
782    Gin(PersistentBTreeMap<alloc::string::String, Vec<RowLocator>>),
783    /// v7.15.0 — `USING gin (col gin_trgm_ops)` over a `TEXT`
784    /// column. Posting lists map `trigram` (PG-compatible 3-byte
785    /// shingle on the lower-cased + space-padded input) to row
786    /// locators. The planner uses this index to accelerate
787    /// `WHERE col LIKE '…'` / `ILIKE '…'` / `similarity(col, q) >
788    /// t` — every literal run of length ≥ 1 in the pattern
789    /// produces a trigram set, the engine intersects the posting
790    /// lists, and the LIKE / similarity predicate is re-evaluated
791    /// per candidate row to filter the over-approximation.
792    /// Persisted via tag-4 index payload in `FILE_VERSION` 24+.
793    GinTrgm(PersistentBTreeMap<alloc::string::String, Vec<RowLocator>>),
794}
795
796/// Multi-layer HNSW graph (v2.13). Each node is assigned a `top_level`;
797/// it appears in layers `0..=top_level`. Higher layers are sparser, so
798/// search starts from the entry at the top layer, greedy-descends to
799/// layer 0, and beam-searches there. Layer 0 keeps a larger neighbour
800/// budget (`m_max_0 = 2 * m` per the HNSW paper); upper layers cap at
801/// `m`. The struct name stays `NswGraph` so external users / on-disk
802/// callers don't have to track a rename — the algorithm changed, the
803/// data slot didn't.
804#[derive(Debug, Clone)]
805pub struct NswGraph {
806    /// Max neighbours per node on layers ≥ 1.
807    pub m: usize,
808    /// Max neighbours on layer 0 (the dense bottom layer). HNSW
809    /// convention: `m_max_0 = 2 * m`.
810    pub m_max_0: usize,
811    /// Entry point — the node that sits on the topmost layer. Search
812    /// always starts here.
813    pub entry: Option<usize>,
814    /// Top layer of the entry node (== `layers.len() - 1` when populated).
815    pub entry_level: u8,
816    /// `levels[i]` = top layer of node `i`. Nodes whose vector cell is
817    /// NULL / non-Vector have `levels[i] = 0` and no neighbour entries.
818    ///
819    /// v5.5.0: backed by `PersistentVec` so `NswGraph::clone` (and the
820    /// `Catalog::clone` on every group-commit write that contains it) is O(1)
821    /// structural-sharing instead of an O(N) element copy.
822    pub levels: PersistentVec<u8>,
823    /// `layers[l][i]` = neighbours of node `i` at layer `l`. Inner vec
824    /// is empty when node `i` doesn't reach layer `l`.
825    ///
826    /// v5.5.0: the per-node middle dimension (the O(N) one) is a
827    /// `PersistentVec`; the outer layer dimension stays a plain `Vec`
828    /// (layer count ≤ 8, so its clone is O(1) in practice) and the inner
829    /// neighbour list stays a `Vec` (bounded by `m_max_0`).
830    ///
831    /// v6.1.x: neighbour slot widened from `usize` (8 B on 64-bit) to
832    /// `u32` (4 B). Row indices are catalog-bounded by `u32::MAX` (4G
833    /// rows per table); the cast at the NSW boundary asserts this. At
834    /// 1M dim-128 SQ8, layer 0 adjacency alone shrinks by ~128 MiB
835    /// — the largest single contribution to the v6.0.5-measured
836    /// 624 MiB ambition gap. On-disk format already used u32 LE, so
837    /// this is a pure in-memory layout change; no `FILE_VERSION` bump.
838    pub layers: Vec<PersistentVec<Vec<u32>>>,
839}
840
841impl NswGraph {
842    fn new(m: usize) -> Self {
843        Self {
844            m,
845            m_max_0: m.saturating_mul(2),
846            entry: None,
847            entry_level: 0,
848            levels: PersistentVec::new(),
849            layers: alloc::vec![PersistentVec::new()],
850        }
851    }
852
853    /// Max-neighbour budget for layer `l`.
854    pub const fn cap_for_layer(&self, layer: u8) -> usize {
855        if layer == 0 { self.m_max_0 } else { self.m }
856    }
857}
858
859/// Deterministic level assignment, seeded on the row index so the same
860/// insert order reproduces the same topology. Distribution is roughly
861/// HNSW-flavoured with `mL ≈ 1/ln(M) ≈ 0.36` for M=16: each 4-bit
862/// chunk that comes up zero promotes the node one layer (so P(level ≥
863/// L) ≈ (1/16)^L).
864#[allow(clippy::verbose_bit_mask)] // clippy suggests trailing_zeros(); we need an explicit MAX cap and a stable distribution shape.
865pub fn nsw_assign_level(row_idx: usize) -> u8 {
866    const MAX_LEVEL: u8 = 7; // 7 ⇒ ~16^7 ≈ 2.7e8 expected nodes between promotions; ample.
867    // SplitMix-style mixer — cheap and seedable.
868    let mut x = (row_idx as u64).wrapping_mul(0x9E37_79B9_7F4A_7C15);
869    x ^= x >> 30;
870    x = x.wrapping_mul(0xBF58_476D_1CE4_E5B9);
871    x ^= x >> 27;
872    x = x.wrapping_mul(0x94D0_49BB_1331_11EB);
873    x ^= x >> 31;
874    // Count contiguous low-end zero nibbles (4-bit chunks). Each zero
875    // nibble has probability 1/16, mirroring HNSW's `mL ≈ 1/ln(M)` for
876    // M=16. `trailing_zeros / 4` would lose the ordering when x = 0, so
877    // a plain loop with a cap is clearer.
878    let mut level: u8 = 0;
879    while x & 0xF == 0 && level < MAX_LEVEL {
880        level += 1;
881        x >>= 4;
882    }
883    level
884}
885
886impl Index {
887    fn new_btree(name: String, column_position: usize) -> Self {
888        Self {
889            name,
890            column_position,
891            kind: IndexKind::BTree(PersistentBTreeMap::new()),
892            included_columns: Vec::new(),
893            partial_predicate: None,
894            expression: None,
895            is_unique: false,
896            extra_column_positions: Vec::new(),
897        }
898    }
899
900    fn new_nsw(name: String, column_position: usize, m: usize) -> Self {
901        Self {
902            name,
903            column_position,
904            kind: IndexKind::Nsw(NswGraph::new(m)),
905            included_columns: Vec::new(),
906            partial_predicate: None,
907            expression: None,
908            is_unique: false,
909            extra_column_positions: Vec::new(),
910        }
911    }
912
913    /// v6.7.1 — BRIN index constructor. BRIN carries no in-memory
914    /// data; the `column_type` snapshot is used by the segment
915    /// encoder + planner for type-checking range predicates.
916    fn new_brin(name: String, column_position: usize, column_type: DataType) -> Self {
917        Self {
918            name,
919            column_position,
920            kind: IndexKind::Brin { column_type },
921            included_columns: Vec::new(),
922            partial_predicate: None,
923            expression: None,
924            is_unique: false,
925            extra_column_positions: Vec::new(),
926        }
927    }
928
929    /// v7.12.3 — GIN inverted-index constructor. Empty posting-list
930    /// map; caller (typically [`Table::add_gin_index`] or
931    /// [`Table::restore_gin_index`]) populates it from existing rows
932    /// or from a deserialised snapshot.
933    fn new_gin(name: String, column_position: usize) -> Self {
934        Self {
935            name,
936            column_position,
937            kind: IndexKind::Gin(PersistentBTreeMap::new()),
938            included_columns: Vec::new(),
939            partial_predicate: None,
940            expression: None,
941            is_unique: false,
942            extra_column_positions: Vec::new(),
943        }
944    }
945
946    /// v7.15.0 — `gin_trgm_ops`-flavoured GIN constructor. Same
947    /// shape as `new_gin` but the posting-list keys are 3-byte
948    /// trigram shingles (`pg_trgm`-compatible) and the column
949    /// type is `TEXT` / `VARCHAR` (not `TSVECTOR`).
950    fn new_gin_trgm(name: String, column_position: usize) -> Self {
951        Self {
952            name,
953            column_position,
954            kind: IndexKind::GinTrgm(PersistentBTreeMap::new()),
955            included_columns: Vec::new(),
956            partial_predicate: None,
957            expression: None,
958            is_unique: false,
959            extra_column_positions: Vec::new(),
960        }
961    }
962
963    /// Look up the locators stored under `key` (B-tree only). Returns
964    /// an empty slice when the key is absent or the index isn't a
965    /// BTree — callers can treat both cases uniformly.
966    ///
967    /// v5.1: return type widened from `&[usize]` to `&[RowLocator]`.
968    /// Pre-v5.2 callers can read the slice and `.as_hot().unwrap()`
969    /// each entry (no `Cold` variants exist until the freezer lands);
970    /// post-v5.2 callers dispatch hot vs. cold per locator.
971    pub fn lookup_eq(&self, key: &IndexKey) -> &[RowLocator] {
972        match &self.kind {
973            IndexKind::BTree(m) => m.get(key).map_or(&[][..], Vec::as_slice),
974            // BRIN / NSW / GIN / trigram-GIN have no IndexKey-keyed
975            // map; lookup is a no-op. GIN uses
976            // [`Index::gin_lookup_word`] instead.
977            IndexKind::Nsw(_)
978            | IndexKind::Brin { .. }
979            | IndexKind::Gin(_)
980            | IndexKind::GinTrgm(_) => &[][..],
981        }
982    }
983
984    /// v7.12.3 — GIN posting-list lookup. Returns the row locators
985    /// whose `tsvector` cell contains `word`. Empty when the word is
986    /// absent from the index or this isn't a GIN index.
987    pub fn gin_lookup_word(&self, word: &str) -> &[RowLocator] {
988        match &self.kind {
989            IndexKind::Gin(m) => m.get(&String::from(word)).map_or(&[][..], Vec::as_slice),
990            IndexKind::BTree(_)
991            | IndexKind::Nsw(_)
992            | IndexKind::Brin { .. }
993            | IndexKind::GinTrgm(_) => &[][..],
994        }
995    }
996
997    /// v7.15.0 — trigram-GIN posting-list lookup. Returns the row
998    /// locators whose indexed `TEXT` cell contains the trigram
999    /// `tri`. Empty when the trigram is absent or this isn't a
1000    /// trigram-GIN index.
1001    pub fn gin_trgm_lookup(&self, tri: &str) -> &[RowLocator] {
1002        match &self.kind {
1003            IndexKind::GinTrgm(m) => m.get(&String::from(tri)).map_or(&[][..], Vec::as_slice),
1004            IndexKind::BTree(_)
1005            | IndexKind::Nsw(_)
1006            | IndexKind::Brin { .. }
1007            | IndexKind::Gin(_) => &[][..],
1008        }
1009    }
1010
1011    /// Borrow the NSW graph (if this is an NSW index). Callers that need
1012    /// the graph for a kNN search go through here.
1013    pub const fn nsw(&self) -> Option<&NswGraph> {
1014        match &self.kind {
1015            IndexKind::Nsw(g) => Some(g),
1016            IndexKind::BTree(_)
1017            | IndexKind::Brin { .. }
1018            | IndexKind::Gin(_)
1019            | IndexKind::GinTrgm(_) => None,
1020        }
1021    }
1022
1023    /// v6.7.1 — true when this index is a BRIN (block range) index.
1024    /// Used by the segment encoder to opt into BRIN sidecar emission
1025    /// at freeze time, and by the planner to opt into page-skipping
1026    /// on range predicates.
1027    pub const fn is_brin(&self) -> bool {
1028        matches!(self.kind, IndexKind::Brin { .. })
1029    }
1030
1031    /// v7.15.0 — true when this index is a trigram GIN
1032    /// (`gin_trgm_ops`-flavoured). Used by the LIKE planner to
1033    /// opt into trigram acceleration.
1034    pub const fn is_gin_trgm(&self) -> bool {
1035        matches!(self.kind, IndexKind::GinTrgm(_))
1036    }
1037
1038    /// v7.12.3 — true when this index is a GIN inverted index.
1039    /// Used by the planner to opt into posting-list acceleration on
1040    /// `WHERE col @@ tsquery` predicates.
1041    pub const fn is_gin(&self) -> bool {
1042        matches!(self.kind, IndexKind::Gin(_))
1043    }
1044}
1045
1046/// In-memory table: schema + a persistent row vector + secondary indices.
1047///
1048/// v4.39: `rows` is a [`PersistentVec`] (Bitmapped Vector Trie, 32-way) so
1049/// `Table::clone()` is `O(1)` — the whole reason for v4.39's existence is
1050/// to make `Catalog::clone()` cheap inside the v4.34 auto-commit wrap.
1051///
1052/// v5.2.1: `hot_bytes` tracks the encoded byte size of every row currently
1053/// in [`Self::rows`], summed over rows. Updated incrementally by `insert`
1054/// (+= encoded row size), `delete_rows` (-= removed rows' encoded sizes),
1055/// and `update_row` (-= old size, += new size). The value is what the
1056/// v5.2 freezer reads to decide when to demote cold rows — when the
1057/// catalog-wide sum crosses `SPG_HOT_TIER_BYTES` (default 4 GiB) the
1058/// freezer thread wakes. v5.2.1 ships measurement only; the freezer
1059/// itself lands in v5.2.2. Stored as `u64` so a single field clone in
1060/// `Catalog::clone` stays at the O(1) invariant v4.39 built.
1061#[derive(Debug, Clone)]
1062pub struct Table {
1063    schema: TableSchema,
1064    rows: PersistentVec<Row>,
1065    indices: Vec<Index>,
1066    hot_bytes: u64,
1067    /// v6.7.0 — cached count of rows currently materialised in the
1068    /// cold tier via `RowLocator::Cold` entries across THIS table's
1069    /// indices. Populated by `ANALYZE` (walks every BTree index and
1070    /// counts Cold locators); the count survives until the next
1071    /// ANALYZE recomputes it. Surfaced via `spg_statistic.cold_row_count`
1072    /// and `spg_stat_segment.table_name`.
1073    ///
1074    /// Honest scope: this is a CACHED count, not a live one.
1075    /// Freezer / promote / DELETE don't currently update the cache
1076    /// incrementally — they invalidate it by setting the
1077    /// `cold_row_count_stale` flag, and the next ANALYZE re-walks.
1078    /// Incremental maintenance is a v6.7.x candidate if observation
1079    /// shows the ANALYZE walk cost dominates.
1080    cold_row_count: u64,
1081    /// v6.7.0 — set when the cached `cold_row_count` may be wrong
1082    /// because rows moved into / out of the cold tier since the last
1083    /// ANALYZE. The virtual-table surface reports the cached value
1084    /// regardless (operators run ANALYZE to refresh).
1085    cold_row_count_stale: bool,
1086}
1087
1088impl Table {
1089    pub fn new(schema: TableSchema) -> Self {
1090        Self {
1091            schema,
1092            rows: PersistentVec::new(),
1093            indices: Vec::new(),
1094            hot_bytes: 0,
1095            cold_row_count: 0,
1096            cold_row_count_stale: false,
1097        }
1098    }
1099
1100    /// Total encoded byte size of every row currently in the hot tier
1101    /// (`self.rows`). See struct docs for the maintenance contract.
1102    /// Returns 0 for an empty table.
1103    #[must_use]
1104    pub const fn hot_bytes(&self) -> u64 {
1105        self.hot_bytes
1106    }
1107
1108    /// v6.7.0 — cached count of cold-tier rows. See struct field
1109    /// docs for the staleness contract.
1110    #[must_use]
1111    pub const fn cold_row_count(&self) -> u64 {
1112        self.cold_row_count
1113    }
1114
1115    /// v6.7.0 — overwrite the cached count. Called by the engine's
1116    /// `analyze_one_table` after walking the indices.
1117    pub fn set_cold_row_count(&mut self, n: u64) {
1118        self.cold_row_count = n;
1119        self.cold_row_count_stale = false;
1120    }
1121
1122    /// v6.7.0 — mark the cached count as potentially out of date.
1123    /// Called by freezer / promote / DELETE paths so a subsequent
1124    /// `spg_statistic` read knows the number may not reflect the
1125    /// current state.
1126    pub fn mark_cold_row_count_stale(&mut self) {
1127        self.cold_row_count_stale = true;
1128    }
1129
1130    /// v6.7.0 — report whether the cached count is known to be out
1131    /// of date. Exposed for completeness; the virtual table surface
1132    /// returns the cached value regardless.
1133    #[must_use]
1134    pub const fn cold_row_count_stale(&self) -> bool {
1135        self.cold_row_count_stale
1136    }
1137
1138    /// v6.7.0 — walk every BTree index and count `RowLocator::Cold`
1139    /// entries; return the MAX across indices. The freeze path
1140    /// (`freeze_oldest_to_cold`) writes cold locators to ONE
1141    /// designated index — that index ends up with the full per-row
1142    /// count. MAX-across-indices yields the precise count when a
1143    /// PK-style index exists; for multi-index tables without a
1144    /// covering index it's a lower bound (rare in practice).
1145    /// Caller responsibility: only invoke under `engine.write()`
1146    /// or after taking ownership; the walk is O(N) over every
1147    /// (key, locator) pair.
1148    #[must_use]
1149    pub fn count_cold_locators(&self) -> u64 {
1150        let mut best: u64 = 0;
1151        for idx in &self.indices {
1152            if let IndexKind::BTree(map) = &idx.kind {
1153                let n: u64 = map
1154                    .iter()
1155                    .map(|(_, locs)| locs.iter().filter(|l| l.is_cold()).count() as u64)
1156                    .sum();
1157                if n > best {
1158                    best = n;
1159                }
1160            }
1161        }
1162        best
1163    }
1164
1165    pub const fn schema(&self) -> &TableSchema {
1166        &self.schema
1167    }
1168
1169    /// v6.7.2 — mutable schema accessor for ALTER TABLE paths.
1170    /// Used by `Engine::exec_alter_table` to flip per-table
1171    /// settings like `hot_tier_bytes`.
1172    pub const fn schema_mut(&mut self) -> &mut TableSchema {
1173        &mut self.schema
1174    }
1175
1176    /// v4.39: returns the persistent row vector by reference. Callers that
1177    /// used to take `&[Row]` should switch to `.iter()` (via
1178    /// `IntoIterator for &PersistentVec`) or `.get(i)` for indexing.
1179    pub const fn rows(&self) -> &PersistentVec<Row> {
1180        &self.rows
1181    }
1182
1183    pub const fn row_count(&self) -> usize {
1184        self.rows.len()
1185    }
1186
1187    /// v6.8.0 — exposed for the engine layer to patch
1188    /// `Index::included_columns` post-creation. Could fold into
1189    /// `add_index` once the engine's IF-NOT-EXISTS guard moves up,
1190    /// but the patch shape is the minimal change for v6.8.0.
1191    pub fn indices_mut(&mut self) -> &mut [Index] {
1192        &mut self.indices
1193    }
1194
1195    pub fn indices(&self) -> &[Index] {
1196        &self.indices
1197    }
1198
1199    /// Compute the next `AUTO_INCREMENT` value for the column at
1200    /// `col_pos`. Defined as `max(existing) + 1`, falling back to `1`
1201    /// when the column currently holds no integer values. NULL / non-
1202    /// integer cells are skipped. Returns `None` when the column isn't
1203    /// an integer type.
1204    pub fn next_auto_value(&self, col_pos: usize) -> Option<i64> {
1205        let ty = self.schema.columns.get(col_pos)?.ty;
1206        if !matches!(ty, DataType::SmallInt | DataType::Int | DataType::BigInt) {
1207            return None;
1208        }
1209        let mut max: Option<i64> = None;
1210        for row in &self.rows {
1211            match row.values.get(col_pos) {
1212                Some(Value::SmallInt(n)) => {
1213                    let v = i64::from(*n);
1214                    max = Some(max.map_or(v, |m| m.max(v)));
1215                }
1216                Some(Value::Int(n)) => {
1217                    let v = i64::from(*n);
1218                    max = Some(max.map_or(v, |m| m.max(v)));
1219                }
1220                Some(Value::BigInt(n)) => {
1221                    max = Some(max.map_or(*n, |m| m.max(*n)));
1222                }
1223                _ => {}
1224            }
1225        }
1226        Some(max.map_or(1, |m| m + 1))
1227    }
1228
1229    /// Return the first index defined over `column_position`, if any.
1230    /// (`v0.8` supports at most one index per column logically; the search
1231    /// just picks the first match.)
1232    pub fn index_on(&self, column_position: usize) -> Option<&Index> {
1233        // v6.7.1 — prefer BTree (has the key→locator map needed
1234        // for `lookup_eq`) over BRIN (metadata-only). When only a
1235        // BRIN exists on the column, return None so the executor
1236        // falls back to the hot-tier row scan instead of trying
1237        // to use BRIN for an equality lookup (which would always
1238        // return an empty slice and look like "no rows matched").
1239        self.indices
1240            .iter()
1241            .find(|i| i.column_position == column_position && matches!(i.kind, IndexKind::BTree(_)))
1242            .or_else(|| {
1243                self.indices.iter().find(|i| {
1244                    i.column_position == column_position && matches!(i.kind, IndexKind::Nsw(_))
1245                })
1246            })
1247    }
1248
1249    /// Insert one row after validating it matches the schema (length + type).
1250    /// Returns `StorageError` on mismatch — the table is left unchanged.
1251    /// Updates every defined index with the new row's key.
1252    pub fn insert(&mut self, row: Row) -> Result<(), StorageError> {
1253        if row.len() != self.schema.columns.len() {
1254            return Err(StorageError::ArityMismatch {
1255                expected: self.schema.columns.len(),
1256                actual: row.len(),
1257            });
1258        }
1259        for (i, (val, col)) in row.values.iter().zip(&self.schema.columns).enumerate() {
1260            if val.is_null() {
1261                if !col.nullable {
1262                    return Err(StorageError::NullInNotNull {
1263                        column: col.name.clone(),
1264                    });
1265                }
1266                continue;
1267            }
1268            let actual = val.data_type().expect("non-null");
1269            // Vector columns require both that the value's variant be Vector
1270            // *and* its dimension match. `actual == col.ty` already encodes
1271            // both because DataType::Vector carries the dim.
1272            //
1273            // VARCHAR(n) / CHAR(n) are storage-equivalent to TEXT — the
1274            // length / padding contract is enforced upstream by
1275            // `coerce_value`. Accept a `Text` value into either.
1276            //
1277            // NUMERIC's `Value::Numeric` carries its actual scale but the
1278            // column declares the *expected* scale (a scale-rescaled
1279            // Value::Numeric is produced upstream by `coerce_value`); the
1280            // structural check here only verifies "value is Numeric and
1281            // its scale equals the column scale".
1282            let compatible = actual == col.ty
1283                || matches!(
1284                    (actual, col.ty),
1285                    (
1286                        DataType::Text,
1287                        DataType::Varchar(_) | DataType::Char(_) | DataType::Json | DataType::Jsonb
1288                    ) | (DataType::Json | DataType::Jsonb, DataType::Text)
1289                        | (DataType::Json, DataType::Jsonb)
1290                        | (DataType::Jsonb, DataType::Json)
1291                        | (DataType::Timestamp, DataType::Timestamptz)
1292                        | (DataType::Timestamptz, DataType::Timestamp)
1293                )
1294                || matches!(
1295                    (actual, col.ty),
1296                    (
1297                        DataType::Numeric { scale: a, .. },
1298                        DataType::Numeric { scale: b, .. },
1299                    ) if a == b
1300                );
1301            if !compatible {
1302                return Err(StorageError::TypeMismatch {
1303                    column: col.name.clone(),
1304                    expected: col.ty,
1305                    actual,
1306                    position: i,
1307                });
1308            }
1309        }
1310        let new_row_idx = self.rows.len();
1311        // Pre-validate before mutating: ensure indices receive an IndexKey.
1312        // For NSW we defer the graph update to *after* the row is pushed
1313        // so the kNN search can see it in `self.rows`.
1314        for idx in &mut self.indices {
1315            match &mut idx.kind {
1316                IndexKind::BTree(map) => {
1317                    if let Some(key) = IndexKey::from_value(&row.values[idx.column_position]) {
1318                        // v4.40: PersistentBTreeMap has no in-place entry-or-default.
1319                        // Clone-then-insert keeps the same semantics — for typical
1320                        // unique-key schemas the Vec is 1-element so the clone is
1321                        // O(1). For dup-heavy columns it's O(M) per insert, traded
1322                        // for the structural-sharing win at clone time.
1323                        let mut entries = map.get(&key).cloned().unwrap_or_default();
1324                        entries.push(RowLocator::Hot(new_row_idx));
1325                        map.insert_mut(key, entries);
1326                    }
1327                }
1328                IndexKind::Gin(map) => {
1329                    // v7.12.3 — extend posting list per lexeme word.
1330                    // NULL or non-TsVector cell → no-op (cell carries
1331                    // no lexemes to index).
1332                    if let Value::TsVector(lexemes) = &row.values[idx.column_position] {
1333                        for lex in lexemes {
1334                            let mut entries = map.get(&lex.word).cloned().unwrap_or_default();
1335                            entries.push(RowLocator::Hot(new_row_idx));
1336                            map.insert_mut(lex.word.clone(), entries);
1337                        }
1338                    }
1339                }
1340                IndexKind::GinTrgm(map) => {
1341                    // v7.15.0 — trigram GIN. Shingle the TEXT cell
1342                    // into PG-compatible 3-byte trigrams and extend
1343                    // each trigram's posting list.
1344                    if let Value::Text(s) = &row.values[idx.column_position] {
1345                        for tri in trgm::extract_trigrams(s) {
1346                            let mut entries = map.get(&tri).cloned().unwrap_or_default();
1347                            entries.push(RowLocator::Hot(new_row_idx));
1348                            map.insert_mut(tri, entries);
1349                        }
1350                    }
1351                }
1352                // NSW handled below after the row push (so the new row
1353                // is visible to the kNN-graph connect step). BRIN
1354                // carries no per-row state.
1355                IndexKind::Nsw(_) | IndexKind::Brin { .. } => {}
1356            }
1357        }
1358        // v5.2.1: maintain incremental hot-tier byte counter. Computed
1359        // before the move so we don't need to borrow `row` after push.
1360        self.hot_bytes = self
1361            .hot_bytes
1362            .saturating_add(row_body_encoded_len(&row, &self.schema) as u64);
1363        // v4.39.1: push_mut keeps streaming inserts at Vec::push speed when
1364        // the table is uniquely owned (the spg-embedded path); inside a TX
1365        // wrap where a Catalog snapshot exists, push_mut path-copies the
1366        // tail just like push() and the snapshot stays valid.
1367        self.rows.push_mut(row);
1368        // NSW updates after the push so the new row is visible to the
1369        // greedy search used during connect.
1370        let new_row_idx = self.rows.len() - 1;
1371        let nsw_targets: Vec<usize> = self
1372            .indices
1373            .iter()
1374            .enumerate()
1375            .filter_map(|(i, idx)| {
1376                if matches!(idx.kind, IndexKind::Nsw(_)) {
1377                    Some(i)
1378                } else {
1379                    None
1380                }
1381            })
1382            .collect();
1383        for idx_pos in nsw_targets {
1384            nsw_insert_at(self, idx_pos, new_row_idx);
1385        }
1386        Ok(())
1387    }
1388
1389    /// Build a new B-tree index over the named column. Rebuilds from
1390    /// existing rows. Errors if `column_name` doesn't exist or the index
1391    /// name is taken.
1392    pub fn add_index(&mut self, name: String, column_name: &str) -> Result<(), StorageError> {
1393        if self.indices.iter().any(|i| i.name == name) {
1394            return Err(StorageError::DuplicateIndex { name });
1395        }
1396        let column_position = self.schema.column_position(column_name).ok_or_else(|| {
1397            StorageError::ColumnNotFound {
1398                column: column_name.into(),
1399            }
1400        })?;
1401        let mut idx = Index::new_btree(name, column_position);
1402        if let IndexKind::BTree(map) = &mut idx.kind {
1403            for (i, row) in self.rows.iter().enumerate() {
1404                if let Some(key) = IndexKey::from_value(&row.values[column_position]) {
1405                    let mut entries = map.get(&key).cloned().unwrap_or_default();
1406                    entries.push(RowLocator::Hot(i));
1407                    map.insert_mut(key, entries);
1408                }
1409            }
1410        }
1411        self.indices.push(idx);
1412        Ok(())
1413    }
1414
1415    /// Build a new NSW (HNSW-flavoured) index over the named column.
1416    /// Required for `ORDER BY col <-> literal LIMIT k` to plan as a
1417    /// graph traversal instead of a full scan. Column must be a Vector
1418    /// type. `m` is the maximum number of neighbours per node.
1419    pub fn add_nsw_index(
1420        &mut self,
1421        name: String,
1422        column_name: &str,
1423        m: usize,
1424    ) -> Result<(), StorageError> {
1425        self.add_nsw_index_inner(name, column_name, m, None)
1426    }
1427
1428    /// v6.0.4 — synchronous rebuild of the named NSW index. If
1429    /// `new_encoding` is `Some(target)` and differs from the column's
1430    /// current encoding, every stored cell at the indexed column is
1431    /// re-coded into the target encoding before the new graph
1432    /// builds. Returns `IndexNotFound` if no index by that name exists
1433    /// and `Unsupported` for non-NSW indexes (`BTree` REBUILD is a no-op
1434    /// the engine layer rejects, not a storage-level concept).
1435    ///
1436    /// Holds the caller's `&mut self` for the duration — no
1437    /// concurrency / staging / WAL-replay machinery in v6.0.4. The
1438    /// "live" optimisation lands as v6.0.4.1.
1439    pub fn rebuild_nsw_index(
1440        &mut self,
1441        name: &str,
1442        new_encoding: Option<VecEncoding>,
1443    ) -> Result<(), StorageError> {
1444        let idx_pos = self
1445            .indices
1446            .iter()
1447            .position(|i| i.name == name)
1448            .ok_or_else(|| StorageError::IndexNotFound {
1449                name: String::from(name),
1450            })?;
1451        let col_pos = self.indices[idx_pos].column_position;
1452        let m = match &self.indices[idx_pos].kind {
1453            IndexKind::Nsw(g) => g.m,
1454            IndexKind::BTree(_)
1455            | IndexKind::Brin { .. }
1456            | IndexKind::Gin(_)
1457            | IndexKind::GinTrgm(_) => {
1458                return Err(StorageError::Unsupported(format!(
1459                    "ALTER INDEX REBUILD on non-NSW index {name:?} — only NSW indexes can rebuild"
1460                )));
1461            }
1462        };
1463        let col_name = self.schema.columns[col_pos].name.clone();
1464        // 1. Optional re-encoding pass. Done first so the cells
1465        //    match the schema before the graph rebuild walks them.
1466        if let Some(target) = new_encoding {
1467            let current = match self.schema.columns[col_pos].ty {
1468                DataType::Vector { encoding, .. } => encoding,
1469                ref other => {
1470                    return Err(StorageError::Unsupported(format!(
1471                        "ALTER INDEX REBUILD WITH (encoding=…) on non-vector column type {other:?}"
1472                    )));
1473                }
1474            };
1475            if target != current {
1476                let DataType::Vector { dim, .. } = self.schema.columns[col_pos].ty else {
1477                    unreachable!("checked above")
1478                };
1479                let n = self.rows.len();
1480                for i in 0..n {
1481                    let row = self
1482                        .rows
1483                        .get_mut(i)
1484                        .expect("row index in bounds (we iterated up to len())");
1485                    let cell = core::mem::replace(&mut row.values[col_pos], Value::Null);
1486                    let recoded = recode_vector_cell(cell, target)?;
1487                    row.values[col_pos] = recoded;
1488                }
1489                self.schema.columns[col_pos].ty = DataType::Vector {
1490                    dim,
1491                    encoding: target,
1492                };
1493            }
1494        }
1495        // 2. Drop the existing index slot + rebuild from row payload.
1496        self.indices.remove(idx_pos);
1497        self.add_nsw_index_inner(String::from(name), &col_name, m, None)?;
1498        Ok(())
1499    }
1500
1501    /// Restore an NSW index from a pre-built graph (used on
1502    /// deserialize). Skips the bulk-build pass since the topology is
1503    /// already known. Returns `DuplicateIndex` or `ColumnNotFound` on
1504    /// schema mismatch as usual.
1505    pub fn restore_nsw_index(
1506        &mut self,
1507        name: String,
1508        column_name: &str,
1509        graph: NswGraph,
1510    ) -> Result<(), StorageError> {
1511        self.add_nsw_index_inner(name, column_name, graph.m, Some(graph))
1512    }
1513
1514    /// Restore a `BTree` index from a pre-built `(IndexKey, Vec<RowLocator>)`
1515    /// map. Used by [`Catalog::deserialize`] when reading a v9 (or later)
1516    /// catalog snapshot — the map travels on disk so cold-tier locators
1517    /// survive a round-trip, instead of being rebuilt from `self.rows`
1518    /// (which would lose every Cold entry). Same error contract as
1519    /// [`Table::add_index`].
1520    pub fn restore_btree_index(
1521        &mut self,
1522        name: String,
1523        column_name: &str,
1524        map: PersistentBTreeMap<IndexKey, Vec<RowLocator>>,
1525    ) -> Result<(), StorageError> {
1526        if self.indices.iter().any(|i| i.name == name) {
1527            return Err(StorageError::DuplicateIndex { name });
1528        }
1529        let column_position = self.schema.column_position(column_name).ok_or_else(|| {
1530            StorageError::ColumnNotFound {
1531                column: column_name.into(),
1532            }
1533        })?;
1534        self.indices.push(Index {
1535            name,
1536            column_position,
1537            kind: IndexKind::BTree(map),
1538            included_columns: Vec::new(),
1539            partial_predicate: None,
1540            expression: None,
1541            is_unique: false,
1542            extra_column_positions: Vec::new(),
1543        });
1544        Ok(())
1545    }
1546
1547    /// v6.7.1 — public restore counterpart for BRIN indices. Used
1548    /// by `Catalog::deserialize` when a v10 snapshot carries a
1549    /// BRIN index entry. BRIN carries no in-memory data — only the
1550    /// `column_type` snapshot is restored.
1551    pub fn restore_brin_index(
1552        &mut self,
1553        name: String,
1554        column_name: &str,
1555        column_type: DataType,
1556    ) -> Result<(), StorageError> {
1557        if self.indices.iter().any(|i| i.name == name) {
1558            return Err(StorageError::DuplicateIndex { name });
1559        }
1560        let column_position = self.schema.column_position(column_name).ok_or_else(|| {
1561            StorageError::ColumnNotFound {
1562                column: column_name.into(),
1563            }
1564        })?;
1565        self.indices
1566            .push(Index::new_brin(name, column_position, column_type));
1567        Ok(())
1568    }
1569
1570    /// v6.7.1 — public CREATE INDEX counterpart for BRIN. Creates
1571    /// the index entry with a snapshot of the indexed column's
1572    /// current `DataType`.
1573    pub fn add_brin_index(&mut self, name: String, column_name: &str) -> Result<(), StorageError> {
1574        if self.indices.iter().any(|i| i.name == name) {
1575            return Err(StorageError::DuplicateIndex { name });
1576        }
1577        let column_position = self.schema.column_position(column_name).ok_or_else(|| {
1578            StorageError::ColumnNotFound {
1579                column: column_name.into(),
1580            }
1581        })?;
1582        let column_type = self.schema.columns[column_position].ty;
1583        self.indices
1584            .push(Index::new_brin(name, column_position, column_type));
1585        Ok(())
1586    }
1587
1588    /// v7.12.3 — Build a new GIN inverted index over a `tsvector`
1589    /// column. Populates posting lists from existing rows. Errors
1590    /// if the column doesn't exist, isn't `TsVector`, or the index
1591    /// name is taken.
1592    pub fn add_gin_index(&mut self, name: String, column_name: &str) -> Result<(), StorageError> {
1593        if self.indices.iter().any(|i| i.name == name) {
1594            return Err(StorageError::DuplicateIndex { name });
1595        }
1596        let column_position = self.schema.column_position(column_name).ok_or_else(|| {
1597            StorageError::ColumnNotFound {
1598                column: column_name.into(),
1599            }
1600        })?;
1601        if self.schema.columns[column_position].ty != DataType::TsVector {
1602            return Err(StorageError::Corrupt(format!(
1603                "GIN index {name:?} requires a tsvector column; \
1604                 {column_name:?} is {:?}",
1605                self.schema.columns[column_position].ty
1606            )));
1607        }
1608        let mut idx = Index::new_gin(name, column_position);
1609        if let IndexKind::Gin(map) = &mut idx.kind {
1610            for (i, row) in self.rows.iter().enumerate() {
1611                if let Value::TsVector(lexemes) = &row.values[column_position] {
1612                    for lex in lexemes {
1613                        let mut entries = map.get(&lex.word).cloned().unwrap_or_default();
1614                        entries.push(RowLocator::Hot(i));
1615                        map.insert_mut(lex.word.clone(), entries);
1616                    }
1617                }
1618            }
1619        }
1620        self.indices.push(idx);
1621        Ok(())
1622    }
1623
1624    /// v7.12.3 — Restore a GIN index from a deserialised snapshot.
1625    /// Mirrors [`Self::restore_btree_index`] but takes the GIN's
1626    /// `word → Vec<RowLocator>` posting-list map (already populated
1627    /// from the catalog stream) instead of an `IndexKey` map.
1628    pub fn restore_gin_index(
1629        &mut self,
1630        name: String,
1631        column_name: &str,
1632        map: PersistentBTreeMap<String, Vec<RowLocator>>,
1633    ) -> Result<(), StorageError> {
1634        if self.indices.iter().any(|i| i.name == name) {
1635            return Err(StorageError::DuplicateIndex { name });
1636        }
1637        let column_position = self.schema.column_position(column_name).ok_or_else(|| {
1638            StorageError::ColumnNotFound {
1639                column: column_name.into(),
1640            }
1641        })?;
1642        let mut idx = Index::new_gin(name, column_position);
1643        idx.kind = IndexKind::Gin(map);
1644        self.indices.push(idx);
1645        Ok(())
1646    }
1647
1648    /// v7.15.0 — `gin_trgm_ops` GIN over a TEXT column. Walks
1649    /// every row, shingles the cell into PG-compatible trigrams,
1650    /// and builds the posting-list map. NULL / non-TEXT cells
1651    /// contribute nothing (no trigrams).
1652    pub fn add_gin_trgm_index(
1653        &mut self,
1654        name: String,
1655        column_name: &str,
1656    ) -> Result<(), StorageError> {
1657        if self.indices.iter().any(|i| i.name == name) {
1658            return Err(StorageError::DuplicateIndex { name });
1659        }
1660        let column_position = self.schema.column_position(column_name).ok_or_else(|| {
1661            StorageError::ColumnNotFound {
1662                column: column_name.into(),
1663            }
1664        })?;
1665        if !matches!(
1666            self.schema.columns[column_position].ty,
1667            DataType::Text | DataType::Varchar(_)
1668        ) {
1669            return Err(StorageError::Corrupt(format!(
1670                "trigram-GIN index {name:?} requires a TEXT/VARCHAR column; \
1671                 {column_name:?} is {:?}",
1672                self.schema.columns[column_position].ty
1673            )));
1674        }
1675        let mut idx = Index::new_gin_trgm(name, column_position);
1676        if let IndexKind::GinTrgm(map) = &mut idx.kind {
1677            for (i, row) in self.rows.iter().enumerate() {
1678                if let Value::Text(s) = &row.values[column_position] {
1679                    for tri in trgm::extract_trigrams(s) {
1680                        let mut entries = map.get(&tri).cloned().unwrap_or_default();
1681                        entries.push(RowLocator::Hot(i));
1682                        map.insert_mut(tri, entries);
1683                    }
1684                }
1685            }
1686        }
1687        self.indices.push(idx);
1688        Ok(())
1689    }
1690
1691    /// v7.15.0 — restore a trigram-GIN from its catalog snapshot
1692    /// payload. Mirrors [`Self::restore_gin_index`].
1693    pub fn restore_gin_trgm_index(
1694        &mut self,
1695        name: String,
1696        column_name: &str,
1697        map: PersistentBTreeMap<String, Vec<RowLocator>>,
1698    ) -> Result<(), StorageError> {
1699        if self.indices.iter().any(|i| i.name == name) {
1700            return Err(StorageError::DuplicateIndex { name });
1701        }
1702        let column_position = self.schema.column_position(column_name).ok_or_else(|| {
1703            StorageError::ColumnNotFound {
1704                column: column_name.into(),
1705            }
1706        })?;
1707        let mut idx = Index::new_gin_trgm(name, column_position);
1708        idx.kind = IndexKind::GinTrgm(map);
1709        self.indices.push(idx);
1710        Ok(())
1711    }
1712
1713    /// v5.1: register cold-tier locators on a `BTree` index. Used
1714    /// after [`Catalog::load_segment_bytes`] to wire every cold-
1715    /// tier row's PK back to its segment so
1716    /// [`Catalog::lookup_by_pk`] can resolve it. Each call
1717    /// appends to the index — keys that already have hot or cold
1718    /// locators keep them. Returns the number of locators
1719    /// registered.
1720    ///
1721    /// Pre-v5.2 (freezer) this is the only path that adds Cold
1722    /// variants to a PB; post-freezer the background freezer
1723    /// thread produces these as a batch under the engine write
1724    /// lock and this API becomes its in-memory primitive.
1725    ///
1726    /// Errors if `index_name` doesn't exist or names an NSW graph
1727    /// (NSW indices don't carry per-key row locators — they're
1728    /// vector-search structures).
1729    pub fn register_cold_locators<I>(
1730        &mut self,
1731        index_name: &str,
1732        locators: I,
1733    ) -> Result<usize, StorageError>
1734    where
1735        I: IntoIterator<Item = (IndexKey, RowLocator)>,
1736    {
1737        let idx = self
1738            .indices
1739            .iter_mut()
1740            .find(|i| i.name == index_name)
1741            .ok_or_else(|| StorageError::Corrupt(format!("index {index_name:?} not found")))?;
1742        let map = match &mut idx.kind {
1743            IndexKind::BTree(map) => map,
1744            IndexKind::Nsw(_)
1745            | IndexKind::Brin { .. }
1746            | IndexKind::Gin(_)
1747            | IndexKind::GinTrgm(_) => {
1748                return Err(StorageError::Corrupt(format!(
1749                    "index {index_name:?} is not BTree; cold locators apply only to BTree indices"
1750                )));
1751            }
1752        };
1753        let mut count = 0usize;
1754        for (key, locator) in locators {
1755            let mut entries = map.get(&key).cloned().unwrap_or_default();
1756            entries.push(locator);
1757            map.insert_mut(key, entries);
1758            count += 1;
1759        }
1760        Ok(count)
1761    }
1762
1763    /// v7.12.3 — GIN-side parallel to [`Self::register_cold_locators`].
1764    /// Re-attaches `word → cold RowLocator` posting-list entries after
1765    /// the from-rows rebuild loop. Errors when the index doesn't
1766    /// exist or isn't a GIN. Both tsvector-GIN and trigram-GIN
1767    /// variants share posting-list shape (`String → Vec<RowLocator>`),
1768    /// so this helper accepts either.
1769    pub fn register_gin_cold_locators<I>(
1770        &mut self,
1771        index_name: &str,
1772        locators: I,
1773    ) -> Result<usize, StorageError>
1774    where
1775        I: IntoIterator<Item = (String, RowLocator)>,
1776    {
1777        let idx = self
1778            .indices
1779            .iter_mut()
1780            .find(|i| i.name == index_name)
1781            .ok_or_else(|| StorageError::Corrupt(format!("index {index_name:?} not found")))?;
1782        let map = match &mut idx.kind {
1783            IndexKind::Gin(map) | IndexKind::GinTrgm(map) => map,
1784            IndexKind::BTree(_) | IndexKind::Nsw(_) | IndexKind::Brin { .. } => {
1785                return Err(StorageError::Corrupt(format!(
1786                    "register_gin_cold_locators: index {index_name:?} is not GIN"
1787                )));
1788            }
1789        };
1790        let mut count = 0usize;
1791        for (word, locator) in locators {
1792            let mut entries = map.get(&word).cloned().unwrap_or_default();
1793            entries.push(locator);
1794            map.insert_mut(word, entries);
1795            count += 1;
1796        }
1797        Ok(count)
1798    }
1799
1800    /// v5.2.3: remove every `Cold` locator currently registered on
1801    /// `index_name` under the given `key`. `Hot` locators for the
1802    /// same key are left in place — useful when a row has just been
1803    /// promoted hot-side and the caller wants the old Cold pointer
1804    /// retired without losing the new hot entry.
1805    ///
1806    /// Returns the number of cold locators removed (0 when the key
1807    /// has only hot entries or the key isn't present at all).
1808    /// Errors when the index doesn't exist or isn't a `BTree`.
1809    pub fn remove_cold_locators_for_key(
1810        &mut self,
1811        index_name: &str,
1812        key: &IndexKey,
1813    ) -> Result<usize, StorageError> {
1814        let idx = self
1815            .indices
1816            .iter_mut()
1817            .find(|i| i.name == index_name)
1818            .ok_or_else(|| {
1819                StorageError::Corrupt(format!(
1820                    "remove_cold_locators_for_key: index {index_name:?} not found"
1821                ))
1822            })?;
1823        let map = match &mut idx.kind {
1824            IndexKind::BTree(map) => map,
1825            IndexKind::Nsw(_)
1826            | IndexKind::Brin { .. }
1827            | IndexKind::Gin(_)
1828            | IndexKind::GinTrgm(_) => {
1829                return Err(StorageError::Corrupt(format!(
1830                    "remove_cold_locators_for_key: index {index_name:?} is not BTree; \
1831                     cold locators apply only to BTree indices"
1832                )));
1833            }
1834        };
1835        let Some(entries) = map.get(key) else {
1836            return Ok(0);
1837        };
1838        let mut kept: Vec<RowLocator> =
1839            entries.iter().copied().filter(RowLocator::is_hot).collect();
1840        let removed = entries.len() - kept.len();
1841        if removed == 0 {
1842            return Ok(0);
1843        }
1844        kept.shrink_to_fit();
1845        // PersistentBTreeMap has no remove API in v5.2; when every
1846        // locator for `key` was Cold, the key keeps an empty Vec
1847        // entry. `Index::lookup_eq` already treats `Some(&[])` and
1848        // `None` as the same empty slice (via `Vec::as_slice`), so
1849        // callers can't distinguish the two. The space cost is one
1850        // empty Vec per shadowed-then-promoted key — bounded and
1851        // recoverable when the future compaction job lands.
1852        map.insert_mut(key.clone(), kept);
1853        Ok(removed)
1854    }
1855
1856    /// v7.13.0 — append a new column to the schema and back-fill
1857    /// every existing row with `fill_value`. Used by the engine's
1858    /// `ALTER TABLE t ADD COLUMN …` handler (mailrs round-5 G1).
1859    /// Indices on existing columns keep working — column positions
1860    /// don't shift since the new column lands at the end — so no
1861    /// index rebuild is needed.
1862    pub fn add_column(&mut self, col: ColumnSchema, fill_value: Value) {
1863        self.schema.columns.push(col);
1864        let mut new_rows: PersistentVec<Row> = PersistentVec::new();
1865        for row in self.rows.iter() {
1866            let mut values = row.values.clone();
1867            values.push(fill_value.clone());
1868            new_rows.push_mut(Row::new(values));
1869        }
1870        self.rows = new_rows;
1871    }
1872
1873    /// v7.15.0 — replace the partial-index predicate source on
1874    /// the index at slot `idx`. Used by `ALTER TABLE … RENAME
1875    /// COLUMN` after the engine rewrites column-identifier
1876    /// references in the predicate source text. Pure metadata
1877    /// edit; index rows are unaffected (they're keyed by
1878    /// column position, not predicate text).
1879    pub fn set_partial_predicate(&mut self, idx: usize, pred: Option<String>) {
1880        debug_assert!(idx < self.indices.len());
1881        self.indices[idx].partial_predicate = pred;
1882    }
1883
1884    /// v7.15.0 — rename the column at `col_pos` to `new_name`.
1885    /// The on-disk row encoding is positional, so no row rewrite
1886    /// is needed; only the schema's column name changes. Indices,
1887    /// UCs, FKs all key off column positions and are unaffected.
1888    /// Source-text references that hold the column name (CHECK
1889    /// predicates, partial-index predicates, runtime DEFAULT
1890    /// expressions, trigger `UPDATE OF` lists) are rewritten by
1891    /// the engine before this helper is called — the storage
1892    /// layer doesn't depend on `spg-sql` and so can't re-parse the
1893    /// predicate sources itself.
1894    pub fn rename_column(&mut self, col_pos: usize, new_name: &str) {
1895        debug_assert!(col_pos < self.schema.columns.len());
1896        self.schema.columns[col_pos].name = new_name.to_string();
1897    }
1898
1899    /// v7.13.3 — drop the column at `col_pos`. Removes the entry
1900    /// from the schema, the value from every row, any index that
1901    /// references the column (pure drop, not shift), and shifts
1902    /// every remaining index/UC/FK column position that pointed
1903    /// past `col_pos` down by one. Used by `ALTER TABLE t DROP
1904    /// COLUMN <c>` (mailrs round-7 S8). FK dependents on this
1905    /// column must already have been removed by the caller (CASCADE
1906    /// path); the helper assumes only same-column index removal is
1907    /// needed.
1908    pub fn drop_column(&mut self, col_pos: usize) {
1909        debug_assert!(col_pos < self.schema.columns.len());
1910        // Strip the column from the schema.
1911        self.schema.columns.remove(col_pos);
1912        // Rewrite every row to omit the cell at col_pos.
1913        let mut new_rows: PersistentVec<Row> = PersistentVec::new();
1914        for row in self.rows.iter() {
1915            let mut values = row.values.clone();
1916            if col_pos < values.len() {
1917                values.remove(col_pos);
1918            }
1919            new_rows.push_mut(Row::new(values));
1920        }
1921        self.rows = new_rows;
1922        // Drop indices on the column outright; shift the rest.
1923        self.indices.retain(|idx| idx.column_position != col_pos);
1924        for idx in &mut self.indices {
1925            if idx.column_position > col_pos {
1926                idx.column_position -= 1;
1927            }
1928            // Same shift for any included-columns reference.
1929            for inc in &mut idx.included_columns {
1930                if *inc as usize > col_pos {
1931                    *inc -= 1;
1932                }
1933            }
1934        }
1935        // Shift uniqueness-constraint column positions (and drop
1936        // entries that lose all columns, though that shouldn't
1937        // happen in practice — caller has already CASCADE-removed
1938        // FKs and there's no general CASCADE for UCs).
1939        let mut surviving_ucs: Vec<UniquenessConstraint> = Vec::new();
1940        for mut uc in core::mem::take(&mut self.schema.uniqueness_constraints) {
1941            uc.columns.retain(|&c| c != col_pos);
1942            if uc.columns.is_empty() {
1943                continue;
1944            }
1945            for c in &mut uc.columns {
1946                if *c > col_pos {
1947                    *c -= 1;
1948                }
1949            }
1950            surviving_ucs.push(uc);
1951        }
1952        self.schema.uniqueness_constraints = surviving_ucs;
1953        // Shift FK local_columns (parent-pointing column positions
1954        // are off-table and untouched).
1955        for fk in &mut self.schema.foreign_keys {
1956            for c in &mut fk.local_columns {
1957                if *c > col_pos {
1958                    *c -= 1;
1959                }
1960            }
1961        }
1962        // Rebuild remaining indices' payload — the column-position
1963        // shift means existing IndexKey entries are still keyed by
1964        // the same column data but the position numbers changed;
1965        // existing key→locator maps stay valid because they're
1966        // keyed by Value not position. The rebuild is conservative
1967        // — same pattern delete_rows uses post-mutation.
1968        self.rebuild_indices();
1969    }
1970
1971    /// v4.4: delete the rows at the given positions in one pass.
1972    /// `positions` must be unique; ordering doesn't matter. Indices
1973    /// are rebuilt from scratch (cheaper than tracking incremental
1974    /// shifts across both B-tree and NSW). Returns the number of
1975    /// rows removed.
1976    pub fn delete_rows(&mut self, positions: &[usize]) -> usize {
1977        if positions.is_empty() {
1978            return 0;
1979        }
1980        // Mark positions; v4.39: PV has no in-place retain, so we rebuild
1981        // a fresh PV by pushing the survivors. Still O(n log₃₂ n); the
1982        // structural-sharing win shows up at `Catalog::clone()`, not here.
1983        let mut to_remove = alloc::vec![false; self.rows.len()];
1984        let mut removed = 0;
1985        for &p in positions {
1986            if p < to_remove.len() && !to_remove[p] {
1987                to_remove[p] = true;
1988                removed += 1;
1989            }
1990        }
1991        let mut new_rows: PersistentVec<Row> = PersistentVec::new();
1992        let mut removed_bytes: u64 = 0;
1993        for (i, row) in self.rows.iter().enumerate() {
1994            if to_remove[i] {
1995                removed_bytes =
1996                    removed_bytes.saturating_add(row_body_encoded_len(row, &self.schema) as u64);
1997            } else {
1998                new_rows.push_mut(row.clone());
1999            }
2000        }
2001        self.rows = new_rows;
2002        self.hot_bytes = self.hot_bytes.saturating_sub(removed_bytes);
2003        self.rebuild_indices();
2004        removed
2005    }
2006
2007    /// v4.4: replace the row at `position` with `new_values` (must
2008    /// match the schema arity + types). Indices are rebuilt for
2009    /// correctness — the affected column might be indexed and its
2010    /// key may have shifted, and a NSW node's vector may have
2011    /// changed, both of which need fresh state.
2012    pub fn update_row(
2013        &mut self,
2014        position: usize,
2015        new_values: Vec<Value>,
2016    ) -> Result<(), StorageError> {
2017        if position >= self.rows.len() {
2018            return Err(StorageError::Corrupt(alloc::format!(
2019                "update_row: position {position} out of bounds (rows={})",
2020                self.rows.len()
2021            )));
2022        }
2023        if new_values.len() != self.schema.columns.len() {
2024            return Err(StorageError::ArityMismatch {
2025                expected: self.schema.columns.len(),
2026                actual: new_values.len(),
2027            });
2028        }
2029        // Reuse the per-cell type-compat validation that `insert`
2030        // applies. The body below mirrors that check intentionally —
2031        // factoring it would be more code than the duplication.
2032        for (i, (val, col)) in new_values.iter().zip(&self.schema.columns).enumerate() {
2033            if val.is_null() {
2034                if !col.nullable {
2035                    return Err(StorageError::NullInNotNull {
2036                        column: col.name.clone(),
2037                    });
2038                }
2039                continue;
2040            }
2041            let actual = val.data_type().expect("non-null");
2042            let compatible = actual == col.ty
2043                || matches!(
2044                    (actual, col.ty),
2045                    (
2046                        DataType::Text,
2047                        DataType::Varchar(_) | DataType::Char(_) | DataType::Json | DataType::Jsonb
2048                    ) | (DataType::Json | DataType::Jsonb, DataType::Text)
2049                        | (DataType::Json, DataType::Jsonb)
2050                        | (DataType::Jsonb, DataType::Json)
2051                        | (DataType::Timestamp, DataType::Timestamptz)
2052                        | (DataType::Timestamptz, DataType::Timestamp)
2053                )
2054                || matches!(
2055                    (actual, col.ty),
2056                    (
2057                        DataType::Numeric { scale: a, .. },
2058                        DataType::Numeric { scale: b, .. },
2059                    ) if a == b
2060                );
2061            if !compatible {
2062                return Err(StorageError::TypeMismatch {
2063                    column: col.name.clone(),
2064                    expected: col.ty,
2065                    actual,
2066                    position: i,
2067                });
2068            }
2069        }
2070        let old_row = self
2071            .rows
2072            .get(position)
2073            .expect("position bounds-checked above");
2074        let old_bytes = row_body_encoded_len(old_row, &self.schema) as u64;
2075        let new_row = Row::new(new_values);
2076        let new_bytes = row_body_encoded_len(&new_row, &self.schema) as u64;
2077        self.rows = self
2078            .rows
2079            .set(position, new_row)
2080            .expect("position bounds-checked above");
2081        self.hot_bytes = self
2082            .hot_bytes
2083            .saturating_sub(old_bytes)
2084            .saturating_add(new_bytes);
2085        self.rebuild_indices();
2086        Ok(())
2087    }
2088
2089    /// v4.4 helper used by `delete_rows` / `update_row`: discard all
2090    /// index payloads and rebuild from `self.rows`. Cheap enough
2091    /// for typical SPG scale (catalogs in the docker-compose
2092    /// deployment shape are small); the alternative — incremental
2093    /// shift bookkeeping across B-tree + NSW — would be far more
2094    /// invasive than the savings justify.
2095    fn rebuild_indices(&mut self) {
2096        // v5.2.3: capture every `Cold` locator on every BTree index
2097        // before the rebuild, so the from-rows re-emission below
2098        // (which only produces `Hot` locators) doesn't drop cold-
2099        // tier entries on keys unrelated to the row that changed.
2100        // Pre-v5.2.3 this was a `freeze_oldest_to_cold` worry only
2101        // and the freezer did its own capture-then-reregister; v5.2.3
2102        // promotes that pattern into the base helper because UPDATE
2103        // / DELETE now run rebuild_indices on tables with cold rows.
2104        let preserved_cold: Vec<(String, Vec<(IndexKey, RowLocator)>)> = self
2105            .indices
2106            .iter()
2107            .filter_map(|idx| match &idx.kind {
2108                IndexKind::BTree(map) => {
2109                    let cold: Vec<(IndexKey, RowLocator)> = map
2110                        .iter()
2111                        .flat_map(|(k, locs)| {
2112                            locs.iter()
2113                                .filter(|l| l.is_cold())
2114                                .copied()
2115                                .map(move |l| (k.clone(), l))
2116                        })
2117                        .collect();
2118                    if cold.is_empty() {
2119                        None
2120                    } else {
2121                        Some((idx.name.clone(), cold))
2122                    }
2123                }
2124                // BRIN / NSW carry no key→locator map. GIN handles
2125                // its own cold preservation below in `preserved_gin_cold`.
2126                IndexKind::Nsw(_)
2127                | IndexKind::Brin { .. }
2128                | IndexKind::Gin(_)
2129                | IndexKind::GinTrgm(_) => None,
2130            })
2131            .collect();
2132
2133        // v7.12.3 — same cold-preservation pattern for GIN's
2134        // `word → Vec<RowLocator>` posting lists. Parallel to the
2135        // BTree pass above (different key type so a separate vec is
2136        // cleaner than a generic merge). v7.15.0: trigram-GIN
2137        // (`gin_trgm_ops`) shares the same posting-list shape, so
2138        // one pass handles both — the `RebuildKind` carries the
2139        // kind tag to drive resurrection.
2140        let preserved_gin_cold: Vec<(String, Vec<(String, RowLocator)>)> = self
2141            .indices
2142            .iter()
2143            .filter_map(|idx| match &idx.kind {
2144                IndexKind::Gin(map) | IndexKind::GinTrgm(map) => {
2145                    let cold: Vec<(String, RowLocator)> = map
2146                        .iter()
2147                        .flat_map(|(w, locs)| {
2148                            locs.iter()
2149                                .filter(|l| l.is_cold())
2150                                .copied()
2151                                .map(move |l| (w.clone(), l))
2152                        })
2153                        .collect();
2154                    if cold.is_empty() {
2155                        None
2156                    } else {
2157                        Some((idx.name.clone(), cold))
2158                    }
2159                }
2160                IndexKind::BTree(_) | IndexKind::Nsw(_) | IndexKind::Brin { .. } => None,
2161            })
2162            .collect();
2163
2164        // v6.7.1 — descriptor needs to capture index kind so the
2165        // rebuild loop can resurrect BTree / NSW / BRIN / GIN exactly
2166        // as they were. (NSW carries m; BRIN carries the column type
2167        // snapshot; BTree / GIN need no extra payload.)
2168        #[derive(Clone)]
2169        enum RebuildKind {
2170            BTree,
2171            Nsw(usize),
2172            Brin(DataType),
2173            Gin,
2174            GinTrgm,
2175        }
2176        let descriptors: Vec<(String, usize, RebuildKind)> = self
2177            .indices
2178            .iter()
2179            .map(|idx| {
2180                let kind = match &idx.kind {
2181                    IndexKind::Nsw(g) => RebuildKind::Nsw(g.m),
2182                    IndexKind::Brin { column_type } => RebuildKind::Brin(*column_type),
2183                    IndexKind::BTree(_) => RebuildKind::BTree,
2184                    IndexKind::Gin(_) => RebuildKind::Gin,
2185                    IndexKind::GinTrgm(_) => RebuildKind::GinTrgm,
2186                };
2187                (idx.name.clone(), idx.column_position, kind)
2188            })
2189            .collect();
2190        self.indices.clear();
2191        for (name, column_position, rebuild_kind) in descriptors {
2192            match rebuild_kind {
2193                RebuildKind::Nsw(m) => {
2194                    let idx = Index::new_nsw(name, column_position, m);
2195                    self.indices.push(idx);
2196                    let idx_pos = self.indices.len() - 1;
2197                    let row_indices: Vec<usize> = (0..self.rows.len()).collect();
2198                    for row_idx in row_indices {
2199                        nsw_insert_at(self, idx_pos, row_idx);
2200                    }
2201                }
2202                RebuildKind::Brin(column_type) => {
2203                    // BRIN has no in-memory rebuild — the summaries
2204                    // live in cold segments which freeze emits.
2205                    self.indices
2206                        .push(Index::new_brin(name, column_position, column_type));
2207                }
2208                RebuildKind::BTree => {
2209                    let mut idx = Index::new_btree(name, column_position);
2210                    if let IndexKind::BTree(map) = &mut idx.kind {
2211                        for (i, row) in self.rows.iter().enumerate() {
2212                            if let Some(key) = IndexKey::from_value(&row.values[column_position]) {
2213                                let mut entries = map.get(&key).cloned().unwrap_or_default();
2214                                entries.push(RowLocator::Hot(i));
2215                                map.insert_mut(key, entries);
2216                            }
2217                        }
2218                    }
2219                    self.indices.push(idx);
2220                }
2221                RebuildKind::Gin => {
2222                    let mut idx = Index::new_gin(name, column_position);
2223                    if let IndexKind::Gin(map) = &mut idx.kind {
2224                        for (i, row) in self.rows.iter().enumerate() {
2225                            if let Value::TsVector(lexemes) = &row.values[column_position] {
2226                                for lex in lexemes {
2227                                    let mut entries =
2228                                        map.get(&lex.word).cloned().unwrap_or_default();
2229                                    entries.push(RowLocator::Hot(i));
2230                                    map.insert_mut(lex.word.clone(), entries);
2231                                }
2232                            }
2233                        }
2234                    }
2235                    self.indices.push(idx);
2236                }
2237                RebuildKind::GinTrgm => {
2238                    let mut idx = Index::new_gin_trgm(name, column_position);
2239                    if let IndexKind::GinTrgm(map) = &mut idx.kind {
2240                        for (i, row) in self.rows.iter().enumerate() {
2241                            if let Value::Text(s) = &row.values[column_position] {
2242                                for tri in trgm::extract_trigrams(s) {
2243                                    let mut entries =
2244                                        map.get(&tri).cloned().unwrap_or_default();
2245                                    entries.push(RowLocator::Hot(i));
2246                                    map.insert_mut(tri, entries);
2247                                }
2248                            }
2249                        }
2250                    }
2251                    self.indices.push(idx);
2252                }
2253            }
2254        }
2255
2256        // Re-attach preserved cold locators after the from-rows
2257        // rebuild. `register_cold_locators` handles the per-key
2258        // entries-vec append; no key collisions arise because the
2259        // rebuild loop above produced only Hot locators.
2260        for (idx_name, locators) in preserved_cold {
2261            // Errors here would only fire if the index disappeared
2262            // between snapshot and rebuild, which can't happen
2263            // because the rebuild restores the same descriptor set.
2264            let _ = self.register_cold_locators(&idx_name, locators);
2265        }
2266        // v7.12.3 — same for GIN posting-list cold locators.
2267        for (idx_name, locators) in preserved_gin_cold {
2268            let _ = self.register_gin_cold_locators(&idx_name, locators);
2269        }
2270    }
2271
2272    fn add_nsw_index_inner(
2273        &mut self,
2274        name: String,
2275        column_name: &str,
2276        m: usize,
2277        restore: Option<NswGraph>,
2278    ) -> Result<(), StorageError> {
2279        if self.indices.iter().any(|i| i.name == name) {
2280            return Err(StorageError::DuplicateIndex { name });
2281        }
2282        let column_position = self.schema.column_position(column_name).ok_or_else(|| {
2283            StorageError::ColumnNotFound {
2284                column: column_name.into(),
2285            }
2286        })?;
2287        if !matches!(
2288            self.schema.columns[column_position].ty,
2289            DataType::Vector { .. }
2290        ) {
2291            return Err(StorageError::TypeMismatch {
2292                column: column_name.into(),
2293                expected: DataType::Vector {
2294                    dim: 0,
2295                    encoding: VecEncoding::F32,
2296                },
2297                actual: self.schema.columns[column_position].ty,
2298                position: column_position,
2299            });
2300        }
2301        if let Some(graph) = restore {
2302            self.indices.push(Index {
2303                name,
2304                column_position,
2305                kind: IndexKind::Nsw(graph),
2306                included_columns: Vec::new(),
2307                partial_predicate: None,
2308                expression: None,
2309                is_unique: false,
2310                extra_column_positions: Vec::new(),
2311            });
2312            return Ok(());
2313        }
2314        let idx = Index::new_nsw(name, column_position, m);
2315        self.indices.push(idx);
2316        let idx_pos = self.indices.len() - 1;
2317        // Bulk-build by walking the existing rows in order — each insert
2318        // sees the partial graph and links into it.
2319        let row_indices: Vec<usize> = (0..self.rows.len()).collect();
2320        for row_idx in row_indices {
2321            nsw_insert_at(self, idx_pos, row_idx);
2322        }
2323        Ok(())
2324    }
2325}
2326
2327/// v6.0.4 — re-encode a single cell to the target `VecEncoding`.
2328/// Used by `Table::rebuild_nsw_index` when ALTER INDEX REBUILD
2329/// includes the optional `WITH (encoding = …)` clause. Round-trip
2330/// goes through f32: `current → Vec<f32> → target`, leaving NULL
2331/// cells untouched. Returns `Unsupported` on a non-vector cell —
2332/// the caller should have rejected the schema before reaching this.
2333fn recode_vector_cell(cell: Value, target: VecEncoding) -> Result<Value, StorageError> {
2334    if matches!(cell, Value::Null) {
2335        return Ok(cell);
2336    }
2337    // Step 1 — extract the f32 representation of the source cell.
2338    let as_f32: Vec<f32> = match &cell {
2339        Value::Vector(v) => v.clone(),
2340        Value::Sq8Vector(q) => quantize::dequantize(q),
2341        Value::HalfVector(h) => h.to_f32_vec(),
2342        other => {
2343            return Err(StorageError::Unsupported(format!(
2344                "ALTER INDEX REBUILD: cannot recode non-vector cell {:?}",
2345                other.data_type()
2346            )));
2347        }
2348    };
2349    // Step 2 — encode into the target shape. `F32` is the identity
2350    // path (saves one alloc round-trip when the source is already
2351    // F32 — but `Value::Vector(as_f32)` is the right answer
2352    // regardless).
2353    Ok(match target {
2354        VecEncoding::F32 => Value::Vector(as_f32),
2355        VecEncoding::Sq8 => Value::Sq8Vector(quantize::quantize(&as_f32)),
2356        VecEncoding::F16 => Value::HalfVector(halfvec::HalfVector::from_f32_slice(&as_f32)),
2357    })
2358}
2359
2360/// Insert one row into the HNSW graph held by index slot `idx_pos`.
2361/// No-op when the row's value at the indexed column isn't a vector.
2362/// v6.0.1: handles `Value::Sq8Vector` by dequantising into an f32
2363/// "query" surface — the existing greedy + beam-search machinery
2364/// then uses `cell_to_query_metric_distance` to route every
2365/// distance call through the cell's actual encoding.
2366fn nsw_insert_at(table: &mut Table, idx_pos: usize, new_row_idx: usize) {
2367    let col_pos = table.indices[idx_pos].column_position;
2368    let cell_dim: Option<usize> = match &table.rows[new_row_idx].values[col_pos] {
2369        Value::Vector(v) => Some(v.len()),
2370        Value::Sq8Vector(q) => Some(q.bytes.len()),
2371        Value::HalfVector(h) => Some(h.dim()),
2372        _ => None,
2373    };
2374    let Some(dim) = cell_dim else {
2375        // Even non-vector rows occupy a level slot so per-node Vec
2376        // lengths stay aligned with `table.rows.len()`.
2377        ensure_node_slot(table, idx_pos, new_row_idx, 0);
2378        return;
2379    };
2380    if dim == 0 {
2381        ensure_node_slot(table, idx_pos, new_row_idx, 0);
2382        return;
2383    }
2384    let level = nsw_assign_level(new_row_idx);
2385    ensure_node_slot(table, idx_pos, new_row_idx, level);
2386    let (entry, entry_level, m) = match &table.indices[idx_pos].kind {
2387        IndexKind::Nsw(g) => (g.entry, g.entry_level, g.m),
2388        IndexKind::BTree(_) | IndexKind::Brin { .. } | IndexKind::Gin(_) | IndexKind::GinTrgm(_) => {
2389            unreachable!("nsw_insert_at on a non-NSW index")
2390        }
2391    };
2392    // First node ever — declare it the entry (it gets its own level).
2393    if entry.is_none() {
2394        if let IndexKind::Nsw(g) = &mut table.indices[idx_pos].kind {
2395            g.entry = Some(new_row_idx);
2396            g.entry_level = level;
2397            *g.levels
2398                .get_mut(new_row_idx)
2399                .expect("levels slot padded by ensure_node_slot") = level;
2400        }
2401        return;
2402    }
2403    // Set the node's recorded level.
2404    if let IndexKind::Nsw(g) = &mut table.indices[idx_pos].kind {
2405        *g.levels
2406            .get_mut(new_row_idx)
2407            .expect("levels slot padded by ensure_node_slot") = level;
2408    }
2409    let query = match &table.rows[new_row_idx].values[col_pos] {
2410        Value::Vector(v) => v.clone(),
2411        // v6.0.1: dequantise the inserted SQ8 cell into an f32 query
2412        // surface so the existing greedy / beam machinery can route
2413        // distances through `cell_to_query_metric_distance`. The
2414        // small dequantisation error is what the recall@10 ≥ 0.95
2415        // envelope already accounts for (V6_DESIGN deliberation #3).
2416        Value::Sq8Vector(q) => quantize::dequantize(q),
2417        // v6.0.3: halfvec dequant is bit-exact at the storage layer,
2418        // so the inserted query is a faithful representation.
2419        Value::HalfVector(h) => h.to_f32_vec(),
2420        _ => return,
2421    };
2422    // Phase 1: greedy descend from `entry` down to `level + 1`, keeping
2423    // exactly one current best so the next layer starts from it.
2424    let mut current = entry.expect("entry was Some above");
2425    let mut current_d = vec_l2_sq(table, col_pos, current, &query);
2426    if entry_level > level {
2427        for layer in (level + 1..=entry_level).rev() {
2428            (current, current_d) =
2429                greedy_layer_walk(table, idx_pos, layer, current, current_d, &query);
2430        }
2431    }
2432    // Phase 2: from `min(level, entry_level)` down to 0, beam-search
2433    // `ef_construction` candidates, run the HNSW §4 heuristic neighbour
2434    // selection over them, and connect bidirectionally.
2435    let top = level.min(entry_level);
2436    let ef = (m * 2).max(8);
2437    for layer in (0..=top).rev() {
2438        let cap = if layer == 0 { m * 2 } else { m };
2439        let mut candidates = layer_beam_search(
2440            table,
2441            idx_pos,
2442            layer,
2443            current,
2444            current_d,
2445            &query,
2446            ef,
2447            NswMetric::L2,
2448        );
2449        candidates.retain(|&(_, n)| n != new_row_idx);
2450        // Take the closest as the entry for the next layer down — done
2451        // before heuristic narrowing because the heuristic can reorder.
2452        if let Some(&(d, n)) = candidates.first() {
2453            current = n;
2454            current_d = d;
2455        }
2456        let peers = select_neighbours_heuristic(&candidates, cap, table, col_pos);
2457        connect_at_layer(table, idx_pos, layer, new_row_idx, &peers);
2458    }
2459    // Phase 3: if the new node climbed above the current entry, take
2460    // over as entry so future inserts/searches start from the new top.
2461    if level > entry_level
2462        && let IndexKind::Nsw(g) = &mut table.indices[idx_pos].kind
2463    {
2464        g.entry = Some(new_row_idx);
2465        g.entry_level = level;
2466    }
2467}
2468
2469/// Make sure `layers[*][new_row_idx]` and `levels[new_row_idx]` exist,
2470/// padding with empty/zero entries as needed. Also grows `layers` to
2471/// accommodate the node's top `level`.
2472fn ensure_node_slot(table: &mut Table, idx_pos: usize, new_row_idx: usize, level: u8) {
2473    let IndexKind::Nsw(g) = &mut table.indices[idx_pos].kind else {
2474        unreachable!("ensure_node_slot on a BTree index");
2475    };
2476    while g.layers.len() <= level as usize {
2477        g.layers.push(PersistentVec::new());
2478    }
2479    while g.levels.len() <= new_row_idx {
2480        g.levels.push_mut(0);
2481    }
2482    for layer_vec in &mut g.layers {
2483        while layer_vec.len() <= new_row_idx {
2484            layer_vec.push_mut(Vec::new());
2485        }
2486    }
2487}
2488
2489/// Single-step greedy walk on one layer: from `current` (with cached
2490/// distance `current_d`), inspect that node's neighbours at `layer` and
2491/// hop to the closest if it beats `current_d`. Repeat until no move
2492/// improves the distance. Cheap variant of beam-search used for the
2493/// "descend" phase that only needs one survivor per layer.
2494fn greedy_layer_walk(
2495    table: &Table,
2496    idx_pos: usize,
2497    layer: u8,
2498    mut current: usize,
2499    mut current_d: f32,
2500    query: &[f32],
2501) -> (usize, f32) {
2502    let g = match &table.indices[idx_pos].kind {
2503        IndexKind::Nsw(g) => g,
2504        IndexKind::BTree(_) | IndexKind::Brin { .. } | IndexKind::Gin(_) | IndexKind::GinTrgm(_) => {
2505            return (current, current_d);
2506        }
2507    };
2508    let col_pos = table.indices[idx_pos].column_position;
2509    loop {
2510        let neighbours: &[u32] = g
2511            .layers
2512            .get(layer as usize)
2513            .and_then(|layer_v| layer_v.get(current))
2514            .map_or(&[][..], Vec::as_slice);
2515        let mut best = current;
2516        let mut best_d = current_d;
2517        for &n in neighbours {
2518            let n = n as usize;
2519            let d = vec_l2_sq(table, col_pos, n, query);
2520            if d < best_d {
2521                best = n;
2522                best_d = d;
2523            }
2524        }
2525        if best == current {
2526            return (current, current_d);
2527        }
2528        current = best;
2529        current_d = best_d;
2530    }
2531}
2532
2533/// Beam search on one layer starting from `entry_node` with cached
2534/// `entry_d`. Returns the top `ef` candidates in ascending-distance
2535/// order. Caller picks the closest as the next layer's entry and / or
2536/// trims to M for connection.
2537///
2538/// v3.0.1: uses two `BinaryHeap`s (min-heap for the open frontier,
2539/// max-heap for the working top-`ef` results) and a `Vec<bool>` visited
2540/// bitmap, replacing the v2.x `Vec` + `partition_point` + `BTreeSet`
2541/// implementation. Same algorithm shape (HNSW search algorithm 2 from
2542/// the paper); the data-structure swap cuts per-visit cost from
2543/// `O(ef + log row_count)` to amortised `O(log ef)`.
2544#[allow(clippy::too_many_arguments)] // Beam search threads layer, entry, query, ef, metric — each is intrinsic. Bundling them into a config struct hides the call sites.
2545fn layer_beam_search(
2546    table: &Table,
2547    idx_pos: usize,
2548    layer: u8,
2549    entry_node: usize,
2550    entry_d: f32,
2551    query: &[f32],
2552    ef: usize,
2553    metric: NswMetric,
2554) -> Vec<(f32, usize)> {
2555    let g = match &table.indices[idx_pos].kind {
2556        IndexKind::Nsw(g) => g,
2557        IndexKind::BTree(_) | IndexKind::Brin { .. } | IndexKind::Gin(_) | IndexKind::GinTrgm(_) => return Vec::new(),
2558    };
2559    let col_pos = table.indices[idx_pos].column_position;
2560    let d0 = if matches!(metric, NswMetric::L2) {
2561        entry_d
2562    } else {
2563        cell_to_query_metric_distance(table, col_pos, entry_node, query, metric)
2564    };
2565    let row_count = table.rows.len();
2566    let mut visited: Vec<bool> = alloc::vec![false; row_count];
2567    if entry_node < row_count {
2568        visited[entry_node] = true;
2569    }
2570    // candidates: min-heap by distance (Closest wrapper) — frontier
2571    // results:    max-heap by distance (Furthest wrapper) — top-ef working set
2572    let mut candidates: alloc::collections::BinaryHeap<NodeClosest> =
2573        alloc::collections::BinaryHeap::with_capacity(ef);
2574    let mut results: alloc::collections::BinaryHeap<NodeFurthest> =
2575        alloc::collections::BinaryHeap::with_capacity(ef);
2576    candidates.push(NodeClosest {
2577        dist: d0,
2578        node: entry_node,
2579    });
2580    results.push(NodeFurthest {
2581        dist: d0,
2582        node: entry_node,
2583    });
2584    while let Some(cur) = candidates.pop() {
2585        let worst = results.peek().map_or(f32::INFINITY, |c| c.dist);
2586        if cur.dist > worst && results.len() >= ef {
2587            break;
2588        }
2589        let neighbours: &[u32] = g
2590            .layers
2591            .get(layer as usize)
2592            .and_then(|layer_v| layer_v.get(cur.node))
2593            .map_or(&[][..], Vec::as_slice);
2594        for &n in neighbours {
2595            let n = n as usize;
2596            if n >= row_count || visited[n] {
2597                continue;
2598            }
2599            visited[n] = true;
2600            // v6.0.1: cell-aware distance — F32 cells take the
2601            // existing scalar metric, SQ8 cells route through
2602            // the asymmetric ADC variant for the same metric.
2603            let dn = cell_to_query_metric_distance(table, col_pos, n, query, metric);
2604            if !dn.is_finite() {
2605                continue;
2606            }
2607            let worst = results.peek().map_or(f32::INFINITY, |c| c.dist);
2608            if results.len() < ef || dn < worst {
2609                results.push(NodeFurthest { dist: dn, node: n });
2610                if results.len() > ef {
2611                    results.pop();
2612                }
2613                candidates.push(NodeClosest { dist: dn, node: n });
2614            }
2615        }
2616    }
2617    // Drain results (max-heap order) and re-sort ascending so callers
2618    // can take `closest = result[0]` without flipping.
2619    let mut out: Vec<(f32, usize)> = results.into_iter().map(|c| (c.dist, c.node)).collect();
2620    out.sort_by(|a, b| a.0.partial_cmp(&b.0).unwrap_or(core::cmp::Ordering::Equal));
2621    out
2622}
2623
2624/// Min-heap wrapper: smaller `dist` → higher priority in a `BinaryHeap`
2625/// (which is a max-heap), so we flip the comparison. NaN sorts last
2626/// (lowest priority) to keep the heap total-ordered.
2627#[derive(Debug, Clone, Copy)]
2628struct NodeClosest {
2629    dist: f32,
2630    node: usize,
2631}
2632impl PartialEq for NodeClosest {
2633    fn eq(&self, other: &Self) -> bool {
2634        self.dist == other.dist && self.node == other.node
2635    }
2636}
2637impl Eq for NodeClosest {}
2638impl PartialOrd for NodeClosest {
2639    fn partial_cmp(&self, other: &Self) -> Option<core::cmp::Ordering> {
2640        Some(self.cmp(other))
2641    }
2642}
2643impl Ord for NodeClosest {
2644    fn cmp(&self, other: &Self) -> core::cmp::Ordering {
2645        // Reversed: smaller dist = greater priority.
2646        other
2647            .dist
2648            .partial_cmp(&self.dist)
2649            .unwrap_or(core::cmp::Ordering::Equal)
2650    }
2651}
2652
2653/// Max-heap wrapper: larger `dist` sits at the top so the worst result
2654/// can be evicted in O(log n) when a better candidate arrives.
2655#[derive(Debug, Clone, Copy)]
2656struct NodeFurthest {
2657    dist: f32,
2658    node: usize,
2659}
2660impl PartialEq for NodeFurthest {
2661    fn eq(&self, other: &Self) -> bool {
2662        self.dist == other.dist && self.node == other.node
2663    }
2664}
2665impl Eq for NodeFurthest {}
2666impl PartialOrd for NodeFurthest {
2667    fn partial_cmp(&self, other: &Self) -> Option<core::cmp::Ordering> {
2668        Some(self.cmp(other))
2669    }
2670}
2671impl Ord for NodeFurthest {
2672    fn cmp(&self, other: &Self) -> core::cmp::Ordering {
2673        self.dist
2674            .partial_cmp(&other.dist)
2675            .unwrap_or(core::cmp::Ordering::Equal)
2676    }
2677}
2678
2679/// HNSW paper §4 algorithm 4: pick `m` neighbours from `candidates` so
2680/// that each chosen point isn't already covered by a closer chosen
2681/// point. Improves graph diversity → fewer hops needed at search time.
2682///
2683/// `candidates` arrives sorted ascending by distance-to-query. We walk
2684/// it in order, keeping a candidate only when no already-chosen point
2685/// is closer to it than the query is. Result is a vector of row
2686/// indices (length ≤ `m`).
2687fn select_neighbours_heuristic(
2688    candidates: &[(f32, usize)],
2689    m: usize,
2690    table: &Table,
2691    col_pos: usize,
2692) -> Vec<usize> {
2693    let mut chosen: Vec<usize> = Vec::with_capacity(m);
2694    for &(d_q, e) in candidates {
2695        if chosen.len() >= m {
2696            break;
2697        }
2698        // v6.0.1: works on either `Value::Vector` (F32) or
2699        // `Value::Sq8Vector` (Sq8) cells — `cell_l2_sq` dispatches
2700        // on encoding. A non-vector cell yields `f32::INFINITY`
2701        // which the `< d_q` test will never accept.
2702        if !matches!(
2703            table.rows.get(e).and_then(|r| r.values.get(col_pos)),
2704            Some(Value::Vector(_) | Value::Sq8Vector(_) | Value::HalfVector(_))
2705        ) {
2706            continue;
2707        }
2708        let mut covered = false;
2709        for &r in &chosen {
2710            // dist(e, r) measured in the same metric the topology was
2711            // built with (L2). If a chosen `r` is closer to `e` than
2712            // the query is, `r` already "covers" `e` for navigation.
2713            if cell_l2_sq(table, col_pos, e, r) < d_q {
2714                covered = true;
2715                break;
2716            }
2717        }
2718        if !covered {
2719            chosen.push(e);
2720        }
2721    }
2722    chosen
2723}
2724
2725/// Bidirectionally connect `new_row_idx` to each of `peers` at `layer`,
2726/// trimming each endpoint's adjacency to that layer's degree cap by
2727/// keeping only the closest neighbours.
2728fn connect_at_layer(
2729    table: &mut Table,
2730    idx_pos: usize,
2731    layer: u8,
2732    new_row_idx: usize,
2733    peers: &[usize],
2734) {
2735    let col_pos = table.indices[idx_pos].column_position;
2736    let cap = match &table.indices[idx_pos].kind {
2737        IndexKind::Nsw(g) => g.cap_for_layer(layer),
2738        IndexKind::BTree(_) | IndexKind::Brin { .. } | IndexKind::Gin(_) | IndexKind::GinTrgm(_) => return,
2739    };
2740    // v6.1.x: NSW adjacency stores neighbour row indices as u32 (4 B
2741    // each) rather than usize (8 B on 64-bit). Boundary casts here
2742    // assert the row count fits in u32 — the catalog already enforces
2743    // ≤ 4G rows per table, so the conversion can't lose data.
2744    let new_row_u32 = u32::try_from(new_row_idx).expect("row index fits in u32");
2745    if let IndexKind::Nsw(g) = &mut table.indices[idx_pos].kind {
2746        let layer_v = &mut g.layers[layer as usize];
2747        if let Some(slot) = layer_v.get_mut(new_row_idx) {
2748            *slot = peers
2749                .iter()
2750                .map(|&p| u32::try_from(p).expect("row index fits in u32"))
2751                .collect();
2752        }
2753    }
2754    for &peer in peers {
2755        // Skip peers whose indexed cell isn't a vector — same fence
2756        // as the F32 path; SQ8 cells flow through `cell_l2_sq`
2757        // below without dequantising.
2758        if !matches!(
2759            &table.rows[peer].values[col_pos],
2760            Value::Vector(_) | Value::Sq8Vector(_) | Value::HalfVector(_)
2761        ) {
2762            continue;
2763        }
2764        // 1. add the new node to peer's adjacency
2765        if let IndexKind::Nsw(g) = &mut table.indices[idx_pos].kind {
2766            let layer_v = &mut g.layers[layer as usize];
2767            if let Some(slot) = layer_v.get_mut(peer)
2768                && !slot.contains(&new_row_u32)
2769            {
2770                slot.push(new_row_u32);
2771            }
2772        }
2773        // 2. if peer is over budget, rebuild its adjacency with the
2774        //    HNSW §4 heuristic — same diversity criterion as the
2775        //    insert path so connectivity stays consistent.
2776        let needs_trim = match &table.indices[idx_pos].kind {
2777            IndexKind::Nsw(g) => g.layers[layer as usize][peer].len() > cap,
2778            IndexKind::BTree(_) | IndexKind::Brin { .. } | IndexKind::Gin(_) | IndexKind::GinTrgm(_) => false,
2779        };
2780        if needs_trim {
2781            let current_peers: Vec<usize> = match &table.indices[idx_pos].kind {
2782                IndexKind::Nsw(g) => g.layers[layer as usize][peer]
2783                    .iter()
2784                    .map(|&n| n as usize)
2785                    .collect(),
2786                IndexKind::BTree(_) | IndexKind::Brin { .. } | IndexKind::Gin(_) | IndexKind::GinTrgm(_) => continue,
2787            };
2788            // Sort by distance from `peer`'s cell ascending so the
2789            // heuristic receives candidates closest-first. `cell_l2_sq`
2790            // dispatches on encoding so SQ8 columns trim using
2791            // symmetric ADC.
2792            let mut tagged: Vec<(f32, usize)> = current_peers
2793                .iter()
2794                .map(|&p| (cell_l2_sq(table, col_pos, peer, p), p))
2795                .collect();
2796            tagged.sort_by(|a, b| a.0.partial_cmp(&b.0).unwrap_or(core::cmp::Ordering::Equal));
2797            let kept = select_neighbours_heuristic(&tagged, cap, table, col_pos);
2798            if let IndexKind::Nsw(g) = &mut table.indices[idx_pos].kind
2799                && let Some(slot) = g.layers[layer as usize].get_mut(peer)
2800            {
2801                *slot = kept
2802                    .into_iter()
2803                    .map(|p| u32::try_from(p).expect("row index fits in u32"))
2804                    .collect();
2805            }
2806        }
2807    }
2808}
2809
2810/// Squared L2 distance from `query` (raw f32) to the cell at
2811/// `(row, col_pos)`. Dispatches on cell encoding: `Value::Vector`
2812/// (F32) uses `l2_distance_sq`; `Value::Sq8Vector` uses
2813/// `sq8_l2_distance_sq_asymmetric` (the v6.0.1 quantised path).
2814/// Returns `f32::INFINITY` for any non-vector cell so callers can
2815/// compare uniformly.
2816fn vec_l2_sq(table: &Table, col_pos: usize, row: usize, query: &[f32]) -> f32 {
2817    match table.rows.get(row).and_then(|r| r.values.get(col_pos)) {
2818        Some(Value::Vector(v)) if v.len() == query.len() => l2_distance_sq(v, query),
2819        Some(Value::Sq8Vector(q)) if q.bytes.len() == query.len() => {
2820            quantize::sq8_l2_distance_sq_asymmetric(q, query)
2821        }
2822        // v6.0.6: halfvec → fused NEON SIMD kernel; no Vec<f32>
2823        // allocation. v6.0.3 used `to_f32_vec()` + f32 NEON which
2824        // was correct but allocated per call (5× slower than F32).
2825        Some(Value::HalfVector(h)) if h.dim() == query.len() => {
2826            halfvec::half_l2_distance_sq_asymmetric(h, query)
2827        }
2828        _ => f32::INFINITY,
2829    }
2830}
2831
2832/// Squared L2 distance between two stored cells (no f32 query in
2833/// sight). Used during HNSW graph build — both endpoints are
2834/// rows already in the table, so symmetric ADC applies for SQ8
2835/// columns. Mixed-encoding cells within one column are a
2836/// schema-level impossibility (INSERT-time coercion enforces
2837/// uniform encoding), so the catch-all is an abort.
2838fn cell_l2_sq(table: &Table, col_pos: usize, row_a: usize, row_b: usize) -> f32 {
2839    let Some(cell_a) = table.rows.get(row_a).and_then(|r| r.values.get(col_pos)) else {
2840        return f32::INFINITY;
2841    };
2842    let Some(cell_b) = table.rows.get(row_b).and_then(|r| r.values.get(col_pos)) else {
2843        return f32::INFINITY;
2844    };
2845    match (cell_a, cell_b) {
2846        (Value::Vector(a), Value::Vector(b)) if a.len() == b.len() => l2_distance_sq(a, b),
2847        (Value::Sq8Vector(a), Value::Sq8Vector(b)) if a.bytes.len() == b.bytes.len() => {
2848            quantize::sq8_l2_distance_sq(a, b)
2849        }
2850        // v6.0.6: halfvec symmetric NEON — fused SIMD kernel that
2851        // loads both cells' raw u16 bits, expands to f32 lanes
2852        // inline, FMA-accumulates the squared diff. No Vec<f32>
2853        // allocation per call.
2854        (Value::HalfVector(a), Value::HalfVector(b)) if a.dim() == b.dim() => {
2855            halfvec::half_l2_distance_sq(a, b)
2856        }
2857        _ => f32::INFINITY,
2858    }
2859}
2860
2861/// kNN-search-time distance: stored cell → f32 query under the
2862/// caller's metric. Dispatches on cell encoding so SQ8 columns
2863/// take the ADC path with the right asymmetric variant. NaN /
2864/// dim-mismatch / non-vector → `f32::INFINITY`.
2865fn cell_to_query_metric_distance(
2866    table: &Table,
2867    col_pos: usize,
2868    row: usize,
2869    query: &[f32],
2870    metric: NswMetric,
2871) -> f32 {
2872    match table.rows.get(row).and_then(|r| r.values.get(col_pos)) {
2873        Some(Value::Vector(v)) if v.len() == query.len() => metric_distance(metric, v, query),
2874        Some(Value::Sq8Vector(q)) if q.bytes.len() == query.len() => match metric {
2875            NswMetric::L2 => quantize::sq8_l2_distance_sq_asymmetric(q, query),
2876            NswMetric::InnerProduct => quantize::sq8_inner_product_asymmetric(q, query),
2877            NswMetric::Cosine => quantize::sq8_cosine_distance_asymmetric(q, query),
2878        },
2879        // v6.0.6: halfvec dispatches by metric to fused NEON
2880        // kernels — no Vec<f32> allocation per call.
2881        Some(Value::HalfVector(h)) if h.dim() == query.len() => match metric {
2882            NswMetric::L2 => halfvec::half_l2_distance_sq_asymmetric(h, query),
2883            NswMetric::InnerProduct => halfvec::half_inner_product_asymmetric(h, query),
2884            NswMetric::Cosine => halfvec::half_cosine_distance_asymmetric(h, query),
2885        },
2886        _ => f32::INFINITY,
2887    }
2888}
2889
2890/// Distance metric used at NSW search time. The graph topology is
2891/// always built with `L2`; querying with `InnerProduct` / `Cosine`
2892/// reuses the same edges but ranks candidates by the chosen metric.
2893/// For the corpus-sized graphs this loses negligible recall vs
2894/// building separate per-metric graphs.
2895#[derive(Debug, Clone, Copy, PartialEq, Eq)]
2896pub enum NswMetric {
2897    /// Squared Euclidean — ranks "smaller = closer" (the sqrt is
2898    /// monotonic so we skip it for ordering).
2899    L2,
2900    /// Negated dot product, matching pgvector `<#>` convention so
2901    /// "smaller = more similar" holds across all three metrics.
2902    InnerProduct,
2903    /// Cosine distance `1 - cos(a, b)`. Zero-norm operand yields
2904    /// `f32::INFINITY` so it sorts last.
2905    Cosine,
2906}
2907
2908/// Multi-layer HNSW kNN search: greedy-descend from the entry to layer 0,
2909/// then beam-search there with the requested `ef` to return the top `k`
2910/// results under the caller-chosen metric. Topology was built with L2 —
2911/// upper-layer descent uses L2 as a coarse heuristic; final beam search
2912/// runs in the requested metric so rankings are correct for `<#>` / `<=>`.
2913fn nsw_search(
2914    table: &Table,
2915    idx_pos: usize,
2916    query: &[f32],
2917    k: usize,
2918    ef: usize,
2919    metric: NswMetric,
2920) -> Vec<(f32, usize)> {
2921    let (entry, entry_level) = match &table.indices[idx_pos].kind {
2922        IndexKind::Nsw(g) => (g.entry, g.entry_level),
2923        IndexKind::BTree(_) | IndexKind::Brin { .. } | IndexKind::Gin(_) | IndexKind::GinTrgm(_) => return Vec::new(),
2924    };
2925    let Some(entry) = entry else {
2926        return Vec::new();
2927    };
2928    let col_pos = table.indices[idx_pos].column_position;
2929    // v6.0.1 step 5: SQ8 columns over-fetch by `SQ8_RERANK_OVER_FETCH`
2930    // so the rerank pass below sees enough candidates to recover
2931    // recall after the ADC re-ordering. F32 + F16 columns skip the
2932    // over-fetch — F32 distances are exact, F16 dequant is
2933    // bit-exact at the storage layer so the beam search already
2934    // ranks under the column's full precision.
2935    let sq8 = matches!(
2936        table.schema.columns.get(col_pos).map(|c| c.ty),
2937        Some(DataType::Vector {
2938            encoding: VecEncoding::Sq8,
2939            ..
2940        })
2941    );
2942    let ef = if sq8 {
2943        ef.max(k).max(k * SQ8_RERANK_OVER_FETCH)
2944    } else {
2945        ef.max(k)
2946    };
2947    // Descend by L2 (the topology metric) so layers prune consistently.
2948    let entry_d = vec_l2_sq(table, col_pos, entry, query);
2949    let mut current = entry;
2950    let mut current_d = entry_d;
2951    for layer in (1..=entry_level).rev() {
2952        (current, current_d) = greedy_layer_walk(table, idx_pos, layer, current, current_d, query);
2953    }
2954    // Final beam search on layer 0 under the caller's metric.
2955    let mut results = layer_beam_search(table, idx_pos, 0, current, current_d, query, ef, metric);
2956    if sq8 {
2957        results = sq8_rerank(table, col_pos, &results, query, metric);
2958    }
2959    results.truncate(k);
2960    results
2961}
2962
2963/// v6.0.1 step 5: re-score ADC top-`K*3` candidates with the
2964/// dequantised cell vs the f32 query, then re-sort. Recovers the
2965/// recall the SQ8 ADC sacrifices for 4× compression — the design's
2966/// "f32 rerank step is on by default" path (deliberation #3).
2967/// `metric` is the same metric the beam search used; the rerank
2968/// arithmetic re-derives the exact distance under that metric.
2969fn sq8_rerank(
2970    table: &Table,
2971    col_pos: usize,
2972    candidates: &[(f32, usize)],
2973    query: &[f32],
2974    metric: NswMetric,
2975) -> Vec<(f32, usize)> {
2976    let mut out: Vec<(f32, usize)> = candidates
2977        .iter()
2978        .filter_map(|&(adc_d, row)| {
2979            let cell = table.rows.get(row).and_then(|r| r.values.get(col_pos))?;
2980            let Value::Sq8Vector(q) = cell else {
2981                // F32 cells shouldn't reach this path (sq8 fence
2982                // above), but stay defensive: pass through with
2983                // the ADC distance unchanged.
2984                return Some((adc_d, row));
2985            };
2986            let deq = quantize::dequantize(q);
2987            if deq.len() != query.len() {
2988                return None;
2989            }
2990            Some((metric_distance(metric, &deq, query), row))
2991        })
2992        .collect();
2993    out.sort_by(|a, b| a.0.partial_cmp(&b.0).unwrap_or(core::cmp::Ordering::Equal));
2994    out
2995}
2996
2997/// Multiplier applied to `k` so the SQ8 rerank pass sees a wider
2998/// candidate set. 3× is the design-stage value; v6.0.5 sweep work
2999/// can re-tune once full corpus profiling is in.
3000const SQ8_RERANK_OVER_FETCH: usize = 3;
3001
3002fn metric_distance(metric: NswMetric, a: &[f32], b: &[f32]) -> f32 {
3003    match metric {
3004        NswMetric::L2 => l2_distance_sq(a, b),
3005        NswMetric::InnerProduct => -inner_product_f32(a, b),
3006        NswMetric::Cosine => {
3007            let (dot, na, nb) = cosine_dot_norms_f32(a, b);
3008            if na == 0.0 || nb == 0.0 {
3009                return f32::INFINITY;
3010            }
3011            // `f32::sqrt` lives in std, so hand-roll Newton-Raphson on
3012            // f64 — same trick the L2 binary op already uses.
3013            let denom = sqrt_newton_f32(na) * sqrt_newton_f32(nb);
3014            1.0 - dot / denom
3015        }
3016    }
3017}
3018
3019/// v6.0.2: dispatch wrapper for the f32 dot product (used by `<#>` +
3020/// the cosine numerator). NEON path when `len % 4 == 0 && len >= 4`,
3021/// scalar fallback otherwise. Returns the positive dot — callers
3022/// negate for the pgvector `<#>` "smaller = closer" convention.
3023///
3024/// Public so perf gates + downstream benches can microbenchmark the
3025/// dispatch directly; not part of the STABILITY contract — internal
3026/// SIMD layout can evolve in any release.
3027#[doc(hidden)]
3028#[inline]
3029pub fn inner_product_f32(a: &[f32], b: &[f32]) -> f32 {
3030    #[cfg(target_arch = "aarch64")]
3031    {
3032        if a.len() == b.len() && a.len() >= 4 && a.len().is_multiple_of(4) {
3033            // SAFETY: NEON is a baseline aarch64 feature; preconditions
3034            // (matching lengths, ≥ 1 full lane group) are checked above.
3035            return unsafe { inner_product_neon(a, b) };
3036        }
3037    }
3038    inner_product_scalar(a, b)
3039}
3040
3041fn inner_product_scalar(a: &[f32], b: &[f32]) -> f32 {
3042    let mut dot: f32 = 0.0;
3043    for (x, y) in a.iter().zip(b.iter()) {
3044        dot += x * y;
3045    }
3046    dot
3047}
3048
3049#[cfg(target_arch = "aarch64")]
3050#[target_feature(enable = "neon")]
3051#[allow(clippy::many_single_char_names)] // NEON intrinsics work in single-letter regs by convention
3052unsafe fn inner_product_neon(a: &[f32], b: &[f32]) -> f32 {
3053    use core::arch::aarch64::{
3054        float32x4_t, vaddq_f32, vaddvq_f32, vdupq_n_f32, vfmaq_f32, vld1q_f32,
3055    };
3056    unsafe {
3057        // Two parallel accumulators (same trick as L2 NEON) so the
3058        // FMA dependency chain doesn't serialise.
3059        let zero: float32x4_t = vdupq_n_f32(0.0);
3060        let mut acc0 = zero;
3061        let mut acc1 = zero;
3062        let n = a.len();
3063        let mut i = 0usize;
3064        while i + 8 <= n {
3065            let av0 = vld1q_f32(a.as_ptr().add(i));
3066            let bv0 = vld1q_f32(b.as_ptr().add(i));
3067            acc0 = vfmaq_f32(acc0, av0, bv0);
3068            let av1 = vld1q_f32(a.as_ptr().add(i + 4));
3069            let bv1 = vld1q_f32(b.as_ptr().add(i + 4));
3070            acc1 = vfmaq_f32(acc1, av1, bv1);
3071            i += 8;
3072        }
3073        while i + 4 <= n {
3074            let av = vld1q_f32(a.as_ptr().add(i));
3075            let bv = vld1q_f32(b.as_ptr().add(i));
3076            acc0 = vfmaq_f32(acc0, av, bv);
3077            i += 4;
3078        }
3079        vaddvq_f32(vaddq_f32(acc0, acc1))
3080    }
3081}
3082
3083/// v6.0.2: dispatch wrapper for the three accumulators (`dot`, `||a||²`,
3084/// `||b||²`) cosine needs. Same NEON pre-condition as the L2 / IP
3085/// paths; same scalar fallback shape.
3086///
3087/// Public for benchmarking only (see `inner_product_f32`); not in the
3088/// STABILITY contract.
3089#[doc(hidden)]
3090#[inline]
3091pub fn cosine_dot_norms_f32(a: &[f32], b: &[f32]) -> (f32, f32, f32) {
3092    #[cfg(target_arch = "aarch64")]
3093    {
3094        if a.len() == b.len() && a.len() >= 4 && a.len().is_multiple_of(4) {
3095            // SAFETY: see `inner_product_neon`.
3096            return unsafe { cosine_dot_norms_neon(a, b) };
3097        }
3098    }
3099    cosine_dot_norms_scalar(a, b)
3100}
3101
3102fn cosine_dot_norms_scalar(a: &[f32], b: &[f32]) -> (f32, f32, f32) {
3103    let mut dot: f32 = 0.0;
3104    let mut na: f32 = 0.0;
3105    let mut nb: f32 = 0.0;
3106    for (x, y) in a.iter().zip(b.iter()) {
3107        dot += x * y;
3108        na += x * x;
3109        nb += y * y;
3110    }
3111    (dot, na, nb)
3112}
3113
3114#[cfg(target_arch = "aarch64")]
3115#[target_feature(enable = "neon")]
3116#[allow(clippy::many_single_char_names, clippy::similar_names)]
3117unsafe fn cosine_dot_norms_neon(a: &[f32], b: &[f32]) -> (f32, f32, f32) {
3118    use core::arch::aarch64::{float32x4_t, vaddvq_f32, vdupq_n_f32, vfmaq_f32, vld1q_f32};
3119    unsafe {
3120        let zero: float32x4_t = vdupq_n_f32(0.0);
3121        let mut acc_dot = zero;
3122        let mut acc_na = zero;
3123        let mut acc_nb = zero;
3124        let n = a.len();
3125        let mut i = 0usize;
3126        while i + 4 <= n {
3127            let av = vld1q_f32(a.as_ptr().add(i));
3128            let bv = vld1q_f32(b.as_ptr().add(i));
3129            acc_dot = vfmaq_f32(acc_dot, av, bv);
3130            acc_na = vfmaq_f32(acc_na, av, av);
3131            acc_nb = vfmaq_f32(acc_nb, bv, bv);
3132            i += 4;
3133        }
3134        (vaddvq_f32(acc_dot), vaddvq_f32(acc_na), vaddvq_f32(acc_nb))
3135    }
3136}
3137
3138fn sqrt_newton_f32(x: f32) -> f32 {
3139    if x <= 0.0 {
3140        return 0.0;
3141    }
3142    let mut g = x;
3143    for _ in 0..10 {
3144        g = 0.5 * (g + x / g);
3145    }
3146    g
3147}
3148
3149/// Squared Euclidean distance — used for ordering inside NSW (the sqrt
3150/// preserves the order). Caller takes sqrt before reporting back to SQL.
3151///
3152/// v3.3.2: aarch64 NEON path for `len % 4 == 0` (which covers every
3153/// HNSW-indexed VECTOR(N) where N is a multiple of 4 — i.e. all
3154/// production-shaped embeddings: 64, 128, 256, 384, 512, 768, 1024,
3155/// 1536, ...). Other shapes fall back to the scalar loop.
3156#[inline]
3157fn l2_distance_sq(a: &[f32], b: &[f32]) -> f32 {
3158    #[cfg(target_arch = "aarch64")]
3159    {
3160        if a.len() == b.len() && a.len() >= 4 && a.len().is_multiple_of(4) {
3161            // SAFETY: NEON is a baseline aarch64 feature (ARMv8);
3162            // the precondition is checked above (matching lengths,
3163            // multiple of 4, at least one 128-bit lane group).
3164            return unsafe { l2_distance_sq_neon(a, b) };
3165        }
3166    }
3167    l2_distance_sq_scalar(a, b)
3168}
3169
3170fn l2_distance_sq_scalar(a: &[f32], b: &[f32]) -> f32 {
3171    let mut sum: f32 = 0.0;
3172    for (x, y) in a.iter().zip(b.iter()) {
3173        let d = *x - *y;
3174        sum += d * d;
3175    }
3176    sum
3177}
3178
3179#[cfg(target_arch = "aarch64")]
3180#[target_feature(enable = "neon")]
3181#[allow(clippy::many_single_char_names)] // NEON intrinsics work in single-letter regs by convention
3182unsafe fn l2_distance_sq_neon(a: &[f32], b: &[f32]) -> f32 {
3183    use core::arch::aarch64::{
3184        float32x4_t, vaddq_f32, vaddvq_f32, vdupq_n_f32, vfmaq_f32, vld1q_f32, vsubq_f32,
3185    };
3186    unsafe {
3187        // Two independent accumulator registers so the FMA dependency
3188        // chain doesn't serialise (each FMA depends on prior FMA).
3189        // Pre-conditions checked by caller: `a.len() == b.len()`,
3190        // `a.len() % 4 == 0`, `a.len() >= 4`.
3191        let zero: float32x4_t = vdupq_n_f32(0.0);
3192        let mut acc0 = zero;
3193        let mut acc1 = zero;
3194        let n = a.len();
3195        let mut i = 0usize;
3196        // Process 8 floats per iter when available (two parallel
3197        // accumulators). Tail of 4 falls into the second loop.
3198        while i + 8 <= n {
3199            let d0 = vsubq_f32(vld1q_f32(a.as_ptr().add(i)), vld1q_f32(b.as_ptr().add(i)));
3200            acc0 = vfmaq_f32(acc0, d0, d0);
3201            let d1 = vsubq_f32(
3202                vld1q_f32(a.as_ptr().add(i + 4)),
3203                vld1q_f32(b.as_ptr().add(i + 4)),
3204            );
3205            acc1 = vfmaq_f32(acc1, d1, d1);
3206            i += 8;
3207        }
3208        while i + 4 <= n {
3209            let d = vsubq_f32(vld1q_f32(a.as_ptr().add(i)), vld1q_f32(b.as_ptr().add(i)));
3210            acc0 = vfmaq_f32(acc0, d, d);
3211            i += 4;
3212        }
3213        vaddvq_f32(vaddq_f32(acc0, acc1))
3214    }
3215}
3216
3217/// Public wrapper: run an NSW kNN search and return the top-k row
3218/// indices ordered by ascending distance under the given metric.
3219pub fn nsw_query(
3220    table: &Table,
3221    idx_name: &str,
3222    query: &[f32],
3223    k: usize,
3224    metric: NswMetric,
3225) -> Vec<usize> {
3226    let Some(idx_pos) = table.indices.iter().position(|i| i.name == idx_name) else {
3227        return Vec::new();
3228    };
3229    let ef = (k * 2).max(NSW_DEFAULT_M);
3230    let mut hits = nsw_search(table, idx_pos, query, k, ef, metric);
3231    hits.truncate(k);
3232    hits.into_iter().map(|(_, idx)| idx).collect()
3233}
3234
3235/// Find any NSW index on a column. Used by the planner to decide
3236/// whether an `ORDER BY col <-> literal LIMIT k` query can skip the
3237/// brute-force scan.
3238pub fn nsw_index_on(table: &Table, column_position: usize) -> Option<&Index> {
3239    table
3240        .indices
3241        .iter()
3242        .find(|i| i.column_position == column_position && matches!(i.kind, IndexKind::Nsw(_)))
3243}
3244
3245/// Catalog: insertion-ordered `Vec<Table>` for stable iter / serialize,
3246/// plus a `BTreeMap<String, usize>` sidecar index so `get` / `get_mut`
3247/// run in O(log n) instead of the old linear scan with per-element
3248/// string compares.
3249///
3250/// A pure `BTreeMap<String, Table>` was tried in an interim version
3251/// of v3.1.2 and regressed the single-table catalog benches by ~10%
3252/// (the per-element `BTreeMap` overhead outweighs the lookup win
3253/// when n is small). The sidecar shape preserves the insertion-order
3254/// iteration the on-disk encoding relies on and keeps `last_mut`
3255/// (used by the deserialize hot path) cheap.
3256#[derive(Debug, Clone, Default)]
3257pub struct Catalog {
3258    tables: Vec<Table>,
3259    /// `name → tables[index]`. Kept in lock-step with `tables`.
3260    /// `create_table` is the only write path.
3261    by_name: BTreeMap<String, usize>,
3262    /// v5.1: in-memory cold-tier segments. Side-loaded via
3263    /// [`Catalog::load_segment_bytes`] — they live outside the
3264    /// catalog snapshot (caller persists them as separate files
3265    /// and re-loads on boot, until v5.3's `CatalogManifest` makes
3266    /// that wiring automatic). `RowLocator::Cold { segment_id, .. }`
3267    /// indexes this `Vec`. Cleared on `Catalog::new` / fresh
3268    /// `deserialize`.
3269    ///
3270    /// `Arc` wrap keeps `Catalog::clone` at O(N segments) bumps
3271    /// (rather than O(total segment bytes) memcpy) so the v4.42
3272    /// group-commit pre-image rollback invariant — clone is
3273    /// effectively free — survives the cold-tier addition.
3274    ///
3275    /// v6.7.3 — slots became `Option<…>` so cold-segment compaction
3276    /// can tombstone merged sources without breaking the
3277    /// `segment_id = index_into_vec` contract that on-disk
3278    /// `RowLocator::Cold { segment_id }` already serialized.
3279    /// `None` slot = the segment was retired by compaction; the
3280    /// physical file may still be on disk (next CHECKPOINT writes
3281    /// a manifest that no longer lists it, and the file becomes
3282    /// an orphan eligible for offline cleanup).
3283    cold_segments: Vec<Option<Arc<OwnedSegment>>>,
3284    /// v7.12.4 — user-defined functions (PL/pgSQL + SQL).
3285    /// Keyed by function name (PG overloading is out of scope).
3286    /// Bodies are stored as the raw source text the parser saw
3287    /// between `$$ ... $$`; the engine re-parses on each
3288    /// invocation. This keeps `spg-storage` free of `spg-sql`
3289    /// dependency — same pattern as partial-index predicates.
3290    functions: BTreeMap<String, FunctionDef>,
3291    /// v7.12.4 — triggers in insertion order. Multiple triggers
3292    /// per table / event fire in this order (matching PG's
3293    /// alphabetical-by-default with insertion-stable tie-break
3294    /// behaviour — we just keep insertion order for now).
3295    triggers: Vec<TriggerDef>,
3296}
3297
3298/// v7.12.4 — catalogued user-defined function. `body` is the raw
3299/// source text between `$$ ... $$`; the engine re-parses it on
3300/// invocation. This keeps the storage codec stable when the
3301/// PL/pgSQL surface grows (no breaking-change risk on the disk
3302/// format).
3303#[derive(Debug, Clone, PartialEq, Eq)]
3304pub struct FunctionDef {
3305    pub name: String,
3306    /// Display form of the argument list, e.g.
3307    /// `"(name TEXT, ts TIMESTAMP)"`. Empty `"()"` for the trigger
3308    /// function shape. Parser-side canonicalised before storage.
3309    pub args_repr: String,
3310    /// Display form of the return type, e.g. `"TRIGGER"` /
3311    /// `"INT"` / `"SETOF text"`. The engine special-cases
3312    /// `"TRIGGER"` (case-insensitive) to gate trigger-only
3313    /// semantics (NEW/OLD).
3314    pub returns: String,
3315    /// `LANGUAGE` clause, lowercased. `"plpgsql"` / `"sql"`.
3316    pub language: String,
3317    /// Source body of the function. PL/pgSQL: includes the
3318    /// surrounding `BEGIN ... END;`. SQL: includes the
3319    /// statement(s). The engine re-parses on invocation; bad
3320    /// bodies surface as a parse error at CALL time, not CREATE.
3321    pub body: String,
3322}
3323
3324/// v7.12.4 — catalogued trigger. References its function by
3325/// name; the function must exist at TRIGGER creation time
3326/// (forward references are deferred to v7.12.5+).
3327#[derive(Debug, Clone, PartialEq, Eq)]
3328pub struct TriggerDef {
3329    pub name: String,
3330    /// Watched table. Trigger is dropped when the table drops.
3331    pub table: String,
3332    /// `"BEFORE"` / `"AFTER"` / `"INSTEAD OF"`. Stored as the
3333    /// uppercased keyword so deserialised catalogs round-trip
3334    /// without canonicalisation surprises.
3335    pub timing: String,
3336    /// Each entry is one of `"INSERT"` / `"UPDATE"` / `"DELETE"`
3337    /// / `"TRUNCATE"`. `INSERT OR UPDATE` parses to two entries.
3338    pub events: Vec<String>,
3339    /// `"ROW"` / `"STATEMENT"`. v7.12.4 ships `"ROW"` only;
3340    /// `"STATEMENT"` parses and persists but the executor
3341    /// refuses it at trigger fire time.
3342    pub for_each: String,
3343    /// Name of the PL/pgSQL function to invoke.
3344    pub function: String,
3345    /// v7.13.0 — `UPDATE OF col, col, …` column-list filter
3346    /// (mailrs round-5 G7). Non-empty means the trigger fires
3347    /// only when at least one of these columns appears in the
3348    /// UPDATE's SET list. Empty = no column filter. Stored in
3349    /// catalog FILE_VERSION 23+; older catalogs deserialise with
3350    /// an empty vec.
3351    pub update_columns: Vec<String>,
3352    /// v7.16.1 — whether the trigger fires when its watched
3353    /// event occurs. Toggled by `ALTER TABLE … { ENABLE |
3354    /// DISABLE } TRIGGER …`; pg_dump --disable-triggers wraps
3355    /// every data block with a DISABLE/ENABLE pair so the
3356    /// rows already-computed in prod don't get re-rewritten.
3357    /// Defaults to `true` at CREATE TRIGGER time. Stored in
3358    /// catalog FILE_VERSION 25+; older catalogs deserialise
3359    /// with `enabled = true`.
3360    pub enabled: bool,
3361}
3362
3363impl Catalog {
3364    pub const fn new() -> Self {
3365        Self {
3366            tables: Vec::new(),
3367            by_name: BTreeMap::new(),
3368            cold_segments: Vec::new(),
3369            functions: BTreeMap::new(),
3370            triggers: Vec::new(),
3371        }
3372    }
3373
3374    /// v7.12.4 — read-only view of catalogued user-defined
3375    /// functions. Engine callers go through here to look up the
3376    /// function body before re-parsing it for invocation.
3377    pub const fn functions(&self) -> &BTreeMap<String, FunctionDef> {
3378        &self.functions
3379    }
3380
3381    /// v7.12.4 — register a new user-defined function. With
3382    /// `or_replace = false`, errors if the name is taken. The
3383    /// engine validates the body before passing it here.
3384    pub fn create_function(
3385        &mut self,
3386        def: FunctionDef,
3387        or_replace: bool,
3388    ) -> Result<(), StorageError> {
3389        if !or_replace && self.functions.contains_key(&def.name) {
3390            return Err(StorageError::Corrupt(format!(
3391                "function {:?} already exists (drop or use CREATE OR REPLACE)",
3392                def.name
3393            )));
3394        }
3395        self.functions.insert(def.name.clone(), def);
3396        Ok(())
3397    }
3398
3399    /// v7.12.4 — remove a user-defined function by name. Returns
3400    /// `true` if a function was removed, `false` if none matched.
3401    /// Caller decides whether to surface `if_exists` semantics.
3402    pub fn drop_function(&mut self, name: &str) -> bool {
3403        self.functions.remove(name).is_some()
3404    }
3405
3406    /// v7.12.4 — read-only slice of all catalogued triggers.
3407    /// Engine row-write paths filter this by (table, event,
3408    /// timing) and fire matches in slice order.
3409    pub fn triggers(&self) -> &[TriggerDef] {
3410        &self.triggers
3411    }
3412
3413    /// v7.15.0 — mutable handle to the trigger slice for
3414    /// `ALTER TABLE … RENAME COLUMN`, which rewrites every
3415    /// `update_columns` entry that referenced the renamed
3416    /// column.
3417    pub fn triggers_mut(&mut self) -> &mut Vec<TriggerDef> {
3418        &mut self.triggers
3419    }
3420
3421    /// v7.12.4 — register a new trigger. With `or_replace = false`,
3422    /// errors when a trigger with the same name already exists on
3423    /// the same table (PG scoping rule — trigger names are
3424    /// per-table, not global). Trigger function must already
3425    /// exist in the catalog at registration time.
3426    pub fn create_trigger(
3427        &mut self,
3428        def: TriggerDef,
3429        or_replace: bool,
3430    ) -> Result<(), StorageError> {
3431        if !self.by_name.contains_key(&def.table) {
3432            return Err(StorageError::TableNotFound {
3433                name: def.table.clone(),
3434            });
3435        }
3436        if !self.functions.contains_key(&def.function) {
3437            return Err(StorageError::Corrupt(format!(
3438                "trigger {:?} references unknown function {:?}",
3439                def.name, def.function
3440            )));
3441        }
3442        let dup = self
3443            .triggers
3444            .iter()
3445            .position(|t| t.name == def.name && t.table == def.table);
3446        match (dup, or_replace) {
3447            (Some(_), false) => Err(StorageError::Corrupt(format!(
3448                "trigger {:?} already exists on table {:?}",
3449                def.name, def.table
3450            ))),
3451            (Some(i), true) => {
3452                self.triggers[i] = def;
3453                Ok(())
3454            }
3455            (None, _) => {
3456                self.triggers.push(def);
3457                Ok(())
3458            }
3459        }
3460    }
3461
3462    /// v7.12.4 — remove a trigger by `(name, table)`. Returns
3463    /// `true` if one was removed.
3464    pub fn drop_trigger(&mut self, name: &str, table: &str) -> bool {
3465        let before = self.triggers.len();
3466        self.triggers
3467            .retain(|t| !(t.name == name && t.table == table));
3468        before != self.triggers.len()
3469    }
3470
3471    pub fn create_table(&mut self, schema: TableSchema) -> Result<(), StorageError> {
3472        if self.by_name.contains_key(&schema.name) {
3473            return Err(StorageError::DuplicateTable {
3474                name: schema.name.clone(),
3475            });
3476        }
3477        let idx = self.tables.len();
3478        let name = schema.name.clone();
3479        self.tables.push(Table::new(schema));
3480        self.by_name.insert(name, idx);
3481        Ok(())
3482    }
3483
3484    pub fn get(&self, name: &str) -> Option<&Table> {
3485        let idx = *self.by_name.get(name)?;
3486        self.tables.get(idx)
3487    }
3488
3489    pub fn get_mut(&mut self, name: &str) -> Option<&mut Table> {
3490        let idx = *self.by_name.get(name)?;
3491        self.tables.get_mut(idx)
3492    }
3493
3494    pub fn table_count(&self) -> usize {
3495        self.tables.len()
3496    }
3497
3498    /// v7.14.0 — remove a table by name. Returns `true` when the
3499    /// table existed (and is now gone), `false` when it didn't.
3500    /// Used by `DROP TABLE` from pg_dump / mysqldump preambles
3501    /// where the dump re-creates schema and starts with
3502    /// `DROP TABLE IF EXISTS`.
3503    pub fn drop_table(&mut self, name: &str) -> bool {
3504        let Some(idx) = self.by_name.remove(name) else {
3505            return false;
3506        };
3507        // swap_remove invalidates the trailing index → rebuild
3508        // by_name for affected entries.
3509        self.tables.swap_remove(idx);
3510        // Re-stamp moved table's index slot in by_name.
3511        if idx < self.tables.len() {
3512            let moved_name = self.tables[idx].schema.name.clone();
3513            self.by_name.insert(moved_name, idx);
3514        }
3515        true
3516    }
3517
3518    /// v7.14.0 — remove a named index across the catalog.
3519    /// Returns `true` when found + dropped.
3520    pub fn drop_named_index(&mut self, name: &str) -> bool {
3521        for t in &mut self.tables {
3522            let before = t.indices.len();
3523            t.indices.retain(|i| i.name != name);
3524            if t.indices.len() != before {
3525                return true;
3526            }
3527        }
3528        false
3529    }
3530
3531    /// Borrow-free copy of every table's name in catalog order
3532    /// (= insertion order, matching the on-disk encoding).
3533    pub fn table_names(&self) -> Vec<String> {
3534        self.tables.iter().map(|t| t.schema.name.clone()).collect()
3535    }
3536
3537    /// v5.1: register a cold-tier segment that already lives in
3538    /// memory (caller did the file read). Returns the
3539    /// `segment_id` that `RowLocator::Cold { segment_id, .. }`
3540    /// will reference — currently this is just the index into
3541    /// `cold_segments`, but treat it as an opaque token.
3542    ///
3543    /// Storage is `no_std`, so file I/O is the caller's
3544    /// responsibility — `spg-server` reads the file and forwards
3545    /// the bytes here. The bytes stay resident in the catalog
3546    /// for the life of the `Catalog`, parsed only once.
3547    pub fn load_segment_bytes(&mut self, bytes: Vec<u8>) -> Result<u32, StorageError> {
3548        let id = u32::try_from(self.cold_segments.len()).map_err(|_| {
3549            StorageError::Corrupt("cold segment count would exceed u32::MAX".into())
3550        })?;
3551        let seg = OwnedSegment::from_bytes(bytes)
3552            .map_err(|e| StorageError::Corrupt(format!("cold segment parse failed: {e}")))?;
3553        self.cold_segments.push(Some(Arc::new(seg)));
3554        Ok(id)
3555    }
3556
3557    /// v6.7.3 — register a cold-tier segment at a specific id. Used
3558    /// by the spg-server manifest-boot path so segments whose
3559    /// neighbouring ids were retired by compaction still get back
3560    /// the same `segment_id` they had pre-restart (the
3561    /// `RowLocator::Cold { segment_id }` baked into the BTree-index
3562    /// snapshot persists across restart and must continue to
3563    /// resolve).
3564    ///
3565    /// Pads the Vec with `None` slots up to `target_id` if needed.
3566    /// Errors when the target slot is already occupied (would
3567    /// stomp another segment), the parse fails, or `target_id`
3568    /// exceeds `u32::MAX`.
3569    pub fn load_segment_bytes_at(
3570        &mut self,
3571        target_id: u32,
3572        bytes: Vec<u8>,
3573    ) -> Result<(), StorageError> {
3574        let seg = OwnedSegment::from_bytes(bytes)
3575            .map_err(|e| StorageError::Corrupt(format!("cold segment parse failed: {e}")))?;
3576        let idx = target_id as usize;
3577        while self.cold_segments.len() <= idx {
3578            self.cold_segments.push(None);
3579        }
3580        if self.cold_segments[idx].is_some() {
3581            return Err(StorageError::Corrupt(format!(
3582                "load_segment_bytes_at: segment_id {target_id} already occupied"
3583            )));
3584        }
3585        self.cold_segments[idx] = Some(Arc::new(seg));
3586        Ok(())
3587    }
3588
3589    /// v6.7.3 — retire a cold-tier segment slot (compaction-driven).
3590    /// The physical file is the caller's concern (typically kept
3591    /// on disk until the next CHECKPOINT writes a manifest that
3592    /// no longer lists it); this just flips the in-memory slot
3593    /// to `None` so later cold lookups for `segment_id` resolve
3594    /// as "unknown" instead of returning a stale row.
3595    ///
3596    /// No-op when the slot is already `None`. Errors only when
3597    /// `segment_id` is out of bounds.
3598    pub fn tombstone_segment(&mut self, segment_id: u32) -> Result<(), StorageError> {
3599        let idx = segment_id as usize;
3600        if idx >= self.cold_segments.len() {
3601            return Err(StorageError::Corrupt(format!(
3602                "tombstone_segment: segment_id {segment_id} out of bounds (len={})",
3603                self.cold_segments.len()
3604            )));
3605        }
3606        self.cold_segments[idx] = None;
3607        Ok(())
3608    }
3609
3610    /// Number of *active* (non-tombstoned) cold segments.
3611    #[must_use]
3612    pub fn cold_segment_count(&self) -> usize {
3613        self.cold_segments.iter().filter(|s| s.is_some()).count()
3614    }
3615
3616    /// Slot count including tombstones (= the next id the
3617    /// no-arg `load_segment_bytes` would allocate).
3618    #[must_use]
3619    pub fn cold_segment_slot_count(&self) -> usize {
3620        self.cold_segments.len()
3621    }
3622
3623    /// v6.2.7 — list every *active* cold-tier segment id known to
3624    /// this catalog (skips compaction tombstones since v6.7.3).
3625    /// Used by EXPLAIN ANALYZE to annotate scan nodes with the
3626    /// segments they could have walked.
3627    #[must_use]
3628    pub fn cold_segment_ids_global(&self) -> Vec<u32> {
3629        self.cold_segments
3630            .iter()
3631            .enumerate()
3632            .filter_map(|(i, s)| s.as_ref().map(|_| i as u32))
3633            .collect()
3634    }
3635
3636    /// v5.2.1: sum of `Table::hot_bytes` across every table. The v5.2
3637    /// freezer compares this against `SPG_HOT_TIER_BYTES` (parsed at
3638    /// server startup; default 4 GiB) and wakes when the budget is
3639    /// crossed. Pre-freezer (v5.2.1) this is measurement-only — the
3640    /// counter exposes whether the budget is being approached without
3641    /// triggering any demotion.
3642    #[must_use]
3643    pub fn hot_tier_bytes(&self) -> u64 {
3644        self.tables
3645            .iter()
3646            .map(Table::hot_bytes)
3647            .fold(0u64, u64::saturating_add)
3648    }
3649
3650    /// v5.2.2: freeze the **first** `max_rows` rows of `table_name`'s
3651    /// hot tier into a brand-new cold-tier segment. The named `BTree`
3652    /// index supplies the per-row PK (its column must be an integer
3653    /// type — v5.2.2 only supports `IndexKey::Int` PKs, matching the
3654    /// `index_key_as_u64` constraint used by the cold-tier lookup
3655    /// path). On success returns a [`FreezeReport`] with the
3656    /// freshly-allocated segment id, the count of rows that moved,
3657    /// the encoded segment bytes (so the caller can persist them to
3658    /// disk for later reload via `SPG_PRELOAD_COLD_SEGMENT`), and the
3659    /// hot-tier byte delta that was reclaimed.
3660    ///
3661    /// **Semantics**:
3662    /// 1. The first `max_rows` rows (by hot-tier position — same as
3663    ///    insertion order under v4.39 `PersistentVec`) are read.
3664    /// 2. Rows are sorted ascending by PK and serialised into a new
3665    ///    segment via [`encode_segment`].
3666    /// 3. The hot rows are dropped via [`Table::delete_rows`]; the
3667    ///    `rebuild_indices` it triggers regenerates `Hot` locators
3668    ///    for every remaining row (their positions shift down by
3669    ///    `max_rows`). Existing `Cold` locators in this index — from
3670    ///    a previous freeze — are also rebuilt **but with empty
3671    ///    payload** since rebuild reads only `self.rows`; this
3672    ///    routine re-registers them at the end of the call so the
3673    ///    user-visible state preserves all prior cold locators.
3674    /// 4. The new segment is loaded into `self.cold_segments` via
3675    ///    [`Catalog::load_segment_bytes`] (allocating a fresh
3676    ///    `segment_id`). New `Cold` locators are registered on the
3677    ///    named index — one per frozen row.
3678    ///
3679    /// **v5.2.2 limits** (relaxed in later sub-versions):
3680    /// - INSERT-only flow: subsequent UPDATE/DELETE on a frozen row
3681    ///   returns a stale-locator error (no promote-on-write until
3682    ///   v5.2.3).
3683    /// - Single-table scope: callers iterate tables themselves.
3684    /// - All-or-nothing: returns `Err` and leaves catalog unchanged
3685    ///   if any step fails before the atomic swap point.
3686    ///
3687    /// Errors:
3688    /// - [`StorageError::Corrupt`] for missing table/index, non-`BTree`
3689    ///   index, non-integer PK column, `max_rows == 0`, or
3690    ///   `max_rows > row_count`.
3691    /// - The encoder's [`SegmentError`] surfaces as `Corrupt` (the
3692    ///   only realistic source is "a single row is larger than the
3693    ///   page size"; SPG schemas don't hit it in practice).
3694    pub fn freeze_oldest_to_cold(
3695        &mut self,
3696        table_name: &str,
3697        index_name: &str,
3698        max_rows: usize,
3699    ) -> Result<FreezeReport, StorageError> {
3700        // --- validation phase: never mutates ---------------------
3701        if max_rows == 0 {
3702            return Err(StorageError::Corrupt(
3703                "freeze_oldest_to_cold: max_rows must be > 0".into(),
3704            ));
3705        }
3706        let table = self.get(table_name).ok_or_else(|| {
3707            StorageError::Corrupt(format!(
3708                "freeze_oldest_to_cold: table {table_name:?} not found"
3709            ))
3710        })?;
3711        if max_rows > table.rows.len() {
3712            return Err(StorageError::Corrupt(format!(
3713                "freeze_oldest_to_cold: max_rows {max_rows} > row_count {}",
3714                table.rows.len()
3715            )));
3716        }
3717        let idx = table
3718            .indices
3719            .iter()
3720            .find(|i| i.name == index_name)
3721            .ok_or_else(|| {
3722                StorageError::Corrupt(format!(
3723                    "freeze_oldest_to_cold: index {index_name:?} not found on {table_name:?}"
3724                ))
3725            })?;
3726        if !matches!(idx.kind, IndexKind::BTree(_)) {
3727            return Err(StorageError::Corrupt(format!(
3728                "freeze_oldest_to_cold: index {index_name:?} is NSW; only BTree indices may freeze"
3729            )));
3730        }
3731        let column_position = idx.column_position;
3732
3733        // --- segment build phase: reads only --------------------
3734        let schema = table.schema.clone();
3735        let mut to_freeze: Vec<(u64, Vec<u8>, IndexKey)> = Vec::with_capacity(max_rows);
3736        for row_idx in 0..max_rows {
3737            let row = table.rows.get(row_idx).expect("bounds-checked above");
3738            let key = IndexKey::from_value(&row.values[column_position]).ok_or_else(|| {
3739                StorageError::Corrupt(format!(
3740                    "freeze_oldest_to_cold: row {row_idx} has NULL / non-key value in index column"
3741                ))
3742            })?;
3743            let pk_u64 = index_key_as_u64(&key).ok_or_else(|| {
3744                StorageError::Corrupt(format!(
3745                    "freeze_oldest_to_cold: index {index_name:?} column type is non-integer; \
3746                     v5.2.2 cold tier requires IndexKey::Int (Text PK lands in v5.5+)"
3747                ))
3748            })?;
3749            to_freeze.push((pk_u64, encode_row_body_dense(row, &schema), key));
3750        }
3751        // encode_segment requires ascending u64 keys. Sort by PK
3752        // before encoding; the caller's row-position order is not
3753        // necessarily PK order (e.g. workloads that insert random
3754        // PKs).
3755        to_freeze.sort_by_key(|(k, _, _)| *k);
3756        // Reject duplicate PKs — encode_segment also rejects them
3757        // (`SegmentError::UnsortedKey`), but the resulting error
3758        // message there is misleading. Surface a clearer one.
3759        for w in to_freeze.windows(2) {
3760            if w[0].0 == w[1].0 {
3761                return Err(StorageError::Corrupt(format!(
3762                    "freeze_oldest_to_cold: duplicate PK {} in freeze batch",
3763                    w[0].0
3764                )));
3765            }
3766        }
3767        // Snapshot the (key, locator) pairs that will be registered
3768        // post-swap. Cloning the IndexKey out before the move makes
3769        // the registration loop borrow-free.
3770        let post_swap_keys: Vec<IndexKey> = to_freeze.iter().map(|(_, _, k)| k.clone()).collect();
3771        // Segment encode is now infallible w.r.t. ordering. Map the
3772        // `SegmentError` into a `StorageError::Corrupt` so the
3773        // public surface stays one error type.
3774        let seg_rows: Vec<(u64, Vec<u8>)> = to_freeze
3775            .into_iter()
3776            .map(|(k, body, _)| (k, body))
3777            .collect();
3778        let frozen_rows = seg_rows.len();
3779        let (seg_bytes, _meta) = encode_segment(seg_rows.into_iter(), 0.01, SEGMENT_PAGE_BYTES)
3780            .map_err(|e| StorageError::Corrupt(format!("freeze_oldest_to_cold: encode: {e}")))?;
3781
3782        // --- atomic swap phase: mutations only past this point ---
3783        // v5.2.3 made `Table::rebuild_indices` preserve every Cold
3784        // locator across the per-table rebuild, so `delete_rows`
3785        // below no longer wipes prior-freeze cold entries. The pre-
3786        // v5.2.3 capture-then-re-register that used to live here
3787        // was removed in v5.3.1 — keeping it would double-count
3788        // every prior-frozen key's Cold locator on each subsequent
3789        // freeze.
3790        let bytes_before = self.get(table_name).expect("just validated").hot_bytes();
3791        let positions: Vec<usize> = (0..max_rows).collect();
3792        let t_mut = self
3793            .get_mut(table_name)
3794            .expect("just validated; still present");
3795        let removed = t_mut.delete_rows(&positions);
3796        debug_assert_eq!(removed, max_rows, "delete_rows count matches request");
3797        let bytes_after = t_mut.hot_bytes();
3798        let bytes_freed = bytes_before.saturating_sub(bytes_after);
3799
3800        let segment_id = self
3801            .load_segment_bytes(seg_bytes.clone())
3802            .map_err(|e| StorageError::Corrupt(format!("freeze_oldest_to_cold: load: {e}")))?;
3803        let new_cold = post_swap_keys.into_iter().map(|k| {
3804            (
3805                k,
3806                RowLocator::Cold {
3807                    segment_id,
3808                    page_offset: 0,
3809                },
3810            )
3811        });
3812        let t_mut = self.get_mut(table_name).expect("still present");
3813        t_mut.register_cold_locators(index_name, new_cold)?;
3814
3815        Ok(FreezeReport {
3816            segment_id,
3817            frozen_rows,
3818            bytes_freed,
3819            segment_bytes: seg_bytes,
3820        })
3821    }
3822
3823    /// v5.1: borrow the cold segment at `segment_id`. Used by the
3824    /// spg-server preload path to enumerate (key, locator) pairs
3825    /// after loading a segment, so it can call
3826    /// [`Table::register_cold_locators`] without re-parsing the
3827    /// bytes.
3828    #[must_use]
3829    pub fn cold_segment(&self, segment_id: u32) -> Option<&OwnedSegment> {
3830        self.cold_segments
3831            .get(segment_id as usize)
3832            .and_then(|s| s.as_deref())
3833    }
3834
3835    /// v5.1: resolve a single `RowLocator::Cold` to its underlying
3836    /// `Row`. Decoupled from [`Catalog::lookup_by_pk`] so callers
3837    /// iterating a multi-locator slice (e.g. the engine's index
3838    /// seek path) can dispatch per locator instead of getting back
3839    /// only the first row for a key. Returns `None` when the
3840    /// segment isn't registered, the key isn't `u64`-coercible, or
3841    /// the segment doesn't actually carry the key (bloom or page-
3842    /// index reject).
3843    pub fn resolve_cold_locator(
3844        &self,
3845        table_name: &str,
3846        segment_id: u32,
3847        key: &IndexKey,
3848    ) -> Option<Row> {
3849        let t = self.get(table_name)?;
3850        let u64_key = index_key_as_u64(key)?;
3851        let seg = self.cold_segments.get(segment_id as usize)?.as_ref()?;
3852        let payload = seg.lookup(u64_key)?;
3853        let (row, _) = decode_row_body_dense(&payload, &t.schema).ok()?;
3854        Some(row)
3855    }
3856
3857    /// v5.1: indexed PK lookup that dispatches per locator,
3858    /// returning the first matching row from either the hot tier
3859    /// (`Table::rows`) or a registered cold segment.
3860    ///
3861    /// The cold path requires the index column to be coercible to
3862    /// a `u64` (the segment's PK type) and the segment payload to
3863    /// be a [`encode_row_body_dense`]-encoded row body for the
3864    /// same schema. v5.1 ships this for BIGINT / INT / SMALLINT
3865    /// PKs; other types fall through to hot-only behavior.
3866    ///
3867    /// Returns `None` if (a) the table or index doesn't exist,
3868    /// (b) the key isn't in the index at all, or (c) the key was
3869    /// resolved to a stale locator (Hot index out of range, Cold
3870    /// segment id unknown, segment lookup miss). Does not surface
3871    /// segment-decode errors — those would indicate corrupted
3872    /// cold-tier files and should be caught at
3873    /// [`Catalog::load_segment_bytes`] time.
3874    pub fn lookup_by_pk(&self, table: &str, index_name: &str, key: &IndexKey) -> Option<Row> {
3875        let t = self.get(table)?;
3876        let idx = t.indices.iter().find(|i| i.name == index_name)?;
3877        let locators = idx.lookup_eq(key);
3878        let cold_u64_key = index_key_as_u64(key);
3879        for loc in locators {
3880            match *loc {
3881                RowLocator::Hot(i) => {
3882                    if let Some(row) = t.rows.get(i) {
3883                        return Some(row.clone());
3884                    }
3885                }
3886                RowLocator::Cold {
3887                    segment_id,
3888                    page_offset: _,
3889                } => {
3890                    let Some(u64_key) = cold_u64_key else {
3891                        // Key type not coercible to u64 — cold tier
3892                        // only handles BIGINT/INT/SMALLINT in v5.1.
3893                        continue;
3894                    };
3895                    let Some(seg) = self
3896                        .cold_segments
3897                        .get(segment_id as usize)
3898                        .and_then(|s| s.as_deref())
3899                    else {
3900                        // v6.7.3 — `None` slot = compaction
3901                        // retired this segment; the live locator
3902                        // on a freshly-compacted index points to
3903                        // the merged segment_id, so a Cold hit
3904                        // here against a tombstone means the BTree
3905                        // entry hasn't been swapped yet (mid-
3906                        // compaction reader race) or the caller is
3907                        // looking up a stale snapshot. Skip — the
3908                        // next locator in the list, if any, is
3909                        // typically the merged segment.
3910                        continue;
3911                    };
3912                    let Some(payload) = seg.lookup(u64_key) else {
3913                        continue;
3914                    };
3915                    let (row, _) = decode_row_body_dense(&payload, &t.schema).ok()?;
3916                    return Some(row);
3917                }
3918            }
3919        }
3920        None
3921    }
3922
3923    /// v5.2.3: promote a frozen row back to the hot tier so an
3924    /// UPDATE / DELETE can mutate it. Reads the cold-tier row body
3925    /// (decoded from its registered segment), pushes it into
3926    /// `table.rows` via [`Table::insert`] (which also adds a fresh
3927    /// `Hot(new_idx)` locator on `index_name`), then retires the
3928    /// shadowed `Cold` locator via
3929    /// [`Table::remove_cold_locators_for_key`]. The cold-tier row
3930    /// in the segment file becomes garbage — recoverable when a
3931    /// future cold-segment compaction job lands.
3932    ///
3933    /// Returns:
3934    /// - `Ok(Some(new_hot_idx))` when the key resolved through a
3935    ///   cold locator and the promote completed. `new_hot_idx` is
3936    ///   the position the row now occupies in `table.rows`.
3937    /// - `Ok(None)` when the key has no Cold locator on the index
3938    ///   (already hot, or wasn't present at all). Callers treat this
3939    ///   as "nothing to do here, fall back to the hot-only path".
3940    ///
3941    /// Errors when the table / index doesn't exist, the index isn't
3942    /// `BTree`, the cold segment is missing / can't decode the row,
3943    /// or the inferred row body fails `Table::insert` validation.
3944    pub fn promote_cold_row(
3945        &mut self,
3946        table_name: &str,
3947        index_name: &str,
3948        key: &IndexKey,
3949    ) -> Result<Option<usize>, StorageError> {
3950        let cold_loc = self.find_cold_locator(table_name, index_name, key)?;
3951        let Some((segment_id, _page_offset)) = cold_loc else {
3952            return Ok(None);
3953        };
3954        let u64_key = index_key_as_u64(key).ok_or_else(|| {
3955            StorageError::Corrupt(
3956                "promote_cold_row: key type not coercible to u64 (cold tier requires integer PK)"
3957                    .into(),
3958            )
3959        })?;
3960        // Read the row body from the segment. Borrow the segment +
3961        // schema short-term so we can then take `&mut self` for the
3962        // hot-side insert.
3963        let schema = self
3964            .get(table_name)
3965            .ok_or_else(|| {
3966                StorageError::Corrupt(format!("promote_cold_row: table {table_name:?} not found"))
3967            })?
3968            .schema
3969            .clone();
3970        let seg = self
3971            .cold_segments
3972            .get(segment_id as usize)
3973            .and_then(|s| s.as_ref())
3974            .ok_or_else(|| {
3975                StorageError::Corrupt(format!(
3976                    "promote_cold_row: segment {segment_id} not registered on catalog"
3977                ))
3978            })?;
3979        let payload = seg.lookup(u64_key).ok_or_else(|| {
3980            StorageError::Corrupt(format!(
3981                "promote_cold_row: key {u64_key} resolves to segment {segment_id} \
3982                 but the segment's bloom/page lookup didn't return a row"
3983            ))
3984        })?;
3985        let (row, _consumed) = decode_row_body_dense(&payload, &schema)?;
3986        // Insert the promoted row into the hot tier. `Table::insert`
3987        // appends to `self.rows`, adds a `Hot(new_idx)` locator to
3988        // every BTree index covering the row's keyed columns, and
3989        // increments `hot_bytes`.
3990        let t = self
3991            .get_mut(table_name)
3992            .expect("table existed at lookup time");
3993        t.insert(row)?;
3994        let new_hot_idx =
3995            t.rows.len().checked_sub(1).ok_or_else(|| {
3996                StorageError::Corrupt("promote_cold_row: empty after insert".into())
3997            })?;
3998        // The hot insert added Hot(new_idx) alongside the still-
3999        // present Cold locator. Drop the Cold entry so future
4000        // lookups return only the fresh hot row.
4001        t.remove_cold_locators_for_key(index_name, key)?;
4002        Ok(Some(new_hot_idx))
4003    }
4004
4005    /// v5.2.3: shadow a frozen row's index entry. Used by DELETE
4006    /// when the row to remove lives in a cold-tier segment — the
4007    /// row body stays in the segment file (becoming garbage) but
4008    /// every `Cold` locator for `key` on `index_name` is removed
4009    /// so PK lookups stop returning it.
4010    ///
4011    /// Returns the number of cold locators retired (0 when the key
4012    /// has no cold entries — the DELETE fell on a hot row or a
4013    /// key that was already absent). Errors when the table /
4014    /// index doesn't exist or the index isn't `BTree`.
4015    ///
4016    /// Cold-segment compaction (which merges shadowed-heavy
4017    /// segments and reclaims their disk footprint) lands in a
4018    /// later v5.x sub-version; until then, repeated UPDATE/DELETE
4019    /// of cold rows can amplify cold-segment disk usage by up to
4020    /// 1-2× — still well under typical LSM-tree shadowing because
4021    /// SPG segments are bulk-baked, not write-merged.
4022    pub fn shadow_cold_row(
4023        &mut self,
4024        table_name: &str,
4025        index_name: &str,
4026        key: &IndexKey,
4027    ) -> Result<usize, StorageError> {
4028        let t = self.get_mut(table_name).ok_or_else(|| {
4029            StorageError::Corrupt(format!("shadow_cold_row: table {table_name:?} not found"))
4030        })?;
4031        t.remove_cold_locators_for_key(index_name, key)
4032    }
4033
4034    /// v6.7.4 — read-only slice preparation for the parallel
4035    /// freezer. Walks rows in `row_range`, builds the
4036    /// `(pk_u64, encoded_body, IndexKey)` triples that the
4037    /// coordinator's k-way merge consumes, sorts the slice by
4038    /// `pk_u64`, and returns a [`FreezeSlice`].
4039    ///
4040    /// Caller invariants:
4041    /// - `row_range.end <= table.rows.len()` (caller's job to
4042    ///   compute the partition).
4043    /// - All slices passed to `commit_freeze_slices` must cover a
4044    ///   contiguous half-open range `[0, total_max_rows)` with no
4045    ///   gaps and no overlaps. The coordinator validates this
4046    ///   invariant before committing.
4047    ///
4048    /// `&self`-only — multiple workers can run this concurrently
4049    /// against the same `Catalog` reference under the engine's
4050    /// write lock (workers don't mutate; the coordinator does).
4051    pub fn prepare_freeze_slice(
4052        &self,
4053        table_name: &str,
4054        index_name: &str,
4055        row_range: core::ops::Range<usize>,
4056    ) -> Result<FreezeSlice, StorageError> {
4057        let table = self.get(table_name).ok_or_else(|| {
4058            StorageError::Corrupt(format!(
4059                "prepare_freeze_slice: table {table_name:?} not found"
4060            ))
4061        })?;
4062        let idx = table
4063            .indices
4064            .iter()
4065            .find(|i| i.name == index_name)
4066            .ok_or_else(|| {
4067                StorageError::Corrupt(format!(
4068                    "prepare_freeze_slice: index {index_name:?} not found on {table_name:?}"
4069                ))
4070            })?;
4071        if !matches!(idx.kind, IndexKind::BTree(_)) {
4072            return Err(StorageError::Corrupt(format!(
4073                "prepare_freeze_slice: index {index_name:?} is NSW; only BTree indices may freeze"
4074            )));
4075        }
4076        if row_range.end > table.rows.len() {
4077            return Err(StorageError::Corrupt(format!(
4078                "prepare_freeze_slice: row_range end {} > row_count {}",
4079                row_range.end,
4080                table.rows.len()
4081            )));
4082        }
4083        let column_position = idx.column_position;
4084        let schema = table.schema.clone();
4085        let mut rows: Vec<(u64, Vec<u8>, IndexKey)> = Vec::with_capacity(row_range.len());
4086        for row_idx in row_range.clone() {
4087            let row = table.rows.get(row_idx).expect("bounds-checked above");
4088            let key = IndexKey::from_value(&row.values[column_position]).ok_or_else(|| {
4089                StorageError::Corrupt(format!(
4090                    "prepare_freeze_slice: row {row_idx} has NULL / non-key value in index column"
4091                ))
4092            })?;
4093            let pk_u64 = index_key_as_u64(&key).ok_or_else(|| {
4094                StorageError::Corrupt(format!(
4095                    "prepare_freeze_slice: index {index_name:?} column type is non-integer; \
4096                     v5.2.2 cold tier requires IndexKey::Int (Text PK lands in v5.5+)"
4097                ))
4098            })?;
4099            rows.push((pk_u64, encode_row_body_dense(row, &schema), key));
4100        }
4101        rows.sort_by_key(|(k, _, _)| *k);
4102        Ok(FreezeSlice { row_range, rows })
4103    }
4104
4105    /// v6.7.4 — coordinator commit step. Merges N
4106    /// [`FreezeSlice`]s into one segment via the standard
4107    /// [`encode_segment`] path, atomically swaps the catalog
4108    /// state (delete the union row range + register Cold
4109    /// locators + load the segment).
4110    ///
4111    /// Validates that the slices cover a contiguous, gap-free,
4112    /// overlap-free half-open range starting at index 0 (the
4113    /// freezer always freezes "oldest first" — same semantics as
4114    /// the single-threaded [`Catalog::freeze_oldest_to_cold`]).
4115    ///
4116    /// Empty `slices` → no-op success (returns a zero-row report
4117    /// without mutating). Total row count = `Σ slice.rows.len()`.
4118    pub fn commit_freeze_slices(
4119        &mut self,
4120        table_name: &str,
4121        index_name: &str,
4122        slices: Vec<FreezeSlice>,
4123    ) -> Result<FreezeReport, StorageError> {
4124        // --- validation phase: never mutates ---------------------
4125        let table = self.get(table_name).ok_or_else(|| {
4126            StorageError::Corrupt(format!(
4127                "commit_freeze_slices: table {table_name:?} not found"
4128            ))
4129        })?;
4130        let idx = table
4131            .indices
4132            .iter()
4133            .find(|i| i.name == index_name)
4134            .ok_or_else(|| {
4135                StorageError::Corrupt(format!(
4136                    "commit_freeze_slices: index {index_name:?} not found on {table_name:?}"
4137                ))
4138            })?;
4139        if !matches!(idx.kind, IndexKind::BTree(_)) {
4140            return Err(StorageError::Corrupt(format!(
4141                "commit_freeze_slices: index {index_name:?} is NSW; only BTree indices may freeze"
4142            )));
4143        }
4144        // Validate slice coverage: contiguous from 0, no gaps, no
4145        // overlaps. Allow the caller to pass slices in any order —
4146        // sort by row_range.start first.
4147        let mut ordered = slices;
4148        ordered.sort_by_key(|s| s.row_range.start);
4149        // Drop fully-empty slices that fell out of an uneven
4150        // partition; they carry no data but contribute to the
4151        // contiguity check, so keep them in line.
4152        let mut expected_start = 0usize;
4153        for s in &ordered {
4154            if s.row_range.start != expected_start {
4155                return Err(StorageError::Corrupt(format!(
4156                    "commit_freeze_slices: gap/overlap at row {}; expected start {}",
4157                    s.row_range.start, expected_start
4158                )));
4159            }
4160            expected_start = s.row_range.end;
4161        }
4162        let max_rows = expected_start;
4163        if max_rows > table.rows.len() {
4164            return Err(StorageError::Corrupt(format!(
4165                "commit_freeze_slices: total row range {} exceeds row_count {}",
4166                max_rows,
4167                table.rows.len()
4168            )));
4169        }
4170        if max_rows == 0 {
4171            return Ok(FreezeReport {
4172                segment_id: u32::MAX,
4173                frozen_rows: 0,
4174                bytes_freed: 0,
4175                segment_bytes: Vec::new(),
4176            });
4177        }
4178
4179        // --- segment build phase: reads only --------------------
4180        // K-way merge of already-sorted slices. Each slice's rows
4181        // are ascending by pk_u64; we keep a per-slice cursor and
4182        // pull the next-smallest head until every cursor drains.
4183        let total_rows: usize = ordered.iter().map(|s| s.rows.len()).sum();
4184        if total_rows != max_rows {
4185            return Err(StorageError::Corrupt(format!(
4186                "commit_freeze_slices: total slice rows {total_rows} ≠ row_range coverage {max_rows}"
4187            )));
4188        }
4189        let mut cursors: Vec<usize> = alloc::vec![0; ordered.len()];
4190        let mut merged: Vec<(u64, Vec<u8>, IndexKey)> = Vec::with_capacity(total_rows);
4191        loop {
4192            // Pick the slice whose head row has the smallest key
4193            // and isn't yet exhausted.
4194            let mut pick: Option<usize> = None;
4195            for (i, c) in cursors.iter().enumerate() {
4196                let slice = &ordered[i];
4197                if *c >= slice.rows.len() {
4198                    continue;
4199                }
4200                match pick {
4201                    None => pick = Some(i),
4202                    Some(j) => {
4203                        if slice.rows[*c].0 < ordered[j].rows[cursors[j]].0 {
4204                            pick = Some(i);
4205                        }
4206                    }
4207                }
4208            }
4209            let Some(i) = pick else { break };
4210            let row = ordered[i].rows[cursors[i]].clone();
4211            cursors[i] += 1;
4212            merged.push(row);
4213        }
4214        // Reject duplicate PKs — same error as the single-threaded
4215        // path so callers get a uniform surface.
4216        for w in merged.windows(2) {
4217            if w[0].0 == w[1].0 {
4218                return Err(StorageError::Corrupt(format!(
4219                    "commit_freeze_slices: duplicate PK {} across slices",
4220                    w[0].0
4221                )));
4222            }
4223        }
4224        let post_swap_keys: Vec<IndexKey> = merged.iter().map(|(_, _, k)| k.clone()).collect();
4225        let seg_rows: Vec<(u64, Vec<u8>)> =
4226            merged.into_iter().map(|(k, body, _)| (k, body)).collect();
4227        let frozen_rows = seg_rows.len();
4228        let (seg_bytes, _meta) = encode_segment(seg_rows.into_iter(), 0.01, SEGMENT_PAGE_BYTES)
4229            .map_err(|e| StorageError::Corrupt(format!("commit_freeze_slices: encode: {e}")))?;
4230
4231        // --- atomic swap phase: mutations only past this point ---
4232        let bytes_before = self.get(table_name).expect("just validated").hot_bytes();
4233        let positions: Vec<usize> = (0..max_rows).collect();
4234        let t_mut = self
4235            .get_mut(table_name)
4236            .expect("just validated; still present");
4237        let removed = t_mut.delete_rows(&positions);
4238        debug_assert_eq!(removed, max_rows, "delete_rows count matches request");
4239        let bytes_after = t_mut.hot_bytes();
4240        let bytes_freed = bytes_before.saturating_sub(bytes_after);
4241
4242        let segment_id = self
4243            .load_segment_bytes(seg_bytes.clone())
4244            .map_err(|e| StorageError::Corrupt(format!("commit_freeze_slices: load: {e}")))?;
4245        let new_cold = post_swap_keys.into_iter().map(|k| {
4246            (
4247                k,
4248                RowLocator::Cold {
4249                    segment_id,
4250                    page_offset: 0,
4251                },
4252            )
4253        });
4254        let t_mut = self.get_mut(table_name).expect("still present");
4255        t_mut.register_cold_locators(index_name, new_cold)?;
4256
4257        Ok(FreezeReport {
4258            segment_id,
4259            frozen_rows,
4260            bytes_freed,
4261            segment_bytes: seg_bytes,
4262        })
4263    }
4264
4265    /// v6.7.3 — compact every cold segment on `(table, index)` whose
4266    /// `OwnedSegment::bytes().len()` is below `target_segment_bytes`
4267    /// into a single larger merged segment. Rows present in source
4268    /// segment payloads but no longer referenced by any
4269    /// `RowLocator::Cold` on the index (DELETE'd + frozen rows
4270    /// retired via [`Catalog::shadow_cold_row`]) are GC'd in the
4271    /// merge.
4272    ///
4273    /// **Semantics**:
4274    /// 1. Walk the BTree index to collect every Cold locator that
4275    ///    targets a small (< threshold) segment. Each such
4276    ///    `(key, segment_id)` becomes a row in the merged segment;
4277    ///    payload is looked up from the source segment in-place.
4278    /// 2. Encode the collected rows into one new segment via
4279    ///    [`encode_segment`]; register it via
4280    ///    [`Catalog::load_segment_bytes`] (allocating a fresh
4281    ///    `merged_segment_id` at the end of `cold_segments`).
4282    /// 3. Rewrite the BTree index in one pass: every
4283    ///    `RowLocator::Cold { segment_id ∈ sources }` becomes
4284    ///    `RowLocator::Cold { segment_id = merged_id, page_offset = 0 }`.
4285    ///    Hot locators are untouched.
4286    /// 4. Tombstone every source slot via
4287    ///    [`Catalog::tombstone_segment`]. Source segment payloads
4288    ///    are no longer reachable through the catalog; the on-disk
4289    ///    files are the caller's concern.
4290    ///
4291    /// On fewer than 2 candidate segments the catalog is **not**
4292    /// mutated and a no-op report (`merged_segment_id: None`,
4293    /// `sources: []`) is returned. This is the routine case — a
4294    /// freshly-frozen table has at most 1 small segment, no merge
4295    /// possible.
4296    ///
4297    /// Atomicity: every mutating step runs after the read-only
4298    /// gather phase, so a panic before the merge encode leaves the
4299    /// catalog unchanged. The mutation block itself (load + rewrite +
4300    /// tombstone) takes only `&mut self` — callers serialise the
4301    /// engine write lock outside this function.
4302    ///
4303    /// Errors when the table / index doesn't exist, the index isn't
4304    /// `BTree`, the index column type isn't u64-coercible (cold-tier
4305    /// pre-condition), or a source segment fails its in-place
4306    /// row-body lookup (would indicate prior catalog corruption).
4307    pub fn compact_cold_segments(
4308        &mut self,
4309        table_name: &str,
4310        index_name: &str,
4311        target_segment_bytes: u64,
4312    ) -> Result<CompactReport, StorageError> {
4313        // --- validation phase ----------------------------------
4314        let t = self.get(table_name).ok_or_else(|| {
4315            StorageError::Corrupt(format!(
4316                "compact_cold_segments: table {table_name:?} not found"
4317            ))
4318        })?;
4319        let idx = t
4320            .indices
4321            .iter()
4322            .find(|i| i.name == index_name)
4323            .ok_or_else(|| {
4324                StorageError::Corrupt(format!(
4325                    "compact_cold_segments: index {index_name:?} not found on {table_name:?}"
4326                ))
4327            })?;
4328        let map = match &idx.kind {
4329            IndexKind::BTree(m) => m,
4330            IndexKind::Nsw(_) | IndexKind::Brin { .. } | IndexKind::Gin(_) | IndexKind::GinTrgm(_) => {
4331                return Err(StorageError::Corrupt(format!(
4332                    "compact_cold_segments: index {index_name:?} is not BTree; \
4333                     compaction applies only to BTree cold-tier indices"
4334                )));
4335            }
4336        };
4337
4338        // --- gather phase --------------------------------------
4339        // Step A: every segment_id this BTree index Cold-references.
4340        let mut referenced_ids: BTreeSet<u32> = BTreeSet::new();
4341        for (_key, locators) in map.iter() {
4342            for loc in locators {
4343                if let RowLocator::Cold { segment_id, .. } = loc {
4344                    referenced_ids.insert(*segment_id);
4345                }
4346            }
4347        }
4348        // Step B: keep only the small + still-active ones.
4349        let candidate_set: BTreeSet<u32> = referenced_ids
4350            .into_iter()
4351            .filter(|id| {
4352                self.cold_segments
4353                    .get(*id as usize)
4354                    .and_then(|s| s.as_deref())
4355                    .is_some_and(|s| (s.bytes().len() as u64) < target_segment_bytes)
4356            })
4357            .collect();
4358        if candidate_set.len() < 2 {
4359            return Ok(CompactReport {
4360                sources: Vec::new(),
4361                merged_segment_id: None,
4362                merged_segment_bytes: Vec::new(),
4363                merged_rows: 0,
4364                deleted_rows_pruned: 0,
4365                bytes_reclaimed_estimate: 0,
4366            });
4367        }
4368        // Step C: pre-count source rows for the deleted-pruned metric.
4369        let mut source_row_count: usize = 0;
4370        let mut source_byte_total: u64 = 0;
4371        for &id in &candidate_set {
4372            let seg = self.cold_segments[id as usize]
4373                .as_ref()
4374                .expect("candidate selected only when slot is Some");
4375            source_row_count = source_row_count.saturating_add(seg.meta().num_rows as usize);
4376            source_byte_total = source_byte_total.saturating_add(seg.bytes().len() as u64);
4377        }
4378        // Step D: collect (key, body) pairs from every live Cold
4379        // locator pointing at a candidate. dedupe by key — one
4380        // BTree key resolves to at most one cold payload (the
4381        // freezer + promote/shadow flow keeps Cold locators
4382        // unique per key).
4383        let mut collected: BTreeMap<u64, (Vec<u8>, IndexKey)> = BTreeMap::new();
4384        for (key, locators) in map.iter() {
4385            for loc in locators {
4386                let RowLocator::Cold { segment_id, .. } = loc else {
4387                    continue;
4388                };
4389                if !candidate_set.contains(segment_id) {
4390                    continue;
4391                }
4392                let u64_key = index_key_as_u64(key).ok_or_else(|| {
4393                    StorageError::Corrupt(format!(
4394                        "compact_cold_segments: index {index_name:?} has non-integer Cold key; \
4395                         cold tier requires IndexKey::Int (Text PK lands in v5.5+)"
4396                    ))
4397                })?;
4398                let seg = self.cold_segments[*segment_id as usize]
4399                    .as_ref()
4400                    .expect("candidate slot guaranteed Some above");
4401                let payload = seg.lookup(u64_key).ok_or_else(|| {
4402                    StorageError::Corrupt(format!(
4403                        "compact_cold_segments: BTree {index_name:?} points key={u64_key} \
4404                         at segment {segment_id} but the segment lookup missed"
4405                    ))
4406                })?;
4407                collected.insert(u64_key, (payload, key.clone()));
4408                break;
4409            }
4410        }
4411        let merged_rows = collected.len();
4412        let deleted_rows_pruned = source_row_count.saturating_sub(merged_rows);
4413
4414        // Step E: encode the merged segment. `BTreeMap<u64, _>`
4415        // iteration is ascending by key, which is what
4416        // `encode_segment` requires.
4417        let seg_rows: Vec<(u64, Vec<u8>)> = collected
4418            .iter()
4419            .map(|(k, (body, _))| (*k, body.clone()))
4420            .collect();
4421        let (seg_bytes, _meta) = encode_segment(seg_rows.into_iter(), 0.01, SEGMENT_PAGE_BYTES)
4422            .map_err(|e| StorageError::Corrupt(format!("compact_cold_segments: encode: {e}")))?;
4423        let merged_bytes_len = seg_bytes.len() as u64;
4424
4425        // --- atomic mutation phase ------------------------------
4426        let merged_segment_id = self
4427            .load_segment_bytes(seg_bytes.clone())
4428            .map_err(|e| StorageError::Corrupt(format!("compact_cold_segments: load: {e}")))?;
4429
4430        // Rewrite the BTree index: every Cold locator pointing at
4431        // a candidate source becomes a Cold locator pointing at
4432        // the merged segment. Use a flat collect-then-replace
4433        // pattern so we never hold a `&self` borrow across the
4434        // `&mut self` write.
4435        let entries: Vec<(IndexKey, Vec<RowLocator>)> = {
4436            let t = self
4437                .get(table_name)
4438                .expect("table existed at the start of this fn");
4439            let idx = t
4440                .indices
4441                .iter()
4442                .find(|i| i.name == index_name)
4443                .expect("index existed at the start of this fn");
4444            let IndexKind::BTree(map) = &idx.kind else {
4445                unreachable!("validated above");
4446            };
4447            map.iter().map(|(k, v)| (k.clone(), v.clone())).collect()
4448        };
4449        let t_mut = self
4450            .get_mut(table_name)
4451            .expect("table existed at the start of this fn");
4452        let idx_mut = t_mut
4453            .indices
4454            .iter_mut()
4455            .find(|i| i.name == index_name)
4456            .expect("index existed at the start of this fn");
4457        let IndexKind::BTree(map_mut) = &mut idx_mut.kind else {
4458            unreachable!("validated above");
4459        };
4460        for (key, locators) in entries {
4461            let mut new_locs: Vec<RowLocator> = Vec::with_capacity(locators.len());
4462            let mut changed = false;
4463            for loc in &locators {
4464                match *loc {
4465                    RowLocator::Cold {
4466                        segment_id,
4467                        page_offset: _,
4468                    } if candidate_set.contains(&segment_id) => {
4469                        let replacement = RowLocator::Cold {
4470                            segment_id: merged_segment_id,
4471                            page_offset: 0,
4472                        };
4473                        if !new_locs.contains(&replacement) {
4474                            new_locs.push(replacement);
4475                        }
4476                        changed = true;
4477                    }
4478                    other => new_locs.push(other),
4479                }
4480            }
4481            if changed {
4482                map_mut.insert_mut(key, new_locs);
4483            }
4484        }
4485
4486        // Tombstone every source slot. Last step — failures here
4487        // would leave the segment double-referenced in both
4488        // memory + manifest, but `tombstone_segment` only errors
4489        // on out-of-bounds, which we've already validated.
4490        for &id in &candidate_set {
4491            self.tombstone_segment(id)?;
4492        }
4493
4494        let bytes_reclaimed_estimate = source_byte_total.saturating_sub(merged_bytes_len);
4495        Ok(CompactReport {
4496            sources: candidate_set.into_iter().collect(),
4497            merged_segment_id: Some(merged_segment_id),
4498            merged_segment_bytes: seg_bytes,
4499            merged_rows,
4500            deleted_rows_pruned,
4501            bytes_reclaimed_estimate,
4502        })
4503    }
4504
4505    /// Internal helper: scan `(table, index)` for a `Cold` locator
4506    /// keyed by `key`. Returns `Ok(Some((segment_id, page_offset)))`
4507    /// when found, `Ok(None)` when the key has only hot entries
4508    /// or no entries at all, `Err` on the same input-validation
4509    /// errors as the public `promote_cold_row` / `shadow_cold_row`.
4510    fn find_cold_locator(
4511        &self,
4512        table_name: &str,
4513        index_name: &str,
4514        key: &IndexKey,
4515    ) -> Result<Option<(u32, u32)>, StorageError> {
4516        let t = self.get(table_name).ok_or_else(|| {
4517            StorageError::Corrupt(format!("find_cold_locator: table {table_name:?} not found"))
4518        })?;
4519        let idx = t
4520            .indices
4521            .iter()
4522            .find(|i| i.name == index_name)
4523            .ok_or_else(|| {
4524                StorageError::Corrupt(format!(
4525                    "find_cold_locator: index {index_name:?} not found on {table_name:?}"
4526                ))
4527            })?;
4528        if !matches!(idx.kind, IndexKind::BTree(_)) {
4529            return Err(StorageError::Corrupt(format!(
4530                "find_cold_locator: index {index_name:?} is NSW; promote-on-write only applies to BTree indices"
4531            )));
4532        }
4533        for loc in idx.lookup_eq(key) {
4534            if let RowLocator::Cold {
4535                segment_id,
4536                page_offset,
4537            } = *loc
4538            {
4539                return Ok(Some((segment_id, page_offset)));
4540            }
4541        }
4542        Ok(None)
4543    }
4544}
4545
4546/// Coerce an [`IndexKey`] to the `u64` that v5.1 cold-tier
4547/// segments use as their on-disk PK. Returns `None` for keys that
4548/// aren't representable as `u64` — Text PKs need a hash mapping
4549/// the segment writer baked in (deferred to v5.2+), Bool PKs are
4550/// almost never wide enough to be sharded into a cold tier.
4551fn index_key_as_u64(key: &IndexKey) -> Option<u64> {
4552    match key {
4553        // Reinterpret the i64 bit pattern as u64. Cold-tier segments
4554        // are sorted by this u64 view, so the chosen interpretation
4555        // only has to match between insert (bake_segment / freezer)
4556        // and lookup — using cast_unsigned keeps both sides honest
4557        // and silences clippy::cast_sign_loss.
4558        IndexKey::Int(n) => Some(n.cast_unsigned()),
4559        IndexKey::Text(_) | IndexKey::Bool(_) => None,
4560    }
4561}
4562
4563#[derive(Debug, Clone, PartialEq, Eq)]
4564#[non_exhaustive]
4565pub enum StorageError {
4566    DuplicateTable {
4567        name: String,
4568    },
4569    TableNotFound {
4570        name: String,
4571    },
4572    ArityMismatch {
4573        expected: usize,
4574        actual: usize,
4575    },
4576    TypeMismatch {
4577        column: String,
4578        expected: DataType,
4579        actual: DataType,
4580        position: usize,
4581    },
4582    NullInNotNull {
4583        column: String,
4584    },
4585    /// Index with this name already exists on the table.
4586    DuplicateIndex {
4587        name: String,
4588    },
4589    /// Column referenced by an index doesn't exist on the table.
4590    ColumnNotFound {
4591        column: String,
4592    },
4593    /// On-disk format failed to parse — corrupted file, wrong magic, truncated
4594    /// payload, or unknown tag bytes.
4595    Corrupt(String),
4596    /// v6.0.4 — ALTER INDEX targeted an index name that doesn't
4597    /// exist on any table in this catalog.
4598    IndexNotFound {
4599        name: String,
4600    },
4601    /// v6.0.4 — operation requested isn't supported on this index
4602    /// kind / column type (e.g. ALTER INDEX REBUILD on a `BTree`
4603    /// index, or REBUILD WITH (encoding=…) on a non-vector column).
4604    Unsupported(String),
4605}
4606
4607impl fmt::Display for StorageError {
4608    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
4609        match self {
4610            Self::DuplicateTable { name } => write!(f, "table already exists: {name}"),
4611            Self::TableNotFound { name } => write!(f, "table not found: {name}"),
4612            Self::ArityMismatch { expected, actual } => write!(
4613                f,
4614                "row arity mismatch: expected {expected} columns, got {actual}"
4615            ),
4616            Self::TypeMismatch {
4617                column,
4618                expected,
4619                actual,
4620                position,
4621            } => write!(
4622                f,
4623                "type mismatch in column {column:?} (position {position}): expected {expected}, got {actual}"
4624            ),
4625            Self::NullInNotNull { column } => {
4626                write!(f, "NULL value in NOT NULL column {column:?}")
4627            }
4628            Self::DuplicateIndex { name } => write!(f, "index already exists: {name}"),
4629            Self::ColumnNotFound { column } => write!(f, "column not found: {column}"),
4630            Self::Corrupt(detail) => write!(f, "corrupt on-disk format: {detail}"),
4631            Self::IndexNotFound { name } => write!(f, "index not found: {name}"),
4632            Self::Unsupported(detail) => write!(f, "unsupported: {detail}"),
4633        }
4634    }
4635}
4636
4637impl ColumnSchema {
4638    pub fn new(name: impl Into<String>, ty: DataType, nullable: bool) -> Self {
4639        Self {
4640            name: name.into(),
4641            ty,
4642            nullable,
4643            default: None,
4644            runtime_default: None,
4645            auto_increment: false,
4646        }
4647    }
4648
4649    /// Builder-style helper to attach a default value to an otherwise
4650    /// plain column schema. Used by the engine when CREATE TABLE
4651    /// specifies `column TYPE DEFAULT <expr>`.
4652    #[must_use]
4653    pub fn with_default(mut self, default: Value) -> Self {
4654        self.default = Some(default);
4655        self
4656    }
4657
4658    /// v7.9.21 — builder for runtime-evaluated defaults
4659    /// (`DEFAULT now()`, `DEFAULT CURRENT_TIMESTAMP`, …).
4660    /// `expr` is the Expr's `Display` form, re-parsed by the
4661    /// engine at each INSERT.
4662    #[must_use]
4663    pub fn with_runtime_default(mut self, expr: impl Into<String>) -> Self {
4664        self.runtime_default = Some(expr.into());
4665        self
4666    }
4667
4668    /// Builder-style helper to mark a column as `AUTO_INCREMENT`.
4669    #[must_use]
4670    pub const fn with_auto_increment(mut self) -> Self {
4671        self.auto_increment = true;
4672        self
4673    }
4674}
4675
4676impl TableSchema {
4677    pub fn new(name: impl Into<String>, columns: Vec<ColumnSchema>) -> Self {
4678        Self {
4679            name: name.into(),
4680            columns,
4681            hot_tier_bytes: None,
4682            foreign_keys: Vec::new(),
4683            uniqueness_constraints: Vec::new(),
4684            checks: Vec::new(),
4685        }
4686    }
4687}
4688
4689// =========================================================================
4690// Persistent binary format for the catalog.
4691//
4692// Layout (little-endian throughout):
4693//
4694//   [magic "SPGDB001" 8 bytes][version u8]
4695//   [table_count u32]
4696//   for each table:
4697//       [name_len u16][name bytes]
4698//       [col_count u16]
4699//       for each col:
4700//           [name_len u16][name bytes]
4701//           [type_tag u8 + optional payload]
4702//               1=Int 2=BigInt 3=Float 4=Text 5=Bool
4703//               6=Vector(u32 dim)
4704//               7=SmallInt
4705//               8=Varchar(u32 max)
4706//               9=Char(u32 size)
4707//               10=Numeric(u8 precision, u8 scale)
4708//               11=Date
4709//               12=Timestamp
4710//           [nullable u8]   0/1
4711//           [default_tag u8] 0=none 1=value (followed by [value_tag u8] + bytes)
4712//       [row_count u32]
4713//       for each row, for each col, one [value_tag u8] + value bytes:
4714//           tag 0 (Null)     → no body
4715//           tag 1 (Int)      → i32 LE
4716//           tag 2 (BigInt)   → i64 LE
4717//           tag 3 (Float)    → f64 LE
4718//           tag 4 (Text)     → u16 LE len + UTF-8 bytes
4719//           tag 5 (Bool)     → u8 0/1
4720//           tag 6 (Vector)   → u32 LE dim + dim×f32 LE
4721//           tag 7 (SmallInt) → i16 LE
4722//           tag 8 (Numeric)  → i128 LE (16 bytes) + u8 scale
4723//           tag 9 (Date)     → i32 LE (days since Unix epoch)
4724//           tag 10 (Timestamp) → i64 LE (microseconds since Unix epoch)
4725//
4726// Bumped to version 3 when NUMERIC was added; to version 4 when
4727// AUTO_INCREMENT (per-column flag) + NSW index `kind` byte landed;
4728// to version 5 when DATE / TIMESTAMP were added; to version 6 when
4729// NSW graph topology started travelling on disk (v2.7); to version 7
4730// when the NSW topology became multi-layer HNSW (v2.13); to version 8
4731// when row encoding switched to schema-driven dense layout (v3.0.2 —
4732// per-row NULL bitmap + per-column fixed-width body, no per-cell type
4733// tag).
4734// =========================================================================
4735
4736const FILE_MAGIC: &[u8; 8] = b"SPGDB001";
4737/// Current catalog snapshot format version emitted by [`Catalog::serialize`].
4738///
4739/// v9 (v5.2) extends v8 by serialising `BTree` index entries directly — every
4740/// `(IndexKey, Vec<RowLocator>)` pair travels on disk with the v5.1
4741/// `RowLocator::write_le` tag-prefixed codec. v8 `BTree` indices stored no
4742/// entries at all (the map was rebuilt from `Table::rows` on load); v9
4743/// preserves on-disk Cold locators so freezer-produced cold-tier index
4744/// entries survive a catalog snapshot round-trip. v8 readers are accepted
4745/// by version dispatch in [`Catalog::deserialize`] — every entry decodes
4746/// as `RowLocator::Hot(_)` via `add_index` rebuild, identical to v5.1
4747/// behaviour.
4748/// v6.7.2 — bumped from 10 to 11 to append per-table
4749/// `hot_tier_bytes: Option<u64>` after the per-table indices
4750/// section. v10 catalogs (v6.7.1) load with `hot_tier_bytes =
4751/// None` for every table (the deserialiser short-circuits when
4752/// version < 11). v11 snapshots written by a pre-v6.7.2 binary
4753/// fail loudly at the version check, matching the v6.1.2 /
4754/// v6.1.4 / v6.2.0 / v6.7.1 envelope-bump upgrade fences.
4755///
4756/// v6.8.0 — bumped from 11 to 12: per-index
4757/// `included_columns: Vec<u16>` appended at the tail of each
4758/// index payload. v11 (= v6.7.2) catalogs load with
4759/// `included_columns = Vec::new()` for every index — same
4760/// "older readers, append-only extension" pattern as the v6.7.2
4761/// hot_tier_bytes byte.
4762/// v7.13.0 — bumped from 22 to 23. mailrs round-5 G3 / G10.
4763/// Per-table appendix gains two new sections:
4764///   * `checks: Vec<String>` — CHECK predicate sources (Display
4765///     form of the AST Expr); re-parsed on INSERT/UPDATE to
4766///     enforce against candidate rows. Same persistence pattern
4767///     as `Index::partial_predicate`.
4768///   * Per `UniquenessConstraint`: trailing `nulls_not_distinct:
4769///     u8` flag for PG 15+ `UNIQUE NULLS NOT DISTINCT (cols)`
4770///     semantics.
4771/// v22 catalogs deserialise with empty `checks` and every UC
4772/// at `nulls_not_distinct = false`.
4773/// v24 introduces:
4774///   * Index kind tag 4 = trigram-GIN (`gin_trgm_ops`-flavoured
4775///     `USING gin` over a TEXT/VARCHAR column). Payload shape is
4776///     identical to tag-3 GIN (String → Vec<RowLocator>); the
4777///     keys are PG-compatible 3-byte trigram shingles instead of
4778///     tsvector lexemes. v23 catalogs deserialise unchanged — no
4779///     v23 writer ever emitted tag 4.
4780/// v25 introduces:
4781///   * Per `TriggerDef`: trailing `enabled: u8` flag (mailrs
4782///     round-9 A.2.b — `ALTER TABLE … { ENABLE | DISABLE }
4783///     TRIGGER …`). v24 catalogs deserialise with every trigger
4784///     `enabled = true`, matching pre-v7.16.1 behaviour.
4785const FILE_VERSION: u8 = 25;
4786/// Oldest format version [`Catalog::deserialize`] still accepts. v8 is the
4787/// v3.0.2 dense-row layout; pre-v8 catalogs require an offline migration.
4788const MIN_SUPPORTED_FILE_VERSION: u8 = 8;
4789
4790// IndexKey wire format (v9):
4791//   tag 0 = Int  → [i64 LE]
4792//   tag 1 = Text → [u16 LE len + UTF-8 bytes] (via write_str / read_str)
4793//   tag 2 = Bool → [u8 0/1]
4794const INDEX_KEY_TAG_INT: u8 = 0;
4795const INDEX_KEY_TAG_TEXT: u8 = 1;
4796const INDEX_KEY_TAG_BOOL: u8 = 2;
4797
4798impl Catalog {
4799    /// Serialize the whole catalog (schema + every row) into a self-contained
4800    /// byte buffer. Format is documented above the impl block.
4801    pub fn serialize(&self) -> Vec<u8> {
4802        let mut out = Vec::with_capacity(64);
4803        out.extend_from_slice(FILE_MAGIC);
4804        out.push(FILE_VERSION);
4805        write_u32(
4806            &mut out,
4807            u32::try_from(self.tables.len()).expect("≤ 4G tables"),
4808        );
4809        for t in &self.tables {
4810            write_str(&mut out, &t.schema.name);
4811            write_u16(
4812                &mut out,
4813                u16::try_from(t.schema.columns.len()).expect("≤ 65k columns/table"),
4814            );
4815            for c in &t.schema.columns {
4816                write_str(&mut out, &c.name);
4817                write_data_type(&mut out, c.ty);
4818                out.push(u8::from(c.nullable));
4819                match &c.default {
4820                    None => out.push(0),
4821                    Some(v) => {
4822                        out.push(1);
4823                        write_value(&mut out, v);
4824                    }
4825                }
4826                out.push(u8::from(c.auto_increment));
4827            }
4828            write_u32(
4829                &mut out,
4830                u32::try_from(t.rows.len()).expect("≤ 4G rows/table"),
4831            );
4832            // v3.0.2 dense row encoding (FILE_VERSION 8): per-row NULL
4833            // bitmap, then tightly-packed bodies. Identical wire format
4834            // as before — extracted into `encode_row_body_dense` so cold-
4835            // tier segments (v5.1+) can share the encoding.
4836            for row in &t.rows {
4837                out.extend_from_slice(&encode_row_body_dense(row, &t.schema));
4838            }
4839            // Index definitions. Per-index payload:
4840            //   [name][col_pos u16][kind u8]
4841            //     kind 0 = B-tree           (no params — rebuilt on load)
4842            //     kind 1 = NSW graph        (u16 M + serialized graph)
4843            // For NSW the graph topology travels on disk so startup
4844            // doesn't re-run the O(n²M) rebuild — see v2.7 notes.
4845            write_u16(
4846                &mut out,
4847                u16::try_from(t.indices.len()).expect("≤ 65k indices/table"),
4848            );
4849            for idx in &t.indices {
4850                write_str(&mut out, &idx.name);
4851                write_u16(
4852                    &mut out,
4853                    u16::try_from(idx.column_position).expect("≤ 65k columns/table"),
4854                );
4855                match &idx.kind {
4856                    IndexKind::BTree(map) => {
4857                        out.push(0);
4858                        // v9: serialise the full PB map. Each entry's
4859                        // RowLocator list travels with the tag-prefixed
4860                        // codec from `row_locator::write_le`, so freezer-
4861                        // produced Cold locators survive a snapshot
4862                        // round-trip. v8 BTree wrote nothing here and
4863                        // rebuilt from rows — v9 readers tolerate v8 by
4864                        // version dispatch in `Catalog::deserialize`.
4865                        write_u32(
4866                            &mut out,
4867                            u32::try_from(map.len()).expect("≤ 4G index entries/index"),
4868                        );
4869                        for (key, locators) in map {
4870                            write_index_key(&mut out, key);
4871                            write_u32(
4872                                &mut out,
4873                                u32::try_from(locators.len()).expect("≤ 4G locators/key"),
4874                            );
4875                            for loc in locators {
4876                                loc.write_le(&mut out);
4877                            }
4878                        }
4879                    }
4880                    IndexKind::Nsw(g) => {
4881                        out.push(1);
4882                        write_u16(&mut out, u16::try_from(g.m).expect("≤ 65k NSW neighbours"));
4883                        write_nsw_graph(&mut out, g);
4884                    }
4885                    IndexKind::Brin { column_type } => {
4886                        // v6.7.1 — tag byte 2 = BRIN. Payload is the
4887                        // column type code (1 byte mapping to the
4888                        // shared DataType numeric encoding); no
4889                        // further data — BRIN summaries live in
4890                        // cold segments, not the catalog.
4891                        out.push(2);
4892                        write_data_type(&mut out, *column_type);
4893                    }
4894                    IndexKind::Gin(map) => {
4895                        // v7.12.3 — tag byte 3 = GIN. Payload mirrors
4896                        // the BTree encoding but with String (lexeme
4897                        // word) keys instead of IndexKey. Tag-prefixed
4898                        // RowLocator codec so freezer-produced Cold
4899                        // locators survive snapshot round-trip.
4900                        // FILE_VERSION 21+; v20 catalogs never wrote a
4901                        // GIN index (the AM degraded to BTree fallback
4902                        // pre-v7.12.3), so no migration shim is needed.
4903                        out.push(3);
4904                        write_u32(
4905                            &mut out,
4906                            u32::try_from(map.len()).expect("≤ 4G GIN posting lists"),
4907                        );
4908                        for (word, locators) in map {
4909                            write_str(&mut out, word);
4910                            write_u32(
4911                                &mut out,
4912                                u32::try_from(locators.len()).expect("≤ 4G locators/posting list"),
4913                            );
4914                            for loc in locators {
4915                                loc.write_le(&mut out);
4916                            }
4917                        }
4918                    }
4919                    IndexKind::GinTrgm(map) => {
4920                        // v7.15.0 — tag byte 4 = GinTrgm
4921                        // (`gin_trgm_ops` GIN over a TEXT column).
4922                        // Payload shape is identical to tag-3 GIN —
4923                        // `String → Vec<RowLocator>` posting lists.
4924                        // The String keys are 3-byte trigrams instead
4925                        // of tsvector lexemes; the deserializer
4926                        // dispatches on the tag, not the key shape.
4927                        // FILE_VERSION 24+; v23 catalogs never wrote
4928                        // a trigram-GIN.
4929                        out.push(4);
4930                        write_u32(
4931                            &mut out,
4932                            u32::try_from(map.len())
4933                                .expect("≤ 4G trigram-GIN posting lists"),
4934                        );
4935                        for (tri, locators) in map {
4936                            write_str(&mut out, tri);
4937                            write_u32(
4938                                &mut out,
4939                                u32::try_from(locators.len())
4940                                    .expect("≤ 4G locators/posting list"),
4941                            );
4942                            for loc in locators {
4943                                loc.write_le(&mut out);
4944                            }
4945                        }
4946                    }
4947                }
4948                // v6.8.0 — included_columns appendix per index.
4949                // Layout: [u16 num_included][num × u16 column_position].
4950                // v11 readers stop before this u16 (deserialise loop
4951                // gated on version >= 12); v12+ readers always
4952                // consume it. Empty Vec serialises as a bare 0u16.
4953                write_u16(
4954                    &mut out,
4955                    u16::try_from(idx.included_columns.len()).expect("≤ 65k INCLUDE columns/index"),
4956                );
4957                for col_pos in &idx.included_columns {
4958                    write_u16(
4959                        &mut out,
4960                        u16::try_from(*col_pos).expect("≤ 65k columns/table"),
4961                    );
4962                }
4963                // v6.8.1 — partial_predicate appendix per index.
4964                // Layout: [u8 has_pred][u16 LE len][bytes (if has_pred)].
4965                // Same v12 gate as included_columns.
4966                match &idx.partial_predicate {
4967                    None => out.push(0),
4968                    Some(pred) => {
4969                        out.push(1);
4970                        write_str(&mut out, pred);
4971                    }
4972                }
4973                // v6.8.2 — expression appendix. Same shape as
4974                // partial_predicate.
4975                match &idx.expression {
4976                    None => out.push(0),
4977                    Some(expr) => {
4978                        out.push(1);
4979                        write_str(&mut out, expr);
4980                    }
4981                }
4982                // v7.9.29 — is_unique appendix (FILE_VERSION 16+).
4983                // Single byte 0/1. v15-and-below readers stop before
4984                // this byte; v16 readers always consume it. mailrs K1.
4985                out.push(u8::from(idx.is_unique));
4986                // v7.9.29 — extra_column_positions appendix.
4987                // Layout: [u16 count][count × u16 column_position].
4988                write_u16(
4989                    &mut out,
4990                    u16::try_from(idx.extra_column_positions.len())
4991                        .expect("≤ 65k extra cols / index"),
4992                );
4993                for cp in &idx.extra_column_positions {
4994                    write_u16(&mut out, u16::try_from(*cp).expect("≤ 65k columns/table"));
4995                }
4996            }
4997            // v6.7.2 — per-table hot_tier_bytes Option<u64>.
4998            // Layout: [u8 has_value][u64 LE value (if has_value)].
4999            // v10 readers stop before this byte (deserialise loop
5000            // gated on version >= 11); v11+ readers always
5001            // consume it.
5002            match t.schema.hot_tier_bytes {
5003                None => out.push(0),
5004                Some(n) => {
5005                    out.push(1);
5006                    out.extend_from_slice(&n.to_le_bytes());
5007                }
5008            }
5009            // v7.6.1 — FOREIGN KEY appendix (catalog FILE_VERSION 13+).
5010            // Layout: [u16 LE fk_count]
5011            //   per fk:
5012            //     [u8 has_name] [str name (if has_name)]
5013            //     [u16 LE local_arity] [u16 LE local_pos]*arity
5014            //     [str parent_table]
5015            //     [u16 LE parent_arity] [u16 LE parent_pos]*arity
5016            //     [u8 on_delete_tag] [u8 on_update_tag]
5017            // Older catalogs (v12 and below) skip this block entirely;
5018            // their reader stops before this byte.
5019            write_u16(
5020                &mut out,
5021                u16::try_from(t.schema.foreign_keys.len()).expect("≤ 65k FKs/table"),
5022            );
5023            for fk in &t.schema.foreign_keys {
5024                match &fk.name {
5025                    None => out.push(0),
5026                    Some(n) => {
5027                        out.push(1);
5028                        write_str(&mut out, n);
5029                    }
5030                }
5031                write_u16(
5032                    &mut out,
5033                    u16::try_from(fk.local_columns.len()).expect("≤ 65k FK columns"),
5034                );
5035                for &p in &fk.local_columns {
5036                    write_u16(&mut out, u16::try_from(p).expect("≤ 65k columns/table"));
5037                }
5038                write_str(&mut out, &fk.parent_table);
5039                write_u16(
5040                    &mut out,
5041                    u16::try_from(fk.parent_columns.len()).expect("≤ 65k FK parent columns"),
5042                );
5043                for &p in &fk.parent_columns {
5044                    write_u16(&mut out, u16::try_from(p).expect("≤ 65k columns/table"));
5045                }
5046                out.push(fk.on_delete.tag());
5047                out.push(fk.on_update.tag());
5048            }
5049            // v7.9.19 — UniquenessConstraint appendix (catalog
5050            // FILE_VERSION 15+). Layout per table after the FK
5051            // block:
5052            //   [u16 count]
5053            //     per constraint:
5054            //       [u8 is_primary_key]
5055            //       [u16 arity][u16 col_pos]*arity
5056            // Older catalogs (v14 and below) skip this block.
5057            write_u16(
5058                &mut out,
5059                u16::try_from(t.schema.uniqueness_constraints.len())
5060                    .expect("≤ 65k uniqueness constraints/table"),
5061            );
5062            for uc in &t.schema.uniqueness_constraints {
5063                out.push(u8::from(uc.is_primary_key));
5064                write_u16(
5065                    &mut out,
5066                    u16::try_from(uc.columns.len()).expect("≤ 65k cols in uniqueness constraint"),
5067                );
5068                for &p in &uc.columns {
5069                    write_u16(&mut out, u16::try_from(p).expect("≤ 65k columns/table"));
5070                }
5071                // v7.13.0 — `nulls_not_distinct` flag
5072                // (FILE_VERSION 23+). Always written by writers at
5073                // version 23+; deserialise gates on `version >= 23`
5074                // so v22-and-below catalogs round-trip cleanly.
5075                out.push(u8::from(uc.nulls_not_distinct));
5076            }
5077            // v7.9.21 — runtime_default appendix per table.
5078            // Layout: [u16 count] then for each:
5079            //   [u16 col_pos][str expr]
5080            // Only columns whose runtime_default is Some land here;
5081            // catalog stays compact for the common literal-default
5082            // case.
5083            let mut rt_defaults: Vec<(usize, &str)> = Vec::new();
5084            for (i, c) in t.schema.columns.iter().enumerate() {
5085                if let Some(e) = &c.runtime_default {
5086                    rt_defaults.push((i, e.as_str()));
5087                }
5088            }
5089            write_u16(
5090                &mut out,
5091                u16::try_from(rt_defaults.len()).expect("≤ 65k runtime defaults/table"),
5092            );
5093            for (pos, expr) in rt_defaults {
5094                write_u16(&mut out, u16::try_from(pos).expect("≤ 65k columns/table"));
5095                write_str(&mut out, expr);
5096            }
5097            // v7.13.0 — CHECK constraint appendix per table.
5098            // Layout: [u16 count] then `count` Display-form
5099            // expression strings. Re-parsed on every INSERT/UPDATE
5100            // by the engine. FILE_VERSION 23+ only; v22 readers
5101            // never reach this block because the writer also moves
5102            // to v23 in lock-step.
5103            write_u16(
5104                &mut out,
5105                u16::try_from(t.schema.checks.len()).expect("≤ 65k CHECK constraints/table"),
5106            );
5107            for c in &t.schema.checks {
5108                write_str(&mut out, c.as_str());
5109            }
5110        }
5111        // v7.12.4 — catalog-wide appendix: user-defined functions
5112        // then triggers. FILE_VERSION 22+ only. v21 and earlier
5113        // readers stop after the last table; v22 readers always
5114        // consume two `u32` counts (possibly zero).
5115        //
5116        // Function entry layout:
5117        //   [str name] [str args_repr] [str returns]
5118        //   [str language] [str body]
5119        // Trigger entry layout:
5120        //   [str name] [str table] [str timing]
5121        //   [u16 event_count] (event_count × str)
5122        //   [str for_each] [str function]
5123        write_u32(
5124            &mut out,
5125            u32::try_from(self.functions.len()).expect("≤ 4G functions"),
5126        );
5127        for fd in self.functions.values() {
5128            write_str(&mut out, &fd.name);
5129            write_str(&mut out, &fd.args_repr);
5130            write_str(&mut out, &fd.returns);
5131            write_str(&mut out, &fd.language);
5132            write_str_long(&mut out, &fd.body);
5133        }
5134        write_u32(
5135            &mut out,
5136            u32::try_from(self.triggers.len()).expect("≤ 4G triggers"),
5137        );
5138        for td in &self.triggers {
5139            write_str(&mut out, &td.name);
5140            write_str(&mut out, &td.table);
5141            write_str(&mut out, &td.timing);
5142            write_u16(
5143                &mut out,
5144                u16::try_from(td.events.len()).expect("≤ 65k events / trigger"),
5145            );
5146            for ev in &td.events {
5147                write_str(&mut out, ev);
5148            }
5149            write_str(&mut out, &td.for_each);
5150            write_str(&mut out, &td.function);
5151            // v7.13.0 — `UPDATE OF cols` filter
5152            // (FILE_VERSION 23+). v22 readers omit; v23 writers
5153            // always emit (possibly zero).
5154            write_u16(
5155                &mut out,
5156                u16::try_from(td.update_columns.len()).expect("≤ 65k cols / trigger"),
5157            );
5158            for c in &td.update_columns {
5159                write_str(&mut out, c);
5160            }
5161            // v7.16.1 — TriggerDef.enabled (FILE_VERSION 25+).
5162            out.push(u8::from(td.enabled));
5163        }
5164        out
5165    }
5166
5167    /// Deserialize a previously-serialized catalog. Rejects bad magic, version
5168    /// mismatch, unknown tags, truncation, and trailing bytes.
5169    pub fn deserialize(buf: &[u8]) -> Result<Self, StorageError> {
5170        let mut cur = Cursor::new(buf);
5171        let magic = cur.take(8)?;
5172        if magic != FILE_MAGIC {
5173            return Err(StorageError::Corrupt(format!(
5174                "bad magic: expected SPGDB001, got {magic:?}"
5175            )));
5176        }
5177        let version = cur.read_u8()?;
5178        if !(MIN_SUPPORTED_FILE_VERSION..=FILE_VERSION).contains(&version) {
5179            return Err(StorageError::Corrupt(format!(
5180                "unsupported file version: {version} (supported: {MIN_SUPPORTED_FILE_VERSION}..={FILE_VERSION})"
5181            )));
5182        }
5183        let table_count = cur.read_u32()? as usize;
5184        let mut cat = Self::new();
5185        for _ in 0..table_count {
5186            deserialize_table(&mut cur, &mut cat, version)?;
5187        }
5188        // v7.12.4 — catalog-wide function + trigger appendix.
5189        // FILE_VERSION 22+ only; v21 and earlier catalogs stop
5190        // after the last table.
5191        if version >= 22 {
5192            let fn_count = cur.read_u32()? as usize;
5193            for _ in 0..fn_count {
5194                let name = cur.read_str()?;
5195                let args_repr = cur.read_str()?;
5196                let returns = cur.read_str()?;
5197                let language = cur.read_str()?;
5198                let body = cur.read_str_long()?;
5199                cat.functions.insert(
5200                    name.clone(),
5201                    FunctionDef {
5202                        name,
5203                        args_repr,
5204                        returns,
5205                        language,
5206                        body,
5207                    },
5208                );
5209            }
5210            let trg_count = cur.read_u32()? as usize;
5211            for _ in 0..trg_count {
5212                let name = cur.read_str()?;
5213                let table = cur.read_str()?;
5214                let timing = cur.read_str()?;
5215                let ev_count = cur.read_u16()? as usize;
5216                let mut events = Vec::with_capacity(ev_count);
5217                for _ in 0..ev_count {
5218                    events.push(cur.read_str()?);
5219                }
5220                let for_each = cur.read_str()?;
5221                let function = cur.read_str()?;
5222                // v7.13.0 — trailing `UPDATE OF cols` filter
5223                // (FILE_VERSION 23+ only; v22 catalogs omit and
5224                // deserialise with an empty vec).
5225                let update_columns = if version >= 23 {
5226                    let n = cur.read_u16()? as usize;
5227                    let mut cols = Vec::with_capacity(n);
5228                    for _ in 0..n {
5229                        cols.push(cur.read_str()?);
5230                    }
5231                    cols
5232                } else {
5233                    Vec::new()
5234                };
5235                // v7.16.1 — TriggerDef.enabled (FILE_VERSION 25+).
5236                // v24-and-below catalogs deserialise with `true`
5237                // — pre-v7.16.1 every trigger always fired.
5238                let enabled = if version >= 25 {
5239                    cur.read_u8()? != 0
5240                } else {
5241                    true
5242                };
5243                cat.triggers.push(TriggerDef {
5244                    name,
5245                    table,
5246                    timing,
5247                    events,
5248                    for_each,
5249                    function,
5250                    update_columns,
5251                    enabled,
5252                });
5253            }
5254        }
5255        if cur.pos < buf.len() {
5256            return Err(StorageError::Corrupt(format!(
5257                "trailing bytes: {} unread",
5258                buf.len() - cur.pos
5259            )));
5260        }
5261        Ok(cat)
5262    }
5263}
5264
5265/// Per-table deserialize body — schema, rows, indices. Pulled out of
5266/// `Catalog::deserialize` to keep the latter under the line-budget lint
5267/// and to give the row hot loop its own scope (so the borrow on `t`
5268/// stays scoped here rather than across the whole catalog loop).
5269fn deserialize_table(
5270    cur: &mut Cursor<'_>,
5271    cat: &mut Catalog,
5272    version: u8,
5273) -> Result<(), StorageError> {
5274    let table_name = cur.read_str()?;
5275    let name = table_name.clone();
5276    let col_count = cur.read_u16()? as usize;
5277    let mut cols = Vec::with_capacity(col_count);
5278    for _ in 0..col_count {
5279        let c_name = cur.read_str()?;
5280        let ty = cur.read_data_type()?;
5281        let nullable = cur.read_u8()? != 0;
5282        let default = match cur.read_u8()? {
5283            0 => None,
5284            1 => Some(cur.read_value()?),
5285            other => {
5286                return Err(StorageError::Corrupt(format!(
5287                    "unknown default tag: {other}"
5288                )));
5289            }
5290        };
5291        let auto_increment = cur.read_u8()? != 0;
5292        // Note: deserialiser sets runtime_default = None for
5293        // older catalogs (≤ v14). v15+ reads it from the
5294        // per-column appendix below.
5295        cols.push(ColumnSchema {
5296            name: c_name,
5297            ty,
5298            nullable,
5299            default,
5300            runtime_default: None,
5301            auto_increment,
5302        });
5303    }
5304    let n_cols = cols.len();
5305    cat.create_table(TableSchema::new(name, cols))?;
5306    // Vec<Table> with insertion-order semantics — the just-pushed
5307    // table is at the end. Sidecar `by_name` is already wired up but
5308    // we skip the map lookup here since we know the position.
5309    let t = cat.tables.last_mut().expect("create_table just pushed");
5310    deserialize_rows(cur, t, n_cols)?;
5311    deserialize_indices(cur, t, version)?;
5312    // v6.7.2 — per-table hot_tier_bytes appendix. v11+ writes
5313    // `[u8 has_value][u64 LE value (if has_value)]`. v10 / v9 / v8
5314    // catalogs skip this entirely (the deserialiser reads no extra
5315    // bytes; the table's hot_tier_bytes stays None from
5316    // TableSchema::new).
5317    if version >= 11 {
5318        let has = cur.read_u8()?;
5319        let hot_tier_bytes = match has {
5320            0 => None,
5321            1 => Some(cur.read_u64()?),
5322            other => {
5323                return Err(StorageError::Corrupt(format!(
5324                    "hot_tier_bytes appendix: unknown has-value byte {other}"
5325                )));
5326            }
5327        };
5328        t.schema_mut().hot_tier_bytes = hot_tier_bytes;
5329    }
5330    // v7.6.1 — FOREIGN KEY appendix (FILE_VERSION 13+). v12 / v11 / …
5331    // catalogs skip this entirely.
5332    if version >= 13 {
5333        let fk_count = cur.read_u16()? as usize;
5334        let mut fks = Vec::with_capacity(fk_count);
5335        for _ in 0..fk_count {
5336            let name = match cur.read_u8()? {
5337                0 => None,
5338                1 => Some(cur.read_str()?),
5339                other => {
5340                    return Err(StorageError::Corrupt(format!(
5341                        "FK appendix: unknown has-name byte {other}"
5342                    )));
5343                }
5344            };
5345            let local_arity = cur.read_u16()? as usize;
5346            let mut local_columns = Vec::with_capacity(local_arity);
5347            for _ in 0..local_arity {
5348                local_columns.push(cur.read_u16()? as usize);
5349            }
5350            let parent_table = cur.read_str()?;
5351            let parent_arity = cur.read_u16()? as usize;
5352            if parent_arity != local_arity {
5353                return Err(StorageError::Corrupt(format!(
5354                    "FK arity mismatch in catalog: local {local_arity} vs parent {parent_arity}"
5355                )));
5356            }
5357            let mut parent_columns = Vec::with_capacity(parent_arity);
5358            for _ in 0..parent_arity {
5359                parent_columns.push(cur.read_u16()? as usize);
5360            }
5361            let on_delete = FkAction::from_tag(cur.read_u8()?).ok_or_else(|| {
5362                StorageError::Corrupt("FK appendix: unknown on_delete tag".into())
5363            })?;
5364            let on_update = FkAction::from_tag(cur.read_u8()?).ok_or_else(|| {
5365                StorageError::Corrupt("FK appendix: unknown on_update tag".into())
5366            })?;
5367            fks.push(ForeignKeyConstraint {
5368                name,
5369                local_columns,
5370                parent_table,
5371                parent_columns,
5372                on_delete,
5373                on_update,
5374            });
5375        }
5376        t.schema_mut().foreign_keys = fks;
5377    }
5378    // v7.9.19 — UniquenessConstraint appendix (FILE_VERSION 15+).
5379    // v14 and below skip this entirely.
5380    if version >= 15 {
5381        let uc_count = cur.read_u16()? as usize;
5382        let mut ucs = Vec::with_capacity(uc_count);
5383        for _ in 0..uc_count {
5384            let is_pk = cur.read_u8()? != 0;
5385            let arity = cur.read_u16()? as usize;
5386            let mut cols = Vec::with_capacity(arity);
5387            for _ in 0..arity {
5388                cols.push(cur.read_u16()? as usize);
5389            }
5390            // v7.13.0 — trailing `nulls_not_distinct` flag
5391            // (FILE_VERSION 23+). v22 and below skip — flag
5392            // defaults to false (= NULLS DISTINCT).
5393            let nulls_not_distinct = if version >= 23 {
5394                cur.read_u8()? != 0
5395            } else {
5396                false
5397            };
5398            ucs.push(UniquenessConstraint {
5399                is_primary_key: is_pk,
5400                columns: cols,
5401                nulls_not_distinct,
5402            });
5403        }
5404        t.schema_mut().uniqueness_constraints = ucs;
5405        // v7.9.21 — runtime_default appendix (FILE_VERSION 15+).
5406        let rt_count = cur.read_u16()? as usize;
5407        for _ in 0..rt_count {
5408            let pos = cur.read_u16()? as usize;
5409            let expr = cur.read_str()?;
5410            if let Some(col) = t.schema_mut().columns.get_mut(pos) {
5411                col.runtime_default = Some(expr);
5412            }
5413        }
5414    }
5415    // v7.13.0 — CHECK constraints appendix (FILE_VERSION 23+).
5416    // v22 and below leave the vec empty.
5417    if version >= 23 {
5418        let check_count = cur.read_u16()? as usize;
5419        let mut checks = Vec::with_capacity(check_count);
5420        for _ in 0..check_count {
5421            checks.push(cur.read_str()?);
5422        }
5423        t.schema_mut().checks = checks;
5424    }
5425    let _ = table_name;
5426    Ok(())
5427}
5428
5429fn deserialize_rows(
5430    cur: &mut Cursor<'_>,
5431    t: &mut Table,
5432    _n_cols: usize,
5433) -> Result<(), StorageError> {
5434    let row_count = cur.read_u32()? as usize;
5435    // v4.39: PV has no `reserve` (the BVT doesn't preallocate a
5436    // contiguous buffer); we just push directly and let the trie
5437    // grow. v5.1: row decode reuses `decode_row_body_dense` so the
5438    // catalog and cold-tier segments share one row codec.
5439    let mut hot_bytes: u64 = 0;
5440    for _ in 0..row_count {
5441        let tail = &cur.buf[cur.pos..];
5442        let (row, consumed) = decode_row_body_dense(tail, &t.schema)?;
5443        cur.pos += consumed;
5444        // v5.2.1: account for hot bytes as we go; the snapshot's row
5445        // block bytes are exactly what `encode_row_body_dense` would
5446        // produce, so `consumed` would do too — but going via the
5447        // helper keeps the counter's definition coupled to the
5448        // encoder rather than the snapshot's row prefix layout.
5449        hot_bytes = hot_bytes.saturating_add(row_body_encoded_len(&row, &t.schema) as u64);
5450        t.rows.push_mut(row);
5451    }
5452    t.hot_bytes = hot_bytes;
5453    Ok(())
5454}
5455
5456fn deserialize_indices(
5457    cur: &mut Cursor<'_>,
5458    t: &mut Table,
5459    version: u8,
5460) -> Result<(), StorageError> {
5461    let index_count = cur.read_u16()? as usize;
5462    for _ in 0..index_count {
5463        let idx_name = cur.read_str()?;
5464        let col_pos = cur.read_u16()? as usize;
5465        let column_name = t
5466            .schema
5467            .columns
5468            .get(col_pos)
5469            .ok_or_else(|| {
5470                StorageError::Corrupt(format!(
5471                    "index {idx_name:?} points at non-existent column position {col_pos}"
5472                ))
5473            })?
5474            .name
5475            .clone();
5476        let kind_tag = cur.read_u8()?;
5477        match kind_tag {
5478            0 => {
5479                if version >= 9 {
5480                    // v9+: BTree entries serialised inline (tag-prefixed
5481                    // locator codec). Restore the map directly so any
5482                    // freezer-produced Cold locators come back exactly
5483                    // as they went out.
5484                    let map = read_btree_map(cur)?;
5485                    t.restore_btree_index(idx_name, &column_name, map)?;
5486                } else {
5487                    // v8: no entries on disk; rebuild from rows. Every
5488                    // entry is materialised as `RowLocator::Hot(i)` —
5489                    // semantically identical to the v5.1 in-memory state
5490                    // since v8 catalogs never produced Cold locators.
5491                    t.add_index(idx_name, &column_name)?;
5492                }
5493            }
5494            1 => {
5495                let m = cur.read_u16()? as usize;
5496                let graph = cur.read_nsw_graph(m)?;
5497                t.restore_nsw_index(idx_name, &column_name, graph)?;
5498            }
5499            2 => {
5500                // v6.7.1 — BRIN tag. Payload is the column type
5501                // tag. No further data — summaries live in cold
5502                // segments.
5503                let column_type = cur.read_data_type()?;
5504                t.restore_brin_index(idx_name, &column_name, column_type)?;
5505            }
5506            3 => {
5507                // v7.12.3 — GIN tag. Payload mirrors the BTree
5508                // encoding but with String (lexeme word) keys.
5509                // Only emitted by FILE_VERSION 21+ writers — v20
5510                // and earlier degraded `USING gin` to BTree.
5511                let map = read_gin_map(cur)?;
5512                t.restore_gin_index(idx_name, &column_name, map)?;
5513            }
5514            4 => {
5515                // v7.15.0 — trigram-GIN tag (`gin_trgm_ops`).
5516                // Same payload shape as tag 3 (String → posting
5517                // list); only emitted by FILE_VERSION 24+ writers.
5518                if version < 24 {
5519                    return Err(StorageError::Corrupt(format!(
5520                        "trigram-GIN index tag 4 found in catalog FILE_VERSION {version}; \
5521                         FILE_VERSION 24+ required (v7.15.0 introduced this tag)"
5522                    )));
5523                }
5524                let map = read_gin_map(cur)?;
5525                t.restore_gin_trgm_index(idx_name, &column_name, map)?;
5526            }
5527            other => {
5528                return Err(StorageError::Corrupt(format!(
5529                    "unknown index kind tag: {other}"
5530                )));
5531            }
5532        }
5533        // v6.8.0 — included_columns appendix per index. v11- snapshots
5534        // stop before this u16; v12+ always carries it (possibly 0).
5535        if version >= 12 {
5536            let num_included = cur.read_u16()? as usize;
5537            if num_included > 0 {
5538                let mut included: Vec<usize> = Vec::with_capacity(num_included);
5539                for _ in 0..num_included {
5540                    let cp = cur.read_u16()? as usize;
5541                    if cp >= t.schema.columns.len() {
5542                        return Err(StorageError::Corrupt(format!(
5543                            "INCLUDE column position {cp} out of range \
5544                             ({} schema columns)",
5545                            t.schema.columns.len()
5546                        )));
5547                    }
5548                    included.push(cp);
5549                }
5550                if let Some(last) = t.indices.last_mut() {
5551                    last.included_columns = included;
5552                }
5553            }
5554            // v6.8.1 — partial_predicate appendix.
5555            match cur.read_u8()? {
5556                0 => {}
5557                1 => {
5558                    let pred = cur.read_str()?;
5559                    if let Some(last) = t.indices.last_mut() {
5560                        last.partial_predicate = Some(pred);
5561                    }
5562                }
5563                other => {
5564                    return Err(StorageError::Corrupt(format!(
5565                        "partial_predicate tag: unknown byte {other}"
5566                    )));
5567                }
5568            }
5569            // v6.8.2 — expression appendix.
5570            match cur.read_u8()? {
5571                0 => {}
5572                1 => {
5573                    let expr = cur.read_str()?;
5574                    if let Some(last) = t.indices.last_mut() {
5575                        last.expression = Some(expr);
5576                    }
5577                }
5578                other => {
5579                    return Err(StorageError::Corrupt(format!(
5580                        "expression tag: unknown byte {other}"
5581                    )));
5582                }
5583            }
5584            // v7.9.29 — is_unique appendix (FILE_VERSION 16+).
5585            // v15-and-below catalogs stop before this byte. mailrs K1.
5586            if version >= 16 {
5587                match cur.read_u8()? {
5588                    0 => {}
5589                    1 => {
5590                        if let Some(last) = t.indices.last_mut() {
5591                            last.is_unique = true;
5592                        }
5593                    }
5594                    other => {
5595                        return Err(StorageError::Corrupt(format!(
5596                            "is_unique tag: unknown byte {other}"
5597                        )));
5598                    }
5599                }
5600                // v7.9.29 — extra_column_positions appendix.
5601                let n = cur.read_u16()? as usize;
5602                if n > 0 {
5603                    let mut extras: Vec<usize> = Vec::with_capacity(n);
5604                    for _ in 0..n {
5605                        let cp = cur.read_u16()? as usize;
5606                        if cp >= t.schema.columns.len() {
5607                            return Err(StorageError::Corrupt(format!(
5608                                "extra column position {cp} out of range \
5609                                 ({} schema columns)",
5610                                t.schema.columns.len()
5611                            )));
5612                        }
5613                        extras.push(cp);
5614                    }
5615                    if let Some(last) = t.indices.last_mut() {
5616                        last.extra_column_positions = extras;
5617                    }
5618                }
5619            }
5620        }
5621    }
5622    Ok(())
5623}
5624
5625/// Parse a v9 `BTree` index payload — `[u32 entry_count]` followed by
5626/// `entry_count` `(IndexKey, Vec<RowLocator>)` pairs. The locator list
5627/// uses the v5.1 tag-prefixed wire format (`RowLocator::read_le`).
5628fn read_btree_map(
5629    cur: &mut Cursor<'_>,
5630) -> Result<PersistentBTreeMap<IndexKey, Vec<RowLocator>>, StorageError> {
5631    let entry_count = cur.read_u32()? as usize;
5632    let mut map = PersistentBTreeMap::new();
5633    for _ in 0..entry_count {
5634        let key = cur.read_index_key()?;
5635        let locator_count = cur.read_u32()? as usize;
5636        let mut locators = Vec::with_capacity(locator_count);
5637        for _ in 0..locator_count {
5638            let tail = &cur.buf[cur.pos..];
5639            let (loc, consumed) = RowLocator::read_le(tail).map_err(|e| {
5640                StorageError::Corrupt(format!("row_locator decode at offset {}: {e}", cur.pos))
5641            })?;
5642            cur.pos += consumed;
5643            locators.push(loc);
5644        }
5645        map.insert_mut(key, locators);
5646    }
5647    Ok(map)
5648}
5649
5650/// v7.12.3 — parse a `Gin` index payload. Mirrors [`read_btree_map`]
5651/// but with `String` (lexeme word) keys instead of `IndexKey`.
5652/// FILE_VERSION 21+ only.
5653fn read_gin_map(
5654    cur: &mut Cursor<'_>,
5655) -> Result<PersistentBTreeMap<String, Vec<RowLocator>>, StorageError> {
5656    let entry_count = cur.read_u32()? as usize;
5657    let mut map = PersistentBTreeMap::new();
5658    for _ in 0..entry_count {
5659        let word = cur.read_str()?;
5660        let locator_count = cur.read_u32()? as usize;
5661        let mut locators = Vec::with_capacity(locator_count);
5662        for _ in 0..locator_count {
5663            let tail = &cur.buf[cur.pos..];
5664            let (loc, consumed) = RowLocator::read_le(tail).map_err(|e| {
5665                StorageError::Corrupt(format!("row_locator decode at offset {}: {e}", cur.pos))
5666            })?;
5667            cur.pos += consumed;
5668            locators.push(loc);
5669        }
5670        map.insert_mut(word, locators);
5671    }
5672    Ok(map)
5673}
5674
5675// --- low-level binary helpers ---------------------------------------------
5676
5677/// Write a `DataType` as a tag byte + optional payload (Vector carries its
5678/// `u32` dimension). Inverse: [`read_data_type`].
5679/// Serialize an HNSW graph after the `[kind=1][u16 M]` header (v7).
5680/// Layout:
5681/// - `[u16 m_max_0]`
5682/// - `[entry u32]` — `u32::MAX` means `None`, else the entry node index
5683/// - `[u8 entry_level]`
5684/// - `[node_count u32]`
5685/// - for each node: `[u8 level]`  (top layer for this node)
5686/// - `[layer_count u8]`
5687/// - for each layer `0..layer_count`:
5688///     - `[u32 layer_node_count]` (== `node_count`; per-layer slot)
5689///     - for each node: `[u16 neighbor_count] [u32 neighbor]*`
5690fn write_nsw_graph(out: &mut Vec<u8>, g: &NswGraph) {
5691    let entry = g.entry.map_or(u32::MAX, |e| {
5692        u32::try_from(e).expect("NSW entry fits in u32")
5693    });
5694    write_u16(
5695        out,
5696        u16::try_from(g.m_max_0).expect("HNSW m_max_0 fits in u16"),
5697    );
5698    out.extend_from_slice(&entry.to_le_bytes());
5699    out.push(g.entry_level);
5700    let node_count = g.levels.len();
5701    write_u32(
5702        out,
5703        u32::try_from(node_count).expect("HNSW node count fits in u32"),
5704    );
5705    for &lvl in &g.levels {
5706        out.push(lvl);
5707    }
5708    let layer_count = u8::try_from(g.layers.len()).expect("HNSW layer count ≤ 255");
5709    out.push(layer_count);
5710    for layer in &g.layers {
5711        write_u32(
5712            out,
5713            u32::try_from(layer.len()).expect("HNSW per-layer node count fits in u32"),
5714        );
5715        for neighbors in layer {
5716            write_u16(
5717                out,
5718                u16::try_from(neighbors.len()).expect("HNSW neighbour list fits in u16"),
5719            );
5720            // v6.1.x: neighbour slot is already u32 in memory; just
5721            // emit the raw bytes. (v6.0 stored usize and converted
5722            // here.)
5723            for &peer in neighbors {
5724                write_u32(out, peer);
5725            }
5726        }
5727    }
5728}
5729
5730fn write_data_type(out: &mut Vec<u8>, t: DataType) {
5731    match t {
5732        DataType::Int => out.push(1),
5733        DataType::BigInt => out.push(2),
5734        DataType::Float => out.push(3),
5735        DataType::Text => out.push(4),
5736        DataType::Bool => out.push(5),
5737        DataType::Vector { dim, encoding } => match encoding {
5738            // Tag 6: pre-v6 F32 vector. Layout unchanged; pre-v6
5739            // binaries continue to deserialise this exactly as
5740            // before.
5741            VecEncoding::F32 => {
5742                out.push(6);
5743                out.extend_from_slice(&dim.to_le_bytes());
5744            }
5745            // v6.0.3: tag 15 for `VECTOR(N) USING HALF`. Same
5746            // forward-compat fence story as SQ8 below.
5747            VecEncoding::F16 => {
5748                out.push(15);
5749                out.extend_from_slice(&dim.to_le_bytes());
5750            }
5751            // v6.0.1: new tag 14 for `VECTOR(N) USING SQ8` column
5752            // type. Pre-v6 readers fall through `read_data_type`'s
5753            // catch-all and surface `Corrupt("unknown data type tag")`
5754            // — the explicit forward-compat fence called out in
5755            // V6_DESIGN deliberation #5.
5756            VecEncoding::Sq8 => {
5757                out.push(14);
5758                out.extend_from_slice(&dim.to_le_bytes());
5759            }
5760        },
5761        DataType::SmallInt => out.push(7),
5762        DataType::Varchar(max) => {
5763            out.push(8);
5764            out.extend_from_slice(&max.to_le_bytes());
5765        }
5766        DataType::Char(size) => {
5767            out.push(9);
5768            out.extend_from_slice(&size.to_le_bytes());
5769        }
5770        DataType::Numeric { precision, scale } => {
5771            out.push(10);
5772            out.push(precision);
5773            out.push(scale);
5774        }
5775        DataType::Date => out.push(11),
5776        DataType::Timestamp => out.push(12),
5777        // v7.9.2 — tag 17 for TIMESTAMPTZ. Body = i64 microseconds
5778        // UTC, identical to tag 12. Only the schema-side type tag
5779        // differs (for wire OID advertisement).
5780        DataType::Timestamptz => out.push(17),
5781        // INTERVAL is runtime-only — CREATE TABLE never produces a
5782        // column with this type, so write_data_type must not be called
5783        // on it. (Disk-format codepoint reserved for a future v3 where
5784        // INTERVAL becomes storable.)
5785        DataType::Interval => {
5786            unreachable!("DataType::Interval has no on-disk encoding in v2.11")
5787        }
5788        DataType::Json => out.push(13),
5789        // v7.9.0: tag 16 for `JSONB`. Same on-disk layout as
5790        // tag 13 — only the wire OID differs.
5791        DataType::Jsonb => out.push(16),
5792        // v7.10.4: tag 18 for `BYTEA`. Body = [u16 len][bytes].
5793        DataType::Bytes => out.push(18),
5794        // v7.10.9: tag 19 for `TEXT[]`. Body = [u16 count][per
5795        // element: u8 null + (if non-null) u16 len + utf-8].
5796        DataType::TextArray => out.push(19),
5797        // v7.11.12: tag 20 for `INT[]`. Body = [u16 count][per
5798        // element: u8 null + (if non-null) i32 LE].
5799        DataType::IntArray => out.push(20),
5800        // v7.11.12: tag 21 for `BIGINT[]`. Body = [u16 count][per
5801        // element: u8 null + (if non-null) i64 LE].
5802        DataType::BigIntArray => out.push(21),
5803        // v7.12.0: tag 22 for `tsvector`. No body — type identity
5804        // alone. Catalog FILE_VERSION 20+.
5805        DataType::TsVector => out.push(22),
5806        // v7.12.0: tag 23 for `tsquery`. No body. Catalog
5807        // FILE_VERSION 20+.
5808        DataType::TsQuery => out.push(23),
5809    }
5810}
5811
5812impl Cursor<'_> {
5813    fn read_data_type(&mut self) -> Result<DataType, StorageError> {
5814        let tag = self.read_u8()?;
5815        match tag {
5816            1 => Ok(DataType::Int),
5817            2 => Ok(DataType::BigInt),
5818            3 => Ok(DataType::Float),
5819            4 => Ok(DataType::Text),
5820            5 => Ok(DataType::Bool),
5821            6 => Ok(DataType::Vector {
5822                dim: self.read_u32()?,
5823                encoding: VecEncoding::F32,
5824            }),
5825            7 => Ok(DataType::SmallInt),
5826            8 => Ok(DataType::Varchar(self.read_u32()?)),
5827            9 => Ok(DataType::Char(self.read_u32()?)),
5828            10 => {
5829                let precision = self.read_u8()?;
5830                let scale = self.read_u8()?;
5831                Ok(DataType::Numeric { precision, scale })
5832            }
5833            11 => Ok(DataType::Date),
5834            12 => Ok(DataType::Timestamp),
5835            13 => Ok(DataType::Json),
5836            14 => Ok(DataType::Vector {
5837                dim: self.read_u32()?,
5838                encoding: VecEncoding::Sq8,
5839            }),
5840            // v6.0.3: tag 15 for `VECTOR(N) USING HALF`. Same
5841            // [u32 dim] type-tag payload as F32 / SQ8; the encoding
5842            // lives in the tag byte itself.
5843            15 => Ok(DataType::Vector {
5844                dim: self.read_u32()?,
5845                encoding: VecEncoding::F16,
5846            }),
5847            // v7.9.0: tag 16 for `JSONB`. Storage shape == Json;
5848            // we only carry the type tag so the wire layer can
5849            // emit PG OID 3802 instead of 114.
5850            16 => Ok(DataType::Jsonb),
5851            // v7.9.2: tag 17 for `TIMESTAMPTZ`. Storage shape ==
5852            // Timestamp (i64 microseconds UTC); only the wire OID
5853            // (1184) differs.
5854            17 => Ok(DataType::Timestamptz),
5855            // v7.10.4: tag 18 for `BYTEA`. Catalog FILE_VERSION 17+.
5856            18 => Ok(DataType::Bytes),
5857            // v7.10.9: tag 19 for `TEXT[]`. Catalog FILE_VERSION 18+.
5858            19 => Ok(DataType::TextArray),
5859            // v7.11.12: tags 20/21 for INT[]/BIGINT[]. FILE_VERSION 19+.
5860            20 => Ok(DataType::IntArray),
5861            21 => Ok(DataType::BigIntArray),
5862            // v7.12.0: tags 22/23 for tsvector / tsquery. Catalog
5863            // FILE_VERSION 20+.
5864            22 => Ok(DataType::TsVector),
5865            23 => Ok(DataType::TsQuery),
5866            other => Err(StorageError::Corrupt(format!(
5867                "unknown data type tag: {other}"
5868            ))),
5869        }
5870    }
5871}
5872
5873/// Fast computation of the byte length [`encode_row_body_dense`]
5874/// would produce, without allocating the output buffer. Mirrors the
5875/// encoder's per-column body sizing so the v5.2.1 `Table::hot_bytes`
5876/// incremental counter doesn't pay an alloc-per-insert tax. Returns
5877/// the exact same `usize` as `encode_row_body_dense(row, schema).len()`.
5878pub fn row_body_encoded_len(row: &Row, schema: &TableSchema) -> usize {
5879    debug_assert_eq!(
5880        row.values.len(),
5881        schema.columns.len(),
5882        "row_body_encoded_len: row arity must match schema"
5883    );
5884    let bitmap_bytes = schema.columns.len().div_ceil(8);
5885    let mut n = bitmap_bytes;
5886    for (col_idx, v) in row.values.iter().enumerate() {
5887        if matches!(v, Value::Null) {
5888            continue;
5889        }
5890        n += value_body_encoded_len(v, schema.columns[col_idx].ty);
5891    }
5892    n
5893}
5894
5895/// Byte length a single cell consumes when written by
5896/// `write_value_body`. Used by [`row_body_encoded_len`]; kept in
5897/// lock-step with the encoder. The `_ty` slot is reserved for future
5898/// type-dependent encodings — every variant currently writes a fixed
5899/// body shape regardless of the declared column type.
5900fn value_body_encoded_len(v: &Value, _ty: DataType) -> usize {
5901    match v {
5902        Value::SmallInt(_) => 2,
5903        // 4-byte body: i32 / Date.
5904        Value::Int(_) | Value::Date(_) => 4,
5905        // 8-byte body: i64 / f64 / Timestamp.
5906        Value::BigInt(_) | Value::Float(_) | Value::Timestamp(_) => 8,
5907        Value::Bool(_) => 1,
5908        // Text/Varchar/Char/Json share the [u16 len][utf-8] layout.
5909        Value::Text(s) | Value::Json(s) => 2 + s.len(),
5910        // [u32 dim][f32 * dim]
5911        Value::Vector(vec) => 4 + 4 * vec.len(),
5912        // v6.0.1: SQ8 cell on-disk shape — [u32 dim][f32 min]
5913        // [f32 max][u8 * dim] = 12 + dim bytes. `hot_bytes`
5914        // tracking on `Table::insert` calls this every row, so
5915        // returning the real size now (even though the actual
5916        // `write_value_body` writer lands in step 6) keeps the
5917        // sizing arithmetic honest for in-memory benches.
5918        Value::Sq8Vector(q) => 4 + 4 + 4 + q.bytes.len(),
5919        // v6.0.3: halfvec on-disk shape — [u32 dim][u16 LE * dim]
5920        // = 4 + 2 * dim bytes.
5921        Value::HalfVector(h) => 4 + h.bytes.len(),
5922        // [i128 scaled][u8 scale]
5923        Value::Numeric { .. } => 16 + 1,
5924        // v7.10.4: BYTEA on-disk shape mirrors Text — [u16 len][bytes].
5925        // The 16-bit length cap is the same TEXT/JSON limit (~65 KB);
5926        // larger blobs need toast-style chunking which is a v7.11
5927        // carve-out (kept aligned with TEXT for now so the catalog
5928        // snapshot stays simple).
5929        Value::Bytes(b) => 2 + b.len(),
5930        // v7.10.9: TEXT[] on-disk shape — [u16 count][per element:
5931        // u8 null flag + (when non-null) u16 len + utf-8 bytes].
5932        Value::TextArray(items) => {
5933            let mut n = 2; // count prefix
5934            for item in items {
5935                n += 1; // null flag
5936                if let Some(s) = item {
5937                    n += 2 + s.len();
5938                }
5939            }
5940            n
5941        }
5942        // v7.11.12: INT[] / BIGINT[] — [u16 count][per element:
5943        // u8 null + (when non-null) fixed-width LE].
5944        Value::IntArray(items) => {
5945            2 + items
5946                .iter()
5947                .map(|x| if x.is_some() { 5 } else { 1 })
5948                .sum::<usize>()
5949        }
5950        Value::BigIntArray(items) => {
5951            2 + items
5952                .iter()
5953                .map(|x| if x.is_some() { 9 } else { 1 })
5954                .sum::<usize>()
5955        }
5956        // v7.12.0: tsvector dense body — [u16 lexeme_count][per
5957        // lex: u16 word_len + utf-8 word + u16 pos_count + (u16
5958        // LE * pos_count) + u8 weight].
5959        Value::TsVector(lexs) => {
5960            let mut n = 2;
5961            for l in lexs {
5962                n += 2 + l.word.len() + 2 + 2 * l.positions.len() + 1;
5963            }
5964            n
5965        }
5966        // v7.12.0: tsquery dense body — prefix-coded tree.
5967        // Sizing must match `write_tsquery_body` walker.
5968        Value::TsQuery(ast) => tsquery_encoded_len(ast),
5969        // NULL is encoded only in the bitmap, never in the body.
5970        Value::Null => 0,
5971        // INTERVAL has no on-disk encoding (see write_value_body).
5972        Value::Interval { .. } => {
5973            unreachable!("Value::Interval has no on-disk encoding")
5974        }
5975    }
5976}
5977
5978/// Encode one row's body in the v3.0.2 dense format (`FILE_VERSION`
5979/// 8): per-row NULL bitmap (1 bit/col, ceil(cols/8) bytes), then
5980/// each non-NULL cell as `write_value_body`. Same wire shape the
5981/// catalog snapshot writes per row inside its rows-block. Exposed
5982/// pub so v5.1+ cold-tier segment writers can produce row payloads
5983/// that the catalog [`decode_row_body_dense`] decodes 1:1.
5984///
5985/// `row.values.len()` must equal `schema.columns.len()` — the row
5986/// is expected to have been validated by `Table::insert` (the
5987/// engine's INSERT path) before reaching this function.
5988pub fn encode_row_body_dense(row: &Row, schema: &TableSchema) -> Vec<u8> {
5989    debug_assert_eq!(
5990        row.values.len(),
5991        schema.columns.len(),
5992        "dense encode: row arity must match schema"
5993    );
5994    let bitmap_bytes = schema.columns.len().div_ceil(8);
5995    // 8 B per fixed-width cell is a reasonable average; the buffer
5996    // grows past this for variable-width Text/Vector cells.
5997    let mut out = Vec::with_capacity(bitmap_bytes + schema.columns.len() * 8);
5998    let bitmap_offset = out.len();
5999    out.resize(bitmap_offset + bitmap_bytes, 0);
6000    for (i, v) in row.values.iter().enumerate() {
6001        if matches!(v, Value::Null) {
6002            out[bitmap_offset + i / 8] |= 1 << (i % 8);
6003        }
6004    }
6005    for (col_idx, v) in row.values.iter().enumerate() {
6006        if matches!(v, Value::Null) {
6007            continue;
6008        }
6009        write_value_body(&mut out, v, schema.columns[col_idx].ty);
6010    }
6011    out
6012}
6013
6014/// Inverse of [`encode_row_body_dense`]. Reads one row's body from
6015/// `bytes` and returns it plus the number of bytes consumed (so a
6016/// caller decoding a back-to-back stream of rows can advance its
6017/// cursor). Returns `StorageError::Corrupt` on truncation, bad
6018/// UTF-8, or unknown cell tags.
6019pub fn decode_row_body_dense(
6020    bytes: &[u8],
6021    schema: &TableSchema,
6022) -> Result<(Row, usize), StorageError> {
6023    let mut cur = Cursor::new(bytes);
6024    let bitmap_bytes = schema.columns.len().div_ceil(8);
6025    let mut bitmap_buf = [0u8; 32];
6026    if bitmap_bytes > bitmap_buf.len() {
6027        return Err(StorageError::Corrupt(format!(
6028            "row NULL bitmap {bitmap_bytes} B exceeds 32 B cap"
6029        )));
6030    }
6031    let slice = cur.take(bitmap_bytes)?;
6032    bitmap_buf[..bitmap_bytes].copy_from_slice(slice);
6033    let mut values = Vec::with_capacity(schema.columns.len());
6034    for (col_idx, col) in schema.columns.iter().enumerate() {
6035        if (bitmap_buf[col_idx / 8] >> (col_idx % 8)) & 1 == 1 {
6036            values.push(Value::Null);
6037        } else {
6038            values.push(cur.read_value_body(col.ty)?);
6039        }
6040    }
6041    Ok((Row { values }, cur.pos))
6042}
6043
6044/// Schema-driven dense value encoding (`FILE_VERSION` 8). Caller already
6045/// knows the column type and has decided this cell is non-NULL, so we
6046/// skip the per-cell type tag the v7 `write_value` was writing. NULL
6047/// is encoded via the per-row bitmap before this function runs, never
6048/// reaches here. Used only inside the row-encoding hot loop; the
6049/// schema-default path still goes through the legacy `write_value` so
6050/// DEFAULT values keep their self-describing tag and remain decodable
6051/// without consulting a column type.
6052fn write_value_body(out: &mut Vec<u8>, v: &Value, ty: DataType) {
6053    match (v, ty) {
6054        (Value::SmallInt(n), DataType::SmallInt) => out.extend_from_slice(&n.to_le_bytes()),
6055        (Value::Int(n), DataType::Int) => out.extend_from_slice(&n.to_le_bytes()),
6056        (Value::BigInt(n), DataType::BigInt) => out.extend_from_slice(&n.to_le_bytes()),
6057        (Value::Float(x), DataType::Float) => out.extend_from_slice(&x.to_le_bytes()),
6058        (Value::Bool(b), DataType::Bool) => out.push(u8::from(*b)),
6059        (Value::Text(s), DataType::Text | DataType::Varchar(_) | DataType::Char(_)) => {
6060            write_str(out, s);
6061        }
6062        (
6063            Value::Vector(v),
6064            DataType::Vector {
6065                encoding: VecEncoding::F32,
6066                ..
6067            },
6068        ) => {
6069            let dim = u32::try_from(v.len()).expect("vector dim fits in u32");
6070            out.extend_from_slice(&dim.to_le_bytes());
6071            for x in v {
6072                out.extend_from_slice(&x.to_le_bytes());
6073            }
6074        }
6075        // v6.0.1: SQ8 dense body — [u32 dim][f32 min][f32 max]
6076        // [u8 * dim]. Self-describes its length so v6 readers
6077        // walking rows of a v6 catalog stay aligned even if the
6078        // declared column dim drifts (defensive, not normally
6079        // possible since CREATE TABLE pins the dim).
6080        (
6081            Value::Sq8Vector(q),
6082            DataType::Vector {
6083                encoding: VecEncoding::Sq8,
6084                ..
6085            },
6086        ) => {
6087            let dim = u32::try_from(q.bytes.len()).expect("vector dim fits in u32");
6088            out.extend_from_slice(&dim.to_le_bytes());
6089            out.extend_from_slice(&q.min.to_le_bytes());
6090            out.extend_from_slice(&q.max.to_le_bytes());
6091            out.extend_from_slice(&q.bytes);
6092        }
6093        // v6.0.3: halfvec dense body — [u32 dim][u16 LE * dim].
6094        // The raw u16 bytes already live in `h.bytes` little-
6095        // endian, so we just splat them.
6096        (
6097            Value::HalfVector(h),
6098            DataType::Vector {
6099                encoding: VecEncoding::F16,
6100                ..
6101            },
6102        ) => {
6103            let dim = u32::try_from(h.dim()).expect("vector dim fits in u32");
6104            out.extend_from_slice(&dim.to_le_bytes());
6105            out.extend_from_slice(&h.bytes);
6106        }
6107        (Value::Numeric { scaled, .. }, DataType::Numeric { scale, .. }) => {
6108            out.extend_from_slice(&scaled.to_le_bytes());
6109            out.push(scale);
6110        }
6111        (Value::Date(d), DataType::Date) => out.extend_from_slice(&d.to_le_bytes()),
6112        (Value::Timestamp(t), DataType::Timestamp | DataType::Timestamptz) => {
6113            out.extend_from_slice(&t.to_le_bytes())
6114        }
6115        // v4.9: JSON stores as length-prefixed text; same shape as
6116        // Text — the type tag lives in the column schema, not the
6117        // per-cell body.
6118        (Value::Json(s), DataType::Json | DataType::Jsonb) => write_str(out, s),
6119        // v7.10.4: BYTEA shares the [u16 len][bytes] shape with
6120        // Text but writes raw bytes (no UTF-8 invariant).
6121        (Value::Bytes(b), DataType::Bytes) => {
6122            let len = u16::try_from(b.len()).expect("BYTEA cell ≤ 64 KiB");
6123            out.extend_from_slice(&len.to_le_bytes());
6124            out.extend_from_slice(b);
6125        }
6126        // v7.10.9: TEXT[] dense body — [u16 count][per element:
6127        // u8 null flag + (when non-null) u16 len + utf-8 bytes].
6128        (Value::TextArray(items), DataType::TextArray) => {
6129            let count = u16::try_from(items.len()).expect("TEXT[] ≤ 65k elements");
6130            out.extend_from_slice(&count.to_le_bytes());
6131            for item in items {
6132                match item {
6133                    None => out.push(1),
6134                    Some(s) => {
6135                        out.push(0);
6136                        let len = u16::try_from(s.len()).expect("TEXT[] element ≤ 64 KiB");
6137                        out.extend_from_slice(&len.to_le_bytes());
6138                        out.extend_from_slice(s.as_bytes());
6139                    }
6140                }
6141            }
6142        }
6143        // v7.11.12: INT[] dense body — [u16 count][per element:
6144        // u8 null + (when non-null) i32 LE].
6145        (Value::IntArray(items), DataType::IntArray) => {
6146            let count = u16::try_from(items.len()).expect("INT[] ≤ 65k elements");
6147            out.extend_from_slice(&count.to_le_bytes());
6148            for item in items {
6149                match item {
6150                    None => out.push(1),
6151                    Some(n) => {
6152                        out.push(0);
6153                        out.extend_from_slice(&n.to_le_bytes());
6154                    }
6155                }
6156            }
6157        }
6158        // v7.11.12: BIGINT[] dense body — [u16 count][per element:
6159        // u8 null + (when non-null) i64 LE].
6160        (Value::BigIntArray(items), DataType::BigIntArray) => {
6161            let count = u16::try_from(items.len()).expect("BIGINT[] ≤ 65k elements");
6162            out.extend_from_slice(&count.to_le_bytes());
6163            for item in items {
6164                match item {
6165                    None => out.push(1),
6166                    Some(n) => {
6167                        out.push(0);
6168                        out.extend_from_slice(&n.to_le_bytes());
6169                    }
6170                }
6171            }
6172        }
6173        // v7.12.0: tsvector dense body — see `value_body_encoded_len`
6174        // for layout. Lexemes are written in their already-sorted order.
6175        (Value::TsVector(lexs), DataType::TsVector) => write_tsvector_body(out, lexs),
6176        // v7.12.0: tsquery dense body — prefix-coded tree.
6177        (Value::TsQuery(ast), DataType::TsQuery) => write_tsquery_body(out, ast),
6178        // Type mismatch shouldn't happen — `Table::insert` validates
6179        // value type against column type before pushing. Treat as a
6180        // bug, not a runtime error.
6181        (other, ty) => unreachable!(
6182            "schema-driven encode received mismatched value/type pair: \
6183             value tag={:?}, column type={:?}",
6184            other.data_type(),
6185            ty
6186        ),
6187    }
6188}
6189
6190fn write_value(out: &mut Vec<u8>, v: &Value) {
6191    match v {
6192        Value::Null => out.push(0),
6193        Value::SmallInt(n) => {
6194            out.push(7);
6195            out.extend_from_slice(&n.to_le_bytes());
6196        }
6197        Value::Int(n) => {
6198            out.push(1);
6199            out.extend_from_slice(&n.to_le_bytes());
6200        }
6201        Value::BigInt(n) => {
6202            out.push(2);
6203            out.extend_from_slice(&n.to_le_bytes());
6204        }
6205        Value::Float(x) => {
6206            out.push(3);
6207            out.extend_from_slice(&x.to_le_bytes());
6208        }
6209        // v4.9: JSON shares the tag-4 (Text) on-disk encoding —
6210        // schema decides which variant comes back on read. The
6211        // bodies are byte-identical so collapsing the match keeps
6212        // clippy::match_same_arms quiet.
6213        Value::Text(s) | Value::Json(s) => {
6214            out.push(4);
6215            write_str(out, s);
6216        }
6217        Value::Bool(b) => {
6218            out.push(5);
6219            out.push(u8::from(*b));
6220        }
6221        Value::Vector(v) => {
6222            out.push(6);
6223            let dim = u32::try_from(v.len()).expect("vector dim fits in u32");
6224            out.extend_from_slice(&dim.to_le_bytes());
6225            for x in v {
6226                out.extend_from_slice(&x.to_le_bytes());
6227            }
6228        }
6229        // v6.0.1: new tag 11 for an SQ8 cell carried with its full
6230        // header. Layout matches the dense row body shape so a
6231        // round-trip through write_value → read_value bit-equals
6232        // the original `Value::Sq8Vector`.
6233        Value::Sq8Vector(q) => {
6234            out.push(11);
6235            let dim = u32::try_from(q.bytes.len()).expect("vector dim fits in u32");
6236            out.extend_from_slice(&dim.to_le_bytes());
6237            out.extend_from_slice(&q.min.to_le_bytes());
6238            out.extend_from_slice(&q.max.to_le_bytes());
6239            out.extend_from_slice(&q.bytes);
6240        }
6241        // v6.0.3: tag 12 for a HalfVector cell.
6242        // Layout: `[u32 dim][u16 LE × dim]` — bit-identical to the
6243        // dense row body so `write_value` / `read_value` bit-equal
6244        // the original `Value::HalfVector`.
6245        Value::HalfVector(h) => {
6246            out.push(12);
6247            let dim = u32::try_from(h.dim()).expect("vector dim fits in u32");
6248            out.extend_from_slice(&dim.to_le_bytes());
6249            out.extend_from_slice(&h.bytes);
6250        }
6251        Value::Numeric { scaled, scale } => {
6252            out.push(8);
6253            out.extend_from_slice(&scaled.to_le_bytes());
6254            out.push(*scale);
6255        }
6256        Value::Date(d) => {
6257            out.push(9);
6258            out.extend_from_slice(&d.to_le_bytes());
6259        }
6260        Value::Timestamp(t) => {
6261            out.push(10);
6262            out.extend_from_slice(&t.to_le_bytes());
6263        }
6264        // Interval is a runtime-only value (no on-disk representation in
6265        // v2.11). CREATE TABLE rejects `DataType::Interval` columns, so a
6266        // Value::Interval here would mean the engine bypassed that gate.
6267        Value::Interval { .. } => {
6268            unreachable!(
6269                "Value::Interval has no on-disk encoding; engine must reject it before write"
6270            )
6271        }
6272        // v7.10.4: BYTEA — [u8 tag=13_b][u16 len][bytes]. Tag
6273        // distinct from Text (4) so the schema-agnostic
6274        // read_value path can disambiguate. (Tag 11 is taken by
6275        // the WAL `auto_commit_sql` shape elsewhere, hence 14.)
6276        Value::Bytes(b) => {
6277            out.push(14);
6278            let len = u16::try_from(b.len()).expect("BYTEA value ≤ 64 KiB");
6279            out.extend_from_slice(&len.to_le_bytes());
6280            out.extend_from_slice(b);
6281        }
6282        // v7.10.9: TEXT[] — [u8 tag=15][u16 count][per elem: u8
6283        // null + (if non-null) u16 len + utf-8 bytes].
6284        Value::TextArray(items) => {
6285            out.push(15);
6286            let count = u16::try_from(items.len()).expect("TEXT[] ≤ 65k elements");
6287            out.extend_from_slice(&count.to_le_bytes());
6288            for item in items {
6289                match item {
6290                    None => out.push(1),
6291                    Some(s) => {
6292                        out.push(0);
6293                        let len = u16::try_from(s.len()).expect("TEXT[] element ≤ 64 KiB");
6294                        out.extend_from_slice(&len.to_le_bytes());
6295                        out.extend_from_slice(s.as_bytes());
6296                    }
6297                }
6298            }
6299        }
6300        // v7.11.12: INT[] — tag 16. [u16 count][per elem: u8 null +
6301        // (if non-null) i32 LE].
6302        Value::IntArray(items) => {
6303            out.push(16);
6304            let count = u16::try_from(items.len()).expect("INT[] ≤ 65k elements");
6305            out.extend_from_slice(&count.to_le_bytes());
6306            for item in items {
6307                match item {
6308                    None => out.push(1),
6309                    Some(n) => {
6310                        out.push(0);
6311                        out.extend_from_slice(&n.to_le_bytes());
6312                    }
6313                }
6314            }
6315        }
6316        // v7.11.12: BIGINT[] — tag 17. [u16 count][per elem: u8 null +
6317        // (if non-null) i64 LE].
6318        Value::BigIntArray(items) => {
6319            out.push(17);
6320            let count = u16::try_from(items.len()).expect("BIGINT[] ≤ 65k elements");
6321            out.extend_from_slice(&count.to_le_bytes());
6322            for item in items {
6323                match item {
6324                    None => out.push(1),
6325                    Some(n) => {
6326                        out.push(0);
6327                        out.extend_from_slice(&n.to_le_bytes());
6328                    }
6329                }
6330            }
6331        }
6332        // v7.12.0: tsvector — tag 18. Body shape matches
6333        // `write_tsvector_body`.
6334        Value::TsVector(lexs) => {
6335            out.push(18);
6336            write_tsvector_body(out, lexs);
6337        }
6338        // v7.12.0: tsquery — tag 19. Body shape matches
6339        // `write_tsquery_body`.
6340        Value::TsQuery(ast) => {
6341            out.push(19);
6342            write_tsquery_body(out, ast);
6343        }
6344    }
6345}
6346
6347/// v7.12.0: shared tsvector body writer (used by both dense and
6348/// schema-agnostic codecs).
6349fn write_tsvector_body(out: &mut Vec<u8>, lexs: &[TsLexeme]) {
6350    let count = u16::try_from(lexs.len()).expect("tsvector ≤ 65k lexemes");
6351    out.extend_from_slice(&count.to_le_bytes());
6352    for l in lexs {
6353        let wlen = u16::try_from(l.word.len()).expect("tsvector word ≤ 64 KiB");
6354        out.extend_from_slice(&wlen.to_le_bytes());
6355        out.extend_from_slice(l.word.as_bytes());
6356        let plen = u16::try_from(l.positions.len()).expect("tsvector pos count ≤ 65k");
6357        out.extend_from_slice(&plen.to_le_bytes());
6358        for p in &l.positions {
6359            out.extend_from_slice(&p.to_le_bytes());
6360        }
6361        out.push(l.weight);
6362    }
6363}
6364
6365/// v7.12.0: shared tsquery body writer. Prefix-coded tree: each
6366/// node starts with `[u8 tag]` then a tag-specific payload. Tags:
6367/// 0=Term, 1=And, 2=Or, 3=Not, 4=Phrase.
6368fn write_tsquery_body(out: &mut Vec<u8>, ast: &TsQueryAst) {
6369    match ast {
6370        TsQueryAst::Term { word, weight_mask } => {
6371            out.push(0);
6372            let len = u16::try_from(word.len()).expect("tsquery term ≤ 64 KiB");
6373            out.extend_from_slice(&len.to_le_bytes());
6374            out.extend_from_slice(word.as_bytes());
6375            out.push(*weight_mask);
6376        }
6377        TsQueryAst::And(a, b) => {
6378            out.push(1);
6379            write_tsquery_body(out, a);
6380            write_tsquery_body(out, b);
6381        }
6382        TsQueryAst::Or(a, b) => {
6383            out.push(2);
6384            write_tsquery_body(out, a);
6385            write_tsquery_body(out, b);
6386        }
6387        TsQueryAst::Not(x) => {
6388            out.push(3);
6389            write_tsquery_body(out, x);
6390        }
6391        TsQueryAst::Phrase {
6392            left,
6393            right,
6394            distance,
6395        } => {
6396            out.push(4);
6397            out.extend_from_slice(&distance.to_le_bytes());
6398            write_tsquery_body(out, left);
6399            write_tsquery_body(out, right);
6400        }
6401    }
6402}
6403
6404/// v7.12.0: byte length that `write_tsquery_body` would emit.
6405fn tsquery_encoded_len(ast: &TsQueryAst) -> usize {
6406    match ast {
6407        TsQueryAst::Term { word, .. } => 1 + 2 + word.len() + 1,
6408        TsQueryAst::And(a, b) | TsQueryAst::Or(a, b) => {
6409            1 + tsquery_encoded_len(a) + tsquery_encoded_len(b)
6410        }
6411        TsQueryAst::Not(x) => 1 + tsquery_encoded_len(x),
6412        TsQueryAst::Phrase { left, right, .. } => {
6413            1 + 2 + tsquery_encoded_len(left) + tsquery_encoded_len(right)
6414        }
6415    }
6416}
6417
6418fn write_u16(out: &mut Vec<u8>, n: u16) {
6419    out.extend_from_slice(&n.to_le_bytes());
6420}
6421fn write_u32(out: &mut Vec<u8>, n: u32) {
6422    out.extend_from_slice(&n.to_le_bytes());
6423}
6424fn write_str(out: &mut Vec<u8>, s: &str) {
6425    let len = u16::try_from(s.len()).expect("identifier / text fits in u16");
6426    write_u16(out, len);
6427    out.extend_from_slice(s.as_bytes());
6428}
6429
6430/// v7.12.4 — long-string variant: `[u32 LE len][bytes]`. For
6431/// payloads that can plausibly exceed 64 KiB (notably PL/pgSQL
6432/// function bodies). Identifiers + short text continue to use
6433/// the u16 [`write_str`] codec.
6434fn write_str_long(out: &mut Vec<u8>, s: &str) {
6435    let len = u32::try_from(s.len()).expect("function body fits in u32");
6436    write_u32(out, len);
6437    out.extend_from_slice(s.as_bytes());
6438}
6439
6440/// Serialise an [`IndexKey`] using the v9 tagged codec. `read_index_key`
6441/// is the inverse. v8 catalogs never wrote index keys (`BTree` entries were
6442/// rebuilt from `Table::rows`), so this codec is v9+ only.
6443fn write_index_key(out: &mut Vec<u8>, key: &IndexKey) {
6444    match key {
6445        IndexKey::Int(n) => {
6446            out.push(INDEX_KEY_TAG_INT);
6447            out.extend_from_slice(&n.to_le_bytes());
6448        }
6449        IndexKey::Text(s) => {
6450            out.push(INDEX_KEY_TAG_TEXT);
6451            write_str(out, s);
6452        }
6453        IndexKey::Bool(b) => {
6454            out.push(INDEX_KEY_TAG_BOOL);
6455            out.push(u8::from(*b));
6456        }
6457    }
6458}
6459
6460struct Cursor<'a> {
6461    buf: &'a [u8],
6462    pos: usize,
6463}
6464
6465impl<'a> Cursor<'a> {
6466    const fn new(buf: &'a [u8]) -> Self {
6467        Self { buf, pos: 0 }
6468    }
6469
6470    fn take(&mut self, n: usize) -> Result<&'a [u8], StorageError> {
6471        let end = self
6472            .pos
6473            .checked_add(n)
6474            .ok_or_else(|| StorageError::Corrupt(format!("length overflow taking {n} bytes")))?;
6475        if end > self.buf.len() {
6476            return Err(StorageError::Corrupt(format!(
6477                "unexpected EOF at offset {} (wanted {n} more bytes)",
6478                self.pos
6479            )));
6480        }
6481        let s = &self.buf[self.pos..end];
6482        self.pos = end;
6483        Ok(s)
6484    }
6485
6486    fn read_u8(&mut self) -> Result<u8, StorageError> {
6487        Ok(self.take(1)?[0])
6488    }
6489    fn read_u16(&mut self) -> Result<u16, StorageError> {
6490        let s = self.take(2)?;
6491        Ok(u16::from_le_bytes([s[0], s[1]]))
6492    }
6493    fn read_u32(&mut self) -> Result<u32, StorageError> {
6494        let s = self.take(4)?;
6495        Ok(u32::from_le_bytes([s[0], s[1], s[2], s[3]]))
6496    }
6497    fn read_i32(&mut self) -> Result<i32, StorageError> {
6498        let s = self.take(4)?;
6499        Ok(i32::from_le_bytes([s[0], s[1], s[2], s[3]]))
6500    }
6501    /// v6.7.2 — u64 LE read for the per-table `hot_tier_bytes`
6502    /// catalog appendix.
6503    fn read_u64(&mut self) -> Result<u64, StorageError> {
6504        let s = self.take(8)?;
6505        Ok(u64::from_le_bytes([
6506            s[0], s[1], s[2], s[3], s[4], s[5], s[6], s[7],
6507        ]))
6508    }
6509    fn read_i64(&mut self) -> Result<i64, StorageError> {
6510        let s = self.take(8)?;
6511        let arr: [u8; 8] = s.try_into().expect("checked");
6512        Ok(i64::from_le_bytes(arr))
6513    }
6514    fn read_f64(&mut self) -> Result<f64, StorageError> {
6515        let s = self.take(8)?;
6516        let arr: [u8; 8] = s.try_into().expect("checked");
6517        Ok(f64::from_le_bytes(arr))
6518    }
6519    fn read_f32(&mut self) -> Result<f32, StorageError> {
6520        let s = self.take(4)?;
6521        Ok(f32::from_le_bytes([s[0], s[1], s[2], s[3]]))
6522    }
6523    fn read_str(&mut self) -> Result<String, StorageError> {
6524        let len = self.read_u16()? as usize;
6525        let bytes = self.take(len)?;
6526        core::str::from_utf8(bytes)
6527            .map(String::from)
6528            .map_err(|_| StorageError::Corrupt("invalid UTF-8 in identifier or text".into()))
6529    }
6530
6531    /// v7.12.4 — long-string variant for payloads written via
6532    /// [`write_str_long`] (u32-length prefix). Used for PL/pgSQL
6533    /// function bodies which can plausibly exceed 64 KiB.
6534    fn read_str_long(&mut self) -> Result<String, StorageError> {
6535        let len = self.read_u32()? as usize;
6536        let bytes = self.take(len)?;
6537        core::str::from_utf8(bytes)
6538            .map(String::from)
6539            .map_err(|_| StorageError::Corrupt("invalid UTF-8 in long-string payload".into()))
6540    }
6541
6542    /// Parse an [`IndexKey`] emitted by `write_index_key` (v9 tagged
6543    /// codec). Returns `StorageError::Corrupt` on unknown tag or
6544    /// truncated payload.
6545    fn read_index_key(&mut self) -> Result<IndexKey, StorageError> {
6546        let tag = self.read_u8()?;
6547        match tag {
6548            INDEX_KEY_TAG_INT => Ok(IndexKey::Int(self.read_i64()?)),
6549            INDEX_KEY_TAG_TEXT => Ok(IndexKey::Text(self.read_str()?)),
6550            INDEX_KEY_TAG_BOOL => Ok(IndexKey::Bool(self.read_u8()? != 0)),
6551            other => Err(StorageError::Corrupt(format!(
6552                "unknown index key tag: {other}"
6553            ))),
6554        }
6555    }
6556    /// Schema-driven dense value decode (`FILE_VERSION` 8). Caller has
6557    /// already cleared the NULL bit from the row bitmap; we read the
6558    /// fixed-width body for the given column type. Used inside the row
6559    /// hot loop; column defaults still go through `read_value` (which
6560    /// reads its own type tag) so DEFAULT round-trips without a schema.
6561    fn read_value_body(&mut self, ty: DataType) -> Result<Value, StorageError> {
6562        match ty {
6563            DataType::SmallInt => {
6564                let s = self.take(2)?;
6565                Ok(Value::SmallInt(i16::from_le_bytes([s[0], s[1]])))
6566            }
6567            DataType::Int => Ok(Value::Int(self.read_i32()?)),
6568            DataType::BigInt => Ok(Value::BigInt(self.read_i64()?)),
6569            DataType::Float => Ok(Value::Float(self.read_f64()?)),
6570            DataType::Bool => Ok(Value::Bool(self.read_u8()? != 0)),
6571            DataType::Text | DataType::Varchar(_) | DataType::Char(_) => {
6572                Ok(Value::Text(self.read_str()?))
6573            }
6574            DataType::Vector {
6575                encoding: VecEncoding::F32,
6576                ..
6577            } => {
6578                let dim = self.read_u32()? as usize;
6579                let mut v = Vec::with_capacity(dim);
6580                for _ in 0..dim {
6581                    let bytes: [u8; 4] = self.take(4)?.try_into().expect("checked");
6582                    v.push(f32::from_le_bytes(bytes));
6583                }
6584                Ok(Value::Vector(v))
6585            }
6586            DataType::Vector {
6587                encoding: VecEncoding::Sq8,
6588                ..
6589            } => {
6590                let dim = self.read_u32()? as usize;
6591                let min = self.read_f32()?;
6592                let max = self.read_f32()?;
6593                let bytes = self.take(dim)?.to_vec();
6594                Ok(Value::Sq8Vector(quantize::Sq8Vector { min, max, bytes }))
6595            }
6596            DataType::Vector {
6597                encoding: VecEncoding::F16,
6598                ..
6599            } => {
6600                let dim = self.read_u32()? as usize;
6601                let bytes = self.take(dim * 2)?.to_vec();
6602                Ok(Value::HalfVector(halfvec::HalfVector { bytes }))
6603            }
6604            DataType::Numeric { .. } => {
6605                let s = self.take(16)?;
6606                let arr: [u8; 16] = s.try_into().expect("checked");
6607                let scaled = i128::from_le_bytes(arr);
6608                let scale = self.read_u8()?;
6609                Ok(Value::Numeric { scaled, scale })
6610            }
6611            DataType::Date => Ok(Value::Date(self.read_i32()?)),
6612            DataType::Timestamp => Ok(Value::Timestamp(self.read_i64()?)),
6613            DataType::Timestamptz => Ok(Value::Timestamp(self.read_i64()?)),
6614            DataType::Jsonb => Ok(Value::Json(self.read_str()?)),
6615            DataType::Interval => {
6616                // Defensive — schema gate (CREATE TABLE rejects Interval
6617                // columns) means this branch can't be hit through normal
6618                // flow; reject corrupt files explicitly rather than
6619                // panic.
6620                Err(StorageError::Corrupt(
6621                    "INTERVAL column found on disk — runtime-only type, v3.0.2 rejects it".into(),
6622                ))
6623            }
6624            DataType::Json => Ok(Value::Json(self.read_str()?)),
6625            // v7.10.4: BYTEA on-disk is [u16 len][bytes]. Same wire
6626            // shape as Text, but read as raw Vec<u8>.
6627            DataType::Bytes => {
6628                let len = self.read_u16()? as usize;
6629                let bytes = self.take(len)?.to_vec();
6630                Ok(Value::Bytes(bytes))
6631            }
6632            // v7.10.9: TEXT[] dense body.
6633            DataType::TextArray => {
6634                let count = self.read_u16()? as usize;
6635                let mut items: Vec<Option<String>> = Vec::with_capacity(count);
6636                for _ in 0..count {
6637                    match self.read_u8()? {
6638                        0 => items.push(Some(self.read_str()?)),
6639                        1 => items.push(None),
6640                        other => {
6641                            return Err(StorageError::Corrupt(format!(
6642                                "TEXT[] null flag: unknown byte {other}"
6643                            )));
6644                        }
6645                    }
6646                }
6647                Ok(Value::TextArray(items))
6648            }
6649            // v7.11.12: INT[] dense body.
6650            DataType::IntArray => {
6651                let count = self.read_u16()? as usize;
6652                let mut items: Vec<Option<i32>> = Vec::with_capacity(count);
6653                for _ in 0..count {
6654                    match self.read_u8()? {
6655                        0 => items.push(Some(self.read_i32()?)),
6656                        1 => items.push(None),
6657                        other => {
6658                            return Err(StorageError::Corrupt(format!(
6659                                "INT[] null flag: unknown byte {other}"
6660                            )));
6661                        }
6662                    }
6663                }
6664                Ok(Value::IntArray(items))
6665            }
6666            // v7.11.12: BIGINT[] dense body.
6667            DataType::BigIntArray => {
6668                let count = self.read_u16()? as usize;
6669                let mut items: Vec<Option<i64>> = Vec::with_capacity(count);
6670                for _ in 0..count {
6671                    match self.read_u8()? {
6672                        0 => items.push(Some(self.read_i64()?)),
6673                        1 => items.push(None),
6674                        other => {
6675                            return Err(StorageError::Corrupt(format!(
6676                                "BIGINT[] null flag: unknown byte {other}"
6677                            )));
6678                        }
6679                    }
6680                }
6681                Ok(Value::BigIntArray(items))
6682            }
6683            // v7.12.0: tsvector dense body — [u16 lex_count]
6684            // [per lex: u16 word_len + utf-8 word + u16 pos_count
6685            // + (u16 LE * pos_count) + u8 weight].
6686            DataType::TsVector => Ok(Value::TsVector(self.read_tsvector_body()?)),
6687            DataType::TsQuery => Ok(Value::TsQuery(self.read_tsquery_body()?)),
6688        }
6689    }
6690
6691    /// v7.12.0 — read a tsvector body emitted by `write_tsvector_body`.
6692    fn read_tsvector_body(&mut self) -> Result<Vec<TsLexeme>, StorageError> {
6693        let count = self.read_u16()? as usize;
6694        let mut out = Vec::with_capacity(count);
6695        for _ in 0..count {
6696            let word = self.read_str()?;
6697            let pos_count = self.read_u16()? as usize;
6698            let mut positions = Vec::with_capacity(pos_count);
6699            for _ in 0..pos_count {
6700                positions.push(self.read_u16()?);
6701            }
6702            let weight = self.read_u8()?;
6703            out.push(TsLexeme {
6704                word,
6705                positions,
6706                weight,
6707            });
6708        }
6709        Ok(out)
6710    }
6711
6712    /// v7.12.0 — read a tsquery body emitted by `write_tsquery_body`.
6713    fn read_tsquery_body(&mut self) -> Result<TsQueryAst, StorageError> {
6714        let tag = self.read_u8()?;
6715        match tag {
6716            0 => {
6717                let word = self.read_str()?;
6718                let weight_mask = self.read_u8()?;
6719                Ok(TsQueryAst::Term { word, weight_mask })
6720            }
6721            1 => {
6722                let a = self.read_tsquery_body()?;
6723                let b = self.read_tsquery_body()?;
6724                Ok(TsQueryAst::And(Box::new(a), Box::new(b)))
6725            }
6726            2 => {
6727                let a = self.read_tsquery_body()?;
6728                let b = self.read_tsquery_body()?;
6729                Ok(TsQueryAst::Or(Box::new(a), Box::new(b)))
6730            }
6731            3 => {
6732                let x = self.read_tsquery_body()?;
6733                Ok(TsQueryAst::Not(Box::new(x)))
6734            }
6735            4 => {
6736                let distance = self.read_u16()?;
6737                let left = self.read_tsquery_body()?;
6738                let right = self.read_tsquery_body()?;
6739                Ok(TsQueryAst::Phrase {
6740                    left: Box::new(left),
6741                    right: Box::new(right),
6742                    distance,
6743                })
6744            }
6745            other => Err(StorageError::Corrupt(format!(
6746                "tsquery: unknown node tag {other}"
6747            ))),
6748        }
6749    }
6750
6751    fn read_value(&mut self) -> Result<Value, StorageError> {
6752        let tag = self.read_u8()?;
6753        match tag {
6754            0 => Ok(Value::Null),
6755            1 => Ok(Value::Int(self.read_i32()?)),
6756            2 => Ok(Value::BigInt(self.read_i64()?)),
6757            3 => Ok(Value::Float(self.read_f64()?)),
6758            4 => Ok(Value::Text(self.read_str()?)),
6759            5 => Ok(Value::Bool(self.read_u8()? != 0)),
6760            6 => {
6761                let dim = self.read_u32()? as usize;
6762                let mut v = Vec::with_capacity(dim);
6763                for _ in 0..dim {
6764                    let bytes: [u8; 4] = self.take(4)?.try_into().expect("checked");
6765                    v.push(f32::from_le_bytes(bytes));
6766                }
6767                Ok(Value::Vector(v))
6768            }
6769            7 => {
6770                let s = self.take(2)?;
6771                Ok(Value::SmallInt(i16::from_le_bytes([s[0], s[1]])))
6772            }
6773            8 => {
6774                let s = self.take(16)?;
6775                let arr: [u8; 16] = s.try_into().expect("checked");
6776                let scaled = i128::from_le_bytes(arr);
6777                let scale = self.read_u8()?;
6778                Ok(Value::Numeric { scaled, scale })
6779            }
6780            9 => Ok(Value::Date(self.read_i32()?)),
6781            10 => Ok(Value::Timestamp(self.read_i64()?)),
6782            // v6.0.1: tag 11 — Sq8Vector. Pre-v6 readers fall
6783            // through to the catch-all and surface
6784            // `Corrupt("unknown value tag")`, matching the
6785            // forward-compat fence on the column-type side.
6786            11 => {
6787                let dim = self.read_u32()? as usize;
6788                let min = self.read_f32()?;
6789                let max = self.read_f32()?;
6790                let bytes = self.take(dim)?.to_vec();
6791                Ok(Value::Sq8Vector(quantize::Sq8Vector { min, max, bytes }))
6792            }
6793            // v6.0.3: tag 12 — HalfVector. Same forward-compat
6794            // fence story as tag 11.
6795            12 => {
6796                let dim = self.read_u32()? as usize;
6797                let bytes = self.take(dim * 2)?.to_vec();
6798                Ok(Value::HalfVector(halfvec::HalfVector { bytes }))
6799            }
6800            // v7.10.4: tag 14 — BYTEA. [u16 len][bytes].
6801            14 => {
6802                let len = self.read_u16()? as usize;
6803                let bytes = self.take(len)?.to_vec();
6804                Ok(Value::Bytes(bytes))
6805            }
6806            // v7.10.9: tag 15 — TEXT[]. [u16 count][per elem: u8
6807            // null + (when non-null) u16 len + utf-8 bytes].
6808            15 => {
6809                let count = self.read_u16()? as usize;
6810                let mut items: Vec<Option<String>> = Vec::with_capacity(count);
6811                for _ in 0..count {
6812                    match self.read_u8()? {
6813                        0 => items.push(Some(self.read_str()?)),
6814                        1 => items.push(None),
6815                        other => {
6816                            return Err(StorageError::Corrupt(format!(
6817                                "TEXT[] null flag in value tag: unknown byte {other}"
6818                            )));
6819                        }
6820                    }
6821                }
6822                Ok(Value::TextArray(items))
6823            }
6824            // v7.11.12: tags 16/17 — INT[] / BIGINT[].
6825            16 => {
6826                let count = self.read_u16()? as usize;
6827                let mut items: Vec<Option<i32>> = Vec::with_capacity(count);
6828                for _ in 0..count {
6829                    match self.read_u8()? {
6830                        0 => items.push(Some(self.read_i32()?)),
6831                        1 => items.push(None),
6832                        other => {
6833                            return Err(StorageError::Corrupt(format!(
6834                                "INT[] null flag in value tag: unknown byte {other}"
6835                            )));
6836                        }
6837                    }
6838                }
6839                Ok(Value::IntArray(items))
6840            }
6841            17 => {
6842                let count = self.read_u16()? as usize;
6843                let mut items: Vec<Option<i64>> = Vec::with_capacity(count);
6844                for _ in 0..count {
6845                    match self.read_u8()? {
6846                        0 => items.push(Some(self.read_i64()?)),
6847                        1 => items.push(None),
6848                        other => {
6849                            return Err(StorageError::Corrupt(format!(
6850                                "BIGINT[] null flag in value tag: unknown byte {other}"
6851                            )));
6852                        }
6853                    }
6854                }
6855                Ok(Value::BigIntArray(items))
6856            }
6857            // v7.12.0: tag 18 — tsvector. Body matches the dense
6858            // form (`read_tsvector_body`).
6859            18 => Ok(Value::TsVector(self.read_tsvector_body()?)),
6860            // v7.12.0: tag 19 — tsquery.
6861            19 => Ok(Value::TsQuery(self.read_tsquery_body()?)),
6862            other => Err(StorageError::Corrupt(format!("unknown value tag: {other}"))),
6863        }
6864    }
6865
6866    /// Read an NSW graph that was emitted via `write_nsw_graph`. `m`
6867    /// is passed in because it was already consumed from the per-
6868    /// index header. Returns the reconstituted `NswGraph`.
6869    fn read_nsw_graph(&mut self, m: usize) -> Result<NswGraph, StorageError> {
6870        let m_max_0 = self.read_u16()? as usize;
6871        let entry_raw = self.read_u32()?;
6872        let entry = if entry_raw == u32::MAX {
6873            None
6874        } else {
6875            Some(entry_raw as usize)
6876        };
6877        let entry_level = self.read_u8()?;
6878        let node_count = self.read_u32()? as usize;
6879        // v5.5.0: levels/per-layer are PV-backed in memory, but the wire
6880        // format is unchanged — decode element-by-element into a PV via
6881        // push_mut (transient in-place, no per-element path-copy here since
6882        // the freshly-built PV is uniquely owned).
6883        let mut levels: PersistentVec<u8> = PersistentVec::new();
6884        for _ in 0..node_count {
6885            levels.push_mut(self.read_u8()?);
6886        }
6887        let layer_count = self.read_u8()? as usize;
6888        let mut layers: Vec<PersistentVec<Vec<u32>>> = Vec::with_capacity(layer_count);
6889        for _ in 0..layer_count {
6890            let n = self.read_u32()? as usize;
6891            let mut per_layer: PersistentVec<Vec<u32>> = PersistentVec::new();
6892            for _ in 0..n {
6893                let cnt = self.read_u16()? as usize;
6894                let mut row: Vec<u32> = Vec::with_capacity(cnt);
6895                for _ in 0..cnt {
6896                    row.push(self.read_u32()?);
6897                }
6898                per_layer.push_mut(row);
6899            }
6900            layers.push(per_layer);
6901        }
6902        Ok(NswGraph {
6903            m,
6904            m_max_0,
6905            entry,
6906            entry_level,
6907            levels,
6908            layers,
6909        })
6910    }
6911}
6912
6913#[cfg(test)]
6914mod tests {
6915    use super::*;
6916    use alloc::string::ToString;
6917    use alloc::vec;
6918
6919    #[cfg(target_arch = "aarch64")]
6920    #[test]
6921    fn neon_l2_matches_scalar() {
6922        // For every dim that's a multiple of 4 (4, 8, 12, 16, 64,
6923        // 128, 256, 384, 512, 768, 1024, 1536), the NEON impl must
6924        // agree with the scalar reference within tight float
6925        // tolerance (FMA rounding differs from separate * + +).
6926        let dims = [4usize, 8, 12, 16, 64, 128, 256, 384, 512, 768, 1024, 1536];
6927        for &d in &dims {
6928            let mut state: u64 = (d as u64).wrapping_mul(0x9E37_79B9_7F4A_7C15);
6929            let mut a = Vec::with_capacity(d);
6930            let mut b = Vec::with_capacity(d);
6931            for _ in 0..d {
6932                state = state
6933                    .wrapping_mul(6_364_136_223_846_793_005)
6934                    .wrapping_add(1);
6935                #[allow(clippy::cast_precision_loss, clippy::cast_possible_truncation)]
6936                let x = (((state >> 32) & 0x00FF_FFFF) as f32) / (0x80_0000_u32 as f32) - 1.0;
6937                state = state
6938                    .wrapping_mul(6_364_136_223_846_793_005)
6939                    .wrapping_add(1);
6940                #[allow(clippy::cast_precision_loss, clippy::cast_possible_truncation)]
6941                let y = (((state >> 32) & 0x00FF_FFFF) as f32) / (0x80_0000_u32 as f32) - 1.0;
6942                a.push(x);
6943                b.push(y);
6944            }
6945            let scalar = l2_distance_sq_scalar(&a, &b);
6946            let neon = unsafe { l2_distance_sq_neon(&a, &b) };
6947            let tol = (scalar.abs().max(1e-6)) * 1e-4;
6948            assert!(
6949                (scalar - neon).abs() <= tol,
6950                "dim={d}: scalar={scalar} neon={neon} diff={}",
6951                (scalar - neon).abs()
6952            );
6953        }
6954    }
6955
6956    #[cfg(target_arch = "aarch64")]
6957    #[test]
6958    fn neon_inner_product_matches_scalar() {
6959        // v6.0.2 step 1: NEON IP must agree with scalar across every
6960        // production-shaped dim. FMA rounding differs from
6961        // separate * + +, so the tolerance scales with magnitude.
6962        let dims = [4usize, 8, 12, 16, 64, 128, 256, 512, 1024];
6963        for &d in &dims {
6964            let mut state: u64 = (d as u64).wrapping_mul(0x9E37_79B9_7F4A_7C15);
6965            let mut a = Vec::with_capacity(d);
6966            let mut b = Vec::with_capacity(d);
6967            for _ in 0..d {
6968                state = state
6969                    .wrapping_mul(6_364_136_223_846_793_005)
6970                    .wrapping_add(1);
6971                #[allow(clippy::cast_precision_loss, clippy::cast_possible_truncation)]
6972                let x = (((state >> 32) & 0x00FF_FFFF) as f32) / (0x80_0000_u32 as f32) - 1.0;
6973                state = state
6974                    .wrapping_mul(6_364_136_223_846_793_005)
6975                    .wrapping_add(1);
6976                #[allow(clippy::cast_precision_loss, clippy::cast_possible_truncation)]
6977                let y = (((state >> 32) & 0x00FF_FFFF) as f32) / (0x80_0000_u32 as f32) - 1.0;
6978                a.push(x);
6979                b.push(y);
6980            }
6981            let scalar = inner_product_scalar(&a, &b);
6982            let neon = unsafe { inner_product_neon(&a, &b) };
6983            #[allow(clippy::cast_precision_loss)]
6984            let tol = (scalar.abs().max(1e-6)) * 1e-4 + (d as f32) * 1e-6;
6985            assert!(
6986                (scalar - neon).abs() <= tol,
6987                "IP dim={d}: scalar={scalar} neon={neon} diff={}",
6988                (scalar - neon).abs()
6989            );
6990        }
6991    }
6992
6993    #[cfg(target_arch = "aarch64")]
6994    #[allow(clippy::similar_names)]
6995    #[test]
6996    fn neon_cosine_dot_norms_matches_scalar() {
6997        let dims = [4usize, 8, 12, 16, 64, 128, 256, 512, 1024];
6998        for &d in &dims {
6999            let mut state: u64 = (d as u64).wrapping_mul(0xBF58_476D_1CE4_E5B9);
7000            let mut a = Vec::with_capacity(d);
7001            let mut b = Vec::with_capacity(d);
7002            for _ in 0..d {
7003                state = state
7004                    .wrapping_mul(6_364_136_223_846_793_005)
7005                    .wrapping_add(1);
7006                #[allow(clippy::cast_precision_loss, clippy::cast_possible_truncation)]
7007                let x = (((state >> 32) & 0x00FF_FFFF) as f32) / (0x80_0000_u32 as f32) - 1.0;
7008                state = state
7009                    .wrapping_mul(6_364_136_223_846_793_005)
7010                    .wrapping_add(1);
7011                #[allow(clippy::cast_precision_loss, clippy::cast_possible_truncation)]
7012                let y = (((state >> 32) & 0x00FF_FFFF) as f32) / (0x80_0000_u32 as f32) - 1.0;
7013                a.push(x);
7014                b.push(y);
7015            }
7016            let (dot_s, na_s, nb_s) = cosine_dot_norms_scalar(&a, &b);
7017            let (dot_n, na_n, nb_n) = unsafe { cosine_dot_norms_neon(&a, &b) };
7018            #[allow(clippy::cast_precision_loss)]
7019            let tol_d = (dot_s.abs().max(1e-6)) * 1e-4 + (d as f32) * 1e-6;
7020            #[allow(clippy::cast_precision_loss)]
7021            let tol_n = (na_s.abs().max(1e-6)) * 1e-4 + (d as f32) * 1e-6;
7022            assert!(
7023                (dot_s - dot_n).abs() <= tol_d,
7024                "cosine dot dim={d}: scalar={dot_s} neon={dot_n}"
7025            );
7026            assert!(
7027                (na_s - na_n).abs() <= tol_n,
7028                "cosine na dim={d}: scalar={na_s} neon={na_n}"
7029            );
7030            assert!(
7031                (nb_s - nb_n).abs() <= tol_n,
7032                "cosine nb dim={d}: scalar={nb_s} neon={nb_n}"
7033            );
7034        }
7035    }
7036
7037    fn make_users_schema() -> TableSchema {
7038        TableSchema::new(
7039            "users",
7040            vec![
7041                ColumnSchema::new("id", DataType::Int, false),
7042                ColumnSchema::new("name", DataType::Text, false),
7043                ColumnSchema::new("score", DataType::Float, true),
7044            ],
7045        )
7046    }
7047
7048    #[test]
7049    fn value_type_tag_matches_variant() {
7050        assert_eq!(Value::Int(1).data_type(), Some(DataType::Int));
7051        assert_eq!(Value::BigInt(1).data_type(), Some(DataType::BigInt));
7052        assert_eq!(Value::Float(1.0).data_type(), Some(DataType::Float));
7053        assert_eq!(Value::Text("x".into()).data_type(), Some(DataType::Text));
7054        assert_eq!(Value::Bool(true).data_type(), Some(DataType::Bool));
7055        assert_eq!(Value::Null.data_type(), None);
7056        assert!(Value::Null.is_null());
7057        assert!(!Value::Int(0).is_null());
7058    }
7059
7060    #[test]
7061    fn sq8_value_reports_sq8_data_type() {
7062        // v6.0.1: a `Value::Sq8Vector` cell surfaces its dim
7063        // (= bytes.len()) and encoding through `data_type()` so
7064        // INSERT-time column type-checks (step 3) can route on
7065        // both shape and encoding.
7066        let q = crate::quantize::quantize(&[0.0, 0.25, 0.5, 0.75, 1.0]);
7067        let v = Value::Sq8Vector(q);
7068        assert_eq!(
7069            v.data_type(),
7070            Some(DataType::Vector {
7071                dim: 5,
7072                encoding: VecEncoding::Sq8,
7073            }),
7074        );
7075    }
7076
7077    #[test]
7078    fn datatype_display_matches_pg_keyword() {
7079        assert_eq!(DataType::Int.to_string(), "INT");
7080        assert_eq!(DataType::BigInt.to_string(), "BIGINT");
7081        assert_eq!(DataType::Float.to_string(), "FLOAT");
7082        assert_eq!(DataType::Text.to_string(), "TEXT");
7083        assert_eq!(DataType::Bool.to_string(), "BOOL");
7084    }
7085
7086    #[test]
7087    fn row_len_and_emptiness() {
7088        let r = Row::new(vec![Value::Int(1), Value::Null]);
7089        assert_eq!(r.len(), 2);
7090        assert!(!r.is_empty());
7091        assert!(Row::new(Vec::new()).is_empty());
7092    }
7093
7094    #[test]
7095    fn table_schema_column_position() {
7096        let s = make_users_schema();
7097        assert_eq!(s.column_position("id"), Some(0));
7098        assert_eq!(s.column_position("score"), Some(2));
7099        assert_eq!(s.column_position("missing"), None);
7100    }
7101
7102    #[test]
7103    fn catalog_create_table_then_lookup() {
7104        let mut cat = Catalog::new();
7105        cat.create_table(make_users_schema()).unwrap();
7106        assert_eq!(cat.table_count(), 1);
7107        assert!(cat.get("users").is_some());
7108        assert!(cat.get("nope").is_none());
7109    }
7110
7111    #[test]
7112    fn catalog_duplicate_table_is_rejected() {
7113        let mut cat = Catalog::new();
7114        cat.create_table(make_users_schema()).unwrap();
7115        let err = cat.create_table(make_users_schema()).unwrap_err();
7116        assert!(matches!(err, StorageError::DuplicateTable { ref name } if name == "users"));
7117    }
7118
7119    #[test]
7120    fn table_insert_happy_path_appends_row() {
7121        let mut cat = Catalog::new();
7122        cat.create_table(make_users_schema()).unwrap();
7123        let t = cat.get_mut("users").unwrap();
7124        t.insert(Row::new(vec![
7125            Value::Int(1),
7126            Value::Text("alice".into()),
7127            Value::Float(99.5),
7128        ]))
7129        .unwrap();
7130        assert_eq!(t.row_count(), 1);
7131        assert_eq!(t.rows()[0].values[1], Value::Text("alice".into()));
7132    }
7133
7134    #[test]
7135    fn table_insert_arity_mismatch() {
7136        let mut cat = Catalog::new();
7137        cat.create_table(make_users_schema()).unwrap();
7138        let t = cat.get_mut("users").unwrap();
7139        let err = t.insert(Row::new(vec![Value::Int(1)])).unwrap_err();
7140        assert!(matches!(
7141            err,
7142            StorageError::ArityMismatch {
7143                expected: 3,
7144                actual: 1
7145            }
7146        ));
7147        assert_eq!(t.row_count(), 0);
7148    }
7149
7150    #[test]
7151    fn table_insert_type_mismatch_reports_column() {
7152        let mut cat = Catalog::new();
7153        cat.create_table(make_users_schema()).unwrap();
7154        let t = cat.get_mut("users").unwrap();
7155        let err = t
7156            .insert(Row::new(vec![
7157                Value::Int(1),
7158                Value::Int(42), // name expects Text
7159                Value::Float(0.0),
7160            ]))
7161            .unwrap_err();
7162        match err {
7163            StorageError::TypeMismatch {
7164                ref column,
7165                expected,
7166                actual,
7167                position,
7168            } => {
7169                assert_eq!(column, "name");
7170                assert_eq!(expected, DataType::Text);
7171                assert_eq!(actual, DataType::Int);
7172                assert_eq!(position, 1);
7173            }
7174            other => panic!("unexpected: {other:?}"),
7175        }
7176        assert_eq!(t.row_count(), 0);
7177    }
7178
7179    #[test]
7180    fn table_insert_null_into_not_null_rejected() {
7181        let mut cat = Catalog::new();
7182        cat.create_table(make_users_schema()).unwrap();
7183        let t = cat.get_mut("users").unwrap();
7184        let err = t
7185            .insert(Row::new(vec![
7186                Value::Int(1),
7187                Value::Null, // name is NOT NULL
7188                Value::Float(1.0),
7189            ]))
7190            .unwrap_err();
7191        assert!(matches!(err, StorageError::NullInNotNull { ref column } if column == "name"));
7192    }
7193
7194    #[test]
7195    fn table_insert_null_into_nullable_ok() {
7196        let mut cat = Catalog::new();
7197        cat.create_table(make_users_schema()).unwrap();
7198        let t = cat.get_mut("users").unwrap();
7199        t.insert(Row::new(vec![
7200            Value::Int(1),
7201            Value::Text("bob".into()),
7202            Value::Null,
7203        ]))
7204        .unwrap();
7205        assert_eq!(t.row_count(), 1);
7206    }
7207
7208    #[test]
7209    fn catalog_get_mut_independent_per_table() {
7210        let mut cat = Catalog::new();
7211        cat.create_table(TableSchema::new(
7212            "a",
7213            vec![ColumnSchema::new("v", DataType::Int, false)],
7214        ))
7215        .unwrap();
7216        cat.create_table(TableSchema::new(
7217            "b",
7218            vec![ColumnSchema::new("v", DataType::Int, false)],
7219        ))
7220        .unwrap();
7221        cat.get_mut("a")
7222            .unwrap()
7223            .insert(Row::new(vec![Value::Int(1)]))
7224            .unwrap();
7225        assert_eq!(cat.get("a").unwrap().row_count(), 1);
7226        assert_eq!(cat.get("b").unwrap().row_count(), 0);
7227    }
7228
7229    // --- v0.6 persistence round-trips --------------------------------------
7230
7231    fn assert_round_trip(cat: &Catalog) {
7232        let bytes = cat.serialize();
7233        let restored = Catalog::deserialize(&bytes).expect("deserialize");
7234        // Compare semantic state: same tables in same order, same schema +
7235        // rows in each.
7236        assert_eq!(restored.table_count(), cat.table_count());
7237        for (a, b) in cat.tables.iter().zip(restored.tables.iter()) {
7238            assert_eq!(a.schema, b.schema);
7239            assert_eq!(a.rows, b.rows);
7240        }
7241    }
7242
7243    #[test]
7244    fn serialize_empty_catalog_round_trips() {
7245        assert_round_trip(&Catalog::new());
7246    }
7247
7248    #[test]
7249    fn serialize_single_empty_table_round_trips() {
7250        let mut cat = Catalog::new();
7251        cat.create_table(make_users_schema()).unwrap();
7252        assert_round_trip(&cat);
7253    }
7254
7255    #[test]
7256    fn nsw_clone_is_o1() {
7257        // v5.5.0: NswGraph::clone must be O(1) structural sharing, not the
7258        // pre-v5.5 O(N) element copy — it rides on Catalog::clone for every
7259        // group-commit write on a vector table. Build a non-trivial multi-
7260        // layer graph, clone it, and prove the clone shares the very same PV
7261        // storage (root+tail Arc) for `levels` and every `layers[l]`. Sharing
7262        // ⇒ no per-node element copy ⇒ clone cost independent of N (node
7263        // count); only the outer layer Vec (len ≤ 8) is copied, O(1) in
7264        // practice.
7265        let mut cat = Catalog::new();
7266        cat.create_table(TableSchema::new(
7267            "docs",
7268            alloc::vec![
7269                ColumnSchema::new("id", DataType::Int, false),
7270                ColumnSchema::new(
7271                    "v",
7272                    DataType::Vector {
7273                        dim: 3,
7274                        encoding: VecEncoding::F32
7275                    },
7276                    true
7277                ),
7278            ],
7279        ))
7280        .unwrap();
7281        let t = cat.get_mut("docs").unwrap();
7282        for i in 0..1500_i32 {
7283            #[allow(clippy::cast_precision_loss)] // 0..1500 — no precision lost
7284            let base = (i as f32) * 0.01;
7285            t.insert(Row::new(alloc::vec![
7286                Value::Int(i),
7287                Value::Vector(alloc::vec![base, base + 0.05, base + 0.1]),
7288            ]))
7289            .unwrap();
7290        }
7291        t.add_nsw_index("docs_nsw".into(), "v", NSW_DEFAULT_M)
7292            .unwrap();
7293        let g = match &cat.get("docs").unwrap().indices()[0].kind {
7294            IndexKind::Nsw(g) => g,
7295            IndexKind::BTree(_) | IndexKind::Brin { .. } | IndexKind::Gin(_) | IndexKind::GinTrgm(_) => {
7296                panic!("expected NSW")
7297            }
7298        };
7299        // Non-trivial graph: one level slot per row, and the geometric level
7300        // distribution puts some nodes above layer 0.
7301        assert_eq!(g.levels.len(), 1500, "one level slot per inserted row");
7302        assert!(
7303            g.layers.len() >= 2,
7304            "1500 nodes should populate at least two HNSW layers, got {}",
7305            g.layers.len()
7306        );
7307
7308        let cloned = g.clone();
7309
7310        assert!(
7311            g.levels.shares_storage_with(&cloned.levels),
7312            "levels PV not shared after clone — clone copied elements (O(N))"
7313        );
7314        assert_eq!(g.layers.len(), cloned.layers.len());
7315        for (l, (orig, cl)) in g.layers.iter().zip(cloned.layers.iter()).enumerate() {
7316            assert!(
7317                orig.shares_storage_with(cl),
7318                "layer {l} PV not shared after clone — clone copied elements (O(N))"
7319            );
7320        }
7321    }
7322
7323    #[test]
7324    fn sq8_catalog_serialise_roundtrip_preserves_cells_and_index() {
7325        // v6.0.1 step 6 verify: a catalog with an `VECTOR(N)
7326        // USING SQ8` column + NSW index survives a full
7327        // serialise → deserialise cycle. Cells re-decode bit-
7328        // identically (per-vector affine triple), the NSW
7329        // topology stays intact, and kNN search still routes
7330        // through the SQ8 ADC dispatcher after the catalog hop.
7331        let mut cat = Catalog::new();
7332        cat.create_table(TableSchema::new(
7333            "vecs",
7334            alloc::vec![
7335                ColumnSchema::new("id", DataType::Int, false),
7336                ColumnSchema::new(
7337                    "v",
7338                    DataType::Vector {
7339                        dim: 8,
7340                        encoding: VecEncoding::Sq8,
7341                    },
7342                    false,
7343                ),
7344            ],
7345        ))
7346        .unwrap();
7347        let t = cat.get_mut("vecs").unwrap();
7348        for i in 0..32_i32 {
7349            #[allow(clippy::cast_precision_loss)]
7350            let base = (i as f32) * 0.03;
7351            let v: Vec<f32> = (0..8_i32)
7352                .map(|j| {
7353                    #[allow(clippy::cast_precision_loss)]
7354                    let off = (j as f32) * 0.01;
7355                    base + off
7356                })
7357                .collect();
7358            t.insert(Row::new(alloc::vec![
7359                Value::Int(i),
7360                Value::Sq8Vector(quantize::quantize(&v)),
7361            ]))
7362            .unwrap();
7363        }
7364        t.add_nsw_index("v_idx".into(), "v", NSW_DEFAULT_M).unwrap();
7365        // Capture a pre-serialise reference cell + nsw hits to
7366        // compare against the restored catalog.
7367        let query = alloc::vec![0.15_f32, 0.16, 0.17, 0.18, 0.19, 0.20, 0.21, 0.22];
7368        let (before_cell, before_ty, before_hits) = {
7369            let t_ref = cat.get("vecs").unwrap();
7370            (
7371                t_ref.rows()[5].values[1].clone(),
7372                t_ref.schema().columns[1].ty,
7373                nsw_query(t_ref, "v_idx", &query, 5, NswMetric::L2),
7374            )
7375        };
7376
7377        let bytes = cat.serialize();
7378        let restored = Catalog::deserialize(&bytes).expect("deserialize ok");
7379        let rt = restored.get("vecs").unwrap();
7380        assert_eq!(rt.schema().columns[1].ty, before_ty);
7381        assert_eq!(rt.rows()[5].values[1], before_cell);
7382        let after_hits = nsw_query(rt, "v_idx", &query, 5, NswMetric::L2);
7383        assert_eq!(before_hits, after_hits);
7384    }
7385
7386    #[test]
7387    fn half_catalog_serialise_roundtrip_preserves_cells_and_index() {
7388        // v6.0.3 step 4 verify: a catalog with a `VECTOR(N) USING
7389        // HALF` column + NSW index survives a full serialise →
7390        // deserialise cycle. Cells re-decode bit-identically (raw
7391        // u16 LE bytes), the NSW topology stays intact, and kNN
7392        // search still returns the same hit IDs against the
7393        // restored catalog.
7394        use crate::halfvec;
7395        let mut cat = Catalog::new();
7396        cat.create_table(TableSchema::new(
7397            "vecs",
7398            alloc::vec![
7399                ColumnSchema::new("id", DataType::Int, false),
7400                ColumnSchema::new(
7401                    "v",
7402                    DataType::Vector {
7403                        dim: 8,
7404                        encoding: VecEncoding::F16,
7405                    },
7406                    false,
7407                ),
7408            ],
7409        ))
7410        .unwrap();
7411        let t = cat.get_mut("vecs").unwrap();
7412        for i in 0..32_i32 {
7413            #[allow(clippy::cast_precision_loss)]
7414            let base = (i as f32) * 0.03;
7415            let v: Vec<f32> = (0..8_i32)
7416                .map(|j| {
7417                    #[allow(clippy::cast_precision_loss)]
7418                    let off = (j as f32) * 0.01;
7419                    base + off
7420                })
7421                .collect();
7422            t.insert(Row::new(alloc::vec![
7423                Value::Int(i),
7424                Value::HalfVector(halfvec::HalfVector::from_f32_slice(&v)),
7425            ]))
7426            .unwrap();
7427        }
7428        t.add_nsw_index("v_idx".into(), "v", NSW_DEFAULT_M).unwrap();
7429        let query = alloc::vec![0.15_f32, 0.16, 0.17, 0.18, 0.19, 0.20, 0.21, 0.22];
7430        let (before_cell, before_ty, before_hits) = {
7431            let t_ref = cat.get("vecs").unwrap();
7432            (
7433                t_ref.rows()[5].values[1].clone(),
7434                t_ref.schema().columns[1].ty,
7435                nsw_query(t_ref, "v_idx", &query, 5, NswMetric::L2),
7436            )
7437        };
7438        let bytes = cat.serialize();
7439        let restored = Catalog::deserialize(&bytes).expect("deserialize ok");
7440        let rt = restored.get("vecs").unwrap();
7441        assert_eq!(rt.schema().columns[1].ty, before_ty);
7442        assert_eq!(rt.rows()[5].values[1], before_cell);
7443        let after_hits = nsw_query(rt, "v_idx", &query, 5, NswMetric::L2);
7444        assert_eq!(before_hits, after_hits);
7445    }
7446
7447    #[test]
7448    #[allow(clippy::similar_names)]
7449    fn hnsw_half_recall_at_10_matches_f32_groundtruth() {
7450        // v6.0.3 step 3 verify: HALF column NSW retrieves ≥ 95%
7451        // top-10 overlap vs brute-force F32 ground truth.
7452        // Half-precision dequantises bit-exactly at the storage
7453        // layer (no rerank pass), so the recall floor is tighter
7454        // than the SQ8 case — only the rounding noise from f32 →
7455        // f16 quantisation contributes.
7456        use crate::halfvec;
7457        fn next(state: &mut u64) -> f32 {
7458            *state = state
7459                .wrapping_add(0x9E37_79B9_7F4A_7C15)
7460                .wrapping_mul(0xBF58_476D_1CE4_E5B9);
7461            #[allow(clippy::cast_precision_loss)]
7462            let u = ((*state >> 32) as u32 as f32) / (u32::MAX as f32);
7463            2.0 * u - 1.0
7464        }
7465        let dim: u32 = 32;
7466        let n: usize = 512;
7467        let dim_us = dim as usize;
7468        let mut seed: u64 = 0xF16_F16_F16_F16_u64;
7469        let corpus: Vec<Vec<f32>> = (0..n)
7470            .map(|_| (0..dim_us).map(|_| next(&mut seed)).collect())
7471            .collect();
7472        let queries: Vec<Vec<f32>> = (0..32)
7473            .map(|_| (0..dim_us).map(|_| next(&mut seed)).collect())
7474            .collect();
7475        let exact_top10: Vec<Vec<usize>> = queries
7476            .iter()
7477            .map(|q| {
7478                let mut scored: Vec<(f32, usize)> = corpus
7479                    .iter()
7480                    .enumerate()
7481                    .map(|(i, v)| (l2_distance_sq(v, q), i))
7482                    .collect();
7483                scored.sort_by(|a, b| a.0.partial_cmp(&b.0).unwrap_or(core::cmp::Ordering::Equal));
7484                scored.into_iter().take(10).map(|(_, i)| i).collect()
7485            })
7486            .collect();
7487        let mut cat = Catalog::new();
7488        cat.create_table(TableSchema::new(
7489            "vecs",
7490            alloc::vec![
7491                ColumnSchema::new("id", DataType::Int, false),
7492                ColumnSchema::new(
7493                    "v",
7494                    DataType::Vector {
7495                        dim,
7496                        encoding: VecEncoding::F16,
7497                    },
7498                    false,
7499                ),
7500            ],
7501        ))
7502        .unwrap();
7503        let t = cat.get_mut("vecs").unwrap();
7504        for (i, v) in corpus.iter().enumerate() {
7505            t.insert(Row::new(alloc::vec![
7506                Value::Int(i32::try_from(i).unwrap()),
7507                Value::HalfVector(halfvec::HalfVector::from_f32_slice(v)),
7508            ]))
7509            .unwrap();
7510        }
7511        t.add_nsw_index("v_idx".into(), "v", NSW_DEFAULT_M).unwrap();
7512        let table = cat.get("vecs").unwrap();
7513        let mut total_overlap = 0_usize;
7514        for (q, exact) in queries.iter().zip(exact_top10.iter()) {
7515            let hits = nsw_query(table, "v_idx", q, 10, NswMetric::L2);
7516            for h in &hits {
7517                if exact.contains(h) {
7518                    total_overlap += 1;
7519                }
7520            }
7521        }
7522        #[allow(clippy::cast_precision_loss)]
7523        let recall = total_overlap as f32 / (10.0 * queries.len() as f32);
7524        assert!(
7525            recall >= 0.95,
7526            "HALF HNSW recall@10 = {recall:.3}, below floor 0.95 — \
7527             check halfvec dispatch in `cell_to_query_metric_distance`"
7528        );
7529    }
7530
7531    #[test]
7532    fn hnsw_sq8_recall_at_10_above_0_95_vs_f32_groundtruth() {
7533        // v6.0.1 step 5 verify: build TWO catalogs over the same
7534        // corpus — one F32, one SQ8 — and confirm SQ8 NSW + f32
7535        // rerank retrieves ≥ 95% top-10 overlap vs brute-force F32
7536        // ground truth. The rerank pass (sq8_rerank) re-scores ADC
7537        // candidates with dequantised cells, recovering recall the
7538        // raw ADC sacrifices for 4× compression.
7539        use crate::quantize;
7540        // Deterministic Gaussian-ish corpus via splitmix64. Vectors
7541        // get normalised so SQ8's per-vector `(min, max)` lives in
7542        // a sensible range; matches the v6.0.0 fuzz harness.
7543        fn next(state: &mut u64) -> f32 {
7544            *state = state
7545                .wrapping_add(0x9E37_79B9_7F4A_7C15)
7546                .wrapping_mul(0xBF58_476D_1CE4_E5B9);
7547            #[allow(clippy::cast_precision_loss)]
7548            let u = ((*state >> 32) as u32 as f32) / (u32::MAX as f32);
7549            2.0 * u - 1.0
7550        }
7551        let dim: u32 = 32;
7552        let n: usize = 512;
7553        let dim_us = dim as usize;
7554        let mut seed: u64 = 0xCAFE_BABE_DEAD_BEEFu64;
7555        let corpus: Vec<Vec<f32>> = (0..n)
7556            .map(|_| (0..dim_us).map(|_| next(&mut seed)).collect())
7557            .collect();
7558        let queries: Vec<Vec<f32>> = (0..32)
7559            .map(|_| (0..dim_us).map(|_| next(&mut seed)).collect())
7560            .collect();
7561        // F32 ground truth — pure exact arithmetic, brute force.
7562        let exact_top10: Vec<Vec<usize>> = queries
7563            .iter()
7564            .map(|q| {
7565                let mut scored: Vec<(f32, usize)> = corpus
7566                    .iter()
7567                    .enumerate()
7568                    .map(|(i, v)| (l2_distance_sq(v, q), i))
7569                    .collect();
7570                scored.sort_by(|a, b| a.0.partial_cmp(&b.0).unwrap_or(core::cmp::Ordering::Equal));
7571                scored.into_iter().take(10).map(|(_, i)| i).collect()
7572            })
7573            .collect();
7574        // SQ8 catalog — INSERTs land as `Value::Sq8Vector` cells;
7575        // HNSW build uses the ADC path verified in step 4.
7576        let mut cat = Catalog::new();
7577        cat.create_table(TableSchema::new(
7578            "vecs",
7579            alloc::vec![
7580                ColumnSchema::new("id", DataType::Int, false),
7581                ColumnSchema::new(
7582                    "v",
7583                    DataType::Vector {
7584                        dim,
7585                        encoding: VecEncoding::Sq8,
7586                    },
7587                    false,
7588                ),
7589            ],
7590        ))
7591        .unwrap();
7592        let t = cat.get_mut("vecs").unwrap();
7593        for (i, v) in corpus.iter().enumerate() {
7594            t.insert(Row::new(alloc::vec![
7595                Value::Int(i32::try_from(i).unwrap()),
7596                Value::Sq8Vector(quantize::quantize(v)),
7597            ]))
7598            .unwrap();
7599        }
7600        t.add_nsw_index("v_idx".into(), "v", NSW_DEFAULT_M).unwrap();
7601        let table = cat.get("vecs").unwrap();
7602        let mut total_overlap = 0_usize;
7603        for (q, exact) in queries.iter().zip(exact_top10.iter()) {
7604            let hits = nsw_query(table, "v_idx", q, 10, NswMetric::L2);
7605            for h in &hits {
7606                if exact.contains(h) {
7607                    total_overlap += 1;
7608                }
7609            }
7610        }
7611        #[allow(clippy::cast_precision_loss)]
7612        let recall = total_overlap as f32 / (10.0 * queries.len() as f32);
7613        assert!(
7614            recall >= 0.95,
7615            "SQ8 HNSW recall@10 = {recall:.3}, below floor 0.95 — \
7616             check `sq8_rerank` is wired in `nsw_search` for SQ8 columns"
7617        );
7618    }
7619
7620    #[test]
7621    fn nsw_index_topology_persists_through_round_trip() {
7622        // Build an NSW index, capture its (entry, neighbors) tuple, do
7623        // a full serialize → deserialize, and verify the restored
7624        // graph is byte-for-byte identical. The point of v2.7 is that
7625        // startup skips the rebuild, so the topology has to survive
7626        // the disk hop.
7627        let mut cat = Catalog::new();
7628        cat.create_table(TableSchema::new(
7629            "docs",
7630            alloc::vec![
7631                ColumnSchema::new("id", DataType::Int, false),
7632                ColumnSchema::new(
7633                    "v",
7634                    DataType::Vector {
7635                        dim: 3,
7636                        encoding: VecEncoding::F32
7637                    },
7638                    true
7639                ),
7640            ],
7641        ))
7642        .unwrap();
7643        let t = cat.get_mut("docs").unwrap();
7644        for i in 0..6_i32 {
7645            #[allow(clippy::cast_precision_loss)] // 0..6 — no precision lost
7646            let base = (i as f32) * 0.1;
7647            let row = Row::new(alloc::vec![
7648                Value::Int(i),
7649                Value::Vector(alloc::vec![base, base + 0.05, base + 0.1]),
7650            ]);
7651            t.insert(row).unwrap();
7652        }
7653        t.add_nsw_index("docs_nsw".into(), "v", NSW_DEFAULT_M)
7654            .unwrap();
7655        let original = match &cat.get("docs").unwrap().indices()[0].kind {
7656            IndexKind::Nsw(g) => g.clone(),
7657            IndexKind::BTree(_) | IndexKind::Brin { .. } | IndexKind::Gin(_) | IndexKind::GinTrgm(_) => {
7658                panic!("expected NSW")
7659            }
7660        };
7661        let bytes = cat.serialize();
7662        let restored = Catalog::deserialize(&bytes).expect("deserialize");
7663        let restored_graph = match &restored.get("docs").unwrap().indices()[0].kind {
7664            IndexKind::Nsw(g) => g.clone(),
7665            IndexKind::BTree(_) | IndexKind::Brin { .. } | IndexKind::Gin(_) | IndexKind::GinTrgm(_) => {
7666                panic!("expected NSW")
7667            }
7668        };
7669        assert_eq!(restored_graph.m, original.m);
7670        assert_eq!(restored_graph.m_max_0, original.m_max_0);
7671        assert_eq!(restored_graph.entry, original.entry);
7672        assert_eq!(restored_graph.entry_level, original.entry_level);
7673        assert_eq!(restored_graph.levels, original.levels);
7674        assert_eq!(restored_graph.layers, original.layers);
7675    }
7676
7677    #[test]
7678    fn hnsw_level_assignment_is_deterministic() {
7679        // Same row index always produces the same level — the topology
7680        // must be reproducible (matters for serialize round-trip).
7681        for i in 0..32usize {
7682            assert_eq!(nsw_assign_level(i), nsw_assign_level(i));
7683        }
7684    }
7685
7686    #[test]
7687    fn hnsw_layer_0_dominates_population() {
7688        // Sanity: out of N inserts, the vast majority should land on
7689        // layer 0. The 4-bit-clear promotion rule gives roughly 1/16
7690        // promotion to layer ≥ 1, so under 50 nodes we expect ~3 on
7691        // layer ≥ 1 and the rest on layer 0.
7692        let on_zero = (0..200usize).filter(|&i| nsw_assign_level(i) == 0).count();
7693        assert!(on_zero > 150, "level-0 nodes too few: {on_zero}");
7694    }
7695
7696    #[test]
7697    fn hnsw_search_matches_brute_force_for_l2_top1() {
7698        // Build a small dataset, query it, and confirm the top result
7699        // matches the brute-force nearest by L2. Topology variability
7700        // shouldn't break recall at k=1 for well-separated vectors.
7701        let mut cat = Catalog::new();
7702        cat.create_table(TableSchema::new(
7703            "vecs",
7704            alloc::vec![
7705                ColumnSchema::new("id", DataType::Int, false),
7706                ColumnSchema::new(
7707                    "v",
7708                    DataType::Vector {
7709                        dim: 3,
7710                        encoding: VecEncoding::F32
7711                    },
7712                    true
7713                ),
7714            ],
7715        ))
7716        .unwrap();
7717        let t = cat.get_mut("vecs").unwrap();
7718        let dataset: alloc::vec::Vec<(i32, [f32; 3])> = alloc::vec![
7719            (1, [0.0, 0.0, 0.0]),
7720            (2, [1.0, 0.0, 0.0]),
7721            (3, [0.0, 1.0, 0.0]),
7722            (4, [0.0, 0.0, 1.0]),
7723            (5, [1.0, 1.0, 0.0]),
7724            (6, [1.0, 0.0, 1.0]),
7725            (7, [0.0, 1.0, 1.0]),
7726            (8, [1.0, 1.0, 1.0]),
7727            (9, [0.5, 0.5, 0.5]),
7728            (10, [0.2, 0.8, 0.5]),
7729        ];
7730        for &(id, v) in &dataset {
7731            t.insert(Row::new(alloc::vec![
7732                Value::Int(id),
7733                Value::Vector(alloc::vec![v[0], v[1], v[2]]),
7734            ]))
7735            .unwrap();
7736        }
7737        t.add_nsw_index("v_idx".into(), "v", NSW_DEFAULT_M).unwrap();
7738        let idx_pos = cat
7739            .get("vecs")
7740            .unwrap()
7741            .indices()
7742            .iter()
7743            .position(|i| i.name == "v_idx")
7744            .unwrap();
7745        for query in [[0.4, 0.4, 0.4], [0.9, 0.1, 0.0], [0.0, 0.9, 0.9]] {
7746            let table = cat.get("vecs").unwrap();
7747            let hnsw_top = nsw_search(table, idx_pos, &query, 1, 16, NswMetric::L2);
7748            let mut brute: alloc::vec::Vec<(f32, usize)> = (0..table.rows.len())
7749                .map(|i| {
7750                    let Value::Vector(v) = &table.rows[i].values[1] else {
7751                        return (f32::INFINITY, i);
7752                    };
7753                    (l2_distance_sq(v, &query), i)
7754                })
7755                .collect();
7756            brute.sort_by(|a, b| a.0.partial_cmp(&b.0).unwrap_or(core::cmp::Ordering::Equal));
7757            assert!(!hnsw_top.is_empty(), "HNSW returned no results");
7758            assert_eq!(
7759                hnsw_top[0].1, brute[0].1,
7760                "HNSW top-1 != brute-force top-1 for {query:?}"
7761            );
7762        }
7763    }
7764
7765    #[test]
7766    fn serialize_table_with_rows_round_trips() {
7767        let mut cat = Catalog::new();
7768        cat.create_table(make_users_schema()).unwrap();
7769        let t = cat.get_mut("users").unwrap();
7770        t.insert(Row::new(vec![
7771            Value::Int(1),
7772            Value::Text("alice".into()),
7773            Value::Float(95.5),
7774        ]))
7775        .unwrap();
7776        t.insert(Row::new(vec![
7777            Value::Int(2),
7778            Value::Text("bob".into()),
7779            Value::Null,
7780        ]))
7781        .unwrap();
7782        assert_round_trip(&cat);
7783    }
7784
7785    #[test]
7786    fn serialize_multiple_tables_round_trips() {
7787        let mut cat = Catalog::new();
7788        cat.create_table(make_users_schema()).unwrap();
7789        cat.create_table(TableSchema::new(
7790            "flags",
7791            vec![
7792                ColumnSchema::new("id", DataType::BigInt, false),
7793                ColumnSchema::new("active", DataType::Bool, false),
7794            ],
7795        ))
7796        .unwrap();
7797        cat.get_mut("flags")
7798            .unwrap()
7799            .insert(Row::new(vec![Value::BigInt(7), Value::Bool(true)]))
7800            .unwrap();
7801        assert_round_trip(&cat);
7802    }
7803
7804    #[test]
7805    fn deserialize_rejects_bad_magic() {
7806        let mut buf = b"BADMAGIC".to_vec();
7807        buf.push(FILE_VERSION);
7808        buf.extend_from_slice(&0u32.to_le_bytes());
7809        let err = Catalog::deserialize(&buf).unwrap_err();
7810        assert!(matches!(err, StorageError::Corrupt(_)));
7811    }
7812
7813    #[test]
7814    fn deserialize_rejects_unsupported_version() {
7815        let mut buf = FILE_MAGIC.to_vec();
7816        buf.push(99); // future version
7817        buf.extend_from_slice(&0u32.to_le_bytes());
7818        let err = Catalog::deserialize(&buf).unwrap_err();
7819        assert!(matches!(err, StorageError::Corrupt(ref s) if s.contains("version")));
7820    }
7821
7822    #[test]
7823    fn deserialize_rejects_truncated_file() {
7824        let mut cat = Catalog::new();
7825        cat.create_table(make_users_schema()).unwrap();
7826        let bytes = cat.serialize();
7827        // Drop the last byte to simulate truncation.
7828        let truncated = &bytes[..bytes.len() - 1];
7829        assert!(matches!(
7830            Catalog::deserialize(truncated),
7831            Err(StorageError::Corrupt(_))
7832        ));
7833    }
7834
7835    #[test]
7836    fn deserialize_rejects_trailing_garbage() {
7837        let cat = Catalog::new();
7838        let mut bytes = cat.serialize();
7839        bytes.push(0xFF);
7840        assert!(matches!(
7841            Catalog::deserialize(&bytes),
7842            Err(StorageError::Corrupt(ref s)) if s.contains("trailing")
7843        ));
7844    }
7845
7846    // --- v0.8 indices ------------------------------------------------------
7847
7848    fn populated_users() -> Catalog {
7849        let mut cat = Catalog::new();
7850        cat.create_table(make_users_schema()).unwrap();
7851        let t = cat.get_mut("users").unwrap();
7852        for (id, name, score) in [
7853            (1, "alice", Some(90.0)),
7854            (2, "bob", None),
7855            (3, "alice", Some(70.0)), // duplicate name → maps to two row idxs
7856        ] {
7857            t.insert(Row::new(vec![
7858                Value::Int(id),
7859                Value::Text(name.into()),
7860                score.map_or(Value::Null, Value::Float),
7861            ]))
7862            .unwrap();
7863        }
7864        cat
7865    }
7866
7867    #[test]
7868    fn add_index_builds_from_existing_rows() {
7869        let mut cat = populated_users();
7870        cat.get_mut("users")
7871            .unwrap()
7872            .add_index("by_id".into(), "id")
7873            .unwrap();
7874        let t = cat.get("users").unwrap();
7875        let idx = t.index_on(0).expect("index_on(0)");
7876        assert_eq!(idx.lookup_eq(&IndexKey::Int(2)), &[RowLocator::Hot(1)]);
7877        assert_eq!(idx.lookup_eq(&IndexKey::Int(99)), &[] as &[RowLocator]);
7878    }
7879
7880    #[test]
7881    fn add_index_dup_name_rejected() {
7882        let mut cat = populated_users();
7883        let t = cat.get_mut("users").unwrap();
7884        t.add_index("ix".into(), "id").unwrap();
7885        let err = t.add_index("ix".into(), "name").unwrap_err();
7886        assert!(matches!(err, StorageError::DuplicateIndex { ref name } if name == "ix"));
7887    }
7888
7889    #[test]
7890    fn add_index_unknown_column_rejected() {
7891        let mut cat = populated_users();
7892        let err = cat
7893            .get_mut("users")
7894            .unwrap()
7895            .add_index("ix".into(), "ghost")
7896            .unwrap_err();
7897        assert!(matches!(err, StorageError::ColumnNotFound { ref column } if column == "ghost"));
7898    }
7899
7900    #[test]
7901    fn insert_after_create_index_updates_it() {
7902        let mut cat = populated_users();
7903        let t = cat.get_mut("users").unwrap();
7904        t.add_index("by_name".into(), "name").unwrap();
7905        t.insert(Row::new(vec![
7906            Value::Int(4),
7907            Value::Text("dave".into()),
7908            Value::Null,
7909        ]))
7910        .unwrap();
7911        let idx = t.index_on(1).unwrap();
7912        assert_eq!(
7913            idx.lookup_eq(&IndexKey::Text("dave".into())),
7914            &[RowLocator::Hot(3)]
7915        );
7916        // Pre-existing duplicates remain mapped to the two original row idxs.
7917        assert_eq!(
7918            idx.lookup_eq(&IndexKey::Text("alice".into())),
7919            &[RowLocator::Hot(0), RowLocator::Hot(2)]
7920        );
7921    }
7922
7923    #[test]
7924    fn null_or_float_values_are_not_indexed() {
7925        let mut cat = populated_users();
7926        let t = cat.get_mut("users").unwrap();
7927        t.add_index("by_score".into(), "score").unwrap();
7928        let idx = t.index_on(2).unwrap();
7929        // bob's score is NULL → no entry for bob.
7930        // Score is Float → the spec says we don't index NaN-prone columns,
7931        // so even the present scores are absent. Lookups via IndexKey::Int(90)
7932        // mis-match the column type and trivially find nothing.
7933        assert_eq!(idx.lookup_eq(&IndexKey::Int(90)), &[] as &[RowLocator]);
7934    }
7935
7936    // --- v0.11 vector type -------------------------------------------------
7937
7938    #[test]
7939    fn vector_value_data_type_carries_dim() {
7940        let v = Value::Vector(vec![1.0, 2.0, 3.0]);
7941        assert_eq!(
7942            v.data_type(),
7943            Some(DataType::Vector {
7944                dim: 3,
7945                encoding: VecEncoding::F32
7946            })
7947        );
7948    }
7949
7950    #[test]
7951    fn vector_column_insert_matching_dim_ok() {
7952        let mut cat = Catalog::new();
7953        cat.create_table(TableSchema::new(
7954            "emb",
7955            vec![ColumnSchema::new(
7956                "v",
7957                DataType::Vector {
7958                    dim: 3,
7959                    encoding: VecEncoding::F32,
7960                },
7961                false,
7962            )],
7963        ))
7964        .unwrap();
7965        cat.get_mut("emb")
7966            .unwrap()
7967            .insert(Row::new(vec![Value::Vector(vec![1.0, 2.0, 3.0])]))
7968            .unwrap();
7969    }
7970
7971    #[test]
7972    fn vector_column_insert_dim_mismatch_rejected() {
7973        let mut cat = Catalog::new();
7974        cat.create_table(TableSchema::new(
7975            "emb",
7976            vec![ColumnSchema::new(
7977                "v",
7978                DataType::Vector {
7979                    dim: 3,
7980                    encoding: VecEncoding::F32,
7981                },
7982                false,
7983            )],
7984        ))
7985        .unwrap();
7986        let err = cat
7987            .get_mut("emb")
7988            .unwrap()
7989            .insert(Row::new(vec![Value::Vector(vec![1.0, 2.0])]))
7990            .unwrap_err();
7991        assert!(matches!(err, StorageError::TypeMismatch { .. }));
7992    }
7993
7994    #[test]
7995    fn vector_value_survives_catalog_round_trip() {
7996        let mut cat = Catalog::new();
7997        cat.create_table(TableSchema::new(
7998            "emb",
7999            vec![
8000                ColumnSchema::new("id", DataType::Int, false),
8001                ColumnSchema::new(
8002                    "v",
8003                    DataType::Vector {
8004                        dim: 4,
8005                        encoding: VecEncoding::F32,
8006                    },
8007                    false,
8008                ),
8009            ],
8010        ))
8011        .unwrap();
8012        cat.get_mut("emb")
8013            .unwrap()
8014            .insert(Row::new(vec![
8015                Value::Int(1),
8016                Value::Vector(vec![0.5, -1.25, 3.0, 7.0]),
8017            ]))
8018            .unwrap();
8019        let restored = Catalog::deserialize(&cat.serialize()).expect("round-trip");
8020        let table = restored.get("emb").unwrap();
8021        assert_eq!(
8022            table.schema().columns[1].ty,
8023            DataType::Vector {
8024                dim: 4,
8025                encoding: VecEncoding::F32
8026            }
8027        );
8028        assert_eq!(
8029            table.rows()[0].values[1],
8030            Value::Vector(vec![0.5, -1.25, 3.0, 7.0])
8031        );
8032    }
8033
8034    #[test]
8035    fn index_survives_serialize_deserialize_round_trip() {
8036        let mut cat = populated_users();
8037        cat.get_mut("users")
8038            .unwrap()
8039            .add_index("by_name".into(), "name")
8040            .unwrap();
8041        let restored = Catalog::deserialize(&cat.serialize()).unwrap();
8042        let idx = restored
8043            .get("users")
8044            .unwrap()
8045            .index_on(1)
8046            .expect("index_on(1) after restore");
8047        assert_eq!(idx.name, "by_name");
8048        // Data was rebuilt from rows, not deserialized directly.
8049        assert_eq!(
8050            idx.lookup_eq(&IndexKey::Text("alice".into())),
8051            &[RowLocator::Hot(0), RowLocator::Hot(2)]
8052        );
8053    }
8054
8055    // --- v5.1 cold-tier integration tests ----------------------
8056
8057    /// Schema with a BIGINT PK column matching what the v5.1 cold-
8058    /// tier path supports (`IndexKey::Int` → `u64` cast).
8059    fn bigint_pk_users_schema() -> TableSchema {
8060        TableSchema::new(
8061            "users",
8062            vec![
8063                ColumnSchema::new("id", DataType::BigInt, false),
8064                ColumnSchema::new("name", DataType::Text, false),
8065            ],
8066        )
8067    }
8068
8069    fn make_user_row(id: i64, name: &str) -> Row {
8070        Row::new(vec![Value::BigInt(id), Value::Text(name.into())])
8071    }
8072
8073    #[test]
8074    fn lookup_by_pk_finds_row_via_hot_index() {
8075        let mut cat = Catalog::new();
8076        cat.create_table(bigint_pk_users_schema()).unwrap();
8077        let t = cat.get_mut("users").unwrap();
8078        for (id, name) in [(1i64, "alice"), (2, "bob"), (3, "carol")] {
8079            t.insert(make_user_row(id, name)).unwrap();
8080        }
8081        t.add_index("by_id".into(), "id").unwrap();
8082        // All locators are Hot; cold_segments is empty.
8083        let got = cat
8084            .lookup_by_pk("users", "by_id", &IndexKey::Int(2))
8085            .unwrap();
8086        assert_eq!(got, make_user_row(2, "bob"));
8087        assert_eq!(cat.cold_segment_count(), 0);
8088    }
8089
8090    #[test]
8091    fn lookup_by_pk_returns_none_when_key_missing() {
8092        let mut cat = Catalog::new();
8093        cat.create_table(bigint_pk_users_schema()).unwrap();
8094        let t = cat.get_mut("users").unwrap();
8095        t.insert(make_user_row(1, "alice")).unwrap();
8096        t.add_index("by_id".into(), "id").unwrap();
8097        assert!(
8098            cat.lookup_by_pk("users", "by_id", &IndexKey::Int(999))
8099                .is_none()
8100        );
8101        // Also: unknown table / unknown index name.
8102        assert!(
8103            cat.lookup_by_pk("other_table", "by_id", &IndexKey::Int(1))
8104                .is_none()
8105        );
8106        assert!(
8107            cat.lookup_by_pk("users", "no_such_index", &IndexKey::Int(1))
8108                .is_none()
8109        );
8110    }
8111
8112    #[test]
8113    fn lookup_by_pk_resolves_cold_locator_via_loaded_segment() {
8114        // Build a cold-tier segment whose payloads are dense-encoded
8115        // BIGINT rows. Wire each PK into the BTree index as a Cold
8116        // locator. The hot tier carries no rows for those PKs.
8117        let mut cat = Catalog::new();
8118        cat.create_table(bigint_pk_users_schema()).unwrap();
8119        let t = cat.get_mut("users").unwrap();
8120        t.add_index("by_id".into(), "id").unwrap();
8121        let schema = t.schema.clone();
8122
8123        let cold_rows: Vec<(i64, &str)> =
8124            vec![(100, "ivy"), (200, "joe"), (300, "kim"), (400, "lin")];
8125        let seg_rows: Vec<(u64, Vec<u8>)> = cold_rows
8126            .iter()
8127            .map(|(id, name)| {
8128                let row = make_user_row(*id, name);
8129                ((*id).cast_unsigned(), encode_row_body_dense(&row, &schema))
8130            })
8131            .collect();
8132        let (seg_bytes, _meta) =
8133            encode_segment(seg_rows.into_iter(), 0.01, SEGMENT_PAGE_BYTES).unwrap();
8134        let seg_id = cat.load_segment_bytes(seg_bytes).unwrap();
8135        assert_eq!(seg_id, 0);
8136        assert_eq!(cat.cold_segment_count(), 1);
8137
8138        let pairs: Vec<(IndexKey, RowLocator)> = cold_rows
8139            .iter()
8140            .map(|(id, _)| {
8141                (
8142                    IndexKey::Int(*id),
8143                    RowLocator::Cold {
8144                        segment_id: seg_id,
8145                        page_offset: 0,
8146                    },
8147                )
8148            })
8149            .collect();
8150        let registered = cat
8151            .get_mut("users")
8152            .unwrap()
8153            .register_cold_locators("by_id", pairs)
8154            .unwrap();
8155        assert_eq!(registered, 4);
8156
8157        for (id, name) in &cold_rows {
8158            let got = cat
8159                .lookup_by_pk("users", "by_id", &IndexKey::Int(*id))
8160                .unwrap_or_else(|| panic!("cold key {id} not found"));
8161            assert_eq!(got, make_user_row(*id, name));
8162        }
8163        // Cold key that isn't in the segment must return None.
8164        assert!(
8165            cat.lookup_by_pk("users", "by_id", &IndexKey::Int(999))
8166                .is_none()
8167        );
8168    }
8169
8170    #[test]
8171    fn lookup_by_pk_mixes_hot_and_cold_tiers() {
8172        // Half the rows live in the hot tier (Table::rows + add_index
8173        // produces Hot locators); half live in a cold segment and have
8174        // Cold locators wired manually. Each lookup hits the right tier.
8175        let mut cat = Catalog::new();
8176        cat.create_table(bigint_pk_users_schema()).unwrap();
8177        let t = cat.get_mut("users").unwrap();
8178        for (id, name) in [(1i64, "alice"), (2, "bob")] {
8179            t.insert(make_user_row(id, name)).unwrap();
8180        }
8181        t.add_index("by_id".into(), "id").unwrap();
8182        let schema = t.schema.clone();
8183
8184        let cold_rows: Vec<(i64, &str)> = vec![(100, "ivy"), (200, "joe")];
8185        let seg_rows: Vec<(u64, Vec<u8>)> = cold_rows
8186            .iter()
8187            .map(|(id, name)| {
8188                let row = make_user_row(*id, name);
8189                ((*id).cast_unsigned(), encode_row_body_dense(&row, &schema))
8190            })
8191            .collect();
8192        let (seg_bytes, _) =
8193            encode_segment(seg_rows.into_iter(), 0.01, SEGMENT_PAGE_BYTES).unwrap();
8194        let seg_id = cat.load_segment_bytes(seg_bytes).unwrap();
8195        let pairs: Vec<(IndexKey, RowLocator)> = cold_rows
8196            .iter()
8197            .map(|(id, _)| {
8198                (
8199                    IndexKey::Int(*id),
8200                    RowLocator::Cold {
8201                        segment_id: seg_id,
8202                        page_offset: 0,
8203                    },
8204                )
8205            })
8206            .collect();
8207        cat.get_mut("users")
8208            .unwrap()
8209            .register_cold_locators("by_id", pairs)
8210            .unwrap();
8211
8212        // Hot tier hits.
8213        assert_eq!(
8214            cat.lookup_by_pk("users", "by_id", &IndexKey::Int(1))
8215                .unwrap(),
8216            make_user_row(1, "alice")
8217        );
8218        assert_eq!(
8219            cat.lookup_by_pk("users", "by_id", &IndexKey::Int(2))
8220                .unwrap(),
8221            make_user_row(2, "bob")
8222        );
8223        // Cold tier hits.
8224        assert_eq!(
8225            cat.lookup_by_pk("users", "by_id", &IndexKey::Int(100))
8226                .unwrap(),
8227            make_user_row(100, "ivy")
8228        );
8229        assert_eq!(
8230            cat.lookup_by_pk("users", "by_id", &IndexKey::Int(200))
8231                .unwrap(),
8232            make_user_row(200, "joe")
8233        );
8234        // Miss in both tiers.
8235        assert!(
8236            cat.lookup_by_pk("users", "by_id", &IndexKey::Int(50))
8237                .is_none()
8238        );
8239    }
8240
8241    #[test]
8242    fn register_cold_locators_rejects_nsw_index() {
8243        let mut cat = Catalog::new();
8244        cat.create_table(TableSchema::new(
8245            "vecs",
8246            vec![
8247                ColumnSchema::new("id", DataType::Int, false),
8248                ColumnSchema::new(
8249                    "v",
8250                    DataType::Vector {
8251                        dim: 4,
8252                        encoding: VecEncoding::F32,
8253                    },
8254                    false,
8255                ),
8256            ],
8257        ))
8258        .unwrap();
8259        let t = cat.get_mut("vecs").unwrap();
8260        t.insert(Row::new(vec![
8261            Value::Int(1),
8262            Value::Vector(vec![1.0, 0.0, 0.0, 0.0]),
8263        ]))
8264        .unwrap();
8265        t.add_nsw_index("by_v".into(), "v", NSW_DEFAULT_M).unwrap();
8266        let err = t
8267            .register_cold_locators(
8268                "by_v",
8269                vec![(
8270                    IndexKey::Int(1),
8271                    RowLocator::Cold {
8272                        segment_id: 0,
8273                        page_offset: 0,
8274                    },
8275                )],
8276            )
8277            .unwrap_err();
8278        // v6.7.1: message switched from "is NSW" to "is not BTree"
8279        // when the Brin variant was added.
8280        assert!(matches!(err, StorageError::Corrupt(ref s) if s.contains("not BTree")));
8281    }
8282
8283    #[test]
8284    fn load_segment_bytes_rejects_garbage() {
8285        let mut cat = Catalog::new();
8286        let err = cat.load_segment_bytes(vec![0u8; 10]).unwrap_err();
8287        assert!(matches!(err, StorageError::Corrupt(ref s) if s.contains("segment")));
8288        // Loader doesn't mutate state on error.
8289        assert_eq!(cat.cold_segment_count(), 0);
8290    }
8291
8292    #[test]
8293    fn load_segment_bytes_returns_sequential_ids() {
8294        let mut cat = Catalog::new();
8295        cat.create_table(bigint_pk_users_schema()).unwrap();
8296        let schema = cat.get("users").unwrap().schema.clone();
8297        for batch in 0u32..3 {
8298            let rows: Vec<(u64, Vec<u8>)> = (0u64..4)
8299                .map(|i| {
8300                    let id = u64::from(batch) * 100 + i;
8301                    let row = make_user_row(id.cast_signed(), "x");
8302                    (id, encode_row_body_dense(&row, &schema))
8303                })
8304                .collect();
8305            let (bytes, _) = encode_segment(rows.into_iter(), 0.01, SEGMENT_PAGE_BYTES).unwrap();
8306            assert_eq!(cat.load_segment_bytes(bytes).unwrap(), batch);
8307        }
8308        assert_eq!(cat.cold_segment_count(), 3);
8309    }
8310
8311    // --- v5.2 catalog format v9 ----------------------------------
8312
8313    /// Hand-craft a v8 catalog byte stream and confirm the v9 reader
8314    /// accepts it and surfaces every `BTree` entry as a Hot locator.
8315    /// Guards the backward-compat read path: existing v3.0.2 / v4.x
8316    /// snapshots on disk must keep loading after the v5.2 bump.
8317    #[test]
8318    fn v8_catalog_decodes_as_all_hot_under_v9_reader() {
8319        // Build a populated catalog in memory, snapshot it with the
8320        // v9 serializer, then patch the version byte back to 8 and
8321        // strip the v9 BTree payload bytes so the layout matches what
8322        // a real v8 snapshot would have produced on disk. The v9
8323        // reader's version dispatch path then rebuilds the index
8324        // from rows (every locator becomes Hot).
8325        let mut cat = populated_users();
8326        cat.get_mut("users")
8327            .unwrap()
8328            .add_index("by_name".into(), "name")
8329            .unwrap();
8330
8331        // To produce a faithful v8 byte stream we re-encode the same
8332        // catalog with the v8 layout: identical bytes up to (and
8333        // including) the per-index kind tag, but no inline BTree
8334        // entries.
8335        let v8_bytes = encode_as_v8(&cat);
8336        assert_eq!(v8_bytes[FILE_MAGIC.len()], 8, "version byte must be 8");
8337
8338        let restored = Catalog::deserialize(&v8_bytes).expect("v9 reader accepts v8 stream");
8339        let idx = restored
8340            .get("users")
8341            .unwrap()
8342            .index_on(1)
8343            .expect("index_on(1) after restore");
8344        // v8 path always materialises Hot locators (no cold tier
8345        // existed pre-v5.2).
8346        assert_eq!(
8347            idx.lookup_eq(&IndexKey::Text("alice".into())),
8348            &[RowLocator::Hot(0), RowLocator::Hot(2)]
8349        );
8350        // No accidental Cold leak.
8351        for entry in idx.lookup_eq(&IndexKey::Text("alice".into())) {
8352            assert!(entry.is_hot(), "v8 → v9 read must yield Hot only");
8353        }
8354    }
8355
8356    /// Encode `cat` using the v8 layout (no inline `BTree` entries,
8357    /// version byte = 8). Pure test helper — duplicates just enough
8358    /// of `Catalog::serialize` to produce a faithful v8 stream that
8359    /// real v3.0.2 / v4.x deployments wrote.
8360    fn encode_as_v8(cat: &Catalog) -> Vec<u8> {
8361        let mut out = Vec::with_capacity(64);
8362        out.extend_from_slice(FILE_MAGIC);
8363        out.push(8u8);
8364        write_u32(&mut out, u32::try_from(cat.tables.len()).unwrap());
8365        for t in &cat.tables {
8366            write_str(&mut out, &t.schema.name);
8367            write_u16(&mut out, u16::try_from(t.schema.columns.len()).unwrap());
8368            for c in &t.schema.columns {
8369                write_str(&mut out, &c.name);
8370                write_data_type(&mut out, c.ty);
8371                out.push(u8::from(c.nullable));
8372                match &c.default {
8373                    None => out.push(0),
8374                    Some(v) => {
8375                        out.push(1);
8376                        write_value(&mut out, v);
8377                    }
8378                }
8379                out.push(u8::from(c.auto_increment));
8380            }
8381            write_u32(&mut out, u32::try_from(t.rows.len()).unwrap());
8382            for row in &t.rows {
8383                out.extend_from_slice(&encode_row_body_dense(row, &t.schema));
8384            }
8385            write_u16(&mut out, u16::try_from(t.indices.len()).unwrap());
8386            for idx in &t.indices {
8387                write_str(&mut out, &idx.name);
8388                write_u16(&mut out, u16::try_from(idx.column_position).unwrap());
8389                match &idx.kind {
8390                    // v8 BTree wrote only the kind tag; entries
8391                    // rebuild from rows on read.
8392                    IndexKind::BTree(_) => out.push(0),
8393                    IndexKind::Nsw(g) => {
8394                        out.push(1);
8395                        write_u16(&mut out, u16::try_from(g.m).unwrap());
8396                        write_nsw_graph(&mut out, g);
8397                    }
8398                    // v8 had no BRIN / GIN; this test-only writer
8399                    // can't serialise either into the legacy format.
8400                    IndexKind::Brin { .. } => panic!(
8401                        "v8 catalog writer cannot serialise BRIN — \
8402                         tests with BRIN indices must use the current writer"
8403                    ),
8404                    IndexKind::Gin(_) => panic!(
8405                        "v8 catalog writer cannot serialise GIN — \
8406                         tests with GIN indices must use the current writer"
8407                    ),
8408                    IndexKind::GinTrgm(_) => panic!(
8409                        "v8 catalog writer cannot serialise trigram-GIN — \
8410                         tests with trgm indices must use the current writer"
8411                    ),
8412                }
8413            }
8414        }
8415        out
8416    }
8417
8418    /// Build a catalog that carries both hot and cold locators on a
8419    /// `BTree` index, snapshot it through `serialize`, then deserialise
8420    /// and confirm every Cold locator round-trips byte-identical and
8421    /// `lookup_by_pk` resolves through the rebuilt cold-segment
8422    /// registry.
8423    #[test]
8424    fn v9_catalog_round_trip_preserves_cold_locators() {
8425        let mut cat = Catalog::new();
8426        cat.create_table(bigint_pk_users_schema()).unwrap();
8427        let t = cat.get_mut("users").unwrap();
8428        // Hot rows: 1, 2
8429        for (id, name) in [(1i64, "alice"), (2, "bob")] {
8430            t.insert(make_user_row(id, name)).unwrap();
8431        }
8432        t.add_index("by_id".into(), "id").unwrap();
8433        let schema = t.schema.clone();
8434
8435        // Cold rows: 100, 200, 300 — sit in a single segment.
8436        let cold_rows: Vec<(i64, &str)> = vec![(100, "ivy"), (200, "joe"), (300, "kim")];
8437        let seg_rows: Vec<(u64, Vec<u8>)> = cold_rows
8438            .iter()
8439            .map(|(id, name)| {
8440                let row = make_user_row(*id, name);
8441                ((*id).cast_unsigned(), encode_row_body_dense(&row, &schema))
8442            })
8443            .collect();
8444        let (seg_bytes, _) =
8445            encode_segment(seg_rows.into_iter(), 0.01, SEGMENT_PAGE_BYTES).unwrap();
8446        let seg_id = cat.load_segment_bytes(seg_bytes.clone()).unwrap();
8447        let pairs: Vec<(IndexKey, RowLocator)> = cold_rows
8448            .iter()
8449            .map(|(id, _)| {
8450                (
8451                    IndexKey::Int(*id),
8452                    RowLocator::Cold {
8453                        segment_id: seg_id,
8454                        page_offset: 0,
8455                    },
8456                )
8457            })
8458            .collect();
8459        cat.get_mut("users")
8460            .unwrap()
8461            .register_cold_locators("by_id", pairs)
8462            .unwrap();
8463
8464        // Snapshot + restore via the v9 codec.
8465        let bytes = cat.serialize();
8466        assert_eq!(bytes[FILE_MAGIC.len()], FILE_VERSION);
8467        let mut restored = Catalog::deserialize(&bytes).expect("v9 round-trip parses");
8468
8469        // Catalog::serialize does not yet emit cold segment file
8470        // bytes (v5.3 manifest is the future home for that). For
8471        // this v9 test the caller side-loads the segment again so
8472        // lookup_by_pk can resolve the Cold locator. The point of
8473        // this assertion is that the locator metadata survived the
8474        // catalog round-trip.
8475        let restored_seg_id = restored.load_segment_bytes(seg_bytes).unwrap();
8476        assert_eq!(restored_seg_id, seg_id);
8477
8478        let idx = restored.get("users").unwrap().index_on(0).unwrap();
8479        // Hot locators round-trip.
8480        assert_eq!(idx.lookup_eq(&IndexKey::Int(1)), &[RowLocator::Hot(0)]);
8481        assert_eq!(idx.lookup_eq(&IndexKey::Int(2)), &[RowLocator::Hot(1)]);
8482        // Cold locators round-trip byte-identical.
8483        for (id, _) in &cold_rows {
8484            assert_eq!(
8485                idx.lookup_eq(&IndexKey::Int(*id)),
8486                &[RowLocator::Cold {
8487                    segment_id: seg_id,
8488                    page_offset: 0,
8489                }]
8490            );
8491        }
8492        // End-to-end: lookup_by_pk resolves both tiers.
8493        assert_eq!(
8494            restored
8495                .lookup_by_pk("users", "by_id", &IndexKey::Int(2))
8496                .unwrap(),
8497            make_user_row(2, "bob")
8498        );
8499        for (id, name) in &cold_rows {
8500            assert_eq!(
8501                restored
8502                    .lookup_by_pk("users", "by_id", &IndexKey::Int(*id))
8503                    .unwrap(),
8504                make_user_row(*id, name)
8505            );
8506        }
8507    }
8508
8509    // --- v5.2.1 hot tier byte tracking ---------------------------
8510
8511    /// `row_body_encoded_len` is the perf-critical fast path; pin it
8512    /// against `encode_row_body_dense(...).len()` for every
8513    /// representative cell type so an encoder change can't silently
8514    /// desync the counter.
8515    #[test]
8516    fn row_body_encoded_len_matches_actual_encode_for_all_types() {
8517        let schema = TableSchema::new(
8518            "wide",
8519            vec![
8520                ColumnSchema::new("a", DataType::SmallInt, true),
8521                ColumnSchema::new("b", DataType::Int, false),
8522                ColumnSchema::new("c", DataType::BigInt, false),
8523                ColumnSchema::new("d", DataType::Float, false),
8524                ColumnSchema::new("e", DataType::Bool, false),
8525                ColumnSchema::new("f", DataType::Text, false),
8526                ColumnSchema::new(
8527                    "g",
8528                    DataType::Vector {
8529                        dim: 3,
8530                        encoding: VecEncoding::F32,
8531                    },
8532                    false,
8533                ),
8534                ColumnSchema::new(
8535                    "h",
8536                    DataType::Numeric {
8537                        precision: 18,
8538                        scale: 2,
8539                    },
8540                    false,
8541                ),
8542                ColumnSchema::new("i", DataType::Date, false),
8543                ColumnSchema::new("j", DataType::Timestamp, false),
8544            ],
8545        );
8546        let cases: &[Row] = &[
8547            Row::new(vec![
8548                Value::SmallInt(7),
8549                Value::Int(42),
8550                Value::BigInt(1_000_000),
8551                Value::Float(1.5),
8552                Value::Bool(true),
8553                Value::Text("hello".into()),
8554                Value::Vector(vec![1.0, 2.0, 3.0]),
8555                Value::Numeric {
8556                    scaled: 12345,
8557                    scale: 2,
8558                },
8559                Value::Date(20_000),
8560                Value::Timestamp(1_700_000_000_000_000),
8561            ]),
8562            // NULL in the bitmap, varied text length.
8563            Row::new(vec![
8564                Value::Null,
8565                Value::Int(0),
8566                Value::BigInt(0),
8567                Value::Float(0.0),
8568                Value::Bool(false),
8569                Value::Text(String::new()),
8570                Value::Vector(vec![]),
8571                Value::Numeric {
8572                    scaled: 0,
8573                    scale: 2,
8574                },
8575                Value::Date(0),
8576                Value::Timestamp(0),
8577            ]),
8578            Row::new(vec![
8579                Value::SmallInt(-1),
8580                Value::Int(-1),
8581                Value::BigInt(-1),
8582                Value::Float(-0.5),
8583                Value::Bool(true),
8584                Value::Text("a much longer payload here".into()),
8585                Value::Vector(vec![0.1, 0.2, 0.3]),
8586                Value::Numeric {
8587                    scaled: -999_999_999,
8588                    scale: 2,
8589                },
8590                Value::Date(-1),
8591                Value::Timestamp(-1),
8592            ]),
8593        ];
8594        for row in cases {
8595            let actual = encode_row_body_dense(row, &schema).len();
8596            let fast = row_body_encoded_len(row, &schema);
8597            assert_eq!(actual, fast, "row {row:?}");
8598        }
8599    }
8600
8601    #[test]
8602    fn hot_bytes_grows_on_insert_and_matches_encoded_sum() {
8603        let mut cat = Catalog::new();
8604        cat.create_table(bigint_pk_users_schema()).unwrap();
8605        let t = cat.get_mut("users").unwrap();
8606        assert_eq!(t.hot_bytes(), 0);
8607        let mut expected: u64 = 0;
8608        for (id, name) in [(1i64, "alice"), (2, "bob"), (3, "carol")] {
8609            let row = make_user_row(id, name);
8610            expected += encode_row_body_dense(&row, &t.schema).len() as u64;
8611            t.insert(row).unwrap();
8612        }
8613        assert_eq!(t.hot_bytes(), expected);
8614        assert_eq!(cat.hot_tier_bytes(), expected);
8615    }
8616
8617    #[test]
8618    fn hot_bytes_shrinks_on_delete() {
8619        let mut cat = Catalog::new();
8620        cat.create_table(bigint_pk_users_schema()).unwrap();
8621        let t = cat.get_mut("users").unwrap();
8622        for (id, name) in [(1i64, "alice"), (2, "bob"), (3, "carol")] {
8623            t.insert(make_user_row(id, name)).unwrap();
8624        }
8625        let before = t.hot_bytes();
8626        // Delete row at position 1 (bob).
8627        let bob_row = make_user_row(2, "bob");
8628        let bob_bytes = encode_row_body_dense(&bob_row, &t.schema).len() as u64;
8629        let removed = t.delete_rows(&[1]);
8630        assert_eq!(removed, 1);
8631        assert_eq!(t.hot_bytes(), before - bob_bytes);
8632    }
8633
8634    #[test]
8635    fn hot_bytes_diffs_on_update_for_variable_width_columns() {
8636        let mut cat = Catalog::new();
8637        cat.create_table(bigint_pk_users_schema()).unwrap();
8638        let t = cat.get_mut("users").unwrap();
8639        t.insert(make_user_row(1, "alice")).unwrap();
8640        let after_insert = t.hot_bytes();
8641        // Update with a longer text payload — bytes must grow exactly
8642        // by the text-length delta.
8643        let new_row = make_user_row(1, "alice-the-longer-name");
8644        let old_len = encode_row_body_dense(&make_user_row(1, "alice"), &t.schema).len() as u64;
8645        let new_len = encode_row_body_dense(&new_row, &t.schema).len() as u64;
8646        t.update_row(0, new_row.values).unwrap();
8647        assert_eq!(t.hot_bytes(), after_insert - old_len + new_len);
8648        assert!(t.hot_bytes() > after_insert, "longer text grew the counter");
8649    }
8650
8651    #[test]
8652    fn hot_bytes_round_trips_through_serialize_deserialize() {
8653        let mut cat = Catalog::new();
8654        cat.create_table(bigint_pk_users_schema()).unwrap();
8655        let t = cat.get_mut("users").unwrap();
8656        for i in 0..10 {
8657            t.insert(make_user_row(i, &alloc::format!("name-{i}")))
8658                .unwrap();
8659        }
8660        let pre = cat.hot_tier_bytes();
8661        let restored = Catalog::deserialize(&cat.serialize()).unwrap();
8662        assert_eq!(restored.hot_tier_bytes(), pre);
8663        assert_eq!(restored.get("users").unwrap().hot_bytes(), pre);
8664    }
8665
8666    // --- v5.2.2 freezer atomic swap -------------------------------
8667
8668    /// Happy path: freeze the first half of a populated hot tier,
8669    /// confirm row counts shift, `hot_bytes` shrinks, and every frozen
8670    /// PK still resolves via `lookup_by_pk` (now through the cold
8671    /// segment registered by the freeze).
8672    #[test]
8673    fn freeze_oldest_to_cold_moves_rows_and_keeps_lookups_working() {
8674        let mut cat = Catalog::new();
8675        cat.create_table(bigint_pk_users_schema()).unwrap();
8676        let t = cat.get_mut("users").unwrap();
8677        for id in 0..10i64 {
8678            t.insert(make_user_row(id, &alloc::format!("u-{id}")))
8679                .unwrap();
8680        }
8681        t.add_index("by_id".into(), "id").unwrap();
8682        let total_bytes_before = t.hot_bytes();
8683
8684        let report = cat
8685            .freeze_oldest_to_cold("users", "by_id", 6)
8686            .expect("freeze succeeds");
8687        assert_eq!(report.frozen_rows, 6);
8688        assert_eq!(report.segment_id, 0);
8689        assert!(report.bytes_freed > 0);
8690        assert!(!report.segment_bytes.is_empty());
8691
8692        let t = cat.get("users").unwrap();
8693        assert_eq!(t.row_count(), 4, "4 hot rows remain (10 - 6 frozen)");
8694        assert_eq!(cat.cold_segment_count(), 1);
8695        // Hot bytes shrank by exactly the freed amount.
8696        assert_eq!(
8697            t.hot_bytes(),
8698            total_bytes_before - report.bytes_freed,
8699            "hot_bytes accounting matches FreezeReport"
8700        );
8701
8702        // Every original PK still resolves — frozen ones via the
8703        // cold segment, kept ones via the (renumbered) hot tier.
8704        for id in 0..10i64 {
8705            let got = cat
8706                .lookup_by_pk("users", "by_id", &IndexKey::Int(id))
8707                .unwrap_or_else(|| panic!("PK {id} disappeared after freeze"));
8708            assert_eq!(got, make_user_row(id, &alloc::format!("u-{id}")));
8709        }
8710    }
8711
8712    /// Two successive freezes on the same index must preserve the
8713    /// first batch's cold locators when the second freeze runs.
8714    /// Catches the `rebuild_indices` wipe-Cold-on-delete bug that
8715    /// `collect_cold_locators` / re-register guards against.
8716    #[test]
8717    fn freeze_twice_preserves_prior_cold_locators() {
8718        let mut cat = Catalog::new();
8719        cat.create_table(bigint_pk_users_schema()).unwrap();
8720        let t = cat.get_mut("users").unwrap();
8721        for id in 0..12i64 {
8722            t.insert(make_user_row(id, &alloc::format!("u-{id}")))
8723                .unwrap();
8724        }
8725        t.add_index("by_id".into(), "id").unwrap();
8726
8727        cat.freeze_oldest_to_cold("users", "by_id", 4)
8728            .expect("first freeze ok");
8729        cat.freeze_oldest_to_cold("users", "by_id", 4)
8730            .expect("second freeze ok");
8731
8732        assert_eq!(cat.get("users").unwrap().row_count(), 4);
8733        assert_eq!(cat.cold_segment_count(), 2);
8734        // All 12 PKs still resolve — first 4 via segment 0,
8735        // next 4 via segment 1, last 4 still hot.
8736        for id in 0..12i64 {
8737            let got = cat
8738                .lookup_by_pk("users", "by_id", &IndexKey::Int(id))
8739                .unwrap_or_else(|| panic!("PK {id} not resolvable after two freezes"));
8740            assert_eq!(got, make_user_row(id, &alloc::format!("u-{id}")));
8741        }
8742    }
8743
8744    /// Validation guard tests. Each must return `Err` and **not
8745    /// mutate the catalog** — the API is all-or-nothing.
8746    #[test]
8747    fn freeze_oldest_to_cold_rejects_invalid_input() {
8748        let mut cat = Catalog::new();
8749        cat.create_table(bigint_pk_users_schema()).unwrap();
8750        let t = cat.get_mut("users").unwrap();
8751        for id in 0..3i64 {
8752            t.insert(make_user_row(id, &alloc::format!("u-{id}")))
8753                .unwrap();
8754        }
8755        t.add_index("by_id".into(), "id").unwrap();
8756
8757        // max_rows == 0
8758        assert!(matches!(
8759            cat.freeze_oldest_to_cold("users", "by_id", 0),
8760            Err(StorageError::Corrupt(_))
8761        ));
8762        // table missing
8763        assert!(matches!(
8764            cat.freeze_oldest_to_cold("missing", "by_id", 1),
8765            Err(StorageError::Corrupt(_))
8766        ));
8767        // index missing
8768        assert!(matches!(
8769            cat.freeze_oldest_to_cold("users", "no_such_index", 1),
8770            Err(StorageError::Corrupt(_))
8771        ));
8772        // max_rows > row_count
8773        assert!(matches!(
8774            cat.freeze_oldest_to_cold("users", "by_id", 999),
8775            Err(StorageError::Corrupt(_))
8776        ));
8777        // Catalog still untouched.
8778        assert_eq!(cat.get("users").unwrap().row_count(), 3);
8779        assert_eq!(cat.cold_segment_count(), 0);
8780    }
8781
8782    /// Freeze with a non-integer PK column must surface a clear
8783    /// error (Text PKs land in v5.5+).
8784    #[test]
8785    fn freeze_oldest_to_cold_rejects_non_integer_pk() {
8786        let mut cat = Catalog::new();
8787        cat.create_table(TableSchema::new(
8788            "by_name",
8789            vec![
8790                ColumnSchema::new("name", DataType::Text, false),
8791                ColumnSchema::new("payload", DataType::BigInt, false),
8792            ],
8793        ))
8794        .unwrap();
8795        let t = cat.get_mut("by_name").unwrap();
8796        t.insert(Row::new(vec![Value::Text("a".into()), Value::BigInt(1)]))
8797            .unwrap();
8798        t.add_index("by_n".into(), "name").unwrap();
8799        let err = cat
8800            .freeze_oldest_to_cold("by_name", "by_n", 1)
8801            .expect_err("non-integer PK rejected");
8802        match err {
8803            StorageError::Corrupt(s) => assert!(
8804                s.contains("non-integer"),
8805                "error message names the constraint: {s}"
8806            ),
8807            other => panic!("expected Corrupt, got {other:?}"),
8808        }
8809        // Catalog untouched.
8810        assert_eq!(cat.get("by_name").unwrap().row_count(), 1);
8811        assert_eq!(cat.cold_segment_count(), 0);
8812    }
8813
8814    /// Hot-tier rows after the freeze must keep their secondary-
8815    /// index lookups working — `delete_rows` shifts positions, and
8816    /// `rebuild_indices` must regenerate Hot locators at the new
8817    /// indices.
8818    #[test]
8819    fn freeze_keeps_remaining_hot_rows_addressable_via_secondary_index() {
8820        let mut cat = Catalog::new();
8821        cat.create_table(bigint_pk_users_schema()).unwrap();
8822        let t = cat.get_mut("users").unwrap();
8823        for id in 0..6i64 {
8824            t.insert(make_user_row(id, &alloc::format!("u-{id}")))
8825                .unwrap();
8826        }
8827        t.add_index("by_id".into(), "id").unwrap();
8828        t.add_index("by_name".into(), "name").unwrap();
8829
8830        cat.freeze_oldest_to_cold("users", "by_id", 3).unwrap();
8831
8832        // Remaining hot rows: id 3, 4, 5. They moved to positions
8833        // 0, 1, 2 inside `self.rows`; the `by_name` index must now
8834        // resolve them via fresh Hot locators.
8835        let idx = cat.get("users").unwrap().index_on(1).unwrap();
8836        let got = idx.lookup_eq(&IndexKey::Text("u-4".into()));
8837        assert_eq!(got.len(), 1);
8838        assert!(got[0].is_hot(), "kept-hot rows still surface as Hot");
8839        match got[0] {
8840            RowLocator::Hot(i) => {
8841                // The 4th-inserted row was at position 4; after
8842                // dropping positions 0..3 it sits at position 1.
8843                assert_eq!(i, 1);
8844            }
8845            RowLocator::Cold { .. } => unreachable!(),
8846        }
8847    }
8848
8849    // --- v5.2.3 promote-on-write primitives ----------------------
8850
8851    /// Build a populated catalog with the first N rows frozen, then
8852    /// run `promote_cold_row` and verify the row crossed tiers
8853    /// correctly: the cold locator is retired, a fresh Hot locator
8854    /// appears, `lookup_by_pk` returns the row from the hot tier, and
8855    /// `hot_bytes` grew by the row's encoded byte length.
8856    #[test]
8857    fn promote_cold_row_pulls_frozen_row_back_to_hot_tier() {
8858        let mut cat = Catalog::new();
8859        cat.create_table(bigint_pk_users_schema()).unwrap();
8860        let t = cat.get_mut("users").unwrap();
8861        for id in 0..6i64 {
8862            t.insert(make_user_row(id, &alloc::format!("u-{id}")))
8863                .unwrap();
8864        }
8865        t.add_index("by_id".into(), "id").unwrap();
8866        // Freeze first 4 rows (ids 0..3). After: hot rows = 4, 5 at
8867        // positions 0, 1; cold locators for keys 0..3.
8868        cat.freeze_oldest_to_cold("users", "by_id", 4).unwrap();
8869        let hot_bytes_before = cat.get("users").unwrap().hot_bytes();
8870
8871        // Promote PK=2 — it lives in segment 0 as a cold row.
8872        let new_idx = cat
8873            .promote_cold_row("users", "by_id", &IndexKey::Int(2))
8874            .expect("promote ok")
8875            .expect("PK 2 was cold");
8876        assert_eq!(
8877            new_idx, 2,
8878            "promoted row appended after the 2 surviving hot rows"
8879        );
8880
8881        let t = cat.get("users").unwrap();
8882        assert_eq!(t.row_count(), 3, "hot tier grew from 2 to 3");
8883        // Hot-bytes climbed by exactly one row's encoded length.
8884        let row = make_user_row(2, "u-2");
8885        let row_len = encode_row_body_dense(&row, &t.schema).len() as u64;
8886        assert_eq!(t.hot_bytes(), hot_bytes_before + row_len);
8887
8888        // The index now reports a Hot locator (the freshly inserted
8889        // row) — no Cold locator left for PK 2.
8890        let entries = t.index_on(0).unwrap().lookup_eq(&IndexKey::Int(2));
8891        assert_eq!(entries.len(), 1, "exactly one locator per key");
8892        assert!(entries[0].is_hot(), "promote retired the Cold locator");
8893        // End-to-end: lookup_by_pk still returns the row body.
8894        assert_eq!(
8895            cat.lookup_by_pk("users", "by_id", &IndexKey::Int(2))
8896                .unwrap(),
8897            row
8898        );
8899        // Other cold rows untouched — still resolvable through the
8900        // segment.
8901        assert_eq!(
8902            cat.lookup_by_pk("users", "by_id", &IndexKey::Int(0))
8903                .unwrap(),
8904            make_user_row(0, "u-0")
8905        );
8906    }
8907
8908    /// `promote_cold_row` on a key that's already hot (or absent)
8909    /// returns `Ok(None)` — not an error. The caller falls back to
8910    /// the hot-only update/delete path.
8911    #[test]
8912    fn promote_cold_row_returns_none_when_key_is_not_cold() {
8913        let mut cat = Catalog::new();
8914        cat.create_table(bigint_pk_users_schema()).unwrap();
8915        let t = cat.get_mut("users").unwrap();
8916        t.insert(make_user_row(7, "alice")).unwrap();
8917        t.add_index("by_id".into(), "id").unwrap();
8918
8919        // Hot-only key.
8920        assert!(
8921            cat.promote_cold_row("users", "by_id", &IndexKey::Int(7))
8922                .unwrap()
8923                .is_none()
8924        );
8925        // Absent key.
8926        assert!(
8927            cat.promote_cold_row("users", "by_id", &IndexKey::Int(99))
8928                .unwrap()
8929                .is_none()
8930        );
8931        // Catalog untouched on both no-op paths.
8932        assert_eq!(cat.get("users").unwrap().row_count(), 1);
8933        assert_eq!(cat.cold_segment_count(), 0);
8934    }
8935
8936    /// `shadow_cold_row` removes every Cold locator for a key on a
8937    /// `BTree` index. After the shadow, `lookup_by_pk` for that key
8938    /// returns None (the row data still sits in the segment file,
8939    /// but it's now garbage; compaction will reclaim it later).
8940    #[test]
8941    fn shadow_cold_row_removes_cold_locators_and_drops_lookup() {
8942        let mut cat = Catalog::new();
8943        cat.create_table(bigint_pk_users_schema()).unwrap();
8944        let t = cat.get_mut("users").unwrap();
8945        for id in 0..5i64 {
8946            t.insert(make_user_row(id, &alloc::format!("u-{id}")))
8947                .unwrap();
8948        }
8949        t.add_index("by_id".into(), "id").unwrap();
8950        cat.freeze_oldest_to_cold("users", "by_id", 3).unwrap();
8951
8952        // Shadow PK=1 — pre-shadow lookup hits the cold tier.
8953        assert!(
8954            cat.lookup_by_pk("users", "by_id", &IndexKey::Int(1))
8955                .is_some(),
8956            "frozen PK resolves before shadow"
8957        );
8958        let removed = cat
8959            .shadow_cold_row("users", "by_id", &IndexKey::Int(1))
8960            .unwrap();
8961        assert_eq!(removed, 1, "exactly one cold locator retired");
8962
8963        // Post-shadow: lookup misses, even though the row still
8964        // exists in segment 0.
8965        assert!(
8966            cat.lookup_by_pk("users", "by_id", &IndexKey::Int(1))
8967                .is_none(),
8968            "shadowed key no longer resolves"
8969        );
8970        // Other cold keys still resolve.
8971        assert_eq!(
8972            cat.lookup_by_pk("users", "by_id", &IndexKey::Int(0))
8973                .unwrap(),
8974            make_user_row(0, "u-0")
8975        );
8976        assert_eq!(
8977            cat.lookup_by_pk("users", "by_id", &IndexKey::Int(2))
8978                .unwrap(),
8979            make_user_row(2, "u-2")
8980        );
8981    }
8982
8983    /// `shadow_cold_row` returns 0 (not Err) for keys with only Hot
8984    /// entries or no entries — the engine's DELETE path uses this
8985    /// signal to decide whether the cold-tier shadow path consumed
8986    /// the work.
8987    #[test]
8988    fn shadow_cold_row_returns_zero_when_key_is_not_cold() {
8989        let mut cat = Catalog::new();
8990        cat.create_table(bigint_pk_users_schema()).unwrap();
8991        let t = cat.get_mut("users").unwrap();
8992        t.insert(make_user_row(1, "alice")).unwrap();
8993        t.add_index("by_id".into(), "id").unwrap();
8994        assert_eq!(
8995            cat.shadow_cold_row("users", "by_id", &IndexKey::Int(1))
8996                .unwrap(),
8997            0,
8998            "hot-only key drops no cold locators"
8999        );
9000        assert_eq!(
9001            cat.shadow_cold_row("users", "by_id", &IndexKey::Int(999))
9002                .unwrap(),
9003            0,
9004            "absent key drops no cold locators"
9005        );
9006        assert_eq!(cat.get("users").unwrap().row_count(), 1);
9007    }
9008
9009    /// Validation guards on both promote / shadow primitives.
9010    #[test]
9011    fn promote_and_shadow_reject_invalid_inputs() {
9012        let mut cat = Catalog::new();
9013        cat.create_table(bigint_pk_users_schema()).unwrap();
9014        let t = cat.get_mut("users").unwrap();
9015        t.insert(make_user_row(1, "alice")).unwrap();
9016        t.add_index("by_id".into(), "id").unwrap();
9017
9018        // Missing table.
9019        assert!(matches!(
9020            cat.promote_cold_row("missing", "by_id", &IndexKey::Int(1)),
9021            Err(StorageError::Corrupt(_))
9022        ));
9023        assert!(matches!(
9024            cat.shadow_cold_row("missing", "by_id", &IndexKey::Int(1)),
9025            Err(StorageError::Corrupt(_))
9026        ));
9027        // Missing index.
9028        assert!(matches!(
9029            cat.promote_cold_row("users", "no_such_index", &IndexKey::Int(1)),
9030            Err(StorageError::Corrupt(_))
9031        ));
9032        assert!(matches!(
9033            cat.shadow_cold_row("users", "no_such_index", &IndexKey::Int(1)),
9034            Err(StorageError::Corrupt(_))
9035        ));
9036    }
9037
9038    // --- v6.7.4 parallel-freezer slice/commit API -----------------
9039
9040    /// One slice covering the entire freeze produces the same
9041    /// catalog state as the single-threaded `freeze_oldest_to_cold`
9042    /// — segment id, frozen row count, hot byte delta, and every
9043    /// post-freeze PK lookup match exactly.
9044    #[test]
9045    fn commit_freeze_slices_single_slice_matches_freeze_oldest() {
9046        let mut a = Catalog::new();
9047        let mut b = Catalog::new();
9048        for cat in [&mut a, &mut b] {
9049            cat.create_table(bigint_pk_users_schema()).unwrap();
9050            let t = cat.get_mut("users").unwrap();
9051            for id in 0..10i64 {
9052                t.insert(make_user_row(id, &alloc::format!("u-{id}")))
9053                    .unwrap();
9054            }
9055            t.add_index("by_id".into(), "id").unwrap();
9056        }
9057        let single = a.freeze_oldest_to_cold("users", "by_id", 6).unwrap();
9058        let slice = b
9059            .prepare_freeze_slice("users", "by_id", 0..6)
9060            .expect("prepare");
9061        let parallel = b
9062            .commit_freeze_slices("users", "by_id", alloc::vec![slice])
9063            .expect("commit");
9064        assert_eq!(single.segment_id, parallel.segment_id);
9065        assert_eq!(single.frozen_rows, parallel.frozen_rows);
9066        assert_eq!(single.bytes_freed, parallel.bytes_freed);
9067        assert_eq!(single.segment_bytes, parallel.segment_bytes);
9068        // Same post-freeze lookup behaviour on both catalogs.
9069        for id in 0..10i64 {
9070            assert_eq!(
9071                a.lookup_by_pk("users", "by_id", &IndexKey::Int(id)),
9072                b.lookup_by_pk("users", "by_id", &IndexKey::Int(id)),
9073                "PK {id} differs after single vs slice freeze"
9074            );
9075        }
9076    }
9077
9078    /// Two slices covering disjoint halves of the freeze produce
9079    /// the same merged segment as one slice covering the full
9080    /// range. The k-way merge preserves PK ordering even when
9081    /// slice halves alternate.
9082    #[test]
9083    fn commit_freeze_slices_two_slices_match_single_slice() {
9084        let mut a = Catalog::new();
9085        let mut b = Catalog::new();
9086        for cat in [&mut a, &mut b] {
9087            cat.create_table(bigint_pk_users_schema()).unwrap();
9088            let t = cat.get_mut("users").unwrap();
9089            // Random-ish PKs so the per-slice sort actually has
9090            // work to do (and slice halves carry interleaved keys).
9091            for id in [3, 7, 1, 9, 5, 0, 8, 4, 2, 6].iter().copied() {
9092                t.insert(make_user_row(id as i64, &alloc::format!("u-{id}")))
9093                    .unwrap();
9094            }
9095            t.add_index("by_id".into(), "id").unwrap();
9096        }
9097        let single = a
9098            .prepare_freeze_slice("users", "by_id", 0..8)
9099            .expect("prepare");
9100        let one = a
9101            .commit_freeze_slices("users", "by_id", alloc::vec![single])
9102            .expect("commit one");
9103        let s1 = b
9104            .prepare_freeze_slice("users", "by_id", 0..4)
9105            .expect("prepare s1");
9106        let s2 = b
9107            .prepare_freeze_slice("users", "by_id", 4..8)
9108            .expect("prepare s2");
9109        let two = b
9110            .commit_freeze_slices("users", "by_id", alloc::vec![s1, s2])
9111            .expect("commit two");
9112        assert_eq!(one.segment_bytes, two.segment_bytes);
9113        assert_eq!(one.frozen_rows, two.frozen_rows);
9114        // Every PK that survived freeze (hot or cold) resolves on
9115        // both catalogs.
9116        for id in 0..10i64 {
9117            assert_eq!(
9118                a.lookup_by_pk("users", "by_id", &IndexKey::Int(id)),
9119                b.lookup_by_pk("users", "by_id", &IndexKey::Int(id)),
9120                "PK {id} differs after one-slice vs two-slice freeze"
9121            );
9122        }
9123    }
9124
9125    /// Gap between slices → error before any mutation lands.
9126    #[test]
9127    fn commit_freeze_slices_rejects_gap() {
9128        let mut cat = Catalog::new();
9129        cat.create_table(bigint_pk_users_schema()).unwrap();
9130        let t = cat.get_mut("users").unwrap();
9131        for id in 0..6i64 {
9132            t.insert(make_user_row(id, &alloc::format!("u-{id}")))
9133                .unwrap();
9134        }
9135        t.add_index("by_id".into(), "id").unwrap();
9136        let s1 = cat.prepare_freeze_slice("users", "by_id", 0..2).unwrap();
9137        let s2 = cat.prepare_freeze_slice("users", "by_id", 3..5).unwrap();
9138        assert!(matches!(
9139            cat.commit_freeze_slices("users", "by_id", alloc::vec![s1, s2]),
9140            Err(StorageError::Corrupt(_))
9141        ));
9142        // Catalog untouched.
9143        assert_eq!(cat.cold_segment_count(), 0);
9144        assert_eq!(cat.get("users").unwrap().row_count(), 6);
9145    }
9146
9147    /// Empty slice list → no-op success, catalog untouched.
9148    #[test]
9149    fn commit_freeze_slices_empty_is_noop() {
9150        let mut cat = Catalog::new();
9151        cat.create_table(bigint_pk_users_schema()).unwrap();
9152        let t = cat.get_mut("users").unwrap();
9153        for id in 0..3i64 {
9154            t.insert(make_user_row(id, &alloc::format!("u-{id}")))
9155                .unwrap();
9156        }
9157        t.add_index("by_id".into(), "id").unwrap();
9158        let report = cat
9159            .commit_freeze_slices("users", "by_id", Vec::new())
9160            .unwrap();
9161        assert_eq!(report.frozen_rows, 0);
9162        assert_eq!(cat.cold_segment_count(), 0);
9163        assert_eq!(cat.get("users").unwrap().row_count(), 3);
9164    }
9165
9166    // --- v6.7.3 cold-segment compaction ---------------------------
9167
9168    /// Two small cold segments merge into a single larger one. The
9169    /// merged segment carries every cold-resident row; the source
9170    /// slots are tombstoned; every PK still resolves through the
9171    /// new merged segment via `lookup_by_pk`.
9172    #[test]
9173    fn compact_merges_small_segments_storage_unit() {
9174        let mut cat = Catalog::new();
9175        cat.create_table(bigint_pk_users_schema()).unwrap();
9176        let t = cat.get_mut("users").unwrap();
9177        for id in 0..8i64 {
9178            t.insert(make_user_row(id, &alloc::format!("u-{id}")))
9179                .unwrap();
9180        }
9181        t.add_index("by_id".into(), "id").unwrap();
9182        // Two freezes of 3 rows each → two small cold segments.
9183        cat.freeze_oldest_to_cold("users", "by_id", 3).unwrap();
9184        cat.freeze_oldest_to_cold("users", "by_id", 3).unwrap();
9185        assert_eq!(cat.cold_segment_count(), 2);
9186        assert_eq!(cat.cold_segment_slot_count(), 2);
9187
9188        // Pick a threshold larger than either segment's size so
9189        // both qualify.
9190        let max_seg_bytes = cat
9191            .cold_segment_ids_global()
9192            .iter()
9193            .map(|id| cat.cold_segment(*id).unwrap().bytes().len() as u64)
9194            .max()
9195            .unwrap();
9196        let target = max_seg_bytes + 1;
9197
9198        let report = cat
9199            .compact_cold_segments("users", "by_id", target)
9200            .expect("compact succeeds");
9201        assert_eq!(report.sources.len(), 2);
9202        let merged_id = report.merged_segment_id.expect("merge happened");
9203        assert_eq!(report.merged_rows, 6);
9204        assert_eq!(report.deleted_rows_pruned, 0);
9205        assert!(!report.merged_segment_bytes.is_empty());
9206
9207        // Active count drops back to 1; slot count grew to 3
9208        // (2 sources tombstoned + 1 merged appended).
9209        assert_eq!(cat.cold_segment_count(), 1);
9210        assert_eq!(cat.cold_segment_slot_count(), 3);
9211        assert_eq!(cat.cold_segment_ids_global(), alloc::vec![merged_id]);
9212
9213        // Every PK that was frozen still resolves (via the merged
9214        // segment); the 2 hot rows still resolve too.
9215        for id in 0..8i64 {
9216            let got = cat
9217                .lookup_by_pk("users", "by_id", &IndexKey::Int(id))
9218                .unwrap_or_else(|| panic!("PK {id} lost after compaction"));
9219            assert_eq!(got, make_user_row(id, &alloc::format!("u-{id}")));
9220        }
9221    }
9222
9223    /// DELETE'd-but-frozen rows are dropped during the merge. Set
9224    /// up two small segments, then shadow one row in each; the
9225    /// merged segment must NOT carry the shadowed rows.
9226    #[test]
9227    fn compact_drops_shadowed_cold_rows() {
9228        let mut cat = Catalog::new();
9229        cat.create_table(bigint_pk_users_schema()).unwrap();
9230        let t = cat.get_mut("users").unwrap();
9231        for id in 0..6i64 {
9232            t.insert(make_user_row(id, &alloc::format!("u-{id}")))
9233                .unwrap();
9234        }
9235        t.add_index("by_id".into(), "id").unwrap();
9236        cat.freeze_oldest_to_cold("users", "by_id", 3).unwrap();
9237        cat.freeze_oldest_to_cold("users", "by_id", 3).unwrap();
9238        // Shadow PK 1 (in seg 0) + PK 4 (in seg 1).
9239        assert_eq!(
9240            cat.shadow_cold_row("users", "by_id", &IndexKey::Int(1))
9241                .unwrap(),
9242            1
9243        );
9244        assert_eq!(
9245            cat.shadow_cold_row("users", "by_id", &IndexKey::Int(4))
9246                .unwrap(),
9247            1
9248        );
9249
9250        let max_seg_bytes = cat
9251            .cold_segment_ids_global()
9252            .iter()
9253            .map(|id| cat.cold_segment(*id).unwrap().bytes().len() as u64)
9254            .max()
9255            .unwrap();
9256        let report = cat
9257            .compact_cold_segments("users", "by_id", max_seg_bytes + 1)
9258            .expect("compact succeeds");
9259        assert_eq!(report.sources.len(), 2);
9260        assert_eq!(report.merged_rows, 4, "6 frozen − 2 shadowed = 4 live");
9261        assert_eq!(report.deleted_rows_pruned, 2);
9262
9263        // PK 1 and 4 stay invisible after compact.
9264        for shadowed in [1i64, 4i64] {
9265            assert!(
9266                cat.lookup_by_pk("users", "by_id", &IndexKey::Int(shadowed))
9267                    .is_none(),
9268                "shadowed PK {shadowed} must remain invisible after compact"
9269            );
9270        }
9271        // The other 4 frozen rows resolve.
9272        for live in [0i64, 2, 3, 5] {
9273            cat.lookup_by_pk("users", "by_id", &IndexKey::Int(live))
9274                .unwrap_or_else(|| panic!("live PK {live} lost after compact"));
9275        }
9276    }
9277
9278    /// No-op cases: 0 or 1 candidate segment under the threshold
9279    /// leaves the catalog untouched.
9280    #[test]
9281    fn compact_is_noop_below_two_candidates() {
9282        let mut cat = Catalog::new();
9283        cat.create_table(bigint_pk_users_schema()).unwrap();
9284        let t = cat.get_mut("users").unwrap();
9285        for id in 0..6i64 {
9286            t.insert(make_user_row(id, &alloc::format!("u-{id}")))
9287                .unwrap();
9288        }
9289        t.add_index("by_id".into(), "id").unwrap();
9290        // 0 cold segments.
9291        let report = cat
9292            .compact_cold_segments("users", "by_id", 1 << 30)
9293            .expect("noop ok");
9294        assert!(report.merged_segment_id.is_none());
9295        assert!(report.sources.is_empty());
9296
9297        // 1 cold segment — still a no-op (need ≥2 to merge).
9298        cat.freeze_oldest_to_cold("users", "by_id", 4).unwrap();
9299        let report = cat
9300            .compact_cold_segments("users", "by_id", 1 << 30)
9301            .expect("noop ok");
9302        assert!(report.merged_segment_id.is_none());
9303        assert_eq!(cat.cold_segment_count(), 1);
9304
9305        // Threshold too small to cover the single segment → still
9306        // no-op.
9307        let report = cat
9308            .compact_cold_segments("users", "by_id", 1)
9309            .expect("noop ok");
9310        assert!(report.merged_segment_id.is_none());
9311        assert_eq!(cat.cold_segment_count(), 1);
9312    }
9313
9314    /// Manifest-style atomicity: a Catalog snapshot taken AFTER
9315    /// `compact_cold_segments` returns must round-trip with the
9316    /// post-compact BTree state, while the cold-tier registry is
9317    /// re-derived from the source-of-truth manifest (=
9318    /// `load_segment_bytes_at` with the merged id + the still-on-
9319    /// disk merged bytes). This mirrors the boot path: catalog
9320    /// snapshot + cold-segment files = full state.
9321    #[test]
9322    fn compact_swap_survives_catalog_roundtrip_via_load_at() {
9323        let mut cat = Catalog::new();
9324        cat.create_table(bigint_pk_users_schema()).unwrap();
9325        let t = cat.get_mut("users").unwrap();
9326        for id in 0..6i64 {
9327            t.insert(make_user_row(id, &alloc::format!("u-{id}")))
9328                .unwrap();
9329        }
9330        t.add_index("by_id".into(), "id").unwrap();
9331        cat.freeze_oldest_to_cold("users", "by_id", 3).unwrap();
9332        cat.freeze_oldest_to_cold("users", "by_id", 3).unwrap();
9333        let max_seg_bytes = cat
9334            .cold_segment_ids_global()
9335            .iter()
9336            .map(|id| cat.cold_segment(*id).unwrap().bytes().len() as u64)
9337            .max()
9338            .unwrap();
9339        let report = cat
9340            .compact_cold_segments("users", "by_id", max_seg_bytes + 1)
9341            .expect("compact ok");
9342        let merged_id = report.merged_segment_id.unwrap();
9343
9344        // Serialise the catalog (BTree index points at merged_id
9345        // now) and the merged segment bytes; pretend to crash; on
9346        // restart, re-hydrate the catalog and reload only the
9347        // merged segment at its baked-in id.
9348        let cat_bytes = cat.serialize();
9349        let merged_bytes = report.merged_segment_bytes.clone();
9350
9351        let mut restored = Catalog::deserialize(&cat_bytes).expect("deserialize ok");
9352        restored
9353            .load_segment_bytes_at(merged_id, merged_bytes)
9354            .expect("reload merged ok");
9355
9356        // All 6 PKs still resolve through the restored merged segment.
9357        for id in 0..6i64 {
9358            let got = restored
9359                .lookup_by_pk("users", "by_id", &IndexKey::Int(id))
9360                .unwrap_or_else(|| panic!("PK {id} lost across roundtrip"));
9361            assert_eq!(got, make_user_row(id, &alloc::format!("u-{id}")));
9362        }
9363        // No source slot ever rehydrates — confirmed by
9364        // `cold_segment_count` matching only the merged segment.
9365        assert_eq!(restored.cold_segment_count(), 1);
9366    }
9367
9368    /// `load_segment_bytes_at` refuses to stomp an occupied slot
9369    /// and pads with `None` when the target id is past the end.
9370    #[test]
9371    fn load_segment_bytes_at_pads_and_rejects_collision() {
9372        let mut cat = Catalog::new();
9373        cat.create_table(bigint_pk_users_schema()).unwrap();
9374        let t = cat.get_mut("users").unwrap();
9375        for id in 0..4i64 {
9376            t.insert(make_user_row(id, &alloc::format!("u-{id}")))
9377                .unwrap();
9378        }
9379        t.add_index("by_id".into(), "id").unwrap();
9380        let report = cat.freeze_oldest_to_cold("users", "by_id", 2).unwrap();
9381        let bytes_seg0 = report.segment_bytes.clone();
9382
9383        // Pad to id=5 (slots 1..5 are None, slot 5 holds the
9384        // segment loaded back). The slot count jumps, the active
9385        // count is now 2 (seg 0 + seg 5).
9386        cat.load_segment_bytes_at(5, bytes_seg0.clone())
9387            .expect("pad + load ok");
9388        assert_eq!(cat.cold_segment_slot_count(), 6);
9389        assert_eq!(cat.cold_segment_count(), 2);
9390
9391        // Re-loading at the same id collides.
9392        assert!(matches!(
9393            cat.load_segment_bytes_at(5, bytes_seg0.clone()),
9394            Err(StorageError::Corrupt(_))
9395        ));
9396        // Re-loading at id 0 (already occupied) also collides.
9397        assert!(matches!(
9398            cat.load_segment_bytes_at(0, bytes_seg0),
9399            Err(StorageError::Corrupt(_))
9400        ));
9401    }
9402
9403    /// Round trip: freeze → promote → re-freeze. The same PK can
9404    /// migrate hot ↔ cold multiple times. After two cycles only the
9405    /// final Hot locator should be live.
9406    #[test]
9407    fn promote_then_refreeze_does_not_leave_orphan_locators() {
9408        let mut cat = Catalog::new();
9409        cat.create_table(bigint_pk_users_schema()).unwrap();
9410        let t = cat.get_mut("users").unwrap();
9411        for id in 0..4i64 {
9412            t.insert(make_user_row(id, &alloc::format!("u-{id}")))
9413                .unwrap();
9414        }
9415        t.add_index("by_id".into(), "id").unwrap();
9416
9417        // Cycle 1: freeze first 2 rows, then promote PK 0.
9418        cat.freeze_oldest_to_cold("users", "by_id", 2).unwrap();
9419        let promoted = cat
9420            .promote_cold_row("users", "by_id", &IndexKey::Int(0))
9421            .unwrap();
9422        assert!(promoted.is_some());
9423        let entries_after_promote = cat
9424            .get("users")
9425            .unwrap()
9426            .index_on(0)
9427            .unwrap()
9428            .lookup_eq(&IndexKey::Int(0))
9429            .to_vec();
9430        assert_eq!(entries_after_promote.len(), 1);
9431        assert!(entries_after_promote[0].is_hot());
9432
9433        // Cycle 2: freeze the front rows again. PK 0 is now at
9434        // position 2 (after the survivors); it could still go cold
9435        // again on a future freeze depending on policy, but the
9436        // current "first N positions" policy leaves it alone here.
9437        // What matters: prior cold locators for PKs 0..1 are gone,
9438        // PKs 2..3 still resolve through their original segments.
9439        for id in [2i64, 3] {
9440            assert_eq!(
9441                cat.lookup_by_pk("users", "by_id", &IndexKey::Int(id))
9442                    .unwrap(),
9443                make_user_row(id, &alloc::format!("u-{id}"))
9444            );
9445        }
9446    }
9447}