Graceful shutdown Hyper connection (#4052)

rustwasm · Aug 6, 2024 · 33ea44f · 33ea44f
1 parent 09e0e6d
commit 33ea44f
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/crates/example-tests/src/lib.rs b/crates/example-tests/src/lib.rs
@@ -354,14 +354,17 @@ pub async fn test_example(
             let conn = builder.serve_connection(TokioIo::new(stream), &service);
             tokio::pin!(conn);
 
-            tokio::select! {
+            let ret = tokio::select! {
                 res = conn.as_mut() => {
                     res.map_err(|e| anyhow::Error::msg(e.to_string()))
                 }
                 _ = rx => {
                     Ok(())
                 }
-            }
+            };
+
+            conn.graceful_shutdown();
+            ret
         },
         async {
             #[derive(Deserialize)]