自己封装一个refcell
parent
f348dcbad4
commit
5628d6f253
@ -0,0 +1,4 @@
|
|||||||
|
//! Synchronization and interior mutability primitives
|
||||||
|
|
||||||
|
mod up;
|
||||||
|
pub use up::UPSafeCell;
|
@ -0,0 +1,31 @@
|
|||||||
|
//! Uniprocessor interior mutability primitives
|
||||||
|
|
||||||
|
use core::cell::{RefCell, RefMut};
|
||||||
|
|
||||||
|
/// Wrap a static data structure inside it so that we are
|
||||||
|
/// able to access it without any `unsafe`.
|
||||||
|
///
|
||||||
|
/// We should only use it in uniprocessor.
|
||||||
|
///
|
||||||
|
/// In order to get mutable reference of inner data, call
|
||||||
|
/// `exclusive_access`.
|
||||||
|
pub struct UPSafeCell<T> {
|
||||||
|
/// inner data
|
||||||
|
inner: RefCell<T>,
|
||||||
|
}
|
||||||
|
|
||||||
|
unsafe impl<T> Sync for UPSafeCell<T> {}
|
||||||
|
|
||||||
|
impl<T> UPSafeCell<T> {
|
||||||
|
/// User is responsible to guarantee that inner struct is only used in
|
||||||
|
/// uniprocessor.
|
||||||
|
pub unsafe fn new(value: T) -> Self {
|
||||||
|
Self {
|
||||||
|
inner: RefCell::new(value),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
/// Exclusive access inner data in UPSafeCell. Panic if the data has been borrowed.
|
||||||
|
pub fn exclusive_access(&self) -> RefMut<'_, T> {
|
||||||
|
self.inner.borrow_mut()
|
||||||
|
}
|
||||||
|
}
|
Loading…
Reference in New Issue