sync/atomic2: support 32 bit platforms (#6308)
							parent
							
								
									3410705974
								
							
						
					
					
						commit
						4d2c8a40d6
					
				|  | @ -26,37 +26,48 @@ $if linux { | |||
| 
 | ||||
| #include <atomic.h> | ||||
| 
 | ||||
| fn C.atomic_fetch_add_explicit(voidptr, i64) i64 | ||||
| fn C.atomic_fetch_sub_explicit(voidptr, i64) i64 | ||||
| 
 | ||||
| [typedef] | ||||
| struct C.atomic_ullong { | ||||
| } | ||||
| 
 | ||||
| [typedef] | ||||
| struct C.atomic_llong { | ||||
| } | ||||
| 
 | ||||
| // add_u64 adds provided delta as an atomic operation
 | ||||
| pub fn add_u64(ptr &u64, delta int) bool { | ||||
| 	res := C.atomic_fetch_add_explicit(&C.atomic_ullong(ptr), delta, C.NULL) | ||||
| 	res := C.atomic_fetch_add_u64(ptr, delta) | ||||
| 	return res == 0 | ||||
| } | ||||
| 
 | ||||
| // sub_u64 subtracts provided delta as an atomic operation
 | ||||
| pub fn sub_u64(ptr &u64, delta int) bool { | ||||
| 	res := C.atomic_fetch_sub_explicit(&C.atomic_ullong(ptr), delta, C.NULL) | ||||
| 	res := C.atomic_fetch_sub_u64(ptr, delta) | ||||
| 	return res == 0 | ||||
| } | ||||
| 
 | ||||
| // add_i64 adds provided delta as an atomic operation
 | ||||
| pub fn add_i64(ptr &i64, delta int) bool { | ||||
| 	res := C.atomic_fetch_add_explicit(&C.atomic_llong(ptr), delta, C.NULL) | ||||
| 	res := C.atomic_fetch_add_u64(ptr, delta) | ||||
| 	return res == 0 | ||||
| } | ||||
| 
 | ||||
| // add_i64 subtracts provided delta as an atomic operation
 | ||||
| pub fn sub_i64(ptr &i64, delta int) bool { | ||||
| 	res := C.atomic_fetch_sub_explicit(&C.atomic_llong(ptr), delta, C.NULL) | ||||
| 	res := C.atomic_fetch_sub_u64(ptr, delta) | ||||
| 	return res == 0 | ||||
| } | ||||
| 
 | ||||
| // atomic store/load operations have to be used when there might be another concurrent access
 | ||||
| 
 | ||||
| // atomicall set a value
 | ||||
| pub fn store_u64(ptr &u64, val u64) { | ||||
| 	C.atomic_store_u64(ptr, val) | ||||
| } | ||||
| 
 | ||||
| // atomicall get a value
 | ||||
| pub fn load_u64(ptr &u64) u64 { | ||||
| 	return C.atomic_load_u64(ptr) | ||||
| } | ||||
| 
 | ||||
| // atomicall set a value
 | ||||
| pub fn store_i64(ptr &i64, val i64) { | ||||
| 	C.atomic_store_u64(ptr, val) | ||||
| } | ||||
| 
 | ||||
| // atomicall get a value
 | ||||
| pub fn load_i64(ptr &i64) i64 { | ||||
| 	return i64(C.atomic_load_u64(ptr)) | ||||
| } | ||||
|  |  | |||
|  | @ -41,49 +41,51 @@ fn test_count_10_times_1_cycle_should_not_be_10_cycles_without_sync() { | |||
| fn test_count_plus_one_u64() { | ||||
| 	mut c := u64(0) | ||||
| 	atomic2.add_u64(&c, 1) | ||||
| 	assert c == 1 | ||||
| 	assert atomic2.load_u64(&c) == 1 | ||||
| } | ||||
| 
 | ||||
| fn test_count_plus_one_i64() { | ||||
| 	mut c := i64(0) | ||||
| 	atomic2.add_i64(&c, 1) | ||||
| 	assert c == 1 | ||||
| 	assert atomic2.load_i64(&c) == 1 | ||||
| } | ||||
| 
 | ||||
| fn test_count_plus_greater_than_one_u64() { | ||||
| 	mut c := u64(0) | ||||
| 	atomic2.add_u64(&c, 10) | ||||
| 	assert c == 10 | ||||
| 	assert atomic2.load_u64(&c) == 10 | ||||
| } | ||||
| 
 | ||||
| fn test_count_plus_greater_than_one_i64() { | ||||
| 	mut c := i64(0) | ||||
| 	atomic2.add_i64(&c, 10) | ||||
| 	assert c == 10 | ||||
| 	assert atomic2.load_i64(&c) == 10 | ||||
| } | ||||
| 
 | ||||
| fn test_count_minus_one_u64() { | ||||
| 	mut c := u64(1) | ||||
| 	atomic2.sub_u64(&c, 1) | ||||
| 	assert c == 0 | ||||
| 	assert atomic2.load_u64(&c) == 0 | ||||
| } | ||||
| 
 | ||||
| fn test_count_minus_one_i64() { | ||||
| 	mut c := i64(0) | ||||
| 	atomic2.sub_i64(&c, 1) | ||||
| 	assert c == -1 | ||||
| 	assert atomic2.load_i64(&c) == -1 | ||||
| } | ||||
| 
 | ||||
| fn test_count_minus_greater_than_one_u64() { | ||||
| 	mut c := u64(10) | ||||
| 	mut c := u64(0) | ||||
| 	atomic2.store_u64(&c, 10) | ||||
| 	atomic2.sub_u64(&c, 10) | ||||
| 	assert c == 0 | ||||
| 	assert atomic2.load_u64(&c) == 0 | ||||
| } | ||||
| 
 | ||||
| fn test_count_minus_greater_than_one_i64() { | ||||
| 	mut c := i64(10) | ||||
| 	mut c := i64(0) | ||||
| 	atomic2.store_i64(&c, 10) | ||||
| 	atomic2.sub_i64(&c, 20) | ||||
| 	assert c == -10 | ||||
| 	assert atomic2.load_i64(&c) == -10 | ||||
| } | ||||
| 
 | ||||
| // count_one_cycle counts the common counter iterations_per_cycle times in thread-safe way
 | ||||
|  |  | |||
		Loading…
	
		Reference in New Issue