1
0
mirror of https://github.com/vlang/v.git synced 2023-08-10 21:13:21 +03:00
v/vlib/sync/many_times.v
2021-12-28 10:16:22 +02:00

37 lines
595 B
V

module sync
import sync.stdatomic
pub struct ManyTimes {
mut:
m RwMutex
pub:
times u64 = 1
count u64
}
// new_many_times return a new ManyTimes struct.
pub fn new_many_times(times u64) &ManyTimes {
mut many_times := &ManyTimes{
times: times
}
many_times.m.init()
return many_times
}
// do execute the function only setting times.
pub fn (mut m ManyTimes) do(f fn ()) {
if stdatomic.load_u64(&m.count) < m.times {
m.do_slow(f)
}
}
fn (mut m ManyTimes) do_slow(f fn ()) {
m.m.@lock()
if m.count < m.times {
stdatomic.store_u64(&m.count, m.count + 1)
f()
}
m.m.unlock()
}