caisin/sea-orm
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

cargo fmt

Browse Source
This commit is contained in:
Billy Chan 2021-08-31 12:49:33 +08:00
parent 8a81e43d2e
commit d1d37fab07
No known key found for this signature in database
GPG Key ID: A2D690CAC7DF3CC7
2 changed files with 141 additions and 55 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/entity/column.rs
View File

@ -1,6 +1,6 @@
use std::str::FromStr;
use crate::{EntityName, IdenStatic, Iterable}; use crate::{EntityName, IdenStatic, Iterable};
use sea_query::{DynIden, Expr, SeaRc, SelectStatement, SimpleExpr, Value}; use sea_query::{DynIden, Expr, SeaRc, SelectStatement, SimpleExpr, Value};
use std::str::FromStr;
#[derive(Debug, Clone)] #[derive(Debug, Clone)]
pub struct ColumnDef { pub struct ColumnDef {

194
src/entity/schema.rs
View File

@ -1,12 +1,17 @@
use crate::{ColumnDef as Sea_Orm_Column_Def, EntityTrait, PrimaryKeyTrait, RelationDef};
use crate::entity::column::ColumnTrait; use crate::entity::column::ColumnTrait;
use crate::entity::relation::RelationTrait;
use crate::entity::primary_key::PrimaryKeyToColumn; use crate::entity::primary_key::PrimaryKeyToColumn;
use sea_query::{Alias, ColumnDef as Sea_Query_Column_Def, ColumnType as Sea_Query_Column_Type, ForeignKeyCreateStatement, IndexCreateStatement, SeaRc, TableCreateStatement, TableRef}; use crate::entity::relation::RelationTrait;
use crate::{ColumnDef as Sea_Orm_Column_Def, EntityTrait, PrimaryKeyTrait, RelationDef};
use sea_query::{
Alias, ColumnDef as Sea_Query_Column_Def, ColumnType as Sea_Query_Column_Type,
ForeignKeyCreateStatement, IndexCreateStatement, SeaRc, TableCreateStatement, TableRef,
};
pub use sea_strum::IntoEnumIterator; pub use sea_strum::IntoEnumIterator;
pub trait CreateStatementOf pub trait CreateStatementOf {
{ fn create_table_statement_of<E>(entity: E) -> TableCreateStatement
fn create_table_statement_of<E>(entity: E) -> TableCreateStatement where E: EntityTrait { where
E: EntityTrait,
{
let mut stmt = TableCreateStatement::new(); let mut stmt = TableCreateStatement::new();
stmt.table(entity); stmt.table(entity);
for relation in E::Relation::iter() { for relation in E::Relation::iter() {
@ -14,40 +19,44 @@ pub trait CreateStatementOf
let relation_trait: RelationDef = relation.def(); let relation_trait: RelationDef = relation.def();
// foreign_key_stmt.name("Temp"); // foreign_key_stmt.name("Temp");
match relation_trait.from_tbl { match relation_trait.from_tbl {
TableRef::Table(tbl) => { foreign_key_stmt.from_tbl(tbl); }, TableRef::Table(tbl) => {
foreign_key_stmt.from_tbl(tbl);
}
_ => todo!(), _ => todo!(),
} }
match relation_trait.to_tbl { match relation_trait.to_tbl {
TableRef::Table(tbl) => { foreign_key_stmt.to_tbl(tbl); }, TableRef::Table(tbl) => {
foreign_key_stmt.to_tbl(tbl);
}
_ => todo!(), _ => todo!(),
} }
match relation_trait.from_col { match relation_trait.from_col {
crate::Identity::Unary(o1) => { crate::Identity::Unary(o1) => {
foreign_key_stmt.from_col(o1); foreign_key_stmt.from_col(o1);
}, }
crate::Identity::Binary(o1, o2) => { crate::Identity::Binary(o1, o2) => {
foreign_key_stmt.from_col(o1); foreign_key_stmt.from_col(o1);
foreign_key_stmt.from_col(o2); foreign_key_stmt.from_col(o2);
}, }
crate::Identity::Ternary(o1, o2, o3) => { crate::Identity::Ternary(o1, o2, o3) => {
foreign_key_stmt.from_col(o1); foreign_key_stmt.from_col(o1);
foreign_key_stmt.from_col(o2); foreign_key_stmt.from_col(o2);
foreign_key_stmt.from_col(o3); foreign_key_stmt.from_col(o3);
}, }
} }
match relation_trait.to_col { match relation_trait.to_col {
crate::Identity::Unary(o1) => { crate::Identity::Unary(o1) => {
foreign_key_stmt.to_col(o1); foreign_key_stmt.to_col(o1);
}, }
crate::Identity::Binary(o1, o2) => { crate::Identity::Binary(o1, o2) => {
foreign_key_stmt.to_col(o1); foreign_key_stmt.to_col(o1);
foreign_key_stmt.to_col(o2); foreign_key_stmt.to_col(o2);
}, }
crate::Identity::Ternary(o1, o2, o3) => { crate::Identity::Ternary(o1, o2, o3) => {
foreign_key_stmt.to_col(o1); foreign_key_stmt.to_col(o1);
foreign_key_stmt.to_col(o2); foreign_key_stmt.to_col(o2);
foreign_key_stmt.to_col(o3); foreign_key_stmt.to_col(o3);
}, }
} }
stmt.foreign_key(&mut foreign_key_stmt); stmt.foreign_key(&mut foreign_key_stmt);
} }
@ -55,8 +64,11 @@ pub trait CreateStatementOf
let sea_orm_column_def: Sea_Orm_Column_Def = col.def().into(); let sea_orm_column_def: Sea_Orm_Column_Def = col.def().into();
let mut index = IndexCreateStatement::new(); let mut index = IndexCreateStatement::new();
let mut sea_query_column_def = Sea_Query_Column_Def::new(col); let mut sea_query_column_def = Sea_Query_Column_Def::new(col);
for key in E::PrimaryKey::iter() { // enum: Id, Name ... for key in E::PrimaryKey::iter() {
if sea_query_column_def.get_column_name() == Sea_Query_Column_Def::new(key.into_column()).get_column_name() { // enum: Id, Name ...
if sea_query_column_def.get_column_name()
== Sea_Query_Column_Def::new(key.into_column()).get_column_name()
{
sea_query_column_def.primary_key(); sea_query_column_def.primary_key();
if E::PrimaryKey::auto_increment() { if E::PrimaryKey::auto_increment() {
sea_query_column_def.auto_increment(); sea_query_column_def.auto_increment();
@ -78,66 +90,136 @@ pub trait CreateStatementOf
} }
match Sea_Query_Column_Type::from(sea_orm_column_def.col_type) { match Sea_Query_Column_Type::from(sea_orm_column_def.col_type) {
Sea_Query_Column_Type::Char(length) => match length { Sea_Query_Column_Type::Char(length) => match length {
Some(length) => { sea_query_column_def.char_len(length); }, Some(length) => {
None => { sea_query_column_def.char(); }, sea_query_column_def.char_len(length);
}
None => {
sea_query_column_def.char();
}
}, },
Sea_Query_Column_Type::String(length) => match length { Sea_Query_Column_Type::String(length) => match length {
Some(length) => { sea_query_column_def.string_len(length); }, Some(length) => {
None => { sea_query_column_def.string(); }, sea_query_column_def.string_len(length);
}
None => {
sea_query_column_def.string();
}
}, },
Sea_Query_Column_Type::Text => { sea_query_column_def.text(); }, Sea_Query_Column_Type::Text => {
sea_query_column_def.text();
}
Sea_Query_Column_Type::TinyInteger(length) => match length { Sea_Query_Column_Type::TinyInteger(length) => match length {
Some(length) => { sea_query_column_def.tiny_integer_len(length); }, Some(length) => {
None => { sea_query_column_def.tiny_integer(); }, sea_query_column_def.tiny_integer_len(length);
}
None => {
sea_query_column_def.tiny_integer();
}
}, },
// Sea_Query_Column_Type::TinyInteger => { sea_query_column_def.tiny_integer(); }, // Sea_Query_Column_Type::TinyInteger => { sea_query_column_def.tiny_integer(); },
Sea_Query_Column_Type::SmallInteger(length) => match length { Sea_Query_Column_Type::SmallInteger(length) => match length {
Some(length) => { sea_query_column_def.small_integer_len(length); }, Some(length) => {
None => { sea_query_column_def.small_integer(); }, sea_query_column_def.small_integer_len(length);
}
None => {
sea_query_column_def.small_integer();
}
}, },
Sea_Query_Column_Type::Integer(length) => match length { Sea_Query_Column_Type::Integer(length) => match length {
Some(length) => { sea_query_column_def.integer_len(length); }, Some(length) => {
None => { sea_query_column_def.integer(); }, sea_query_column_def.integer_len(length);
}
None => {
sea_query_column_def.integer();
}
}, },
Sea_Query_Column_Type::BigInteger(length) => match length { Sea_Query_Column_Type::BigInteger(length) => match length {
Some(length) => { sea_query_column_def.big_integer_len(length); }, Some(length) => {
None => { sea_query_column_def.big_integer(); }, sea_query_column_def.big_integer_len(length);
}
None => {
sea_query_column_def.big_integer();
}
}, },
Sea_Query_Column_Type::Float(precision) => match precision { Sea_Query_Column_Type::Float(precision) => match precision {
Some(precision) => { sea_query_column_def.float_len(precision); }, Some(precision) => {
None => { sea_query_column_def.float(); }, sea_query_column_def.float_len(precision);
}
None => {
sea_query_column_def.float();
}
}, },
Sea_Query_Column_Type::Double(precision) => match precision { Sea_Query_Column_Type::Double(precision) => match precision {
Some(precision) => { sea_query_column_def.double_len(precision); }, Some(precision) => {
None => { sea_query_column_def.double(); }, sea_query_column_def.double_len(precision);
}
None => {
sea_query_column_def.double();
}
}, },
Sea_Query_Column_Type::Decimal(_) => { sea_query_column_def.decimal(); }, Sea_Query_Column_Type::Decimal(_) => {
sea_query_column_def.decimal();
}
Sea_Query_Column_Type::DateTime(precision) => match precision { Sea_Query_Column_Type::DateTime(precision) => match precision {
Some(precision) => { sea_query_column_def.date_time_len(precision); }, Some(precision) => {
None => { sea_query_column_def.date_time(); }, sea_query_column_def.date_time_len(precision);
}
None => {
sea_query_column_def.date_time();
}
}, },
Sea_Query_Column_Type::Timestamp(precision) => match precision { Sea_Query_Column_Type::Timestamp(precision) => match precision {
Some(precision) => { sea_query_column_def.timestamp_len(precision); }, Some(precision) => {
None => { sea_query_column_def.timestamp(); }, sea_query_column_def.timestamp_len(precision);
}
None => {
sea_query_column_def.timestamp();
}
}, },
Sea_Query_Column_Type::Time(precision) => match precision { Sea_Query_Column_Type::Time(precision) => match precision {
Some(precision) => { sea_query_column_def.time_len(precision); }, Some(precision) => {
None => { sea_query_column_def.time(); }, sea_query_column_def.time_len(precision);
}
None => {
sea_query_column_def.time();
}
}, },
Sea_Query_Column_Type::Date => { sea_query_column_def.date(); }, Sea_Query_Column_Type::Date => {
sea_query_column_def.date();
}
Sea_Query_Column_Type::Binary(length) => match length { Sea_Query_Column_Type::Binary(length) => match length {
Some(length) => { sea_query_column_def.binary_len(length); }, Some(length) => {
None => { sea_query_column_def.binary(); }, sea_query_column_def.binary_len(length);
}
None => {
sea_query_column_def.binary();
}
}, },
Sea_Query_Column_Type::Boolean => { sea_query_column_def.boolean(); }, Sea_Query_Column_Type::Boolean => {
Sea_Query_Column_Type::Money(_) => { sea_query_column_def.money(); }, sea_query_column_def.boolean();
Sea_Query_Column_Type::Json => { sea_query_column_def.json(); }, }
Sea_Query_Column_Type::JsonBinary => { sea_query_column_def.json_binary(); }, Sea_Query_Column_Type::Money(_) => {
Sea_Query_Column_Type::Custom(iden) => { sea_query_column_def.custom(Alias::new(&iden.to_string())); }, sea_query_column_def.money();
Sea_Query_Column_Type::Uuid => { sea_query_column_def.uuid(); }, }
Sea_Query_Column_Type::Json => {
sea_query_column_def.json();
}
Sea_Query_Column_Type::JsonBinary => {
sea_query_column_def.json_binary();
}
Sea_Query_Column_Type::Custom(iden) => {
sea_query_column_def.custom(Alias::new(&iden.to_string()));
}
Sea_Query_Column_Type::Uuid => {
sea_query_column_def.uuid();
}
Sea_Query_Column_Type::TimestampWithTimeZone(length) => match length { Sea_Query_Column_Type::TimestampWithTimeZone(length) => match length {
Some(length) => { sea_query_column_def.timestamp_with_time_zone_len(length); }, Some(length) => {
None => { sea_query_column_def.timestamp_with_time_zone(); }, sea_query_column_def.timestamp_with_time_zone_len(length);
}
None => {
sea_query_column_def.timestamp_with_time_zone();
}
}, },
} }
stmt.col(&mut sea_query_column_def); stmt.col(&mut sea_query_column_def);
@ -152,11 +234,12 @@ impl<EntityTrait> CreateStatementOf for EntityTrait {}
#[cfg(test)] #[cfg(test)]
mod tests { mod tests {
use crate::{CreateStatementOf, tests_cfg}; use crate::{tests_cfg, CreateStatementOf};
#[test] #[test]
fn test_create_statement_tests_cfg_cake() { fn test_create_statement_tests_cfg_cake() {
let create_statement = tests_cfg::cake::Entity::create_table_statement_of(tests_cfg::cake::Entity); let create_statement =
tests_cfg::cake::Entity::create_table_statement_of(tests_cfg::cake::Entity);
let table = format!("{:?}", create_statement.get_table_name()); let table = format!("{:?}", create_statement.get_table_name());
let columns = format!("{:?}", create_statement.get_columns()); let columns = format!("{:?}", create_statement.get_columns());
let relations = format!("{:?}", create_statement.get_foreign_key_create_stmts()); let relations = format!("{:?}", create_statement.get_foreign_key_create_stmts());
@ -165,7 +248,10 @@ mod tests {
assert_eq!("TableCreateStatement { table: Some(cake), columns: [ColumnDef { table: Some(cake), name: id, types: Some(Integer(None)), spec: [PrimaryKey, AutoIncrement, NotNull] }, ColumnDef { table: Some(cake), name: name, types: Some(String(None)), spec: [NotNull] }], options: [], partitions: [], indexes: [], foreign_keys: [ForeignKeyCreateStatement { foreign_key: TableForeignKey { name: None, table: Some(cake), ref_table: Some(fruit), columns: [id], ref_columns: [cake_id], on_delete: None, on_update: None } }], if_not_exists: true }", result); assert_eq!("TableCreateStatement { table: Some(cake), columns: [ColumnDef { table: Some(cake), name: id, types: Some(Integer(None)), spec: [PrimaryKey, AutoIncrement, NotNull] }, ColumnDef { table: Some(cake), name: name, types: Some(String(None)), spec: [NotNull] }], options: [], partitions: [], indexes: [], foreign_keys: [ForeignKeyCreateStatement { foreign_key: TableForeignKey { name: None, table: Some(cake), ref_table: Some(fruit), columns: [id], ref_columns: [cake_id], on_delete: None, on_update: None } }], if_not_exists: true }", result);
assert_eq!(r#"Some("cake")"#, table); assert_eq!(r#"Some("cake")"#, table);
assert_eq!("[ForeignKeyCreateStatement { foreign_key: TableForeignKey { name: None, table: Some(cake), ref_table: Some(fruit), columns: [id], ref_columns: [cake_id], on_delete: None, on_update: None } }]", relations); assert_eq!("[ForeignKeyCreateStatement { foreign_key: TableForeignKey { name: None, table: Some(cake), ref_table: Some(fruit), columns: [id], ref_columns: [cake_id], on_delete: None, on_update: None } }]", relations);
assert_eq!(r#"[ColumnDef { table: Some(cake), name: id, types: Some(Integer(None)), spec: [PrimaryKey, AutoIncrement, NotNull] }, ColumnDef { table: Some(cake), name: name, types: Some(String(None)), spec: [NotNull] }]"#, columns); assert_eq!(
r#"[ColumnDef { table: Some(cake), name: id, types: Some(Integer(None)), spec: [PrimaryKey, AutoIncrement, NotNull] }, ColumnDef { table: Some(cake), name: name, types: Some(String(None)), spec: [NotNull] }]"#,
columns
);
assert_eq!("[]", indexs); assert_eq!("[]", indexs);
} }
} }