Fix: Bind postMessage (#671)

* Bind postMessage when passing into onMessage * Change files * Fix
microsoft · May 30, 2023 · 88d7269 · 88d7269
1 parent 461581d
commit 88d7269
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/change/@lage-run-scheduler-5d05a85f-ab8d-4a75-af7d-c329304d03bc.json b/change/@lage-run-scheduler-5d05a85f-ab8d-4a75-af7d-c329304d03bc.json
@@ -0,0 +1,7 @@
+{
+  "type": "patch",
+  "comment": "Bind postMessage when passing into onMessage",
+  "packageName": "@lage-run/scheduler",
+  "email": "[email protected]",
+  "dependentChangeType": "patch"
+}
diff --git a/packages/scheduler/src/WrappedTarget.ts b/packages/scheduler/src/WrappedTarget.ts
@@ -206,6 +206,8 @@ export class WrappedTarget implements TargetRun {
       { target },
       target.weight ?? 1,
       (worker, stdout, stderr) => {
+        const postMessage = worker.postMessage.bind(worker);
+
         msgHandler = (data) => {
           if (data.type === "log") {
             logger.log(data.level, data.msg, { target, threadId: worker.threadId });
@@ -214,7 +216,7 @@ export class WrappedTarget implements TargetRun {
               worker.postMessage({ type: "hash", hash });
             });
           } else if (this.options.onMessage) {
-            this.options.onMessage(data, worker.postMessage);
+            this.options.onMessage(data, postMessage);
           }
         };