aljaz/turso
1
0
Fork 0
mirror of https://github.com/aljazceru/turso.git synced 2026-01-31 13:54:27 +01:00
Code Issues Packages Projects Releases Wiki Activity

Handle renaming child foreign keys on op_rename_table

Browse Source
This commit is contained in:
PThorpe92
2025-11-02 17:10:01 -05:00
parent 2c49c47300
commit ef24911824
1 changed files with 24 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
core/vdbe/execute.rs
View File

@@ -8443,12 +8443,35 @@ pub fn op_rename_table(
let Table::BTree(btree) = table else {
panic!("only btree tables can be renamed");
};
let btree = Arc::make_mut(btree);
// update this table's own foreign keys
for fk_arc in &mut btree.foreign_keys {
let fk = Arc::make_mut(fk_arc);
if normalize_ident(&fk.parent_table) == normalized_from {
fk.parent_table = normalized_to.clone();
}
}
btree.name = normalized_to.to_owned();
}
schema.tables.insert(normalized_to.to_owned(), table);
for (tname, t_arc) in schema.tables.iter_mut() {
// skip the table we just renamed
if normalize_ident(tname) == normalized_to {
continue;
}
if let Table::BTree(ref mut child_btree_arc) = Arc::make_mut(t_arc) {
let child_btree = Arc::make_mut(child_btree_arc);
for fk_arc in &mut child_btree.foreign_keys {
if normalize_ident(&fk_arc.parent_table) == normalized_from {
let fk = Arc::make_mut(fk_arc);
fk.parent_table = normalized_to.clone();
}
}
}
}
});
state.pc += 1;