diff options
Diffstat (limited to 'ext')
-rw-r--r-- | ext/web/message_port.rs | 16 |
1 files changed, 10 insertions, 6 deletions
diff --git a/ext/web/message_port.rs b/ext/web/message_port.rs index ac33145b1..c069037f8 100644 --- a/ext/web/message_port.rs +++ b/ext/web/message_port.rs @@ -15,6 +15,7 @@ use deno_core::OpState; use deno_core::RcRef; use deno_core::Resource; use deno_core::ResourceId; +use futures::future::poll_fn; use serde::Deserialize; use serde::Serialize; use tokio::sync::mpsc::error::TryRecvError; @@ -52,16 +53,19 @@ impl MessagePort { Ok(()) } - #[allow(clippy::await_holding_refcell_ref)] // TODO(ry) remove! pub async fn recv( &self, state: Rc<RefCell<OpState>>, ) -> Result<Option<JsMessageData>, AnyError> { - let mut rx = self - .rx - .try_borrow_mut() - .map_err(|_| type_error("Port receiver is already borrowed"))?; - if let Some((data, transferables)) = rx.recv().await { + let rx = &self.rx; + + let maybe_data = poll_fn(|cx| { + let mut rx = rx.borrow_mut(); + rx.poll_recv(cx) + }) + .await; + + if let Some((data, transferables)) = maybe_data { let js_transferables = serialize_transferables(&mut state.borrow_mut(), transferables); return Ok(Some(JsMessageData { |