use crate::sync::{AtomicUsize, Backoff, fence};
use core::{mem, sync::atomic::Ordering};
#[derive(Debug)]
pub(crate) struct SeqLock {
state: AtomicUsize,
}
impl SeqLock {
#[inline]
pub(crate) const fn new() -> Self {
Self { state: AtomicUsize::new(0) }
}
#[inline]
pub(crate) fn optimistic_read(&self) -> Option<usize> {
let state = self.state.load(Ordering::Acquire);
if state == 1 { None } else { Some(state) }
}
#[inline]
pub(crate) fn validate_read(&self, stamp: usize) -> bool {
fence(Ordering::Acquire);
self.state.load(Ordering::Relaxed) == stamp
}
#[inline]
pub(crate) fn write(&'static self) -> SeqLockWriteGuard {
let backoff = Backoff::new();
loop {
let previous = self.state.swap(1, Ordering::Acquire);
if previous != 1 {
fence(Ordering::Release);
return SeqLockWriteGuard { lock: self, state: previous };
}
backoff.snooze();
}
}
}
pub(crate) struct SeqLockWriteGuard {
lock: &'static SeqLock,
state: usize,
}
impl SeqLockWriteGuard {
pub(crate) fn abort(self) {
self.lock.state.store(self.state, Ordering::Release);
#[expect(clippy::mem_forget, reason = "avoids incrementing the stamp")]
mem::forget(self);
}
}
impl Drop for SeqLockWriteGuard {
fn drop(&mut self) {
self.lock.state.store(self.state.wrapping_add(2), Ordering::Release);
}
}