actix/actix-net
1
0
Fork 0
mirror of https://github.com/fafhrd91/actix-net synced 2024-11-23 22:51: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]
#[cfg(not(test))] // Work around for rust-lang/rust#62127
pub fn main(_: TokenStream, item: TokenStream) -> TokenStream {
let 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 mut input = syn::parse_macro_input!(item as syn::ItemFn);
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() {
return syn::Error::new_spanned(input.sig.fn_token, "only async fn is supported")
if sig.asyncness.is_none() {
return syn::Error::new_spanned(sig.fn_token, "only async fn is supported")
.to_compile_error()
.into();
}
sig.asyncness = None;
(quote! {
#(#attrs)*
fn #name(#inputs) #ret {
actix_rt::System::new("main")
.block_on(async { #body })
#vis #sig {
actix_rt::System::new(stringify!(#name))
.block_on(async move { #body })
}
})
.into()