v/vlib/sync/sync_nix.c.v

155 lines
3.3 KiB
V
Raw Normal View History

2020-01-23 21:04:46 +01:00
// Copyright (c) 2019-2020 Alexander Medvednikov. All rights reserved.
2019-06-23 04:21:30 +02:00
// Use of this source code is governed by an MIT license
// that can be found in the LICENSE file.
2019-06-22 21:53:22 +02:00
module sync
2020-07-15 10:22:33 +02:00
import time
#flag -lpthread
$if macos {
#include <dispatch/dispatch.h>
} $else {
#include <semaphore.h>
}
2019-12-21 23:41:42 +01:00
// [init_with=new_mutex] // TODO: implement support for this struct attribute, and disallow Mutex{} from outside the sync.new_mutex() function.
[ref_only]
pub struct Mutex {
2019-06-22 21:53:22 +02:00
mutex C.pthread_mutex_t
}
[ref_only]
pub struct RwMutex {
mutex C.pthread_rwlock_t
}
[ref_only]
struct RwMutexAttr {
attr C.pthread_rwlockattr_t
}
2020-07-15 10:22:33 +02:00
/* MacOSX has no unnamed semaphores and no `timed_wait()` at all
so we emulate the behaviour with other devices */
// [ref_only]
// struct MacOSX_Semaphore {
// sem C.dispatch_semaphore_t
// }
2020-07-15 10:22:33 +02:00
[ref_only]
struct PosixSemaphore {
sem C.sem_t
}
pub struct Semaphore {
2020-08-06 15:28:19 +02:00
mut:
sem voidptr
2020-07-15 10:22:33 +02:00
}
pub fn new_mutex() &Mutex {
m := &Mutex{}
2019-12-21 23:41:42 +01:00
C.pthread_mutex_init(&m.mutex, C.NULL)
2019-10-25 16:24:40 +02:00
return m
}
pub fn new_rwmutex() &RwMutex {
m := &RwMutex{}
a := &RwMutexAttr{}
C.pthread_rwlockattr_init(&a.attr)
// Give writer priority over readers
C.pthread_rwlockattr_setkind_np(&a.attr, C.PTHREAD_RWLOCK_PREFER_WRITER_NONRECURSIVE_NP)
C.pthread_rwlockattr_setpshared(&a.attr, C.PTHREAD_PROCESS_PRIVATE)
C.pthread_rwlock_init(&m.mutex, &a.attr)
return m
}
// m_lock(), for *manual* mutex handling, since `lock` is a keyword
pub fn (mut m Mutex) m_lock() {
2019-06-22 21:53:22 +02:00
C.pthread_mutex_lock(&m.mutex)
}
2020-05-17 13:51:18 +02:00
pub fn (mut m Mutex) unlock() {
2019-06-22 21:53:22 +02:00
C.pthread_mutex_unlock(&m.mutex)
}
// RwMutex has separate read- and write locks
pub fn (mut m RwMutex) r_lock() {
C.pthread_rwlock_rdlock(&m.mutex)
}
pub fn (mut m RwMutex) w_lock() {
C.pthread_rwlock_wrlock(&m.mutex)
}
// Windows SRWLocks have different function to unlock
// So provide two functions here, too, to have a common interface
pub fn (mut m RwMutex) r_unlock() {
C.pthread_rwlock_unlock(&m.mutex)
}
pub fn (mut m RwMutex) w_unlock() {
C.pthread_rwlock_unlock(&m.mutex)
}
2020-07-15 10:22:33 +02:00
2020-08-06 15:28:19 +02:00
[inline]
2020-07-15 10:22:33 +02:00
pub fn new_semaphore() Semaphore {
2020-08-06 15:28:19 +02:00
return new_semaphore_init(0)
}
pub fn new_semaphore_init(n u32) Semaphore {
2020-07-15 10:22:33 +02:00
$if macos {
s := Semaphore{
sem: C.dispatch_semaphore_create(n)
2020-07-15 10:22:33 +02:00
}
return s
} $else {
s := Semaphore{
sem: &PosixSemaphore{}
}
2020-08-06 15:28:19 +02:00
unsafe { C.sem_init(&&PosixSemaphore(s.sem).sem, 0, n) }
2020-07-15 10:22:33 +02:00
return s
}
}
pub fn (s Semaphore) post() {
$if macos {
C.dispatch_semaphore_signal(s.sem)
2020-07-15 10:22:33 +02:00
} $else {
unsafe { C.sem_post(&&PosixSemaphore(s.sem).sem) }
2020-07-15 10:22:33 +02:00
}
}
pub fn (s Semaphore) wait() {
$if macos {
C.dispatch_semaphore_wait(s.sem, C.DISPATCH_TIME_FOREVER)
2020-07-15 10:22:33 +02:00
} $else {
unsafe { C.sem_wait(&&PosixSemaphore(s.sem).sem) }
2020-07-15 10:22:33 +02:00
}
}
pub fn (s Semaphore) try_wait() bool {
$if macos {
return C.dispatch_semaphore_wait(s.sem, C.DISPATCH_TIME_NOW) == 0
2020-07-15 10:22:33 +02:00
} $else {
return unsafe { C.sem_trywait(&&PosixSemaphore(s.sem).sem) == 0 }
2020-07-15 10:22:33 +02:00
}
}
pub fn (s Semaphore) timed_wait(timeout time.Duration) bool {
$if macos {
return C.dispatch_semaphore_wait(s.sem, C.dispatch_time(C.DISPATCH_TIME_NOW, i64(timeout))) == 0
2020-07-15 10:22:33 +02:00
} $else {
t_spec := timeout.timespec()
return unsafe { C.sem_timedwait(&&PosixSemaphore(s.sem).sem, &t_spec) == 0 }
2020-07-15 10:22:33 +02:00
}
}
2020-08-06 15:28:19 +02:00
pub fn (s Semaphore) destroy() bool {
$if macos {
for s.try_wait() {}
C.dispatch_release(s.sem)
return true
2020-08-06 15:28:19 +02:00
} $else {
return unsafe { C.sem_destroy(&&PosixSemaphore(s.sem).sem) == 0 }
}
}