1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
mod machine;
mod processor;
mod system;
mod task;
pub use system::get_num_cpus;
pub use system::set_num_cpus;
pub use system::detach_current_thread;
use std::future::Future;
use std::pin::Pin;
use std::task::{Context, Poll};
use self::task::TaskTag;
type Task = async_task::Task<TaskTag>;
pub fn spawn<T, R>(task: T) -> JoinHandle<R>
where
T: Future<Output = R> + Send + 'static,
R: Send + 'static,
{
let system = system::get();
let (task, handle) = async_task::spawn(task, move |task| system.push(task), TaskTag::new());
task.schedule();
JoinHandle(handle)
}
pub struct JoinHandle<R>(async_task::JoinHandle<R, TaskTag>);
impl<R> Future for JoinHandle<R> {
type Output = R;
#[inline(always)]
fn poll(mut self: Pin<&mut Self>, cx: &mut Context) -> Poll<Self::Output> {
match Pin::new(&mut self.0).poll(cx) {
Poll::Pending => Poll::Pending,
Poll::Ready(Some(val)) => Poll::Ready(val),
Poll::Ready(None) => unreachable!(),
}
}
}