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

Add indexed and unique support

Browse Source
This commit is contained in:
Marco Napetti 2021-09-06 11:25:05 +02:00
parent 8d8b3e7813
commit 5179fb257a
1 changed files with 23 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
sea-orm-macros/src/derives/entity_model.rs
View File

@ -46,8 +46,11 @@ impl sea_orm::prelude::EntityName for Entity {
columns_enum.push(quote! { #field_name }); columns_enum.push(quote! { #field_name });
let mut nullable = false; let mut nullable = false;
let mut default_value = None;
let mut indexed = false;
let mut unique = false;
let mut sql_type = None; let mut sql_type = None;
// search for #[sea_orm(primary_key, type = "String", nullable)] // search for #[sea_orm(primary_key, column_type = "String", nullable, default_value = "something", indexed, unique)]
field.attrs.iter().for_each(|attr| { field.attrs.iter().for_each(|attr| {
if let Some(ident) = attr.path.get_ident() { if let Some(ident) = attr.path.get_ident() {
if ident != "sea_orm" { if ident != "sea_orm" {
@ -64,12 +67,15 @@ impl sea_orm::prelude::EntityName for Entity {
match meta { match meta {
Meta::NameValue(nv) => { Meta::NameValue(nv) => {
if let Some(name) = nv.path.get_ident() { if let Some(name) = nv.path.get_ident() {
if name == "type" { if name == "column_type" {
if let Lit::Str(litstr) = &nv.lit { if let Lit::Str(litstr) = &nv.lit {
let ty: TokenStream = syn::parse_str(&litstr.value()).unwrap(); let ty: TokenStream = syn::parse_str(&litstr.value()).unwrap();
sql_type = Some(ty); sql_type = Some(ty);
} }
} }
else if name == "default_value" {
default_value = Some(nv.lit.to_owned());
}
} }
}, },
Meta::Path(p) => { Meta::Path(p) => {
@ -80,6 +86,12 @@ impl sea_orm::prelude::EntityName for Entity {
else if name == "nullable" { else if name == "nullable" {
nullable = true; nullable = true;
} }
else if name == "indexed" {
indexed = true;
}
else if name == "unique" {
unique = true;
}
} }
}, },
_ => {}, _ => {},
@ -90,7 +102,7 @@ impl sea_orm::prelude::EntityName for Entity {
let field_type = sql_type.unwrap_or_else(|| { let field_type = sql_type.unwrap_or_else(|| {
let field_type = &field.ty; let field_type = &field.ty;
let temp = quote! { #field_type } let temp = quote! { #field_type }
.to_string()//Example: "Option < String >" .to_string()//E.g.: "Option < String >"
.replace(" ", ""); .replace(" ", "");
let temp = if temp.starts_with("Option<") { let temp = if temp.starts_with("Option<") {
nullable = true; nullable = true;
@ -118,12 +130,17 @@ impl sea_orm::prelude::EntityName for Entity {
} }
}); });
let mut match_row = quote! { Self::#field_name => sea_orm::prelude::ColumnType::#field_type.def() };
if nullable { if nullable {
columns_trait.push(quote! { Self::#field_name => sea_orm::prelude::ColumnType::#field_type.def().null() }); match_row = quote! { #match_row.null() };
} }
else { if indexed {
columns_trait.push(quote! { Self::#field_name => sea_orm::prelude::ColumnType::#field_type.def() }); match_row = quote! { #match_row.indexed() };
} }
if unique {
match_row = quote! { #match_row.unique() };
}
columns_trait.push(match_row);
} }
} }
} }