rdma 0.3.0

Low-level RDMA API
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

use crate::utils::{ptr_from_addr, ptr_to_addr};

use std::marker::PhantomData;
use std::sync::Weak;

use fnv::FnvHashSet;

pub struct WeakSet<T> {
    set: FnvHashSet<usize>,
    _marker: PhantomData<Weak<T>>,
}

impl<T> WeakSet<T> {
    pub fn new() -> Self {
        Self {
            set: FnvHashSet::default(),
            _marker: PhantomData,
        }
    }

    pub fn insert(&mut self, weak: Weak<T>) {
        self.set.insert(ptr_to_addr(Weak::into_raw(weak)));
    }

    pub fn remove(&mut self, p: *const T) -> bool {
        let is_removed = self.set.remove(&ptr_to_addr(p));
        if is_removed {
            // SAFETY: guaranteed by `WeakSet::insert`
            unsafe { drop(Weak::from_raw(p)) }
        }
        is_removed
    }
}

impl<T> Drop for WeakSet<T> {
    fn drop(&mut self) {
        for addr in self.set.drain() {
            // SAFETY: guaranteed by `WeakSet::insert`
            unsafe {
                let p = ptr_from_addr::<T>(addr);
                drop(Weak::from_raw(p));
            }
        }
    }
}