aljaz/turso
1
0
Fork 0
mirror of https://github.com/aljazceru/turso.git synced 2026-01-09 19:24:21 +01:00
Code Issues Packages Projects Releases Wiki Activity

impl ToSqlString for CREATE TABLE stmt

Browse Source
This commit is contained in:
pedrocarlo
2025-05-27 21:35:44 -03:00
parent a8f5257240
commit 8a7cc7669d
3 changed files with 131 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
vendored/sqlite3-parser/src/to_sql_string/stmt/alter_table.rs
View File

@@ -65,7 +65,7 @@ impl ToSqlString for ast::ColumnConstraint {
clause,
deref_clause,
} => format!(
"{}{}",
"FOREIGN KEY {}{}",
clause.to_sql_string(context),
if let Some(deref) = deref_clause {
deref.to_sql_string(context)

115
vendored/sqlite3-parser/src/to_sql_string/stmt/create_table.rs Normal file
View File

@@ -0,0 +1,115 @@
use crate::{ast, to_sql_string::ToSqlString};
impl ToSqlString for ast::CreateTableBody {
fn to_sql_string<C: crate::to_sql_string::ToSqlContext>(&self, context: &C) -> String {
match self {
Self::AsSelect(select) => format!("AS {}", select.to_sql_string(context)),
Self::ColumnsAndConstraints {
columns,
constraints,
options,
} => {
format!(
"({}) {}{}",
columns
.iter()
.map(|(_, col)| col.to_sql_string(context))
.collect::<Vec<_>>()
.join(", "),
constraints
.as_ref()
.map_or("".to_string(), |constraints| format!(
" {}",
constraints
.iter()
.map(|constraint| constraint.to_sql_string(context))
.collect::<Vec<_>>()
.join(", ")
)),
options.to_sql_string(context)
)
}
}
}
}
impl ToSqlString for ast::NamedTableConstraint {
fn to_sql_string<C: crate::to_sql_string::ToSqlContext>(&self, context: &C) -> String {
if let Some(name) = &self.name {
format!("{} {}", name.0, self.constraint.to_sql_string(context))
} else {
format!("{}", self.constraint.to_sql_string(context))
}
}
}
impl ToSqlString for ast::TableConstraint {
fn to_sql_string<C: crate::to_sql_string::ToSqlContext>(&self, context: &C) -> String {
match self {
Self::Check(expr) => format!("CHECK ({})", expr.to_sql_string(context)),
Self::ForeignKey {
columns,
clause,
deref_clause,
} => format!(
"FOREIGN KEY ({}){}{}",
columns
.iter()
.map(|col| col.to_sql_string(context))
.collect::<Vec<_>>()
.join(", "),
clause.to_sql_string(context),
if let Some(deref) = deref_clause {
deref.to_sql_string(context)
} else {
"".to_string()
}
),
Self::PrimaryKey {
columns,
auto_increment,
conflict_clause,
} => format!(
"PRIMARY KEY ({}){}{}",
columns
.iter()
.map(|col| col.to_sql_string(context))
.collect::<Vec<_>>()
.join(", "),
conflict_clause.map_or("".to_string(), |conflict| format!(
" {}",
conflict.to_sql_string(context)
)),
auto_increment.then_some(" AUTOINCREMENT").unwrap_or("")
),
Self::Unique {
columns,
conflict_clause,
} => format!(
"UNIQUE ({}){}",
columns
.iter()
.map(|col| col.to_sql_string(context))
.collect::<Vec<_>>()
.join(", "),
conflict_clause.map_or("".to_string(), |conflict| format!(
" {}",
conflict.to_sql_string(context)
))
),
}
}
}
impl ToSqlString for ast::TableOptions {
fn to_sql_string<C: crate::to_sql_string::ToSqlContext>(&self, _context: &C) -> String {
if *self == Self::NONE {
""
} else if *self == Self::STRICT {
"STRICT"
} else {
"WITHOUT ROWID"
}
.to_string()
}
}

15
vendored/sqlite3-parser/src/to_sql_string/stmt/mod.rs
View File

@@ -3,6 +3,7 @@ use crate::ast;
use super::ToSqlString;
mod alter_table;
mod create_table;
mod select;
impl ToSqlString for ast::Stmt {
@@ -75,6 +76,20 @@ impl ToSqlString for ast::Stmt {
))
)
}
Self::CreateTable {
temporary,
if_not_exists,
tbl_name,
body,
} => {
format!(
"CREATE{} TABLE {}{} {}",
temporary.then_some(" TEMP").unwrap_or(""),
if_not_exists.then_some("IF NOT EXISTS ").unwrap_or(""),
tbl_name.to_sql_string(context),
body.to_sql_string(context)
)
}
Self::Select(select) => select.to_sql_string(context),
_ => todo!(),
}