local-ex 0.1.0

Thread local async executor
Documentation 
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

use std::{
    future::Future,
    task::{Context, Poll},
};

use async_executor::{LocalExecutor, Task};

pub struct LocalEx;
impl LocalEx {
    thread_local!(static EX: LocalExecutor<'static> = LocalExecutor::new());
    pub fn spawn<T: 'static>(fut: impl Future<Output = T> + 'static) -> Task<T> {
        Self::EX.with(|e| e.spawn(fut))
    }

    pub fn run<F: Future>(task: F) -> F::Output {
        Self::EX.with(|ex| {
            futures_lite::pin!(task);

            let this = std::thread::current();
            let waker = waker_fn::waker_fn(move || {
                this.unpark();
            });
            let mut cx = Context::from_waker(&waker);

            loop {
                if let Poll::Ready(r) = task.as_mut().poll(&mut cx) {
                    return r;
                }
                while ex.try_tick() {}

                let fut = ex.tick();
                futures_lite::pin!(fut);

                match fut.poll(&mut cx) {
                    Poll::Ready(_) => (),
                    Poll::Pending => std::thread::park(),
                }
            }
        })
    }
}