TheCloud/v
1
0
Fork 0
mirror of https://github.com/vlang/v.git synced 2025-08-03 17:57:59 -04:00
Code Issues Packages Projects Releases Wiki Activity

sync.stdatomic: add atomic types (#24561)

Browse Source
This commit is contained in:
kbkpbot 2025-05-25 18:10:08 +08:00 committed by GitHub
parent f19e8a7fa3
commit ed7a8a34f1
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
4 changed files with 398 additions and 37 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

190
thirdparty/stdatomic/win/atomic.h vendored
View File

@ -40,7 +40,9 @@
#define kill_dependency(y) ((void)0)
#define atomic_thread_fence(order) \
MemoryBarrier();
((order) == memory_order_seq_cst ? MemoryBarrier() : \
(order) == memory_order_release ? WriteBarrier() : \
(order) == memory_order_acquire ? ReadBarrier() : (void)0);
#define atomic_signal_fence(order) \
((void)0)
@ -48,18 +50,18 @@
#define atomic_is_lock_free(obj) 0
typedef intptr_t atomic_flag;
typedef intptr_t atomic_bool;
typedef intptr_t atomic_char;
typedef intptr_t atomic_schar;
typedef intptr_t atomic_uchar;
typedef intptr_t atomic_short;
typedef intptr_t atomic_ushort;
typedef intptr_t atomic_int;
typedef intptr_t atomic_uint;
typedef intptr_t atomic_long;
typedef intptr_t atomic_ulong;
typedef intptr_t atomic_llong;
typedef intptr_t atomic_ullong;
typedef _Atomic bool atomic_bool;
typedef _Atomic char atomic_char;
typedef _Atomic signed char atomic_schar;
typedef _Atomic unsigned char atomic_uchar;
typedef _Atomic short atomic_short;
typedef _Atomic unsigned short atomic_ushort;
typedef _Atomic int atomic_int;
typedef _Atomic unsigned int atomic_uint;
typedef _Atomic long atomic_long;
typedef _Atomic unsigned long atomic_ulong;
typedef _Atomic long long atomic_llong;
typedef _Atomic unsigned long long atomic_ullong;
typedef intptr_t atomic_wchar_t;
typedef intptr_t atomic_int_least8_t;
typedef intptr_t atomic_uint_least8_t;
@ -77,9 +79,9 @@ typedef intptr_t atomic_int_fast32_t;
typedef intptr_t atomic_uint_fast32_t;
typedef intptr_t atomic_int_fast64_t;
typedef intptr_t atomic_uint_fast64_t;
typedef intptr_t atomic_intptr_t;
typedef intptr_t atomic_uintptr_t;
typedef intptr_t atomic_size_t;
typedef _Atomic intptr_t atomic_intptr_t;
typedef _Atomic uintptr_t atomic_uintptr_t;
typedef _Atomic size_t atomic_size_t;
typedef intptr_t atomic_ptrdiff_t;
typedef intptr_t atomic_intmax_t;
typedef intptr_t atomic_uintmax_t;
@ -90,6 +92,7 @@ typedef intptr_t atomic_uintmax_t;
fake it (works the same) with InterlockedCompareExchange64 until it
succeeds
*/
__CRT_INLINE LONGLONG _InterlockedExchangeAdd64(LONGLONG volatile *Addend, LONGLONG Value)
{
LONGLONG Old;
@ -110,16 +113,6 @@ __CRT_INLINE LONG _InterlockedExchangeAdd(LONG volatile *Addend, LONG Value)
return Old;
}
__CRT_INLINE SHORT _InterlockedExchangeAdd16(SHORT volatile *Addend, SHORT Value)
{
SHORT Old;
do
{
Old = *Addend;
} while (InterlockedCompareExchange16(Addend, Old + Value, Old) != Old);
return Old;
}
#define InterlockedIncrement64 _InterlockedExchangeAdd64
#endif
@ -127,6 +120,7 @@ __CRT_INLINE SHORT _InterlockedExchangeAdd16(SHORT volatile *Addend, SHORT Value
#define atomic_store(object, desired) \
do \
{ \
MemoryBarrier(); \
*(object) = (desired); \
MemoryBarrier(); \
} while (0)
@ -146,7 +140,7 @@ __CRT_INLINE SHORT _InterlockedExchangeAdd16(SHORT volatile *Addend, SHORT Value
#define atomic_exchange_explicit(object, desired, order) \
atomic_exchange(object, desired)
static inline int atomic_compare_exchange_strong(intptr_t *object, intptr_t *expected,
static inline int atomic_compare_exchange_strong(intptr_t volatile *object, intptr_t volatile *expected,
intptr_t desired)
{
intptr_t old = *expected;
@ -210,21 +204,22 @@ static inline int atomic_compare_exchange_strong(intptr_t *object, intptr_t *exp
#define atomic_fetch_or_ptr atomic_fetch_or
#define atomic_fetch_xor_ptr atomic_fetch_xor
static inline void atomic_store_u64(unsigned long long* object, unsigned long long desired) {
static inline void atomic_store_u64(unsigned long long volatile * object, unsigned long long desired) {
do {
MemoryBarrier();
*(object) = (desired);
MemoryBarrier();
} while (0);
}
static inline unsigned long long atomic_load_u64(unsigned long long* object) {
static inline unsigned long long atomic_load_u64(unsigned long long volatile * object) {
return (MemoryBarrier(), *(object));
}
#define atomic_exchange_u64(object, desired) \
InterlockedExchange64(object, desired)
static inline int atomic_compare_exchange_strong_u64(unsigned long long* object, unsigned long long* expected,
static inline int atomic_compare_exchange_strong_u64(unsigned long long volatile * object, unsigned long long volatile * expected,
unsigned long long desired)
{
unsigned long long old = *expected;
@ -252,21 +247,22 @@ static inline int atomic_compare_exchange_strong_u64(unsigned long long* object,
static inline void atomic_store_u32(unsigned* object, unsigned desired) {
static inline void atomic_store_u32(unsigned volatile * object, unsigned desired) {
do {
MemoryBarrier();
*(object) = (desired);
MemoryBarrier();
} while (0);
}
static inline unsigned atomic_load_u32(unsigned* object) {
static inline unsigned atomic_load_u32(unsigned volatile * object) {
return (MemoryBarrier(), *(object));
}
#define atomic_exchange_u32(object, desired) \
InterlockedExchange(object, desired)
static inline int atomic_compare_exchange_strong_u32(unsigned* object, unsigned* expected,
static inline int atomic_compare_exchange_strong_u32(unsigned volatile * object, unsigned volatile * expected,
unsigned desired)
{
unsigned old = *expected;
@ -292,23 +288,40 @@ static inline int atomic_compare_exchange_strong_u32(unsigned* object, unsigned*
#define atomic_fetch_and_u32(object, operand) \
InterlockedAnd(object, operand)
#ifdef _MSC_VER
#define InterlockedExchangeAdd16 _InterlockedExchangeAdd16
static inline void atomic_store_u16(unsigned short* object, unsigned short desired) {
#else
#define InterlockedExchangeAdd16 ManualInterlockedExchangeAdd16
static inline unsigned short ManualInterlockedExchangeAdd16(unsigned short volatile* Addend, unsigned short Value) {
__asm__ __volatile__ (
"lock xaddw %w[value], %[mem]"
: [mem] "+m" (*Addend), [value] "+r" (Value)
: : "memory"
);
return Value;
}
#endif
static inline void atomic_store_u16(unsigned short volatile * object, unsigned short desired) {
do {
MemoryBarrier();
*(object) = (desired);
MemoryBarrier();
} while (0);
}
static inline unsigned short atomic_load_u16(unsigned short* object) {
static inline unsigned short atomic_load_u16(unsigned short volatile * object) {
return (MemoryBarrier(), *(object));
}
#define atomic_exchange_u16(object, desired) \
InterlockedExchange16(object, desired)
static inline int atomic_compare_exchange_strong_u16(unsigned short* object, unsigned short* expected,
static inline int atomic_compare_exchange_strong_u16(unsigned short volatile * object, unsigned short volatile * expected,
unsigned short desired)
{
unsigned short old = *expected;
@ -335,7 +348,6 @@ static inline int atomic_compare_exchange_strong_u16(unsigned short* object, uns
InterlockedAnd16(object, operand)
#define atomic_fetch_add_explicit(object, operand, order) \
atomic_fetch_add(object, operand)
@ -363,4 +375,108 @@ static inline int atomic_compare_exchange_strong_u16(unsigned short* object, uns
#define atomic_flag_clear_explicit(object, order) \
atomic_flag_clear(object)
#ifdef _MSC_VER
#define InterlockedCompareExchange8 _InterlockedCompareExchange8
#define InterlockedExchangeAdd8 _InterlockedExchangeAdd8
#define InterlockedOr8 _InterlockedOr8
#define InterlockedXor8 _InterlockedXor8
#define InterlockedAnd8 _InterlockedAnd8
#else
#define InterlockedCompareExchange8 ManualInterlockedCompareExchange8
#define InterlockedExchangeAdd8 ManualInterlockedExchangeAdd8
#define InterlockedOr8 ManualInterlockedOr8
#define InterlockedXor8 ManualInterlockedXor8
#define InterlockedAnd8 ManualInterlockedAnd8
static inline unsigned char ManualInterlockedCompareExchange8(unsigned char volatile * dest, unsigned char exchange, unsigned char comparand) {
unsigned char result;
__asm__ volatile (
"lock cmpxchgb %[exchange], %[dest]"
: "=a" (result), [dest] "+m" (*dest)
: [exchange] "q" (exchange), "a" (comparand)
: "memory"
);
return result;
}
static inline unsigned char ManualInterlockedExchangeAdd8(unsigned char volatile * dest, unsigned char value) {
unsigned char oldValue;
unsigned char newValue;
do {
oldValue = *dest;
} while (ManualInterlockedCompareExchange8(dest, oldValue + value, oldValue) != oldValue);
return oldValue;
}
static inline unsigned char ManualInterlockedOr8(unsigned char volatile * dest, unsigned char value) {
unsigned char oldValue;
do {
oldValue = *dest;
} while (ManualInterlockedCompareExchange8(dest, oldValue | value, oldValue) != oldValue);
return oldValue;
}
static inline unsigned char ManualInterlockedXor8(unsigned char volatile * dest, unsigned char value) {
unsigned char oldValue;
do {
oldValue = *dest;
} while (ManualInterlockedCompareExchange8(dest, oldValue ^ value, oldValue) != oldValue);
return oldValue;
}
static inline unsigned char ManualInterlockedAnd8(unsigned char volatile * dest, unsigned char value) {
unsigned char oldValue;
do {
oldValue = *dest;
} while (ManualInterlockedCompareExchange8(dest, oldValue & value, oldValue) != oldValue);
return oldValue;
}
#endif
static inline void atomic_store_byte(unsigned char volatile * object, unsigned char desired) {
do {
MemoryBarrier();
*(object) = (desired);
MemoryBarrier();
} while (0);
}
static inline unsigned char atomic_load_byte(unsigned char volatile * object) {
return (MemoryBarrier(), *(object));
}
#define atomic_exchange_byte(object, desired) \
InterlockedExchange8(object, desired)
static inline int atomic_compare_exchange_strong_byte(unsigned char volatile * object, unsigned char volatile * expected,
unsigned char desired)
{
unsigned char old = *expected;
*expected = InterlockedCompareExchange8(object, desired, old);
return *expected == old;
}
#define atomic_compare_exchange_weak_byte(object, expected, desired) \
atomic_compare_exchange_strong_byte(object, expected, desired)
#define atomic_fetch_add_byte(object, operand) \
InterlockedExchangeAdd8(object, operand)
#define atomic_fetch_sub_byte(object, operand) \
InterlockedExchangeAdd8(object, -(operand))
#define atomic_fetch_or_byte(object, operand) \
InterlockedOr8(object, operand)
#define atomic_fetch_xor_byte(object, operand) \
InterlockedXor8(object, operand)
#define atomic_fetch_and_byte(object, operand) \
InterlockedAnd8(object, operand)
#endif /* COMPAT_ATOMICS_WIN32_STDATOMIC_H */

8
vlib/sync/stdatomic/1.declarations.c.v
View File

@ -77,6 +77,14 @@ fn C.atomic_exchange_ptr(voidptr, voidptr) voidptr
fn C.atomic_fetch_add_ptr(voidptr, voidptr) voidptr
fn C.atomic_fetch_sub_ptr(voidptr, voidptr) voidptr
fn C.atomic_load_byte(voidptr) u8
fn C.atomic_store_byte(voidptr, u8)
fn C.atomic_compare_exchange_weak_byte(voidptr, voidptr, u8) bool
fn C.atomic_compare_exchange_strong_byte(voidptr, voidptr, u8) bool
fn C.atomic_exchange_byte(voidptr, u8) u8
fn C.atomic_fetch_add_byte(voidptr, u8) u8
fn C.atomic_fetch_sub_byte(voidptr, u8) u8
fn C.atomic_load_u16(voidptr) u16
fn C.atomic_store_u16(voidptr, u16)
fn C.atomic_compare_exchange_weak_u16(voidptr, voidptr, u16) bool

141
vlib/sync/stdatomic/atomic.c.v
View File

@ -60,3 +60,144 @@ pub fn store_i64(ptr &i64, val i64) {
pub fn load_i64(ptr &i64) i64 {
return i64(C.atomic_load_u64(voidptr(ptr)))
}
@[heap]
pub struct AtomicVal[T] {
val T
}
// new_atomic creates a new atomic value of `T` type
@[inline]
pub fn new_atomic[T](val T) &AtomicVal[T] {
$if T is $int || T is bool {
return &AtomicVal[T]{
val: val
}
} $else {
panic('atomic: only support number and bool types')
}
return unsafe { nil }
}
// load returns current value of the atomic value
@[inline]
pub fn (mut a AtomicVal[T]) load() T {
$if T is bool {
return C.atomic_load_byte(voidptr(&a.val)) != 0
} $else $if T is u8 || T is i8 {
return T(C.atomic_load_byte(voidptr(&a.val)))
} $else $if T is u16 || T is i16 {
return T(C.atomic_load_u16(voidptr(&a.val)))
} $else $if T is u32 || T is i32 {
return T(C.atomic_load_u32(voidptr(&a.val)))
} $else $if T is u64 || T is i64 {
return T(C.atomic_load_u64(voidptr(&a.val)))
} $else $if T is int {
// TODO: remove this test or a compile time support $if sizeof() ==
if sizeof(int) == 4 {
return int(C.atomic_load_u32(voidptr(&a.val)))
} else {
return int(C.atomic_load_u64(voidptr(&a.val)))
}
} $else $if T is isize || T is usize {
// TODO: remove this test or a compile time support $if sizeof() ==
if sizeof(isize) == 4 {
return T(C.atomic_load_u32(voidptr(&a.val)))
} else {
return T(C.atomic_load_u64(voidptr(&a.val)))
}
}
return a.val
}
// store updates the atomic value with `val`
@[inline]
pub fn (mut a AtomicVal[T]) store(val T) {
$if T is bool {
C.atomic_store_byte(voidptr(&a.val), val)
} $else $if T is u8 || T is i8 {
C.atomic_store_byte(voidptr(&a.val), val)
} $else $if T is u16 || T is i16 {
C.atomic_store_u16(voidptr(&a.val), val)
} $else $if T is u32 || T is i32 {
C.atomic_store_u32(voidptr(&a.val), val)
} $else $if T is u64 || T is i64 {
C.atomic_store_u64(voidptr(&a.val), val)
} $else $if T is int {
// TODO: remove this test or a compile time support $if sizeof() ==
if sizeof(int) == 4 {
C.atomic_store_u32(voidptr(&a.val), val)
} else {
C.atomic_store_u64(voidptr(&a.val), val)
}
} $else $if T is isize || T is usize {
// TODO: remove this test or a compile time support $if sizeof() ==
if sizeof(isize) == 4 {
C.atomic_store_u32(voidptr(&a.val), val)
} else {
C.atomic_store_u64(voidptr(&a.val), val)
}
}
}
// add adds the atomic value with `delta`
@[inline]
pub fn (mut a AtomicVal[T]) add(delta T) T {
$if T is bool {
panic('atomic: can not add() a bool type')
} $else $if T is u8 || T is i8 {
C.atomic_fetch_add_byte(voidptr(&a.val), delta)
} $else $if T is u16 || T is i16 {
C.atomic_fetch_add_u16(voidptr(&a.val), delta)
} $else $if T is u32 || T is i32 {
C.atomic_fetch_add_u32(voidptr(&a.val), delta)
} $else $if T is u64 || T is i64 {
C.atomic_fetch_add_u64(voidptr(&a.val), delta)
} $else $if T is int {
// TODO: remove this test or a compile time support $if sizeof() ==
if sizeof(int) == 4 {
C.atomic_fetch_add_u32(voidptr(&a.val), delta)
} else {
C.atomic_fetch_add_u64(voidptr(&a.val), delta)
}
} $else $if T is isize || T is usize {
// TODO: remove this test or a compile time support $if sizeof() ==
if sizeof(isize) == 4 {
C.atomic_fetch_add_u32(voidptr(&a.val), delta)
} else {
C.atomic_fetch_add_u64(voidptr(&a.val), delta)
}
}
return T(a.val)
}
// sub subs the atomic value with `delta`
@[inline]
pub fn (mut a AtomicVal[T]) sub(delta T) T {
$if T is bool {
panic('atomic: can not sub() a bool type')
} $else $if T is u8 || T is i8 {
C.atomic_fetch_sub_byte(voidptr(&a.val), delta)
} $else $if T is u16 || T is i16 {
C.atomic_fetch_sub_u16(voidptr(&a.val), delta)
} $else $if T is u32 || T is i32 {
C.atomic_fetch_sub_u32(voidptr(&a.val), delta)
} $else $if T is u64 || T is i64 {
C.atomic_fetch_sub_u64(voidptr(&a.val), delta)
} $else $if T is int {
// TODO: remove this test or a compile time support $if sizeof() ==
if sizeof(int) == 4 {
C.atomic_fetch_sub_u32(voidptr(&a.val), delta)
} else {
C.atomic_fetch_sub_u64(voidptr(&a.val), delta)
}
} $else $if T is isize || T is usize {
// TODO: remove this test or a compile time support $if sizeof() ==
if sizeof(isize) == 4 {
C.atomic_fetch_sub_u32(voidptr(&a.val), delta)
} else {
C.atomic_fetch_sub_u64(voidptr(&a.val), delta)
}
}
return T(a.val)
}

96
vlib/sync/stdatomic/atomic_test.v
View File

@ -93,3 +93,99 @@ fn count_one_cycle_without_sync(mut counter Counter, mut group sync.WaitGroup) {
}
group.done()
}
fn test_atomic_vals() {
mut v_bool := stdatomic.new_atomic(false)
v_bool.store(true)
assert v_bool.load() == true
v_bool.store(false)
assert v_bool.load() == false
mut v_i8 := stdatomic.new_atomic(i8(-33))
v_i8.store(-34)
assert v_i8.load() == -34
v_i8.add(10)
assert v_i8.load() == -24
v_i8.sub(7)
assert v_i8.load() == -31
mut v_u8 := stdatomic.new_atomic(u8(33))
v_u8.store(34)
assert v_u8.load() == 34
v_u8.add(10)
assert v_u8.load() == 44
v_u8.sub(7)
assert v_u8.load() == 37
mut v_i16 := stdatomic.new_atomic(i16(-333))
v_i16.store(-334)
assert v_i16.load() == -334
v_i16.add(10)
assert v_i16.load() == -324
v_i16.sub(7)
assert v_i16.load() == -331
mut v_u16 := stdatomic.new_atomic(u16(333))
v_u16.store(334)
assert v_u16.load() == 334
v_u16.add(10)
assert v_u16.load() == 344
v_u16.sub(7)
assert v_u16.load() == 337
mut v_i32 := stdatomic.new_atomic(i32(-3333))
v_i32.store(-3334)
assert v_i32.load() == -3334
v_i32.add(10)
assert v_i32.load() == -3324
v_i32.sub(7)
assert v_i32.load() == -3331
mut v_u32 := stdatomic.new_atomic(u32(3333))
v_u32.store(3334)
assert v_u32.load() == 3334
v_u32.add(10)
assert v_u32.load() == 3344
v_u32.sub(7)
assert v_u32.load() == 3337
mut v_i64 := stdatomic.new_atomic(i64(-33333))
v_i64.store(-33334)
assert v_i64.load() == -33334
v_i64.add(10)
assert v_i64.load() == -33324
v_i64.sub(7)
assert v_i64.load() == -33331
mut v_u64 := stdatomic.new_atomic(u64(33333))
v_u64.store(33334)
assert v_u64.load() == 33334
v_u64.add(10)
assert v_u64.load() == 33344
v_u64.sub(7)
assert v_u64.load() == 33337
mut v_int := stdatomic.new_atomic(int(-44))
v_int.store(-45)
assert v_int.load() == -45
v_int.add(10)
assert v_int.load() == -35
v_int.sub(7)
assert v_int.load() == -42
mut v_isize := stdatomic.new_atomic(isize(-55))
v_isize.store(-56)
assert v_isize.load() == -56
v_isize.add(10)
assert v_isize.load() == -46
v_isize.sub(7)
assert v_isize.load() == -53
mut v_usize := stdatomic.new_atomic(usize(55))
v_usize.store(56)
assert v_usize.load() == 56
v_usize.add(10)
assert v_usize.load() == 66
v_usize.sub(7)
assert v_usize.load() == 59
}