tokio-uring-xitca 0.2.0

a fork of tokio-uring with miniaml maintenance
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
42
43
44
45
46
47
48
49

use core::cell::RefCell;

use crate::runtime::driver::{Handle, WeakHandle};

/// Owns the driver and resides in thread-local storage.
pub(crate) struct RuntimeContext {
    driver: RefCell<Option<Handle>>,
}

impl RuntimeContext {
    /// Construct the context with an uninitialized driver.
    pub(crate) const fn new() -> Self {
        Self {
            driver: RefCell::new(None),
        }
    }

    /// Initialize the driver.
    pub(crate) fn set_handle(&self, handle: Handle) {
        let mut guard = self.driver.borrow_mut();

        assert!(guard.is_none(), "Attempted to initialize the driver twice");

        *guard = Some(handle);
    }

    pub(crate) fn unset_driver(&self) {
        let mut guard = self.driver.borrow_mut();

        assert!(guard.is_some(), "Attempted to clear nonexistent driver");

        *guard = None;
    }

    /// Check if driver is initialized
    #[allow(dead_code)]
    pub(crate) fn is_set(&self) -> bool {
        self.driver.try_borrow().map(|b| b.is_some()).unwrap_or(false)
    }

    pub(crate) fn handle(&self) -> Option<Handle> {
        self.driver.borrow().clone()
    }

    #[allow(dead_code)]
    pub(crate) fn weak(&self) -> Option<WeakHandle> {
        self.driver.borrow().as_ref().map(Into::into)
    }
}