From 38326a5c2addceec135675f7462bb17e5a9b076b Mon Sep 17 00:00:00 2001
From: Taiki Endo <te316e89@gmail.com>
Date: Sun, 9 May 2021 00:20:30 +0900
Subject: [PATCH] fix

---
 tokio-macros/src/entry.rs | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/tokio-macros/src/entry.rs b/tokio-macros/src/entry.rs
index 6e251d7c69b..e26e89ba6f5 100644
--- 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())