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

Incorporate requested CR changes

Browse Source
This commit is contained in:
Tim Eggert 2021-10-12 13:06:56 +02:00
parent b6c5d71fe2
commit d9306126cf
2 changed files with 53 additions and 91 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
sea-orm-cli/Cargo.toml
View File

@ -59,4 +59,3 @@ runtime-tokio-rustls = [
"sqlx/runtime-tokio-rustls", "sqlx/runtime-tokio-rustls",
"sea-schema/runtime-tokio-rustls", "sea-schema/runtime-tokio-rustls",
] ]

143
sea-orm-codegen/src/entity/writer.rs
View File

@ -27,6 +27,37 @@ pub enum WithSerde {
Both, Both,
} }
impl WithSerde {
pub fn extra_derive(&self) -> TokenStream {
let mut extra_derive = match self {
Self::None => {
quote! {}
}
Self::Serialize => {
quote! {
Serialize
}
}
Self::Deserialize => {
quote! {
Deserialize
}
}
Self::Both => {
quote! {
Serialize, Deserialize
}
}
};
if !extra_derive.is_empty() {
extra_derive = quote! { , #extra_derive }
}
extra_derive
}
}
impl FromStr for WithSerde { impl FromStr for WithSerde {
type Err = crate::Error; type Err = crate::Error;
@ -171,29 +202,29 @@ impl EntityWriter {
} }
pub fn gen_import(with_serde: &WithSerde) -> TokenStream { pub fn gen_import(with_serde: &WithSerde) -> TokenStream {
let prelude_import = quote!(
use sea_orm::entity::prelude::*;
);
match with_serde { match with_serde {
WithSerde::None => { WithSerde::None => prelude_import,
quote! {
use sea_orm::entity::prelude::*;
}
}
WithSerde::Serialize => { WithSerde::Serialize => {
quote! { quote! {
use sea_orm::entity::prelude::*; #prelude_import
use serde::Serialize; use serde::Serialize;
} }
} }
WithSerde::Deserialize => { WithSerde::Deserialize => {
quote! { quote! {
use sea_orm::entity::prelude::*; #prelude_import
use serde::Deserialize; use serde::Deserialize;
} }
} }
WithSerde::Both => { WithSerde::Both => {
quote! { quote! {
use sea_orm::entity::prelude::*; #prelude_import
use serde::{Deserialize,Serialize}; use serde::{Deserialize,Serialize};
} }
} }
@ -222,40 +253,12 @@ impl EntityWriter {
let column_names_snake_case = entity.get_column_names_snake_case(); let column_names_snake_case = entity.get_column_names_snake_case();
let column_rs_types = entity.get_column_rs_types(); let column_rs_types = entity.get_column_rs_types();
match with_serde { let extra_derive = with_serde.extra_derive();
WithSerde::None => {
quote! {
#[derive(Clone, Debug, PartialEq, DeriveModel, DeriveActiveModel)]
pub struct Model {
#(pub #column_names_snake_case: #column_rs_types,)*
}
}
}
WithSerde::Serialize => {
quote! {
#[derive(Clone, Debug, PartialEq, DeriveModel, DeriveActiveModel, Serialize)]
pub struct Model {
#(pub #column_names_snake_case: #column_rs_types,)*
}
}
}
WithSerde::Deserialize => { quote! {
quote! { #[derive(Clone, Debug, PartialEq, DeriveModel, DeriveActiveModel #extra_derive)]
#[derive(Clone, Debug, PartialEq, DeriveModel, DeriveActiveModel, Deserialize)] pub struct Model {
pub struct Model { #(pub #column_names_snake_case: #column_rs_types,)*
#(pub #column_names_snake_case: #column_rs_types,)*
}
}
}
WithSerde::Both => {
quote! {
#[derive(Clone, Debug, PartialEq, DeriveModel, DeriveActiveModel, Serialize, Deserialize)]
pub struct Model {
#(pub #column_names_snake_case: #column_rs_types,)*
}
}
} }
} }
} }
@ -453,56 +456,16 @@ impl EntityWriter {
}) })
.collect(); .collect();
match with_serde { let extra_derive = with_serde.extra_derive();
WithSerde::None => {
quote! {
#[derive(Clone, Debug, PartialEq, DeriveEntityModel)]
#[sea_orm(table_name = #table_name)]
pub struct Model {
#(
#attrs
pub #column_names_snake_case: #column_rs_types,
)*
}
}
}
WithSerde::Serialize => {
quote! {
#[derive(Clone, Debug, PartialEq, DeriveEntityModel, Serialize)]
#[sea_orm(table_name = #table_name)]
pub struct Model {
#(
#attrs
pub #column_names_snake_case: #column_rs_types,
)*
}
}
}
WithSerde::Deserialize => { quote! {
quote! { #[derive(Clone, Debug, PartialEq, DeriveEntityModel #extra_derive)]
#[derive(Clone, Debug, PartialEq, DeriveEntityModel, Deserialize)] #[sea_orm(table_name = #table_name)]
#[sea_orm(table_name = #table_name)] pub struct Model {
pub struct Model { #(
#( #attrs
#attrs pub #column_names_snake_case: #column_rs_types,
pub #column_names_snake_case: #column_rs_types, )*
)*
}
}
}
WithSerde::Both => {
quote! {
#[derive(Clone, Debug, PartialEq, DeriveEntityModel, Serialize, Deserialize)]
#[sea_orm(table_name = #table_name)]
pub struct Model {
#(
#attrs
pub #column_names_snake_case: #column_rs_types,
)*
}
}
} }
} }
} }