Get rid of useless DeserializeWith wrapper

This commit is contained in:
Mingun 2021-02-27 18:52:32 +05:00
parent 7b840897a9
commit ca772a14f9

View File

@ -1728,6 +1728,15 @@ fn deserialize_externally_tagged_variant(
} }
} }
/// Generates significant part of the `Visitor::visit_seq` and `Visitor::visit_map`
/// bodies of visitors for deserializing variants of internally tagged enum:
///
/// ```ignore
/// #[serde(tag = "...")]
/// enum InternallyTagged {
/// ...
/// }
/// ```
fn deserialize_internally_tagged_variant( fn deserialize_internally_tagged_variant(
params: &Parameters, params: &Parameters,
variant: &Variant, variant: &Variant,
@ -1779,11 +1788,9 @@ fn deserialize_untagged_variant(
deserializer: TokenStream, deserializer: TokenStream,
) -> Fragment { ) -> Fragment {
if let Some(path) = variant.attrs.deserialize_with() { if let Some(path) = variant.attrs.deserialize_with() {
let (wrapper, wrapper_ty, unwrap_fn) = wrap_deserialize_variant_with(params, variant, path); let unwrap_fn = unwrap_to_variant_closure(params, variant, quote!(__wrap));
return quote_block! { return quote_block! {
#wrapper _serde::__private::Result::map(#path(#deserializer), #unwrap_fn)
_serde::__private::Result::map(
<#wrapper_ty as _serde::Deserialize>::deserialize(#deserializer), #unwrap_fn)
}; };
} }
@ -2883,44 +2890,54 @@ fn wrap_deserialize_variant_with(
variant: &Variant, variant: &Variant,
deserialize_with: &syn::ExprPath, deserialize_with: &syn::ExprPath,
) -> (TokenStream, TokenStream, TokenStream) { ) -> (TokenStream, TokenStream, TokenStream) {
let this = &params.this;
let variant_ident = &variant.ident;
let field_tys = variant.fields.iter().map(|field| field.ty); let field_tys = variant.fields.iter().map(|field| field.ty);
let (wrapper, wrapper_ty) = let (wrapper, wrapper_ty) =
wrap_deserialize_with(params, &quote!((#(#field_tys),*)), deserialize_with); wrap_deserialize_with(params, &quote!((#(#field_tys),*)), deserialize_with);
let unwrap_fn = unwrap_to_variant_closure(params, variant, quote!(__wrap.value));
(wrapper, wrapper_ty, unwrap_fn)
}
/// Generates closure that converts single input parameter to the final value
fn unwrap_to_variant_closure(
params: &Parameters,
variant: &Variant,
wrapper: TokenStream,
) -> TokenStream {
let this = &params.this;
let variant_ident = &variant.ident;
let field_access = (0..variant.fields.len()).map(|n| { let field_access = (0..variant.fields.len()).map(|n| {
Member::Unnamed(Index { Member::Unnamed(Index {
index: n as u32, index: n as u32,
span: Span::call_site(), span: Span::call_site(),
}) })
}); });
let unwrap_fn = match variant.style {
match variant.style {
Style::Struct if variant.fields.len() == 1 => { Style::Struct if variant.fields.len() == 1 => {
let member = &variant.fields[0].member; let member = &variant.fields[0].member;
quote! { quote! {
|__wrap| #this::#variant_ident { #member: __wrap.value } |__wrap| #this::#variant_ident { #member: #wrapper }
} }
} }
Style::Struct => { Style::Struct => {
let members = variant.fields.iter().map(|field| &field.member); let members = variant.fields.iter().map(|field| &field.member);
quote! { quote! {
|__wrap| #this::#variant_ident { #(#members: __wrap.value.#field_access),* } |__wrap| #this::#variant_ident { #(#members: #wrapper.#field_access),* }
} }
} }
Style::Tuple => quote! { Style::Tuple => quote! {
|__wrap| #this::#variant_ident(#(__wrap.value.#field_access),*) |__wrap| #this::#variant_ident(#(#wrapper.#field_access),*)
}, },
Style::Newtype => quote! { Style::Newtype => quote! {
|__wrap| #this::#variant_ident(__wrap.value) |__wrap| #this::#variant_ident(#wrapper)
}, },
Style::Unit => quote! { Style::Unit => quote! {
|__wrap| #this::#variant_ident |__wrap| #this::#variant_ident
}, },
}; }
(wrapper, wrapper_ty, unwrap_fn)
} }
fn expr_is_missing(field: &Field, cattrs: &attr::Container) -> Fragment { fn expr_is_missing(field: &Field, cattrs: &attr::Container) -> Fragment {