foster/denoland-deno
1
0
Fork 0
mirror of https://github.com/denoland/deno.git synced 2025-01-11 08:33:43 -05:00
Code Issues Packages 1 Wiki Activity

fix(ext/node): patch MessagePort if provided as workerData (#23198)

Browse source 
MessagePort if directly assigned to workerData property instead of
embedding it in an object then it is not patched to a NodeMessagePort.
This commit fixes the bug.
This commit is contained in:
Satya Rohith 2024-04-03 16:42:16 +05:30 committed by GitHub
parent 92a8ada719
commit 86bc7a4381
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 19 additions and 22 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

37
ext/node/polyfills/worker_threads.ts
View file

@ -343,14 +343,7 @@ internals.__initWorkerThreads = (
defaultExport.parentPort = parentPort;
defaultExport.threadId = threadId;
for (const obj in workerData as Record<string, unknown>) {
if (ObjectPrototypeIsPrototypeOf(MessagePortPrototype, workerData[obj])) {
workerData[obj] = webMessagePortToNodeMessagePort(
workerData[obj] as MessagePort,
);
break;
}
}
workerData = patchMessagePortIfFound(workerData);
parentPort.off = parentPort.removeListener = function (
this: ParentPort,
@ -369,18 +362,7 @@ internals.__initWorkerThreads = (
// deno-lint-ignore no-explicit-any
const _listener = (ev: any) => {
let message = ev.data;
if (ObjectPrototypeIsPrototypeOf(MessagePortPrototype, message)) {
message = webMessagePortToNodeMessagePort(message);
} else {
for (const obj in message) {
if (
ObjectPrototypeIsPrototypeOf(MessagePortPrototype, message[obj])
) {
message[obj] = webMessagePortToNodeMessagePort(message[obj]);
break;
}
}
}
message = patchMessagePortIfFound(message);
return listener(message);
};
listeners.set(listener, _listener);
@ -481,6 +463,21 @@ function webMessagePortToNodeMessagePort(port: MessagePort) {
return port;
}
// deno-lint-ignore no-explicit-any
function patchMessagePortIfFound(data: any) {
if (ObjectPrototypeIsPrototypeOf(MessagePortPrototype, data)) {
data = webMessagePortToNodeMessagePort(data);
} else {
for (const obj in data as Record<string, unknown>) {
if (ObjectPrototypeIsPrototypeOf(MessagePortPrototype, data[obj])) {
data[obj] = webMessagePortToNodeMessagePort(data[obj] as MessagePort);
break;
}
}
}
return data;
}
export {
BroadcastChannel,
MessagePort,

2
tests/testdata/workers/node_worker_message_port.mjs vendored
View file

@ -9,7 +9,7 @@ const deferred = createDeferred();
const worker = new workerThreads.Worker(
import.meta.resolve("./node_worker_message_port_1.cjs"),
{
workerData: { workerPort },
workerData: workerPort,
transferList: [workerPort],
},
);

2
tests/testdata/workers/node_worker_message_port_1.cjs vendored
View file

@ -1,7 +1,7 @@
const { parentPort, workerData } = require("worker_threads");
parentPort.on("message", (msg) => {
const workerPort = workerData.workerPort;
const workerPort = workerData;
parentPort.postMessage("Hello from worker on parentPort!");
workerPort.postMessage("Hello from worker on workerPort!");
workerPort.on("close", () => console.log("worker port closed"));