nordic-dev.net/embassy
2
0
Fork 0
mirror of https://github.com/embassy-rs/embassy.git synced 2024-11-21 22:32:29 +00:00
Code Issues Packages Projects Releases Wiki Activity

task codegen: use local items instead of non-locals with prefixed names.

Browse Source
This commit is contained in:
Dario Nieuwenhuis 2020-10-31 16:36:09 +01:00
parent 53eb594878
commit 57abd7064d
1 changed files with 5 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
embassy-macros/src/lib.rs
View File

@ -85,29 +85,15 @@ pub fn task(args: TokenStream, item: TokenStream) -> TokenStream {
let name = task_fn.sig.ident.clone();
let type_name = format_ident!("__embassy_executor_type_{}", name);
let pool_name = format_ident!("__embassy_executor_pool_{}", name);
let task_fn_name = format_ident!("__embassy_executor_task_{}", name);
let create_fn_name = format_ident!("__embassy_executor_create_{}", name);
let visibility = &task_fn.vis;
task_fn.sig.ident = task_fn_name.clone();
task_fn.sig.ident = format_ident!("task");
let result = quote! {
#task_fn
#[allow(non_camel_case_types)]
type #type_name = impl ::core::future::Future + 'static;
fn #create_fn_name(#args) -> #type_name {
#task_fn_name(#arg_names)
}
#[allow(non_upper_case_globals)]
static #pool_name: [::embassy::executor::Task<#type_name>; #pool_size] = [::embassy::executor::Task::new(); #pool_size];
#visibility fn #name(#args) -> ::embassy::executor::SpawnToken {
unsafe { ::embassy::executor::Task::spawn(&#pool_name, || #create_fn_name(#arg_names)) }
#task_fn
type F = impl ::core::future::Future + 'static;
static POOL: [::embassy::executor::Task<F>; #pool_size] = [::embassy::executor::Task::new(); #pool_size];
unsafe { ::embassy::executor::Task::spawn(&POOL, || task(#arg_names)) }
}
};
result.into()