From 8e007cc137a2210a3a0b47177b55e4b850bbb12e Mon Sep 17 00:00:00 2001 From: Evan Almloff Date: Sun, 11 Feb 2024 13:35:34 -0600 Subject: [PATCH] fix the web renderer panicing when the hot reload connection fails --- packages/web/src/lib.rs | 23 ++++++++++++----------- 1 file changed, 12 insertions(+), 11 deletions(-) diff --git a/packages/web/src/lib.rs b/packages/web/src/lib.rs index 7e365d89b..324b3a8a9 100644 --- a/packages/web/src/lib.rs +++ b/packages/web/src/lib.rs @@ -61,10 +61,7 @@ pub use crate::cfg::Config; #[cfg(feature = "file_engine")] pub use crate::file_engine::WebFileEngineExt; use dioxus_core::VirtualDom; -use futures_util::{ - future::{select, Either}, - pin_mut, FutureExt, StreamExt, -}; +use futures_util::{pin_mut, select, FutureExt, StreamExt}; mod cfg; mod dom; @@ -159,17 +156,21 @@ pub async fn run(virtual_dom: VirtualDom, web_config: Config) { let (mut res, template) = { let work = dom.wait_for_work().fuse(); pin_mut!(work); + let mut rx_next = rx.select_next_some(); #[cfg(all(feature = "hot_reload", debug_assertions))] - match select(work, select(hotreload_rx.next(), rx.next())).await { - Either::Left((_, _)) => (None, None), - Either::Right((Either::Left((new_template, _)), _)) => (None, new_template), - Either::Right((Either::Right((evt, _)), _)) => (evt, None), + { + let mut hot_reload_next = hotreload_rx.select_next_some(); + select! { + _ = work => (None, None), + new_template = hot_reload_next => (None, Some(new_template)), + evt = rx_next => (Some(evt), None), + } } #[cfg(not(all(feature = "hot_reload", debug_assertions)))] - match select(work, rx.next()).await { - Either::Left((_, _)) => (None, None), - Either::Right((evt, _)) => (evt, None), + select! { + _ = work => (None, None), + evt = rx_next => (Some(evt), None), } };