summaryrefslogtreecommitdiffstats
path: root/tokio/src/runtime/io.rs
blob: dca7310ff2653efba934264479de8ec339d8b6cc (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.

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

pub(crate) use variant::*;

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

    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 = Either<driver::Driver, ParkThread>;

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

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

            Ok((Either::A(driver), Some(handle)))
        } else {
            let driver = ParkThread::new();
            Ok((Either::B(driver), None))
        }
    }
}

#[cfg(any(not(feature = "io-driver"), loom))]
mod variant {
    use crate::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(_enable: bool) -> io::Result<(Driver, Handle)> {
        let driver = ParkThread::new();

        Ok((driver, ()))
    }
}