Skip to content

Commit

Permalink
Revert "macros: run runtime inside LocalSet when using macro (#4027)"
Browse files Browse the repository at this point in the history
This reverts commit 33f0a1f.
  • Loading branch information
Darksonn committed Sep 30, 2021
1 parent 44cfe10 commit f889d66
Show file tree
Hide file tree
Showing 4 changed files with 34 additions and 51 deletions.
8 changes: 4 additions & 4 deletions tests-build/tests/fail/macros_type_mismatch.stderr
Original file line number Diff line number Diff line change
@@ -1,8 +1,8 @@
error[E0308]: mismatched types
--> $DIR/macros_type_mismatch.rs:5:7
--> $DIR/macros_type_mismatch.rs:5:5
|
5 | Ok(())
| ^^^^ expected `()`, found enum `Result`
| ^^^^^^ expected `()`, found enum `Result`
|
= note: expected unit type `()`
found enum `Result<(), _>`
Expand All @@ -16,12 +16,12 @@ help: try adding a return type
| ^^^^^^^^^^^^^^^^

error[E0308]: mismatched types
--> $DIR/macros_type_mismatch.rs:10:18
--> $DIR/macros_type_mismatch.rs:10:5
|
9 | async fn missing_return_type() {
| - help: try adding a return type: `-> Result<(), _>`
10 | return Ok(());
| ^ expected `()`, found enum `Result`
| ^^^^^^^^^^^^^^ expected `()`, found enum `Result`
|
= note: expected unit type `()`
found enum `Result<(), _>`
Expand Down
9 changes: 5 additions & 4 deletions tokio-macros/src/entry.rs
Original file line number Diff line number Diff line change
Expand Up @@ -339,10 +339,11 @@ fn parse_knobs(
{
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");
Expand Down
58 changes: 25 additions & 33 deletions tokio-macros/src/lib.rs
Original file line number Diff line number Diff line change
Expand Up @@ -27,9 +27,6 @@ use proc_macro::TokenStream;
/// helps set up a `Runtime` without requiring the user to use
/// [Runtime](../tokio/runtime/struct.Runtime.html) or
/// [Builder](../tokio/runtime/struct.Builder.html) directly.
/// The function executes in the context of a
/// [LocalSet](../tokio/task/struct.LocalSet.html), allowing calls to
/// [spawn_local](../tokio/task/fn.spawn_local.html) without further setup.
///
/// Note: This macro is designed to be simplistic and targets applications that
/// do not require a complex setup. If the provided functionality is not
Expand Down Expand Up @@ -87,14 +84,13 @@ use proc_macro::TokenStream;
///
/// ```rust
/// fn main() {
/// let ls = tokio::task::LocalSet::new();
/// let rt = tokio::runtime::Builder::new_multi_thread()
/// tokio::runtime::Builder::new_multi_thread()
/// .enable_all()
/// .build()
/// .unwrap();
/// ls.block_on(&rt, async {
/// println!("Hello world");
/// })
/// .unwrap()
/// .block_on(async {
/// println!("Hello world");
/// })
/// }
/// ```
///
Expand All @@ -113,14 +109,13 @@ use proc_macro::TokenStream;
///
/// ```rust
/// fn main() {
/// let ls = tokio::task::LocalSet::new();
/// let rt = tokio::runtime::Builder::new_current_thread()
/// tokio::runtime::Builder::new_current_thread()
/// .enable_all()
/// .build()
/// .unwrap();
/// ls.block_on(&rt, async {
/// println!("Hello world");
/// })
/// .unwrap()
/// .block_on(async {
/// println!("Hello world");
/// })
/// }
/// ```
///
Expand All @@ -137,15 +132,14 @@ use proc_macro::TokenStream;
///
/// ```rust
/// fn main() {
/// let ls = tokio::task::LocalSet::new();
/// let rt = tokio::runtime::Builder::new_multi_thread()
/// tokio::runtime::Builder::new_multi_thread()
/// .worker_threads(2)
/// .enable_all()
/// .build()
/// .unwrap();
/// ls.block_on(&rt, async {
/// println!("Hello world");
/// })
/// .unwrap()
/// .block_on(async {
/// println!("Hello world");
/// })
/// }
/// ```
///
Expand All @@ -162,15 +156,14 @@ use proc_macro::TokenStream;
///
/// ```rust
/// fn main() {
/// let ls = tokio::task::LocalSet::new();
/// let rt = tokio::runtime::Builder::new_current_thread()
/// tokio::runtime::Builder::new_current_thread()
/// .enable_all()
/// .start_paused(true)
/// .build()
/// .unwrap();
/// ls.block_on(&rt, async {
/// println!("Hello world");
/// })
/// .unwrap()
/// .block_on(async {
/// println!("Hello world");
/// })
/// }
/// ```
///
Expand Down Expand Up @@ -211,14 +204,13 @@ pub fn main(args: TokenStream, item: TokenStream) -> TokenStream {
///
/// ```rust
/// fn main() {
/// let ls = tokio::task::LocalSet::new();
/// let rt = tokio::runtime::Builder::new_current_thread()
/// tokio::runtime::Builder::new_current_thread()
/// .enable_all()
/// .build()
/// .unwrap();
/// ls.block_on(&rt, async {
/// println!("Hello world");
/// })
/// .unwrap()
/// .block_on(async {
/// println!("Hello world");
/// })
/// }
/// ```
///
Expand Down
10 changes: 0 additions & 10 deletions tokio/tests/task_local_set.rs
Original file line number Diff line number Diff line change
Expand Up @@ -16,16 +16,6 @@ use std::sync::atomic::Ordering::{self, SeqCst};
use std::sync::atomic::{AtomicBool, AtomicUsize};
use std::time::Duration;

#[tokio::test(flavor = "current_thread")]
async fn localset_implicit_current_thread() {
task::spawn_local(async {}).await.unwrap();
}

#[tokio::test(flavor = "multi_thread")]
async fn localset_implicit_multi_thread() {
task::spawn_local(async {}).await.unwrap();
}

#[tokio::test(flavor = "current_thread")]
async fn local_basic_scheduler() {
LocalSet::new()
Expand Down

0 comments on commit f889d66

Please sign in to comment.