VienDesu/schemars
3
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Merge metadata setters and transform applications if they're both specified

Browse source
This commit is contained in:
Graham Esau 2024-08-12 18:24:38 +01:00
parent eb3077742f
commit faf15e7859
1 changed files with 3 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

12
schemars_derive/src/metadata.rs
View file

@ -17,14 +17,7 @@ pub struct SchemaMetadata<'a> {
impl<'a> SchemaMetadata<'a> { impl<'a> SchemaMetadata<'a> {
pub fn apply_to_schema(&self, schema_expr: &mut TokenStream) { pub fn apply_to_schema(&self, schema_expr: &mut TokenStream) {
let setters = self.make_setters(); let setters = self.make_setters();
if !setters.is_empty() { if !setters.is_empty() || !self.transforms.is_empty() {
*schema_expr = quote! {{
let mut schema = #schema_expr;
#(#setters)*
schema
}}
}
if !self.transforms.is_empty() {
let apply_transforms = self.transforms.iter().map(|t| { let apply_transforms = self.transforms.iter().map(|t| {
quote_spanned! {t.span()=> quote_spanned! {t.span()=>
schemars::transform::Transform::transform(&mut #t, &mut schema); schemars::transform::Transform::transform(&mut #t, &mut schema);
@ -32,9 +25,10 @@ impl<'a> SchemaMetadata<'a> {
}); });
*schema_expr = quote! {{ *schema_expr = quote! {{
let mut schema = #schema_expr; let mut schema = #schema_expr;
#(#setters)*
#(#apply_transforms)* #(#apply_transforms)*
schema schema
}}; }}
} }
} }