v/vlib/sync/many_times.v

37 lines
589 B
V

module sync
import sync.atomic2
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 atomic2.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 {
atomic2.store_u64(&m.count, m.count + 1)
f()
}
m.m.unlock()
}