37 lines
589 B
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()
|
||
|
|
}
|