weird things

This commit is contained in:
nora 2022-05-10 14:17:34 +02:00
parent 1a76116ed0
commit 2a458890bc
3 changed files with 43 additions and 7 deletions

View file

@ -1,2 +1,3 @@
mod linked_list;
mod mutex;
mod spsc;

View file

@ -13,7 +13,7 @@ struct Mutex<T> {
impl<T> Mutex<T> {
pub fn new(value: T) -> Self {
Self {
value,
value: UnsafeCell::new(value),
status: AtomicU8::new(INIT),
}
}
@ -22,6 +22,7 @@ impl<T> Mutex<T> {
if self
.status
.compare_exchange(INIT, ACQUIRED, Ordering::Acquire, Ordering::Relaxed)
.is_ok()
{
Some(MutexGuard { mutex: self })
} else {
@ -34,12 +35,11 @@ impl<T> Mutex<T> {
// don't use spin loops
// but I can't be bothered with the proper solution
loop {
if self.status.compare_exchange_weak(
INIT,
ACQUIRED,
Ordering::Acquire,
Ordering::Relaxed,
) {
if self
.status
.compare_exchange_weak(INIT, ACQUIRED, Ordering::Acquire, Ordering::Relaxed)
.is_ok()
{
return MutexGuard { mutex: self };
} else {
std::hint::spin_loop();

35
src/spsc.rs Normal file
View file

@ -0,0 +1,35 @@
use std::cell::UnsafeCell;
use std::mem::MaybeUninit;
use std::sync::atomic::{AtomicPtr, AtomicUsize, Ordering};
struct Spsc<const N: usize, T> {
start: AtomicUsize,
end: AtomicUsize,
buffer: [UnsafeCell<MaybeUninit<T>>; N],
}
impl<const N: usize, T> Spsc<N, T> {
pub fn new() -> Self {
Self {
start: AtomicUsize::default(),
end: AtomicUsize::default(),
buffer: [UnsafeCell::new(MaybeUninit::uninit()); N],
}
}
pub fn push(&self, value: T) {
let end = self.end.load(Ordering::Acquire);
let start = self.start.load(Ordering::Acquire);
if end != 0 && end == start {}w
if end < self.buffer.len() {
let end = unsafe { &self.buffer[end] };
unsafe { end.get().write(value) };
self.end.fetch_and(1, Ordering::Release);
}
}
}
unsafe impl<const N: usize, T> Sync for Spsc<N, T> {}
unsafe impl<const N: usize, T> Send for Spsc<N, T> {}