| ... | ... | @@ -32,19 +32,21 @@ pub fn stop(self: *WaitGroup) void { |
| 32 | 32 | } |
| 33 | 33 | |
| 34 | 34 | pub fn wait(self: *WaitGroup) void { |
| 35 | | const held = self.lock.acquire(); |
| 35 | while (true) { |
| 36 | const held = self.lock.acquire(); |
| 36 | 37 | |
| 37 | | if (self.counter == 0) { |
| 38 | | held.release(); |
| 39 | | return; |
| 40 | | } |
| 38 | if (self.counter == 0) { |
| 39 | held.release(); |
| 40 | return; |
| 41 | } |
| 41 | 42 | |
| 42 | | var event = std.ResetEvent.init(); |
| 43 | | defer event.deinit(); |
| 43 | var event = std.ResetEvent.init(); |
| 44 | defer event.deinit(); |
| 44 | 45 | |
| 45 | | std.debug.assert(self.event == null); |
| 46 | | self.event = &event; |
| 46 | std.debug.assert(self.event == null); |
| 47 | self.event = &event; |
| 47 | 48 | |
| 48 | | held.release(); |
| 49 | | event.wait(); |
| 49 | held.release(); |
| 50 | event.wait(); |
| 51 | } |
| 50 | 52 | } |