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

macros: fix handling of arguments of #[tokio::main] attribute (#1578)

Browse Source
This commit is contained in:
Taiki Endo
2019-09-23 03:05:04 +09:00
committed by GitHub
parent 376d63867a
commit ddbb0c3836
2 changed files with 9 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tokio-macros/Cargo.toml
+1 -1
View File
@@ -26,7 +26,7 @@ proc-macro = true
[dependencies]
quote = "1"
syn = { version = "1", features = ["full"] }
syn = { version = "1.0.3", features = ["full"] }
[dev-dependencies]
tokio = { version = "=0.2.0-alpha.5", path = "../tokio", default-features = false, features = ["rt-full"] }
tokio-macros/src/lib.rs
+8 -6
View File
@@ -74,16 +74,18 @@ pub fn main(args: TokenStream, item: TokenStream) -> TokenStream {
let mut runtime = RuntimeType::Multi;
for arg in args {
if let syn::NestedMeta::Meta(syn::Meta::Path(ident)) = arg {
let seg = ident.segments.first().expect("Must have specified ident");
match seg.ident.to_string().to_lowercase().as_str() {
if let syn::NestedMeta::Meta(syn::Meta::Path(path)) = arg {
let ident = path.get_ident();
if ident.is_none() {
let msg = "Must have specified ident";
return syn::Error::new_spanned(path, msg).to_compile_error().into();
}
match ident.unwrap().to_string().to_lowercase().as_str() {
"multi_thread" => runtime = RuntimeType::Multi,
"single_thread" => runtime = RuntimeType::Single,
name => {
let msg = format!("Unknown attribute {} is specified", name);
return syn::Error::new_spanned(ident, msg)
.to_compile_error()
.into();
return syn::Error::new_spanned(path, msg).to_compile_error().into();
}
}
}