src/WaitGroup.zig


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34

const std = @import("std");
const WaitGroup = @This();

lock: std.Mutex = .{},
counter: usize = 0,
event: std.AutoResetEvent = .{},

pub fn start(self: *WaitGroup) void {
    const held = self.lock.acquire();
    defer held.release();

    self.counter += 1;
}

pub fn stop(self: *WaitGroup) void {
    const held = self.lock.acquire();
    defer held.release();

    self.counter -= 1;
    if (self.counter == 0)
        self.event.set();
}

pub fn wait(self: *WaitGroup) void {
    {
        const held = self.lock.acquire();
        defer held.release();

        if (self.counter == 0)
            return;
    }

    self.event.wait();
}