32 lines
524 B
V
32 lines
524 B
V
// 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
|
|
|
|
pub struct WaitGroup {
|
|
mut:
|
|
mu Mutex
|
|
active int
|
|
}
|
|
|
|
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')
|
|
}
|
|
}
|
|
|
|
pub fn (wg mut WaitGroup) done() {
|
|
wg.add(-1)
|
|
}
|
|
|
|
pub fn (wg mut WaitGroup) wait() {
|
|
for wg.active > 0 {
|
|
// waiting
|
|
}
|
|
}
|
|
|