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