worker: allow BroadcastChannel in receiveMessageOnPort

PR-URL: #37535
Reviewed-By: Colin Ihrig <[email protected]>
Reviewed-By: James M Snell <[email protected]>
Reviewed-By: Michaël Zasso <[email protected]>

Author: addaleax

doc/api/worker_threads.md

@@ -179,9 +179,13 @@ if (isMainThread) {
 ## `worker.receiveMessageOnPort(port)`
 <!-- YAML
 added: v12.3.0
+changes:
+  - version: REPLACEME
+    pr-url: https://github.com/nodejs/node/pull/37535
+    description: The port argument can also refer to a `BroadcastChannel` now.
 -->
 
-* `port` {MessagePort}
+* `port` {MessagePort|BroadcastChannel}
 
 * Returns: {Object|undefined}
 

lib/internal/worker/io.js

@@ -333,7 +333,7 @@ function createWorkerStdio() {
 }
 
 function receiveMessageOnPort(port) {
-  const message = receiveMessageOnPort_(port);
+  const message = receiveMessageOnPort_(port?.[kHandle] ?? port);
   if (message === noMessageSymbol) return undefined;
   return { message };
 }

test/parallel/test-worker-broadcastchannel.js

@@ -4,6 +4,7 @@ const common = require('../common');
 const {
   BroadcastChannel,
   Worker,
+  receiveMessageOnPort
 } = require('worker_threads');
 const assert = require('assert');
 
@@ -140,3 +141,13 @@ assert.throws(() => new BroadcastChannel(), {
     message: /BroadcastChannel is closed/
   });
 }
+
+{
+  const bc1 = new BroadcastChannel('channel4');
+  const bc2 = new BroadcastChannel('channel4');
+  bc1.postMessage('some data');
+  assert.strictEqual(receiveMessageOnPort(bc2).message, 'some data');
+  assert.strictEqual(receiveMessageOnPort(bc2), undefined);
+  bc1.close();
+  bc2.close();
+}