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
65
66
67
68
use core::future::Future;
use core::pin::Pin;
use core::task::{Context, Poll};
#[cfg(feature = "tokio_asyncs")]
pub mod task_impl {
use super::{Context, Future, Pin, Poll};
pub struct JoinHandleImpl<T>(tokio::task::JoinHandle<T>);
impl<T> Future for JoinHandleImpl<T> {
type Output = T;
fn poll(self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Self::Output> {
match unsafe { self.map_unchecked_mut(|s| &mut s.0) }.poll(cx) {
Poll::Ready(r) => Poll::Ready(r.expect("task join failed")),
Poll::Pending => Poll::Pending,
}
}
}
pub fn spawn<T: Send + 'static, F: Future<Output = T> + Send + 'static>(
future: F,
) -> JoinHandleImpl<T> {
JoinHandleImpl(tokio::task::spawn(future))
}
pub fn spawn_local<T: 'static, F: Future<Output = T> + 'static>(
future: F,
) -> JoinHandleImpl<T> {
JoinHandleImpl(tokio::task::spawn_local(future))
}
}
pub fn spawn<T: Send + 'static, F: Future<Output = T> + Send + 'static>(
future: F,
) -> JoinHandle<T> {
JoinHandle(task_impl::spawn(future))
}
pub fn spawn_local<T: 'static, F: Future<Output = T> + 'static>(future: F) -> JoinHandle<T> {
JoinHandle(task_impl::spawn_local(future))
}
pub struct JoinHandle<T>(task_impl::JoinHandleImpl<T>);
impl<T> Future for JoinHandle<T> {
type Output = T;
fn poll(self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Self::Output> {
unsafe { self.map_unchecked_mut(|s| &mut s.0) }.poll(cx)
}
}
pub async fn yield_now() {
struct YieldNow {
yielded: bool,
}
impl Future for YieldNow {
type Output = ();
fn poll(mut self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<()> {
if self.yielded {
return Poll::Ready(());
}
self.yielded = true;
cx.waker().wake_by_ref();
Poll::Pending
}
}
YieldNow { yielded: false }.await
}