Avoid flow checks in ASGI websocket close (#416)

emmett-framework · Oct 15, 2024 · 6513f6e · 6513f6e
1 parent c9e2f26
commit 6513f6e
Showing 1 changed file with 6 additions and 9 deletions.
diff --git a/src/asgi/io.rs b/src/asgi/io.rs
@@ -376,16 +376,13 @@ impl ASGIWebsocketProtocol {
         let ws_tx = self.ws_tx.clone();
 
         future_into_py_iter(self.rt.clone(), py, async move {
-            match ws_tx.lock().await.take() {
-                Some(tx) => {
-                    closed.store(true, atomic::Ordering::Relaxed);
-                    WebsocketDetachedTransport::new(true, ws_rx.lock().await.take(), Some(tx))
-                        .close()
-                        .await;
-                    Ok(())
-                }
-                _ => error_flow!(),
+            if let Some(tx) = ws_tx.lock().await.take() {
+                closed.store(true, atomic::Ordering::Relaxed);
+                WebsocketDetachedTransport::new(true, ws_rx.lock().await.take(), Some(tx))
+                    .close()
+                    .await;
             }
+            Ok(())
         })
     }