compio-executor 0.1.0-rc.1

Executor for compio
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

use std::{
    mem::ManuallyDrop,
    task::{RawWaker, RawWakerVTable, Waker},
};

use crate::task::Task;

impl Task {
    const VTABLE: &'static RawWakerVTable = {
        &RawWakerVTable::new(
            Self::clone_waker,
            Self::wake,
            Self::wake_by_ref,
            Self::drop_waker,
        )
    };

    #[inline(always)]
    unsafe fn clone_waker(ptr: *const ()) -> RawWaker {
        unsafe { Task::increment_count(ptr) };
        RawWaker::new(ptr, Self::VTABLE)
    }

    unsafe fn wake(ptr: *const ()) {
        unsafe { Task::from_raw(ptr) }.schedule();
    }

    unsafe fn wake_by_ref(ptr: *const ()) {
        ManuallyDrop::new(unsafe { Task::from_raw(ptr) }).schedule();
    }

    unsafe fn drop_waker(ptr: *const ()) {
        drop(unsafe { Task::from_raw(ptr) });
    }

    /// Get a reference to [`Waker`] and run a closure on it.
    ///
    /// This will not increase the reference counter and is very cheap to call.
    pub fn with_waker<R, F: FnOnce(&Waker) -> R>(&self, f: F) -> R {
        let raw = RawWaker::new(self.as_raw(), Self::VTABLE);
        f(&ManuallyDrop::new(unsafe { Waker::from_raw(raw) }))
    }
}