itsscb/tokio
1
0
Fork 0
mirror of https://github.com/tokio-rs/tokio.git synced 2025-09-25 12:00:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

macros: inherit visibility

Browse Source
This commit is contained in:
Douman 2019-12-13 17:53:11 +01:00
parent 8abaf89e5f
commit 91ecb4b4c2
1 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
tokio-macros/src/lib.rs
View File

@ -65,6 +65,7 @@ pub fn main(args: TokenStream, item: TokenStream) -> TokenStream {
let inputs = &input.sig.inputs;
let body = &input.block;
let attrs = &input.attrs;
let vis = input.vis;
if input.sig.asyncness.is_none() {
let msg = "the async keyword is missing from the function declaration";
@ -101,13 +102,13 @@ pub fn main(args: TokenStream, item: TokenStream) -> TokenStream {
let result = match runtime {
Runtime::Threaded | Runtime::Auto => quote! {
#(#attrs)*
fn #name(#inputs) #ret {
#vis fn #name(#inputs) #ret {
tokio::runtime::Runtime::new().unwrap().block_on(async { #body })
}
},
Runtime::Basic => quote! {
#(#attrs)*
fn #name(#inputs) #ret {
#vis fn #name(#inputs) #ret {
tokio::runtime::Builder::new()
.basic_scheduler()
.enable_all()
@ -156,6 +157,7 @@ pub fn test(args: TokenStream, item: TokenStream) -> TokenStream {
let name = &input.sig.ident;
let body = &input.block;
let attrs = &input.attrs;
let vis = input.vis;
for attr in attrs {
if attr.path.is_ident("test") {
@ -202,14 +204,14 @@ pub fn test(args: TokenStream, item: TokenStream) -> TokenStream {
Runtime::Threaded => quote! {
#[test]
#(#attrs)*
fn #name() #ret {
#vis fn #name() #ret {
tokio::runtime::Runtime::new().unwrap().block_on(async { #body })
}
},
Runtime::Basic | Runtime::Auto => quote! {
#[test]
#(#attrs)*
fn #name() #ret {
#vis fn #name() #ret {
tokio::runtime::Builder::new()
.basic_scheduler()
.enable_all()