|
@@ -2,8 +2,6 @@
|
|
|
//+private
|
|
|
package sync2
|
|
|
|
|
|
-when #config(ODIN_SYNC_USE_PTHREADS, true) {
|
|
|
-
|
|
|
import "core:time"
|
|
|
import "core:runtime"
|
|
|
import "core:sys/unix"
|
|
@@ -32,142 +30,6 @@ _mutex_try_lock :: proc(m: ^Mutex) -> bool {
|
|
|
return err == 0
|
|
|
}
|
|
|
|
|
|
-
|
|
|
-
|
|
|
-RW_Mutex_State :: distinct uint
|
|
|
-RW_Mutex_State_Half_Width :: size_of(RW_Mutex_State)*8/2
|
|
|
-RW_Mutex_State_Is_Writing :: RW_Mutex_State(1)
|
|
|
-RW_Mutex_State_Writer :: RW_Mutex_State(1)<<1
|
|
|
-RW_Mutex_State_Reader :: RW_Mutex_State(1)<<RW_Mutex_State_Half_Width
|
|
|
-
|
|
|
-RW_Mutex_State_Writer_Mask :: RW_Mutex_State(1<<(RW_Mutex_State_Half_Width-1) - 1) << 1
|
|
|
-RW_Mutex_State_Reader_Mask :: RW_Mutex_State(1<<(RW_Mutex_State_Half_Width-1) - 1) << RW_Mutex_State_Half_Width
|
|
|
-
|
|
|
-
|
|
|
-_RW_Mutex :: struct {
|
|
|
- // NOTE(bill): pthread_rwlock_t cannot be used since pthread_rwlock_destroy is required on some platforms
|
|
|
- // TODO(bill): Can we determine which platforms exactly?
|
|
|
- state: RW_Mutex_State,
|
|
|
- mutex: Mutex,
|
|
|
- sema: Sema,
|
|
|
-}
|
|
|
-
|
|
|
-_rw_mutex_lock :: proc(rw: ^RW_Mutex) {
|
|
|
- _ = atomic_add(&rw.impl.state, RW_Mutex_State_Writer)
|
|
|
- mutex_lock(&rw.impl.mutex)
|
|
|
-
|
|
|
- state := atomic_or(&rw.impl.state, RW_Mutex_State_Writer)
|
|
|
- if state & RW_Mutex_State_Reader_Mask != 0 {
|
|
|
- sema_wait(&rw.impl.sema)
|
|
|
- }
|
|
|
-}
|
|
|
-
|
|
|
-_rw_mutex_unlock :: proc(rw: ^RW_Mutex) {
|
|
|
- _ = atomic_and(&rw.impl.state, ~RW_Mutex_State_Is_Writing)
|
|
|
- mutex_unlock(&rw.impl.mutex)
|
|
|
-}
|
|
|
-
|
|
|
-_rw_mutex_try_lock :: proc(rw: ^RW_Mutex) -> bool {
|
|
|
- if mutex_try_lock(&rw.impl.mutex) {
|
|
|
- state := atomic_load(&rw.impl.state)
|
|
|
- if state & RW_Mutex_State_Reader_Mask == 0 {
|
|
|
- _ = atomic_or(&rw.impl.state, RW_Mutex_State_Is_Writing)
|
|
|
- return true
|
|
|
- }
|
|
|
-
|
|
|
- mutex_unlock(&rw.impl.mutex)
|
|
|
- }
|
|
|
- return false
|
|
|
-}
|
|
|
-
|
|
|
-_rw_mutex_shared_lock :: proc(rw: ^RW_Mutex) {
|
|
|
- state := atomic_load(&rw.impl.state)
|
|
|
- for state & (RW_Mutex_State_Is_Writing|RW_Mutex_State_Writer_Mask) == 0 {
|
|
|
- ok: bool
|
|
|
- state, ok = atomic_compare_exchange_weak(&rw.impl.state, state, state + RW_Mutex_State_Reader)
|
|
|
- if ok {
|
|
|
- return
|
|
|
- }
|
|
|
- }
|
|
|
-
|
|
|
- mutex_lock(&rw.impl.mutex)
|
|
|
- _ = atomic_add(&rw.impl.state, RW_Mutex_State_Reader)
|
|
|
- mutex_unlock(&rw.impl.mutex)
|
|
|
-}
|
|
|
-
|
|
|
-_rw_mutex_shared_unlock :: proc(rw: ^RW_Mutex) {
|
|
|
- state := atomic_sub(&rw.impl.state, RW_Mutex_State_Reader)
|
|
|
-
|
|
|
- if (state & RW_Mutex_State_Reader_Mask == RW_Mutex_State_Reader) &&
|
|
|
- (state & RW_Mutex_State_Is_Writing != 0) {
|
|
|
- sema_post(&rw.impl.sema)
|
|
|
- }
|
|
|
-}
|
|
|
-
|
|
|
-_rw_mutex_try_shared_lock :: proc(rw: ^RW_Mutex) -> bool {
|
|
|
- state := atomic_load(&rw.impl.state)
|
|
|
- if state & (RW_Mutex_State_Is_Writing|RW_Mutex_State_Writer_Mask) == 0 {
|
|
|
- _, ok := atomic_compare_exchange_strong(&rw.impl.state, state, state + RW_Mutex_State_Reader)
|
|
|
- if ok {
|
|
|
- return true
|
|
|
- }
|
|
|
- }
|
|
|
- if mutex_try_lock(&rw.impl.mutex) {
|
|
|
- _ = atomic_add(&rw.impl.state, RW_Mutex_State_Reader)
|
|
|
- mutex_unlock(&rw.impl.mutex)
|
|
|
- return true
|
|
|
- }
|
|
|
-
|
|
|
- return false
|
|
|
-}
|
|
|
-
|
|
|
-
|
|
|
-_Recursive_Mutex :: struct {
|
|
|
- owner: int,
|
|
|
- recursion: int,
|
|
|
- mutex: Mutex,
|
|
|
-}
|
|
|
-
|
|
|
-_recursive_mutex_lock :: proc(m: ^Recursive_Mutex) {
|
|
|
- tid := _current_thread_id()
|
|
|
- if tid != m.impl.owner {
|
|
|
- mutex_lock(&m.impl.mutex)
|
|
|
- }
|
|
|
- // inside the lock
|
|
|
- m.impl.owner = tid
|
|
|
- m.impl.recursion += 1
|
|
|
-}
|
|
|
-
|
|
|
-_recursive_mutex_unlock :: proc(m: ^Recursive_Mutex) {
|
|
|
- tid := _current_thread_id()
|
|
|
- assert(tid == m.impl.owner)
|
|
|
- m.impl.recursion -= 1
|
|
|
- recursion := m.impl.recursion
|
|
|
- if recursion == 0 {
|
|
|
- m.impl.owner = 0
|
|
|
- }
|
|
|
- if recursion == 0 {
|
|
|
- mutex_unlock(&m.impl.mutex)
|
|
|
- }
|
|
|
- // outside the lock
|
|
|
-
|
|
|
-}
|
|
|
-
|
|
|
-_recursive_mutex_try_lock :: proc(m: ^Recursive_Mutex) -> bool {
|
|
|
- tid := _current_thread_id()
|
|
|
- if m.impl.owner == tid {
|
|
|
- return mutex_try_lock(&m.impl.mutex)
|
|
|
- }
|
|
|
- if !mutex_try_lock(&m.impl.mutex) {
|
|
|
- return false
|
|
|
- }
|
|
|
- // inside the lock
|
|
|
- m.impl.owner = tid
|
|
|
- m.impl.recursion += 1
|
|
|
- return true
|
|
|
-}
|
|
|
-
|
|
|
-
|
|
|
_Cond :: struct {
|
|
|
pthread_cond: unix.pthread_cond_t,
|
|
|
}
|
|
@@ -177,17 +39,6 @@ _cond_wait :: proc(c: ^Cond, m: ^Mutex) {
|
|
|
assert(err == 0)
|
|
|
}
|
|
|
|
|
|
-_cond_wait_with_timeout :: proc(c: ^Cond, m: ^Mutex, timeout: time.Duration) -> bool {
|
|
|
- ns := time.duration_nanoseconds(timeout)
|
|
|
- timeout_timespec := &time.TimeSpec{
|
|
|
- tv_sec = ns / 1e9,
|
|
|
- tv_nsec = ns % 1e9,
|
|
|
- }
|
|
|
- err := unix.pthread_cond_timedwait(&c.impl.pthread_cond, &m.impl.pthread_mutex, timeout_timespec)
|
|
|
- // TODO(bill):
|
|
|
- return err == 0
|
|
|
-}
|
|
|
-
|
|
|
_cond_signal :: proc(c: ^Cond) {
|
|
|
err := unix.pthread_cond_signal(&c.impl.pthread_cond)
|
|
|
assert(err == 0)
|
|
@@ -197,35 +48,3 @@ _cond_broadcast :: proc(c: ^Cond) {
|
|
|
err := unix.pthread_cond_broadcast(&c.impl.pthread_cond)
|
|
|
assert(err == 0)
|
|
|
}
|
|
|
-
|
|
|
-_Sema :: struct {
|
|
|
- mutex: Mutex,
|
|
|
- cond: Cond,
|
|
|
- count: int,
|
|
|
-}
|
|
|
-
|
|
|
-_sema_wait :: proc(s: ^Sema) {
|
|
|
- mutex_lock(&s.impl.mutex)
|
|
|
- defer mutex_unlock(&s.impl.mutex)
|
|
|
-
|
|
|
- for s.impl.count == 0 {
|
|
|
- cond_wait(&s.impl.cond, &s.impl.mutex)
|
|
|
- }
|
|
|
-
|
|
|
- s.impl.count -= 1
|
|
|
- if s.impl.count > 0 {
|
|
|
- cond_signal(&s.impl.cond)
|
|
|
- }
|
|
|
-}
|
|
|
-
|
|
|
-_sema_post :: proc(s: ^Sema, count := 1) {
|
|
|
- mutex_lock(&s.impl.mutex)
|
|
|
- defer mutex_unlock(&s.impl.mutex)
|
|
|
-
|
|
|
- s.impl.count += count
|
|
|
- cond_signal(&s.impl.cond)
|
|
|
-}
|
|
|
-
|
|
|
-
|
|
|
-
|
|
|
-} // ODIN_SYNC_USE_PTHREADS
|