diff options
author | mash-graz <mash-graz@users.noreply.github.com> | 2024-03-11 00:23:06 +0100 |
---|---|---|
committer | GitHub <noreply@github.com> | 2024-03-11 00:23:06 +0100 |
commit | 80dbcd3ddf589d20ae7786c716959c6d48ea1f1e (patch) | |
tree | 55967de66db8f4654539f462e6eea33385179aa3 /ext/web/message_port.rs | |
parent | 16dbbfa64a5d2905580535c52c1db51d1cf5b89f (diff) |
fix(ext/node) implement receiveMessageOnPort for node:worker_threads (#22766)
Implementation of `receiveMessageOnPort` for `node:worker_threads`
Fixes: #22702
Diffstat (limited to 'ext/web/message_port.rs')
-rw-r--r-- | ext/web/message_port.rs | 20 |
1 files changed, 20 insertions, 0 deletions
diff --git a/ext/web/message_port.rs b/ext/web/message_port.rs index 556030965..18429a179 100644 --- a/ext/web/message_port.rs +++ b/ext/web/message_port.rs @@ -17,6 +17,7 @@ use deno_core::Resource; use deno_core::ResourceId; use serde::Deserialize; use serde::Serialize; +use tokio::sync::mpsc::error::TryRecvError; use tokio::sync::mpsc::unbounded_channel; use tokio::sync::mpsc::UnboundedReceiver; use tokio::sync::mpsc::UnboundedSender; @@ -227,3 +228,22 @@ pub async fn op_message_port_recv_message( let cancel = RcRef::map(resource.clone(), |r| &r.cancel); resource.port.recv(state).or_cancel(cancel).await? } + +#[op2] +#[serde] +pub fn op_message_port_recv_message_sync( + state: &mut OpState, // Rc<RefCell<OpState>>, + #[smi] rid: ResourceId, +) -> Result<Option<JsMessageData>, AnyError> { + let resource = state.resource_table.get::<MessagePortResource>(rid)?; + let mut rx = resource.port.rx.borrow_mut(); + + match rx.try_recv() { + Ok((d, t)) => Ok(Some(JsMessageData { + data: d, + transferables: serialize_transferables(state, t), + })), + Err(TryRecvError::Empty) => Ok(None), + Err(TryRecvError::Disconnected) => Ok(None), + } +} |