waitmap 1.1.0

an awaitable concurrent hash map
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

use std::task::Waker;

use smallvec::SmallVec;

pub struct WakerSet {
    wakers: SmallVec<[Option<Waker>; 1]>,
}

impl WakerSet {
    pub fn new() -> WakerSet {
        WakerSet {
            wakers: SmallVec::new(),
        }
    }

    pub fn replace(&mut self, waker: Waker, idx: &mut usize) {
        let len = self.wakers.len();
        if *idx >= len {
            debug_assert!(len != std::usize::MAX); // usize::MAX is used as a sentinel
            *idx = len;
            self.wakers.push(Some(waker));
        } else {
            self.wakers[*idx] = Some(waker);
        }
    }

    pub fn remove(&mut self, idx: usize) {
        self.wakers[idx] = None;
    }

    pub fn wake(self) {
        for waker in self.wakers {
            if let Some(waker) = waker {
                waker.wake()
            }
        }
    }
}