1
0
mirror of https://github.com/vlang/v.git synced 2023-08-10 21:13:21 +03:00
v/vlib/sync/waitgroup.v

39 lines
779 B
V
Raw Normal View History

2019-07-30 16:06:16 +03:00
// Copyright (c) 2019 Alexander Medvednikov. All rights reserved.
// Use of this source code is governed by an MIT license
// that can be found in the LICENSE file.
module sync
2019-10-25 17:24:40 +03:00
//[init_with=new_waitgroup] // TODO: implement support for init_with struct attribute, and disallow WaitGroup{} from outside the sync.new_waitgroup() function.
pub struct WaitGroup {
2019-07-30 16:06:16 +03:00
mut:
mu Mutex
active int
2019-07-30 16:06:16 +03:00
}
2019-10-25 17:24:40 +03:00
pub fn new_waitgroup() WaitGroup {
mut w := WaitGroup{}
w.mu = sync.new_mutex()
return w
}
2019-07-30 16:06:16 +03:00
pub fn (wg mut WaitGroup) add(delta int) {
wg.mu.lock()
wg.active += delta
wg.mu.unlock()
if wg.active < 0 {
panic('Negative number of jobs in waitgroup')
}
2019-07-30 16:06:16 +03:00
}
pub fn (wg mut WaitGroup) done() {
wg.add(-1)
2019-07-30 16:06:16 +03:00
}
pub fn (wg mut WaitGroup) wait() {
for wg.active > 0 {
// waiting
}
2019-07-30 16:06:16 +03:00
}