fix hello_world example

jgraef · Dec 3, 2024 · 40e0d91 · 40e0d91
1 parent cbbf05a
commit 40e0d91
Showing 1 changed file with 17 additions and 13 deletions.
diff --git a/examples/hello_world.rs b/examples/hello_world.rs
@@ -14,19 +14,23 @@ async fn main() -> Result<(), Error> {
 
     let (mut tx, mut rx) = websocket.split();
 
-    tokio::task::spawn_local(async move {
-        for i in 1..11 {
-            tx.send(Message::Text(format!("Hello, World! #{i}")))
-                .await
-                .unwrap();
-        }
-    });
-
-    while let Some(message) = rx.try_next().await? {
-        if let Message::Text(text) = message {
-            println!("received: {text}");
-        }
-    }
+    futures_util::future::join(
+        async move {
+            for i in 1..11 {
+                tx.send(Message::Text(format!("Hello, World! #{i}")))
+                    .await
+                    .unwrap();
+            }
+        },
+        async move {
+            while let Some(message) = rx.try_next().await.unwrap() {
+                if let Message::Text(text) = message {
+                    println!("received: {text}");
+                }
+            }
+        },
+    )
+    .await;
 
     Ok(())
 }