Make sure WaitGroup updates and checks atomically

CS4215-OOGA · Apr 20, 2024 · 97dd1f0 · 97dd1f0
1 parent d30edb5
commit 97dd1f0
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/std/ooga-std.ooga b/std/ooga-std.ooga
@@ -23,17 +23,25 @@ func (s *Sync) NewWaitGroup(counter int) WaitGroup {
 }
 
 func (wg *WaitGroup) Add(delta int) {
+    startAtomic();
     wg.counter += delta
+    endAtomic();
 }
 
 func (wg *WaitGroup) Done() {
+    // update expressions are atomic
     wg.counter--
 }
 
 func (wg *WaitGroup) Wait() {
-    for wg.counter > 0 {
+    for {
+        startAtomic();
+        if wg.counter <= 0 {
+            endAtomic();
+            break;
+        }
+        endAtomic();
         yieldThread(); // yield to avoid wasteful loop
-        "YIELD";
     }
 }