Add Forever
This commit is contained in:
parent
03bd11ce0d
commit
53eb594878
2 changed files with 35 additions and 0 deletions
33
embassy/src/util/forever.rs
Normal file
33
embassy/src/util/forever.rs
Normal file
|
@ -0,0 +1,33 @@
|
||||||
|
use core::cell::UnsafeCell;
|
||||||
|
use core::mem::MaybeUninit;
|
||||||
|
use core::sync::atomic::{AtomicBool, Ordering};
|
||||||
|
|
||||||
|
pub struct Forever<T> {
|
||||||
|
used: AtomicBool,
|
||||||
|
t: UnsafeCell<MaybeUninit<T>>,
|
||||||
|
}
|
||||||
|
|
||||||
|
unsafe impl<T> Send for Forever<T> {}
|
||||||
|
unsafe impl<T> Sync for Forever<T> {}
|
||||||
|
|
||||||
|
impl<T> Forever<T> {
|
||||||
|
pub const fn new() -> Self {
|
||||||
|
Self {
|
||||||
|
used: AtomicBool::new(false),
|
||||||
|
t: UnsafeCell::new(MaybeUninit::uninit()),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn put(&self, val: T) -> &'static mut T {
|
||||||
|
if self.used.compare_and_swap(false, true, Ordering::SeqCst) {
|
||||||
|
panic!("Forever.put() called multiple times");
|
||||||
|
}
|
||||||
|
|
||||||
|
unsafe {
|
||||||
|
let p = self.t.get();
|
||||||
|
let p = (&mut *p).as_mut_ptr();
|
||||||
|
p.write(val);
|
||||||
|
&mut *p
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
|
@ -10,6 +10,8 @@ mod waker_store;
|
||||||
pub use waker_store::*;
|
pub use waker_store::*;
|
||||||
mod drop_bomb;
|
mod drop_bomb;
|
||||||
pub use drop_bomb::*;
|
pub use drop_bomb::*;
|
||||||
|
mod forever;
|
||||||
|
pub use forever::*;
|
||||||
|
|
||||||
use defmt::{debug, error, info, intern, trace, warn};
|
use defmt::{debug, error, info, intern, trace, warn};
|
||||||
|
|
||||||
|
|
Loading…
Reference in a new issue