Merge metadata setters and transform applications if they're both spe…

…cified
GREsau · Aug 12, 2024 · faf15e7 · faf15e7
1 parent eb30777
commit faf15e7
Showing 1 changed file with 3 additions and 9 deletions.
diff --git a/schemars_derive/src/metadata.rs b/schemars_derive/src/metadata.rs
@@ -17,24 +17,18 @@ pub struct SchemaMetadata<'a> {
 impl<'a> SchemaMetadata<'a> {
     pub fn apply_to_schema(&self, schema_expr: &mut TokenStream) {
         let setters = self.make_setters();
-        if !setters.is_empty() {
-            *schema_expr = quote! {{
-                let mut schema = #schema_expr;
-                #(#setters)*
-                schema
-            }}
-        }
-        if !self.transforms.is_empty() {
+        if !setters.is_empty() || !self.transforms.is_empty() {
             let apply_transforms = self.transforms.iter().map(|t| {
                 quote_spanned! {t.span()=>
                     schemars::transform::Transform::transform(&mut #t, &mut schema);
                 }
             });
             *schema_expr = quote! {{
                 let mut schema = #schema_expr;
+                #(#setters)*
                 #(#apply_transforms)*
                 schema
-            }};
+            }}
         }
     }