vibesql-executor 0.2.0

Query execution engine for vibesql SQL database
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80

//! Index predicate extraction
//!
//! Extracts range and IN predicates from WHERE clauses for index optimization.
//! Supports both column-based and expression-based (functional) indexes.
//!
//! # Module Organization
//!
//! - `range`: Range predicate extraction (>, <, >=, <=, BETWEEN, AND combinations)
//! - `in_list`: IN predicate extraction and main entry point for column-based indexes
//! - `composite`: Multi-column composite index predicate extraction
//! - `expression_index`: Expression-based (functional) index support
//!
//! # Public API
//!
//! The main entry points are:
//! - `extract_index_predicate()`: Extract predicate for single-column index
//! - `extract_index_predicate_for_indexed_column()`: Unified entry for any index column type
//! - `extract_composite_predicates_with_in()`: Extract predicates for composite indexes
//! - `where_clause_fully_satisfied_by_indexed_column()`: Check if index covers WHERE clause

use vibesql_types::SqlValue;

mod composite;
mod expression_index;
mod in_list;
pub(super) mod range;
mod temporal_coercion;

// Re-export types
pub(crate) use composite::{CompositePredicateType, PrefixPredicateResult, PrefixWithRangeResult};

/// Range predicate information extracted from WHERE clause
#[derive(Debug)]
pub(crate) struct RangePredicate {
    pub start: Option<SqlValue>,
    pub end: Option<SqlValue>,
    pub inclusive_start: bool,
    pub inclusive_end: bool,
    /// Whether NULL values should be excluded from the range scan
    /// This is true for all SQL inequality comparisons (<, <=, >, >=, BETWEEN)
    /// since NULL comparisons in SQL return NULL (not true)
    pub exclude_nulls: bool,
}

/// Index predicate types that can be pushed down to storage layer
#[derive(Debug)]
pub(crate) enum IndexPredicate {
    /// Range scan with optional bounds (>, <, >=, <=, BETWEEN)
    Range(RangePredicate),
    /// Multi-value lookup (IN predicate)
    In(Vec<SqlValue>),
}

// Re-export composite index functions
pub(crate) use composite::{
    build_residual_where_clause, extract_composite_predicates_with_in,
    extract_prefix_equality_predicates, extract_prefix_with_trailing_range,
    generate_composite_keys, where_clause_fully_satisfied_by_composite_key,
};

// Test-only re-export (function kept for backward compatibility with tests)
#[cfg(test)]
pub(crate) use composite::extract_composite_equality_predicates;

// Re-export expression index functions
pub(crate) use expression_index::{
    extract_index_predicate_for_indexed_column, where_clause_fully_satisfied_by_indexed_column,
};

// Re-export main extraction function
pub(crate) use in_list::{extract_index_predicate, where_clause_fully_satisfied_by_index};

// Re-export temporal probe-bound coercion (issue #5333)
pub(crate) use temporal_coercion::{
    coerce_equality_key, coerce_index_predicate_for_temporal_keys, EqualityKeyCoercion,
};

#[cfg(test)]
#[path = "tests.rs"]
mod tests;