powdb_query/executor/

prepared.rs

//! PreparedQuery struct and related Engine methods.

use crate::ast::*;
use crate::plan::*;
use crate::result::{QueryError, QueryResult};
use powdb_storage::catalog::Catalog;
use powdb_storage::row::{ROW_MAGIC, ROW_PREFIX_SIZE};
use powdb_storage::types::*;

use super::compiled::*;
use super::eval::*;
use super::Engine;

pub struct PreparedQuery {
    plan_template: PlanNode,
    /// Total number of `Expr::Literal` slots reachable from the plan.
    /// Callers must supply exactly this many literals per execution.
    pub param_count: usize,
    /// Fast-path metadata for `PlanNode::Insert`. `Some` when:
    ///   * the template is an Insert, and
    ///   * every assignment RHS is `Expr::Literal(_)` (no computed exprs),
    ///     which means param_count == assignments.len() and the caller's
    ///     literal slice maps 1:1 to schema column indices.
    ///
    /// Mission C Phase 15: upgraded from a bare `Vec<usize>` to a
    /// dedicated [`InsertFast`] struct so the execute path can skip the
    /// second `catalog.schema(table)` HashMap lookup just to read
    /// `n_cols`, and can dispatch through `get_table_mut` + `tbl.insert`
    /// instead of going via the generic `catalog.insert` wrapper.
    insert_fast: Option<InsertFast>,
    /// Mission C Phase 14: fast-path metadata for point updates by primary
    /// key — `T filter .pk = <lit> update { col := <lit> }` where `pk` is
    /// an indexed column and `col` is fixed-size and not indexed. At
    /// execute time we skip plan clone, substitute walk, schema re-lookup,
    /// `resolved_assignments` + `FastPatch` + `matching_rids` Vec allocs,
    /// and the whole `PlanNode::Update` arm. Just a btree lookup and a
    /// byte patch.
    update_pk_fast: Option<UpdatePkFast>,
}

/// Mission C Phase 15: precomputed insert fast-path metadata. Built once
/// in [`Engine::prepare`] from a `PlanNode::Insert` template whose every
/// assignment RHS is a raw literal. The execute path reads `n_cols` and
/// `col_indices` directly — no catalog schema lookup needed.
#[derive(Clone)]
struct InsertFast {
    /// Mission C Phase 18: cached slot index into `Catalog::tables`.
    /// DROP/ALTER/index DDL can invalidate the slot or row contract, so every
    /// execution compares the O(1) catalog structure generation below.
    table_slot: usize,
    structure_generation: u64,
    /// Schema column index for each positional literal, in the order the
    /// caller passes them.
    col_indices: Vec<usize>,
    /// Total number of schema columns — the size `insert_values_scratch`
    /// must be resized to before filling positions via `col_indices`.
    /// Cached here so the hot loop skips `catalog.schema(table)` entirely.
    n_cols: usize,
    /// Schema slots omitted by this prepared INSERT. The scratch row is shared
    /// by all prepared inserts on an engine, so these positions must be reset
    /// to NULL before each execution. Precomputing the complement avoids an
    /// O(columns × assignments) membership scan on the write hot path.
    omitted_col_indices: Vec<usize>,
    /// Assigned column definitions in parameter order. Runtime literals still
    /// pass through the same coercion rules as the generic INSERT executor;
    /// preparing with an integer placeholder must not permit a later string to
    /// be stored in an integer column.
    assigned_columns: Vec<ColumnDef>,
    /// Required slots are checked after coercion so a bound NULL cannot bypass
    /// the generic INSERT required-column contract.
    required_col_indices: Vec<usize>,
    /// Needed to mark dependent materialized views dirty in both prepared
    /// execution variants.
    table_name: String,
    /// Prepare-time schema names used only to preserve canonical required-field
    /// errors. Structural validity is the O(1) generation check above.
    schema_columns: Vec<ColumnDef>,
}

/// Mission C Phase 14: precomputed fast-path for `update_by_pk` shaped
/// prepared queries. Built once in [`Engine::prepare`] and reused on every
/// `execute_prepared` call.
#[derive(Clone)]
struct UpdatePkFast {
    /// Mission C Phase 18: cached slot index into `Catalog::tables`, guarded
    /// by the O(1) catalog structure generation on every execution.
    table_slot: usize,
    structure_generation: u64,
    /// Name of the key column (the `.id = ?` side). We look this up in
    /// the owning table's `indexed_cols` at execute time rather than
    /// caching a raw `&BTree` — the engine owns the catalog and can't
    /// hand out long-lived borrows anyway, and the n≤5 linear scan is
    /// a handful of ns.
    key_col: String,
    /// Target column position. A later ALTER ADD INDEX on this column must
    /// disable the raw byte-patch path so live secondary indexes and unique
    /// constraints are maintained by the generic update executor.
    target_col_idx: usize,
    /// Byte offset of the target fixed column in the row encoding:
    /// `2 + bitmap_size + layout.fixed_offsets[target_col]`.
    field_off: usize,
    /// Byte offset of the bitmap byte containing the target column's null
    /// bit (`2 + target_col / 8`).
    bitmap_byte_off: usize,
    /// Bit mask for the target column's null bit.
    bit_mask: u8,
    /// Type of the target fixed column — drives the literal-to-bytes
    /// encoding at execute time.
    target_type: TypeId,
    /// Index into the caller's `literals` slice that holds the filter key.
    /// Always 0 today (filter literal is visited before the assignment
    /// RHS), but stored explicitly so the contract is obvious.
    key_literal_idx: usize,
    /// Index into the caller's `literals` slice that holds the new value.
    value_literal_idx: usize,
}

fn cached_table_matches(catalog: &Catalog, structure_generation: u64) -> bool {
    catalog.structure_generation() == structure_generation
}

fn literal_can_take_without_error(literal: &Literal, column: &ColumnDef) -> bool {
    matches!(
        (literal, column.type_id),
        (
            Literal::Int(_),
            TypeId::Int | TypeId::Float | TypeId::DateTime
        ) | (Literal::Float(_), TypeId::Float | TypeId::Int)
            | (Literal::String(_), TypeId::Str)
            | (Literal::Bool(_), TypeId::Bool)
    )
}

fn restore_taken_strings(fast: &InsertFast, literals: &mut [Literal], values: &mut [Value]) {
    for (position, literal) in literals.iter_mut().enumerate() {
        if let Literal::String(destination) = literal {
            if let Value::Str(source) = &mut values[fast.col_indices[position]] {
                *destination = std::mem::take(source);
            }
        }
    }
}

impl Engine {
    pub fn prepare(&mut self, query: &str) -> Result<PreparedQuery, QueryError> {
        // The stored template is the RAW plan, exactly like the plan cache's
        // entries: lowering is a function of catalog state, and a prepared
        // statement outlives the DDL that changes it. Lowering therefore
        // happens per execution, below.
        let (plan, _) = self.plan_text_and_lower(query)?;
        // Same walk-order restriction as the plan cache: a nested block that
        // wrote `offset` before `limit` cannot have its slots rebound in
        // source order.
        if crate::plan_cache::nested_projection_defeats_cache(&plan) {
            return Err(QueryError::Execution(
                "cannot prepare a nested projection that writes `offset` before \
                 `limit`; write `limit` before `offset` in the nested block"
                    .into(),
            ));
        }
        let param_count = crate::plan_cache::count_literal_slots(&plan);

        // Insert fast path: if the template is Insert and every assignment
        // RHS is a literal, resolve column indices once here and store
        // them. execute_prepared will skip the plan-clone + substitute
        // walk on this path.
        //
        // Mission C Phase 15: also cache `n_cols` and the target table
        // name so execute_prepared doesn't need a second HashMap lookup
        // on `self.catalog.schema(table)` just to size the scratch Vec.
        let insert_fast = match &plan {
            // Single-row inserts only: the byte-level fast path patches one
            // row's worth of scratch. Multi-row `insert T {..},{..}` falls
            // through to the generic plan path (always correct).
            PlanNode::Insert {
                table,
                rows,
                returning,
            } if !returning
                && rows.len() == 1
                && rows[0].iter().all(|a| matches!(a.value, Expr::Literal(_)))
                && param_count == rows[0].len() =>
            {
                let assignments = &rows[0];
                let table_slot = self
                    .catalog
                    .table_slot(table)
                    .ok_or_else(|| QueryError::TableNotFound(table.clone()))?;
                let schema = self.catalog.table_by_slot(table_slot).schema();
                let n_cols = schema.columns.len();
                let indices: Result<Vec<usize>, QueryError> = assignments
                    .iter()
                    .map(|a| {
                        schema
                            .column_index(&a.field)
                            .ok_or_else(|| QueryError::ColumnNotFound {
                                table: table.clone(),
                                column: a.field.clone(),
                            })
                    })
                    .collect();
                let indices = indices?;
                let defaults = self.catalog.column_defaults(table).unwrap_or(&[]);
                let auto = self.catalog.auto_columns(table).unwrap_or(&[]);
                let omitted_required = schema
                    .columns
                    .iter()
                    .enumerate()
                    .any(|(index, column)| column.required && !indices.contains(&index));
                // Defaults and auto columns require table-owned state updates.
                // Keep those shapes on the generic executor, which applies the
                // full schema contract. An omitted required column must also
                // take the generic path so it returns the canonical error.
                if defaults.iter().any(Option::is_some)
                    || auto.iter().any(|is_auto| *is_auto)
                    || omitted_required
                {
                    None
                } else {
                    let omitted_col_indices = (0..n_cols)
                        .filter(|index| !indices.contains(index))
                        .collect();
                    let assigned_columns = indices
                        .iter()
                        .map(|&index| schema.columns[index].clone())
                        .collect();
                    let required_col_indices = schema
                        .columns
                        .iter()
                        .enumerate()
                        .filter_map(|(index, column)| column.required.then_some(index))
                        .collect();
                    Some(InsertFast {
                        table_slot,
                        structure_generation: self.catalog.structure_generation(),
                        col_indices: indices,
                        n_cols,
                        omitted_col_indices,
                        assigned_columns,
                        required_col_indices,
                        table_name: table.clone(),
                        schema_columns: schema.columns.clone(),
                    })
                }
            }
            _ => None,
        };

        // Mission C Phase 14: update-by-pk fast path. Match on the shape
        // planner::plan_update builds for `T filter .pk = ? update
        // { col := ? }` — `Update { input: IndexScan(pk), assignments:
        // [{col, Literal}] }` — and only if every precondition holds:
        //   * `pk` is an indexed column (so the executor would take the
        //     btree.lookup path at run time regardless)
        //   * there's exactly one assignment
        //   * the assigned column is fixed-size and *not* indexed (so we
        //     don't have to maintain any secondary index on write)
        //   * both literal slots are already `Expr::Literal` (no computed
        //     expressions)
        // If any of these fail we fall through to the standard substitute
        // + execute path.
        let update_pk_fast = Self::try_build_update_pk_fast(&self.catalog, &plan);

        Ok(PreparedQuery {
            plan_template: plan,
            param_count,
            insert_fast,
            update_pk_fast,
        })
    }

    /// Mission C Phase 14: inspect a planned tree and, if it matches the
    /// `update_by_pk` fast-path shape, return the precomputed byte-patch
    /// metadata. Returns `None` on any mismatch — the caller falls through
    /// to the substitute-and-execute path, which is always correct.
    fn try_build_update_pk_fast(catalog: &Catalog, plan: &PlanNode) -> Option<UpdatePkFast> {
        // Top level must be `Update { input: IndexScan(...), ... }`.
        let (table, input, assignments) = match plan {
            // `returning` must materialize the post-update row image, which the
            // byte-patch fast path can't produce — fall through to the generic
            // executor arm.
            PlanNode::Update {
                table,
                input,
                assignments,
                returning: false,
            } => (table, input.as_ref(), assignments),
            _ => return None,
        };
        // Exactly one assignment — the bench hot path and the only case
        // where a single byte-patch covers the whole mutation.
        if assignments.len() != 1 {
            return None;
        }
        let assn = &assignments[0];
        // Assignment RHS must be a raw literal, not a computed expr.
        if !matches!(assn.value, Expr::Literal(_)) {
            return None;
        }
        // Input must be an IndexScan on the same table with a literal key.
        let (key_col, key_table) = match input {
            PlanNode::IndexScan {
                table: t,
                column,
                key: Expr::Literal(_),
            } => (column.clone(), t.clone()),
            _ => return None,
        };
        if &key_table != table {
            return None;
        }

        // Look up schema + index state from the live catalog, caching
        // the slot so the execute path skips the name probe.
        let table_slot = catalog.table_slot(table)?;
        let tbl = catalog.table_by_slot(table_slot);
        let schema = tbl.schema();

        // Key column must be a UNIQUE INT index, because the execute path
        // probes it with `BTree::lookup_int`, which binary-searches assuming
        // every key is a `Value::Int` and treats any other variant as `Less`.
        //
        // That probe is only equivalent to the lowered plan's probe under both
        // conditions at once:
        //
        //   * `TypeId::Int` — a float / datetime / str / bool column stores its
        //     keys in a different lane, so `lookup_int` addresses nothing and
        //     the mutation silently reported `Modified(0)` while the same text
        //     reported `Modified(1)` (or a typed error, for str and bool);
        //   * unique — a non-unique index does not store one bare `Value::Int`
        //     per key, so the same probe missed there too.
        //
        // Both are also what `plan_exec::lowering::coerce_column_index_key`
        // decides for the text path, which is the definition this fast path has
        // to match. Anything else falls through to the substitute-and-lower
        // path below, which is always correct.
        let key_col_idx = schema.column_index(&key_col)?;
        if schema.columns[key_col_idx].type_id != TypeId::Int {
            return None;
        }
        if tbl.is_index_unique(&key_col) != Some(true) {
            return None;
        }

        // Target column must exist, be fixed-size, and NOT be indexed (so
        // we don't have to maintain any secondary index here).
        let target_col_idx = schema.column_index(&assn.field)?;
        let target_type = schema.columns[target_col_idx].type_id;
        if !is_fixed_size(target_type) {
            return None;
        }
        if tbl.has_indexed_col(target_col_idx) {
            return None;
        }

        // Precompute byte offsets from the cached row layout.
        let layout = tbl.row_layout();
        let fixed_off = layout.fixed_offset(target_col_idx)?;
        let bitmap_size = layout.bitmap_size();
        let field_off = 2 + bitmap_size + fixed_off;
        let bitmap_byte_off = 2 + target_col_idx / 8;
        let bit_mask = 1u8 << (target_col_idx % 8);

        // Literal walk order for `Update { IndexScan(key), [{value}] }`
        // (see `plan_cache::substitute_plan` — input first, then the
        // assignments). The filter key is literal 0, the assignment RHS
        // is literal 1.
        Some(UpdatePkFast {
            table_slot,
            structure_generation: catalog.structure_generation(),
            key_col,
            target_col_idx,
            field_off,
            bitmap_byte_off,
            bit_mask,
            target_type,
            key_literal_idx: 0,
            value_literal_idx: 1,
        })
    }

    /// Execute a [`PreparedQuery`] with the given literal values.
    ///
    /// The literals are substituted into a clone of the template plan in
    /// the same deterministic walk order that [`crate::canonicalize`]
    /// produces (filter predicate first, then projection, then assignment
    /// RHS, and so on). Substitution errors here mean the caller passed
    /// the wrong number of literals for this query shape.
    pub fn execute_prepared(
        &mut self,
        prep: &PreparedQuery,
        literals: &[Literal],
    ) -> Result<QueryResult, QueryError> {
        if literals.len() != prep.param_count {
            return Err(QueryError::Execution(format!(
                "prepared query expects {} literal(s), got {}",
                prep.param_count,
                literals.len(),
            )));
        }

        // Mission C Phase 14: update-by-pk fast path. Skip plan clone,
        // substitute walk, resolved_assignments, FastPatch, Vec<RowId>,
        // RowLayout::new — straight to btree.lookup_int + byte patch.
        // On rare mismatches (wrong literal type, index dropped after
        // prepare) the helper returns `Ok(None)` and we fall through to
        // the generic substitute-and-execute path below.
        if let Some(fast) = prep
            .update_pk_fast
            .as_ref()
            .filter(|_| !self.generic_path_forced("prepared-update-pk"))
        {
            if let Some(result) = self.try_execute_update_pk_fast(fast, literals)? {
                // Mark dependent views dirty for prepared update fast path.
                // Before the commit below, so the flag is on disk before the
                // mutation it describes is durable.
                if let PlanNode::Update { table, .. } = &prep.plan_template {
                    self.view_registry
                        .mark_dependents_dirty(table)
                        .map_err(QueryError::from_storage_io)?;
                }
                // Mission B (post-review): statement-boundary WAL group
                // commit. The fast path appended an Update record but did
                // not flush — flush it now so the executor's contract is
                // "WAL is on disk before this returns".
                self.catalog
                    .commit_autocommit()
                    .map_err(QueryError::from_storage_io)?;
                return Ok(result);
            }
        }

        // Insert fast path: skip plan-clone + substitute walk + PlanNode::Insert
        // arm's column-index resolution. Build the Row directly from the
        // caller's literal slice using indices we resolved at prepare time.
        // Saves ~300-500ns per insert on the bench.
        //
        // Mission C Phase 13: the scratch `Vec<Value>` is reused across
        // calls — no fresh allocation per insert. We split the borrow
        // between `self.catalog` and `self.insert_values_scratch` by
        // moving the scratch into a local, filling it, passing to the
        // catalog, and putting it back.
        //
        // Mission C Phase 15: the cached `InsertFast` carries `n_cols`
        // and the table name, so the hot path makes exactly one catalog
        // HashMap lookup (`get_table_mut`) and dispatches straight into
        // `tbl.insert` — no intermediate schema lookup, no generic
        // `Catalog::insert` wrapper.
        if let Some(fast) = prep.insert_fast.as_ref().filter(|fast| {
            !self.generic_path_forced("prepared-insert")
                && cached_table_matches(&self.catalog, fast.structure_generation)
        }) {
            let mut values = std::mem::take(&mut self.insert_values_scratch);
            values.resize(fast.n_cols, Value::Empty);
            // Columns omitted by the prepared INSERT must return to NULL on
            // every execution. Assigned string slots keep their allocation so
            // repeated prepared inserts copy into stable buffers instead of
            // allocating one String per field per row.
            for &index in &fast.omitted_col_indices {
                values[index] = Value::Empty;
            }
            for (pos, lit) in literals.iter().enumerate() {
                let value = &mut values[fast.col_indices[pos]];
                let column = &fast.assigned_columns[pos];
                match (value, lit, column.type_id) {
                    (Value::Str(buffer), Literal::String(text), TypeId::Str) => {
                        buffer.clear();
                        buffer.push_str(text);
                    }
                    (value, literal, _) => {
                        let raw = literal_value_from(literal);
                        match coerce_value(raw, column) {
                            Ok(coerced) => *value = coerced,
                            Err(error) => {
                                self.insert_values_scratch = values;
                                return Err(QueryError::Execution(error));
                            }
                        }
                    }
                }
            }
            for &index in &fast.required_col_indices {
                if matches!(values[index], Value::Empty) {
                    let column = &fast.schema_columns[index];
                    self.insert_values_scratch = values;
                    return Err(QueryError::Execution(format!(
                        "column '{}' is required but no value was provided",
                        column.name
                    )));
                }
            }
            // Mission C Phase 18: direct O(1) slot index — no
            // catalog hash probe. Slot was resolved at prepare time.
            // Durability fix: route through the WAL-logging `insert_by_slot`
            // (was the raw `Table::insert`, which bypassed the WAL and lost
            // every prepared insert on a crash).
            let res = self
                .catalog
                .insert_by_slot(fast.table_slot, &values)
                .map_err(|e| e.to_string());
            // Retain ordinary row buffers, but do not pin an overflow-sized
            // client string in the engine forever after one prepared insert.
            for value in &mut values {
                if matches!(value, Value::Str(buffer) if buffer.capacity() > powdb_storage::page::MAX_ROW_DATA_SIZE)
                {
                    *value = Value::Empty;
                }
            }
            // Keep one row's string buffers for the next prepared execution.
            // This is bounded by the prepared row width and never escapes the
            // engine; the catalog has already encoded/copied the values.
            self.insert_values_scratch = values;
            res?;
            // Mark dependent views dirty for prepared insert fast path.
            self.view_registry
                .mark_dependents_dirty(&fast.table_name)
                .map_err(QueryError::from_storage_io)?;
            // Mission B (post-review): statement-boundary WAL group commit.
            self.catalog
                .commit_autocommit()
                .map_err(QueryError::from_storage_io)?;
            return Ok(QueryResult::Modified(1));
        }

        let mut plan = prep.plan_template.clone();
        let mut idx = 0usize;
        crate::plan_cache::substitute_plan(&mut plan, literals, &mut idx);
        debug_assert_eq!(idx, literals.len());
        // The template is raw planner output and the substituted literals are
        // new, so this is the first and only chance to lower. Executing the
        // template directly is what made a prepared `.price < $1` answer
        // differently from the same query executed as text.
        let plan = self.lower(&plan);
        let result = self.execute_lowered(&plan);
        // Mission B (post-review): statement-boundary WAL group commit.
        // No-op when nothing was buffered (read-only plans).
        self.catalog
            .commit_autocommit()
            .map_err(QueryError::from_storage_io)?;
        result
    }

    /// Mission C Phase 14: point-update fast path for prepared
    /// `T filter .pk = ? update { col := ? }` queries. The caller has
    /// already verified this is a UNIQUE INT-indexed pk with a fixed-size,
    /// non-indexed target column; all we do here is pluck the two
    /// literals out of the caller's slice, run one `btree.lookup_int`,
    /// and patch 1–8 bytes of the row. No plan clone, no allocations.
    ///
    /// Returns:
    ///   * `Ok(Some(result))` — fast path took the mutation.
    ///   * `Ok(None)` — can't take the fast path this call (wrong
    ///     literal type, index dropped since prepare, etc.). Caller
    ///     falls through to the generic substitute-and-execute path.
    ///   * `Err(_)` — real error (table gone, I/O, etc.).
    #[inline]
    fn try_execute_update_pk_fast(
        &mut self,
        fast: &UpdatePkFast,
        literals: &[Literal],
    ) -> Result<Option<QueryResult>, QueryError> {
        if !cached_table_matches(&self.catalog, fast.structure_generation) {
            return Ok(None);
        }
        let current_table = self.catalog.table_by_slot(fast.table_slot);
        // Re-check the two properties `lookup_int` depends on rather than
        // trusting the prepare-time decision. The structure generation above
        // already catches index DDL, so this is defence in depth against a
        // future mutation of the index set that forgets to bump it: losing the
        // fast path costs speed, taking it on a non-unique or non-Int index
        // loses writes.
        if current_table.has_indexed_col(fast.target_col_idx)
            || current_table.is_index_unique(&fast.key_col) != Some(true)
        {
            return Ok(None);
        }
        // 1) Extract the key literal. The fast path is only built for
        //    int key columns; any other literal type means the caller
        //    is violating the prepared-query contract or the schema
        //    changed — either way, fall back.
        let key_int = match &literals[fast.key_literal_idx] {
            Literal::Int(v) => *v,
            _ => return Ok(None),
        };

        // 2) Encode the new value as little-endian bytes matching the
        //    target column's fixed encoding.
        let bytes: FixedBytes = match (fast.target_type, &literals[fast.value_literal_idx]) {
            (TypeId::Int, Literal::Int(v)) => FixedBytes::I64(v.to_le_bytes()),
            (TypeId::DateTime, Literal::Int(v)) => FixedBytes::I64(v.to_le_bytes()),
            (TypeId::Float, Literal::Float(v)) => FixedBytes::F64(v.to_le_bytes()),
            (TypeId::Bool, Literal::Bool(v)) => FixedBytes::Bool(if *v { 1 } else { 0 }),
            // Type mismatch — fall back to the generic path for a
            // consistent error shape.
            _ => return Ok(None),
        };

        // 3) Look up the table + btree, do the int lookup, patch the row
        //    in place. Phase 18: table dispatch is a direct slot index;
        //    the btree lookup is the linear scan over `indexed_cols`.
        //    Single btree.lookup_int + one `with_row_bytes_mut` call.
        //    No Vec allocations at all.
        //
        // Mission B2: route the in-place patch through the catalog's
        // WAL-logged wrapper so crash recovery sees the update. The
        // extra cost is one WAL append + fsync per query — the hot
        // loop structure is unchanged.
        let tbl = self.catalog.table_by_slot_mut(fast.table_slot);
        let btree = tbl
            .index(&fast.key_col)
            .expect("prepared update index was revalidated above");
        let Some(rid) = btree.lookup_int(key_int) else {
            return Ok(Some(QueryResult::Modified(0)));
        };

        let fast_table_slot = fast.table_slot;
        let bitmap_byte_off = fast.bitmap_byte_off;
        let bit_mask = fast.bit_mask;
        let field_off = fast.field_off;
        let ok = self
            .catalog
            .update_row_bytes_logged_by_slot(fast_table_slot, rid, |row| {
                let base = if row.len() >= ROW_PREFIX_SIZE && &row[0..4] == ROW_MAGIC {
                    ROW_PREFIX_SIZE
                } else {
                    0
                };
                // Idempotent null-bit clear — safe even when the column was
                // already non-null (the overwhelmingly common case).
                row[base + bitmap_byte_off] &= !bit_mask;
                let field_bytes = bytes.as_slice();
                row[base + field_off..base + field_off + field_bytes.len()]
                    .copy_from_slice(field_bytes);
            })
            .map_err(QueryError::from_storage_io)?;

        Ok(Some(QueryResult::Modified(if ok { 1 } else { 0 })))
    }

    /// Mission C Phase 13: moving variant of [`Engine::execute_prepared`]
    /// for the insert fast path. Takes `literals` by mutable reference
    /// so that each `Literal::String` can be consumed via `mem::take`
    /// instead of cloned into a `Value::Str`. On `insert_batch_1k` that
    /// removes three per-row heap allocations (name, status, email),
    /// bringing the workload over the line vs SQLite's amortized
    /// prepare+execute loop.
    ///
    /// The caller's `Literal::String` entries are replaced with empty
    /// strings on successful inserts — the `literals` slice is *not*
    /// left in a valid-for-reuse state except for `Int`/`Float`/`Bool`
    /// values. Non-insert templates fall through to the standard
    /// substitute-and-execute path.
    pub fn execute_prepared_take(
        &mut self,
        prep: &PreparedQuery,
        literals: &mut [Literal],
    ) -> Result<QueryResult, QueryError> {
        if literals.len() != prep.param_count {
            return Err(QueryError::Execution(format!(
                "prepared query expects {} literal(s), got {}",
                prep.param_count,
                literals.len(),
            )));
        }

        if let Some(fast) = prep
            .insert_fast
            .as_ref()
            .filter(|fast| cached_table_matches(&self.catalog, fast.structure_generation))
        {
            // Moving strings is only safe when coercion cannot fail or replace
            // the string with another representation. Complex/coercing shapes
            // use the borrowed path; on success we still honor this method's
            // consume-on-success contract.
            if !literals
                .iter()
                .zip(&fast.assigned_columns)
                .all(|(literal, column)| literal_can_take_without_error(literal, column))
            {
                let result = self.execute_prepared(prep, literals);
                if result.is_ok() {
                    for literal in literals {
                        if let Literal::String(value) = literal {
                            value.clear();
                        }
                    }
                }
                return result;
            }
            let mut values = std::mem::take(&mut self.insert_values_scratch);
            values.clear();
            values.resize(fast.n_cols, Value::Empty);
            for (pos, lit) in literals.iter_mut().enumerate() {
                let raw = literal_value_take(lit);
                match coerce_value(raw, &fast.assigned_columns[pos]) {
                    Ok(coerced) => values[fast.col_indices[pos]] = coerced,
                    Err(error) => {
                        restore_taken_strings(fast, literals, &mut values);
                        values.clear();
                        self.insert_values_scratch = values;
                        return Err(QueryError::Execution(error));
                    }
                }
            }
            for &index in &fast.required_col_indices {
                if matches!(values[index], Value::Empty) {
                    let column = &fast.schema_columns[index];
                    let error = format!(
                        "column '{}' is required but no value was provided",
                        column.name
                    );
                    restore_taken_strings(fast, literals, &mut values);
                    values.clear();
                    self.insert_values_scratch = values;
                    return Err(QueryError::Execution(error));
                }
            }
            // Mission C Phase 18: direct O(1) slot index — see
            // `execute_prepared` for rationale. This is the hot path
            // for `insert_batch_1k`. Durability fix: WAL-logging
            // `insert_by_slot` (was the raw `Table::insert`).
            if let Err(error) = self.catalog.insert_by_slot(fast.table_slot, &values) {
                restore_taken_strings(fast, literals, &mut values);
                values.clear();
                self.insert_values_scratch = values;
                return Err(QueryError::from_storage_io(error));
            }
            if let Err(error) = self.view_registry.mark_dependents_dirty(&fast.table_name) {
                restore_taken_strings(fast, literals, &mut values);
                values.clear();
                self.insert_values_scratch = values;
                return Err(QueryError::from_storage_io(error));
            }
            // Mission B (post-review): statement-boundary WAL group commit.
            if let Err(error) = self.catalog.commit_autocommit() {
                restore_taken_strings(fast, literals, &mut values);
                values.clear();
                self.insert_values_scratch = values;
                return Err(QueryError::from_storage_io(error));
            }
            values.clear();
            self.insert_values_scratch = values;
            return Ok(QueryResult::Modified(1));
        }

        // Non-insert templates — fall back to the standard path. We
        // can't usefully move the literals because `substitute_plan`
        // still expects an immutable slice, and the non-insert hot
        // paths are dominated by plan walks anyway.
        let result = self.execute_prepared(prep, literals);
        if result.is_ok() && matches!(prep.plan_template, PlanNode::Insert { .. }) {
            for literal in literals {
                if let Literal::String(value) = literal {
                    value.clear();
                }
            }
        }
        result
    }

    /// Walk an expression tree and replace every `InSubquery` node with
    /// an `InList` by executing the subquery and collecting its first
    /// column as literal values. This must be called before entering
    /// the row-by-row scan loop because the scan closure can't call back
    /// into the engine.
    pub(super) fn materialize_subqueries(&mut self, expr: &Expr) -> Result<Expr, QueryError> {
        match expr {
            Expr::InSubquery {
                expr: inner,
                subquery,
                negated,
            } => {
                if is_correlated_subquery(subquery, &self.catalog) {
                    let inner = self.materialize_subqueries(inner)?;
                    return Ok(Expr::InSubquery {
                        expr: Box::new(inner),
                        subquery: subquery.clone(),
                        negated: *negated,
                    });
                }
                let inner = self.materialize_subqueries(inner)?;
                // Plan and execute the subquery.
                let sub_plan = self.plan_and_lower(Statement::Query(*subquery.clone()))?;
                let result = self.execute_lowered(&sub_plan)?;
                let values = match result {
                    QueryResult::Rows { rows, .. } => {
                        let mut values = Vec::with_capacity(rows.len());
                        let mut cancel = crate::cancel::CancelCheck::new();
                        for mut row in rows {
                            cancel.tick()?;
                            if !row.is_empty() {
                                values.push(value_to_expr(row.swap_remove(0)));
                            }
                        }
                        values
                    }
                    _ => Vec::new(),
                };
                // WS2: byte-budget guard on the materialized IN-list.
                self.charge_in_list(&values)?;
                Ok(Expr::InList {
                    expr: Box::new(inner),
                    list: values,
                    negated: *negated,
                })
            }
            Expr::ExistsSubquery { subquery, negated } => {
                if is_correlated_subquery(subquery, &self.catalog) {
                    return Ok(expr.clone());
                }
                // Uncorrelated EXISTS: run the subquery once and collapse
                // into a Bool literal.
                let sub_plan = self.plan_and_lower(Statement::Query(*subquery.clone()))?;
                let result = self.execute_lowered(&sub_plan)?;
                let has_rows = match result {
                    QueryResult::Rows { rows, .. } => !rows.is_empty(),
                    _ => false,
                };
                let truth = if *negated { !has_rows } else { has_rows };
                Ok(Expr::Literal(Literal::Bool(truth)))
            }
            Expr::BinaryOp(l, op, r) => {
                let l = self.materialize_subqueries(l)?;
                let r = self.materialize_subqueries(r)?;
                Ok(Expr::BinaryOp(Box::new(l), *op, Box::new(r)))
            }
            Expr::UnaryOp(op, inner) => {
                let inner = self.materialize_subqueries(inner)?;
                Ok(Expr::UnaryOp(*op, Box::new(inner)))
            }
            Expr::Case { whens, else_expr } => {
                let whens = whens
                    .iter()
                    .map(|(c, r)| {
                        let c = self.materialize_subqueries(c)?;
                        let r = self.materialize_subqueries(r)?;
                        Ok((Box::new(c), Box::new(r)))
                    })
                    .collect::<Result<Vec<_>, QueryError>>()?;
                let else_expr = match else_expr {
                    Some(e) => Some(Box::new(self.materialize_subqueries(e)?)),
                    None => None,
                };
                Ok(Expr::Case { whens, else_expr })
            }
            // Leaf nodes: no subqueries possible.
            other => Ok(other.clone()),
        }
    }

    /// Write-path per-row materialisation of correlated subqueries.
    pub(super) fn materialize_correlated_for_row(
        &mut self,
        expr: &Expr,
        outer_row: &[Value],
        outer_columns: &[String],
    ) -> Result<Expr, QueryError> {
        match expr {
            Expr::InSubquery {
                expr: inner,
                subquery,
                negated,
            } => {
                let inner = self.materialize_correlated_for_row(inner, outer_row, outer_columns)?;
                let mut sub = *subquery.clone();
                if let Some(ref filter) = sub.filter {
                    sub.filter = Some(substitute_outer_refs(
                        filter,
                        &sub.source,
                        &self.catalog,
                        outer_row,
                        outer_columns,
                    ));
                }
                let sub_plan = self.plan_and_lower(Statement::Query(sub))?;
                let result = self.execute_lowered(&sub_plan)?;
                let values = match result {
                    QueryResult::Rows { rows, .. } => {
                        let mut values = Vec::with_capacity(rows.len());
                        let mut cancel = crate::cancel::CancelCheck::new();
                        for mut row in rows {
                            cancel.tick()?;
                            if !row.is_empty() {
                                values.push(value_to_expr(row.swap_remove(0)));
                            }
                        }
                        values
                    }
                    _ => Vec::new(),
                };
                Ok(Expr::InList {
                    expr: Box::new(inner),
                    list: values,
                    negated: *negated,
                })
            }
            Expr::ExistsSubquery { subquery, negated } => {
                let mut sub = *subquery.clone();
                if let Some(ref filter) = sub.filter {
                    sub.filter = Some(substitute_outer_refs(
                        filter,
                        &sub.source,
                        &self.catalog,
                        outer_row,
                        outer_columns,
                    ));
                }
                let sub_plan = self.plan_and_lower(Statement::Query(sub))?;
                let result = self.execute_lowered(&sub_plan)?;
                let has_rows = match result {
                    QueryResult::Rows { rows, .. } => !rows.is_empty(),
                    _ => false,
                };
                let truth = if *negated { !has_rows } else { has_rows };
                Ok(Expr::Literal(Literal::Bool(truth)))
            }
            Expr::BinaryOp(l, op, r) => {
                let l = self.materialize_correlated_for_row(l, outer_row, outer_columns)?;
                let r = self.materialize_correlated_for_row(r, outer_row, outer_columns)?;
                Ok(Expr::BinaryOp(Box::new(l), *op, Box::new(r)))
            }
            Expr::UnaryOp(op, inner) => {
                let inner = self.materialize_correlated_for_row(inner, outer_row, outer_columns)?;
                Ok(Expr::UnaryOp(*op, Box::new(inner)))
            }
            other => Ok(other.clone()),
        }
    }
}