summaryrefslogtreecommitdiffstats
path: root/tokio/src/runtime/io.rs
blob: 35009a67f313b583862c1878441c066e2c760a90 (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
//! Abstracts out the APIs necessary to `Runtime` for integrating the I/O
//! 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 self::variant::*;

/// Re-exported for convenience.
pub(crate) use std::io::Result;

#[cfg(feature = "io-driver")]
mod variant {
    use crate::net::driver;

    use std::io;

    /// The driver value the runtime passes to the `timer` layer.
    ///
    /// When the `io-driver` feature is enabled, this is the "real" I/O driver
    /// backed by Mio. Without the `io-driver` feature, this is a thread parker
    /// backed by a condition variable.
    pub(crate) type Driver = driver::Reactor;

    /// The handle the runtime stores for future use.
    ///
    /// When the `io-driver` feature is **not** enabled, this is `()`.
    pub(crate) type Handle = driver::Handle;

    pub(crate) fn create_driver() -> io::Result<(Driver, Handle)> {
        let driver = driver::Reactor::new()?;
        let handle = driver.handle();

        Ok((driver, handle))
    }

    pub(crate) fn set_default(handle: &Handle) -> driver::DefaultGuard<'_> {
        driver::set_default(handle)
    }
}

#[cfg(not(feature = "io-driver"))]
mod variant {
    use crate::runtime::park::ParkThread;

    use std::io;

    /// I/O is not enabled, use a condition variable based parker
    pub(crate) type Driver = ParkThread;

    /// There is no handle
    pub(crate) type Handle = ();

    pub(crate) fn create_driver() -> io::Result<(Driver, Handle)> {
        let driver = ParkThread::new();

        Ok((driver, ()))
    }

    pub(crate) fn set_default(_handle: &Handle) {}
}