Merge pull request #17 from CS4215-OOGA/optimize-waitgroup

Optimise waitgroup
CS4215-OOGA · Apr 15, 2024 · f0f5e69 · f0f5e69
2 parents cab1fda + fb04d89
commit f0f5e69
Show file tree

Hide file tree

Showing 4 changed files with 18 additions and 4 deletions.
diff --git a/src/parser/ooga.pegjs b/src/parser/ooga.pegjs
@@ -1005,7 +1005,6 @@ FunctionDeclaration
       };
     }
 
-// Arnav: Not sure what this is for!!
 FunctionExpression
   = FunctionToken __ id:(Identifier __)?
     "(" __ params:(FormalParameterList __)? ")" __
@@ -1325,4 +1324,4 @@ BreakpointStatement
     return {
       tag: "BreakpointStatement"
     };
-  }
+  }
diff --git a/src/vm/oogavm-machine.ts b/src/vm/oogavm-machine.ts
@@ -245,6 +245,9 @@ function timeoutThread() {
 // Handbook for developing new built-in functions
 // Builtins need to return heap addresses, if they dont return any
 // useful value, return either heap.True or heap.Undefined
+
+// Used to either block the thread or voluntarily yield thread without signalling blocked
+let blockedThreadState = false;
 let yieldThreadState = false;
 
 export const builtinMappings = {
@@ -277,6 +280,10 @@ export const builtinMappings = {
     getThreadID: () => {
         return TSValueToAddress(currentThreadId);
     },
+    blockThread: () => {
+        blockedThreadState = true;
+        return null;
+    },
     yieldThread: () => {
         yieldThreadState = true;
         return null;
@@ -1207,9 +1214,15 @@ function runInstruction() {
     const instr = instrs[PC++];
     log(instr);
     microcode[instr.tag](instr);
+
+    if (blockedThreadState) {
+        blockedThreadState = false;
+        blockThread();
+    }
+
     if (yieldThreadState) {
         yieldThreadState = false;
-        blockThread();
+        timeoutThread();
     }
 
     if (!isAtomicSection) {

diff --git a/src/vm/oogavm-typechecker.ts b/src/vm/oogavm-typechecker.ts
@@ -117,6 +117,7 @@ const global_type_frame = {
     startAtomic: new FunctionType([], new NullType()),
     endAtomic: new FunctionType([], new NullType()),
     yieldThread: new FunctionType([], new NullType()),
+    blockThread: new FunctionType([], new NullType()),
     getThreadID: new FunctionType([], new IntegerType()),
     oogaError: new FunctionType([], new NullType()),
 };

diff --git a/std/ooga-std.ooga b/std/ooga-std.ooga
@@ -24,6 +24,7 @@ func (wg *WaitGroup) Done() {
 
 func (wg *WaitGroup) Wait() {
     for wg.counter > 0 {
+        yieldThread(); // yield to avoid wasteful loop
     }
 }
 
@@ -45,7 +46,7 @@ func (m *Mutex) Lock() {
         }
         // else
         endAtomic();
-        yieldThread(); // block and loop until can pick up lock
+        blockThread(); // block and loop until can pick up lock
     }
     m.currentThread = getThreadID();
     endAtomic();