fix

tokio-rs · May 8, 2021 · 38326a5 · 38326a5
1 parent f73eef9
commit 38326a5
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/tokio-macros/src/entry.rs b/tokio-macros/src/entry.rs
@@ -325,7 +325,7 @@ fn parse_knobs(
     let body =
         proc_macro2::TokenStream::from(TokenStream::from(input_tokens.into_iter().last().unwrap()));
 
-    input.block = syn::parse_quote! {
+    let body = quote! {
         {
             #rt
                 .enable_all()
@@ -335,9 +335,15 @@ fn parse_knobs(
         }
     };
 
+    let attrs = input.attrs;
+    let vis = input.vis;
+    let sig = input.sig;
     let result = quote! {
         #header
-        #input
+        #(#attrs)*
+        #vis
+        #sig
+        #body
     };
 
     Ok(result.into())