| 1 | #[cfg (not(all(test, loom)))] |
| 2 | pub(crate) mod sync { |
| 3 | pub(crate) mod atomic { |
| 4 | pub(crate) use core::sync::atomic::{AtomicPtr, AtomicUsize, Ordering}; |
| 5 | |
| 6 | pub(crate) trait AtomicMut<T> { |
| 7 | fn with_mut<F, R>(&mut self, f: F) -> R |
| 8 | where |
| 9 | F: FnOnce(&mut *mut T) -> R; |
| 10 | } |
| 11 | |
| 12 | impl<T> AtomicMut<T> for AtomicPtr<T> { |
| 13 | fn with_mut<F, R>(&mut self, f: F) -> R |
| 14 | where |
| 15 | F: FnOnce(&mut *mut T) -> R, |
| 16 | { |
| 17 | f(self.get_mut()) |
| 18 | } |
| 19 | } |
| 20 | } |
| 21 | } |
| 22 | |
| 23 | #[cfg (all(test, loom))] |
| 24 | pub(crate) mod sync { |
| 25 | pub(crate) mod atomic { |
| 26 | pub(crate) use loom::sync::atomic::{AtomicPtr, AtomicUsize, Ordering}; |
| 27 | |
| 28 | pub(crate) trait AtomicMut<T> {} |
| 29 | } |
| 30 | } |
| 31 | |