chore: fix output of macro after new rustc release (#4189)

tokio-rs · Nov 16, 2021 · 8303b5a · 8303b5a
1 parent 01a58e0
commit 8303b5a
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/tokio-macros/src/entry.rs b/tokio-macros/src/entry.rs
@@ -356,10 +356,11 @@ fn parse_knobs(mut input: syn::ItemFn, is_test: bool, config: FinalConfig) -> To
         {
             let body = async #body;
             #[allow(clippy::expect_used)]
-            #tail_return tokio::task::LocalSet::new().block_on(
-              &#rt.enable_all().build().expect("Failed building the Runtime"),
-              body,
-            )#tail_semicolon
+            #tail_return #rt
+                .enable_all()
+                .build()
+                .expect("Failed building the Runtime")
+                .block_on(body)#tail_semicolon
         }
     })
     .expect("Parsing failure");