aljaz/turso
1
0
Fork 0
mirror of https://github.com/aljazceru/turso.git synced 2026-02-10 02:34:20 +01:00
Code Issues Packages Projects Releases Wiki Activity

refactor: simplification and better names

Browse Source
This commit is contained in:
Levy A.
2025-05-25 22:49:50 -03:00
parent a7761e431b
commit d65fe0f978
1 changed files with 12 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
core/translate/mod.rs
View File

@@ -115,15 +115,16 @@ pub fn translate_inner(
let (table_name, alter_table) = *a;
let ast::Name(table_name) = table_name.name;
let Some(table) = schema.tables.get(&table_name) else {
let Some(original_btree) = schema
.tables
.get(&table_name)
.and_then(|table| table.btree())
else {
return Err(LimboError::ParseError(format!(
"no such table: {table_name}"
)));
};
let Some(original_btree) = table.btree() else {
todo!()
};
let mut btree = (*original_btree).clone();
match alter_table {
@@ -139,22 +140,17 @@ pub fn translate_inner(
)));
}
let (dropped_col, col) = btree
.columns
.iter()
.enumerate()
.find(|(_, Column { name, .. })| name.as_ref() == Some(&column))
.ok_or_else(|| {
LimboError::ParseError(format!("no such column: \"{column}\""))
})?;
let (dropped_index, column) = btree.get_column(name).ok_or_else(|| {
LimboError::ParseError(format!("no such column: \"{column}\""))
})?;
if col.primary_key {
if column.primary_key {
return Err(LimboError::ParseError(format!(
"cannot drop column \"{column}\": PRIMARY KEY"
)));
}
if col.unique
if column.unique
|| btree.unique_sets.as_ref().is_some_and(|set| {
set.iter().any(|set| {
set.iter().any(|(column_name, _)| column_name == &column)
@@ -166,7 +162,7 @@ pub fn translate_inner(
)));
}
btree.columns.remove(dropped_col);
btree.columns.remove(dropped_index);
let sql = btree.to_sql();
@@ -218,7 +214,7 @@ pub fn translate_inner(
let mut iter = first_column;
for i in 0..(column_count + 1) {
if i == dropped_col {
if i == dropped_index {
continue;
}