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
#![no_std]

use task::{RunState, TaskRef};

/// An object that holds a blocked `Task` that will be automatically unblocked
/// upon drop.
pub struct WaitGuard {
    task: TaskRef,
}

impl WaitGuard {
    /// Blocks the given `Task` and returns a new `WaitGuard` object that will
    /// automatically unblock that Task when it is dropped.
    ///
    /// Returns an error if the task cannot be blocked; see
    /// [`task::Task::block`] for more details.
    pub fn new(task: TaskRef) -> Result<Self, RunState> {
        task.block()?;
        Ok(WaitGuard { task })
    }

    /// Blocks the task guarded by this guard, which is useful to re-block a
    /// task after it spuriously woken up.
    ///
    /// Returns an error if the task cannot be blocked; see
    /// [`task::Task::block`] for more details.
    pub fn block_again(&self) -> Result<RunState, RunState> {
        self.task.block()
    }

    /// Returns a reference to the task being blocked by this guard.
    pub fn task(&self) -> &TaskRef {
        &self.task
    }
}

impl Drop for WaitGuard {
    fn drop(&mut self) {
        let _ = self.task.unblock();
    }
}