actix/actix-net
1
0
Fork 0
mirror of https://github.com/fafhrd91/actix-net synced 2024-11-24 01:11:07 +01:00
Code Issues Releases Wiki Activity

Macro improvements. (#74)

Browse Source 
* Macro improvements.

* Fix usage in `fn main`.
This commit is contained in:
daxpedda 2019-12-10 03:47:35 +01:00 committed by Nikolay Kim
parent 8b3062cd6e
commit 0913badd61
1 changed files with 12 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
actix-macros/src/lib.rs
View File

@ -17,25 +17,26 @@ 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 ret = &input.sig.output;
let name = &input.sig.ident;
let inputs = &input.sig.inputs;
let body = &input.block;
let attrs = &input.attrs; let attrs = &input.attrs;
let vis = &input.vis;
let sig = &mut input.sig;
let body = &input.block;
let name = &sig.ident;
if input.sig.asyncness.is_none() { if sig.asyncness.is_none() {
return syn::Error::new_spanned(input.sig.fn_token, "only async fn is supported") return syn::Error::new_spanned(sig.fn_token, "only async fn is supported")
.to_compile_error() .to_compile_error()
.into(); .into();
} }
sig.asyncness = None;
(quote! { (quote! {
#(#attrs)* #(#attrs)*
fn #name(#inputs) #ret { #vis #sig {
actix_rt::System::new("main") actix_rt::System::new(stringify!(#name))
.block_on(async { #body }) .block_on(async move { #body })
} }
}) })
.into() .into()