Fix the test_stream function running after the stream is closed (#3396)

DioxusLabs · Jan 8, 2025 · 5ea0867 · 5ea0867
1 parent 5815e05
commit 5ea0867
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/examples/fullstack-streaming/src/main.rs b/examples/fullstack-streaming/src/main.rs
@@ -30,7 +30,10 @@ pub async fn test_stream() -> Result<TextStream, ServerFnError> {
         loop {
             tokio::time::sleep(tokio::time::Duration::from_millis(500)).await;
             dioxus::logger::tracing::info!("Sending new chunk!");
-            let _ = tx.unbounded_send(Ok("Hello, world!".to_string()));
+            if tx.unbounded_send(Ok("Hello, world!".to_string())).is_err() {
+                // If the channel is closed, stop sending chunks
+                break;
+            }
         }
     });