diff options
Diffstat (limited to 'src/lock.rs')
| -rw-r--r-- | src/lock.rs | 24 |
1 files changed, 24 insertions, 0 deletions
diff --git a/src/lock.rs b/src/lock.rs new file mode 100644 index 0000000..becdfaa --- /dev/null +++ b/src/lock.rs @@ -0,0 +1,24 @@ +use std::sync::atomic::{AtomicBool, Ordering}; + +#[derive(Debug, Default)] +pub struct Lock { + is_locked: AtomicBool, +} + +impl Lock { + pub const fn new() -> Self { + Self { + is_locked: AtomicBool::new(false), + } + } + + pub fn try_lock(&self) -> bool { + self.is_locked + .compare_exchange_weak(false, true, Ordering::Acquire, Ordering::Relaxed) + .is_ok() + } + + pub fn unlock(&self) { + self.is_locked.store(false, Ordering::Release) + } +} |
