From 88d72690b8e3444290d6be5c7072188cf85b4c39 Mon Sep 17 00:00:00 2001
From: Altinok Darici <altinokd@microsoft.com>
Date: Tue, 30 May 2023 11:16:22 -0700
Subject: [PATCH] Fix: Bind postMessage (#671)

* Bind postMessage when passing into onMessage

* Change files

* Fix
---
 ...run-scheduler-5d05a85f-ab8d-4a75-af7d-c329304d03bc.json | 7 +++++++
 packages/scheduler/src/WrappedTarget.ts                    | 4 +++-
 2 files changed, 10 insertions(+), 1 deletion(-)
 create mode 100644 change/@lage-run-scheduler-5d05a85f-ab8d-4a75-af7d-c329304d03bc.json

diff --git a/change/@lage-run-scheduler-5d05a85f-ab8d-4a75-af7d-c329304d03bc.json b/change/@lage-run-scheduler-5d05a85f-ab8d-4a75-af7d-c329304d03bc.json
new file mode 100644
index 000000000..a43970903
--- /dev/null
+++ 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": "altinokd@microsoft.com",
+  "dependentChangeType": "patch"
+}
diff --git a/packages/scheduler/src/WrappedTarget.ts b/packages/scheduler/src/WrappedTarget.ts
index b4ac38174..ab7a46622 100644
--- 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);
           }
         };