either-slot 1.2.0

An atomic slot whose senders can either place their value into the slot, or retrive all the data from the slot.
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

pub use core::{
    alloc::{AllocError, Allocator},
    mem::{self, ManuallyDrop},
    ptr::NonNull,
};

pub use ::alloc::alloc::handle_alloc_error;
pub use loom::{
    alloc::{alloc, dealloc, Layout},
    cell::UnsafeCell,
    hint,
    sync::atomic::{self, AtomicBool, AtomicU8, AtomicUsize, Ordering::*},
};

pub struct Global;

unsafe impl Allocator for Global {
    fn allocate(&self, layout: Layout) -> Result<NonNull<[u8]>, AllocError> {
        match layout.size() {
            0 => Ok(NonNull::slice_from_raw_parts(layout.dangling(), 0)),
            // SAFETY: `layout` is non-zero in size,
            size => unsafe {
                let raw_ptr = alloc(layout);
                let ptr = NonNull::new(raw_ptr).ok_or(AllocError)?;
                Ok(NonNull::slice_from_raw_parts(ptr, size))
            },
        }
    }

    unsafe fn deallocate(&self, ptr: NonNull<u8>, layout: Layout) {
        if layout.size() != 0 {
            // SAFETY: `layout` is non-zero in size,
            // other conditions must be upheld by the caller
            unsafe { dealloc(ptr.as_ptr(), layout) }
        }
    }
}