Merge pull request #2341 from dusk-network/rues-close-ws

dusk-network · Sep 10, 2024 · e9a07a3 · e9a07a3
2 parents db65642 + fa1a517
commit e9a07a3
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/rusk/src/lib/http.rs b/rusk/src/lib/http.rs
@@ -483,6 +483,12 @@ async fn handle_stream_rues<H: HandleRequest>(
 
     loop {
         tokio::select! {
+            recv = stream.next() => {
+                if let Some(Ok(Message::Close(msg))) = recv {
+                    let _ = stream.close(msg).await;
+                    break;
+                }
+            }
             _ = shutdown.recv() => {
                 let _ = stream.close(Some(CloseFrame {
                     code: CloseCode::Away,