Fix usage in `fn main`.

This commit is contained in:
daxpedda 2019-12-09 12:08:38 +01:00
parent d7800612a5
commit cd6007d483
No known key found for this signature in database
GPG Key ID: 43D62A3EA388E46F
1 changed files with 4 additions and 2 deletions

View File

@ -17,10 +17,10 @@ use quote::quote;
#[proc_macro_attribute] #[proc_macro_attribute]
#[cfg(not(test))] // Work around for rust-lang/rust#62127 #[cfg(not(test))] // Work around for rust-lang/rust#62127
pub fn main(_: TokenStream, item: TokenStream) -> TokenStream { pub fn main(_: TokenStream, item: TokenStream) -> TokenStream {
let input = syn::parse_macro_input!(item as syn::ItemFn); let mut input = syn::parse_macro_input!(item as syn::ItemFn);
let attrs = &input.attrs; let attrs = &input.attrs;
let vis = &input.vis; let vis = &input.vis;
let sig = &input.sig; let sig = &mut input.sig;
let body = &input.block; let body = &input.block;
let name = &sig.ident; let name = &sig.ident;
@ -30,6 +30,8 @@ pub fn main(_: TokenStream, item: TokenStream) -> TokenStream {
.into(); .into();
} }
sig.asyncness = None;
(quote! { (quote! {
#(#attrs)* #(#attrs)*
#vis #sig { #vis #sig {