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) => {
42                columns::execute_add_column(add_column, database)
43            }
44            AlterTableStmt::DropColumn(drop_column) => {
45                columns::execute_drop_column(drop_column, database)
46            }
47            AlterTableStmt::AlterColumn(alter_column) => {
48                columns::execute_alter_column(alter_column, database)
49            }
50            AlterTableStmt::AddConstraint(add_constraint) => {
51                constraints::execute_add_constraint(add_constraint, database)
52            }
53            AlterTableStmt::DropConstraint(drop_constraint) => {
54                constraints::execute_drop_constraint(drop_constraint, database)
55            }
56            AlterTableStmt::RenameTable(rename_table) => {
57                table_options::execute_rename_table(rename_table, database)
58            }
59            AlterTableStmt::ModifyColumn(modify_column) => {
60                columns::execute_modify_column(modify_column, database)
61            }
62            AlterTableStmt::ChangeColumn(change_column) => {
63                columns::execute_change_column(change_column, database)
64            }
65        }
66    }
67}
vibesql_executor/alter/mod.rs

vibesql_executor/alter/
mod.rs
