summaryrefslogtreecommitdiffstats
path: root/tokio/src/runtime/time.rs
blob: 1cd58cd6f5ca57acdacdc9a6201d96a95cdd733b (plain)
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
73
74
//! Abstracts out the APIs necessary to `Runtime` for integrating the time
//! driver. When the `time` feature flag is **not** enabled. These APIs are
//! shells. This isolates the complexity of dealing with conditional
//! compilation.

pub(crate) use variant::*;

#[cfg(all(feature = "time", not(loom)))]
mod variant {
    use crate::park::Either;
    use crate::runtime::io;
    use crate::time::{self, driver};

    pub(crate) type Clock = time::Clock;
    pub(crate) type Driver = Either<driver::Driver<io::Driver>, io::Driver>;
    pub(crate) type Handle = Option<driver::Handle>;

    pub(crate) fn create_clock() -> Clock {
        Clock::new()
    }

    /// Create a new timer driver / handle pair
    pub(crate) fn create_driver(
        enable: bool,
        io_driver: io::Driver,
        clock: Clock,
    ) -> (Driver, Handle) {
        if enable {
            let driver = driver::Driver::new(io_driver, clock);
            let handle = driver.handle();

            (Either::A(driver), Some(handle))
        } else {
            (Either::B(io_driver), None)
        }
    }

    pub(crate) fn with_default<F, R>(handle: &Handle, clock: &Clock, f: F) -> R
    where
        F: FnOnce() -> R,
    {
        let _time = handle.as_ref().map(|handle| driver::set_default(handle));
        clock.enter(f)
    }
}

#[cfg(any(not(feature = "time"), loom))]
mod variant {
    use crate::runtime::io;

    pub(crate) type Clock = ();
    pub(crate) type Driver = io::Driver;
    pub(crate) type Handle = ();

    pub(crate) fn create_clock() -> Clock {
        ()
    }

    /// Create a new timer driver / handle pair
    pub(crate) fn create_driver(
        _enable: bool,
        io_driver: io::Driver,
        _clock: Clock,
    ) -> (Driver, Handle) {
        (io_driver, ())
    }

    pub(crate) fn with_default<F, R>(_handler: &Handle, _clock: &Clock, f: F) -> R
    where
        F: FnOnce() -> R,
    {
        f()
    }
}