tokio-rs/tokio
1
0
Fork 0
mirror of https://github.com/tokio-rs/tokio.git synced 2026-09-06 00:00:10 +02:00
Code Issues Packages Projects Releases Wiki Activity

macros: do not automatically pull rt-core (#2038)

Browse Source
This commit is contained in:
Artem Vorotnikov
2020-01-02 11:22:15 -08:00
committed by Carl Lerche
parent 3cf91db4b6
commit e43f28f6a8
3 changed files with 17 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tokio-macros/src/lib.rs
+5 -15
View File
@@ -20,6 +20,7 @@ use proc_macro::TokenStream;
use quote::quote;
use std::num::NonZeroUsize;
#[derive(Clone, Copy, PartialEq)]
enum Runtime {
Basic,
Threaded,
@@ -145,21 +146,10 @@ fn parse_knobs(
}
}
let mut rt = quote! { tokio::runtime::Builder::new() };
match (runtime, is_test) {
(Some(Runtime::Threaded), _) => {
if rt_threaded {
rt = quote! { #rt.threaded_scheduler() }
}
}
(Some(Runtime::Basic), _) => rt = quote! { #rt.basic_scheduler() },
(None, true) => rt = quote! { #rt.basic_scheduler() },
(None, false) => {
if rt_threaded {
rt = quote! { #rt.threaded_scheduler() }
}
}
};
let mut rt = quote! { tokio::runtime::Builder::new().basic_scheduler() };
if rt_threaded && (runtime == Some(Runtime::Threaded) || (runtime.is_none() && !is_test)) {
rt = quote! { #rt.threaded_scheduler() };
}
if let Some(v) = core_threads.map(|v| v.get()) {
rt = quote! { #rt.core_threads(#v) };
}