uasync 0.1.1

fast, safe, async executor
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
69
70
71
72

use super::{
    builder::Builder,
    enter::EnterGuard,
    handle::Handle,
    task::{JoinHandle, Task},
};
use std::{fmt, future::Future, io, mem::replace, time::Duration};

pub struct Runtime {
    pub(super) handle: Handle,
    pub(super) shutdown_on_drop: bool,
}

impl fmt::Debug for Runtime {
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
        f.debug_struct("Runtime").finish()
    }
}

impl Runtime {
    pub fn new() -> io::Result<Self> {
        Builder::new_multi_thread().build()
    }

    pub fn handle(&self) -> &Handle {
        &self.handle
    }

    pub fn enter(&self) -> EnterGuard<'_> {
        self.handle.enter()
    }

    pub fn block_on<F: Future>(&self, future: F) -> F::Output {
        let future = Box::pin(future);
        Task::block_on(&self.handle.scheduler, future)
    }

    pub fn spawn<F>(&self, future: F) -> JoinHandle<F::Output>
    where
        F: Future + Send + 'static,
        F::Output: Send + 'static,
    {
        let future = Box::pin(future);
        Task::spawn(&self.handle.scheduler, None, future)
    }

    pub fn shutdown_background(self) {
        let mut mut_self = self;
        mut_self.shutdown_and_join(None);
    }

    pub fn shutdown_timeout(self, timeout: Duration) {
        let mut mut_self = self;
        mut_self.shutdown_and_join(Some(Some(timeout)));
    }

    fn shutdown_and_join(&mut self, join_timeout: Option<Option<Duration>>) {
        if replace(&mut self.shutdown_on_drop, false) {
            self.handle.scheduler.shutdown();

            if let Some(timeout) = join_timeout {
                self.handle.scheduler.join(timeout);
            }
        }
    }
}

impl Drop for Runtime {
    fn drop(&mut self) {
        self.shutdown_and_join(Some(None))
    }
}