vibesql_executor/alter/
mod.rs

1//! ALTER TABLE executor
2
3mod columns;
4mod constraints;
5mod table_options;
6mod validation;
7
8use vibesql_ast::*;
9use vibesql_storage::Database;
10
11use crate::{errors::ExecutorError, privilege_checker::PrivilegeChecker};
12
13/// Executor for ALTER TABLE statements
14pub struct AlterTableExecutor;
15
16impl AlterTableExecutor {
17    /// Execute an ALTER TABLE statement
18    pub fn execute(
19        stmt: &AlterTableStmt,
20        database: &mut Database,
21    ) -> Result<String, ExecutorError> {
22        // Get table name from the statement and check ALTER privilege
23        let table_name = match stmt {
24            AlterTableStmt::AddColumn(s) => &s.table_name,
25            AlterTableStmt::DropColumn(s) => &s.table_name,
26            AlterTableStmt::AlterColumn(s) => match s {
27                AlterColumnStmt::SetDefault { table_name, .. } => table_name,
28                AlterColumnStmt::DropDefault { table_name, .. } => table_name,
29                AlterColumnStmt::SetNotNull { table_name, .. } => table_name,
30                AlterColumnStmt::DropNotNull { table_name, .. } => table_name,
31            },
32            AlterTableStmt::AddConstraint(s) => &s.table_name,
33            AlterTableStmt::DropConstraint(s) => &s.table_name,
34            AlterTableStmt::RenameTable(s) => &s.table_name,
35            AlterTableStmt::ModifyColumn(s) => &s.table_name,
36            AlterTableStmt::ChangeColumn(s) => &s.table_name,
37        };
38        PrivilegeChecker::check_alter(database, table_name)?;
39
40        match stmt {
41            AlterTableStmt::AddColumn(add_column) => columns::execute_add_column(add_column, database),
42            AlterTableStmt::DropColumn(drop_column) => {
43                columns::execute_drop_column(drop_column, database)
44            }
45            AlterTableStmt::AlterColumn(alter_column) => {
46                columns::execute_alter_column(alter_column, database)
47            }
48            AlterTableStmt::AddConstraint(add_constraint) => {
49                constraints::execute_add_constraint(add_constraint, database)
50            }
51            AlterTableStmt::DropConstraint(drop_constraint) => {
52                constraints::execute_drop_constraint(drop_constraint, database)
53            }
54            AlterTableStmt::RenameTable(rename_table) => {
55                table_options::execute_rename_table(rename_table, database)
56            }
57            AlterTableStmt::ModifyColumn(modify_column) => {
58                columns::execute_modify_column(modify_column, database)
59            }
60            AlterTableStmt::ChangeColumn(change_column) => {
61                columns::execute_change_column(change_column, database)
62            }
63        }
64    }
65}
vibesql_executor/alter/mod.rs

vibesql_executor/alter/
mod.rs
