address new review

libp2p · Mar 22, 2024 · af8f693 · af8f693
1 parent e154864
commit af8f693
Showing 1 changed file with 9 additions and 9 deletions.
diff --git a/yamux/src/connection/closing.rs b/yamux/src/connection/closing.rs
@@ -49,6 +49,14 @@ where
 
         loop {
             match this.state {
+                State::FlushingPendingFrames => {
+                    ready!(this.socket.poll_ready_unpin(cx))?;
+
+                    match this.pending_frames.pop_front() {
+                        Some(frame) => this.socket.start_send_unpin(frame)?,
+                        None => this.state = State::ClosingStreamReceiver,
+                    }
+                }
                 State::ClosingStreamReceiver => {
                     for stream in this.stream_receivers.iter_mut() {
                         stream.inner_mut().close();
@@ -74,14 +82,6 @@ where
                         }
                     }
                 }
-                State::FlushingPendingFrames => {
-                    ready!(this.socket.poll_ready_unpin(cx))?;
-
-                    match this.pending_frames.pop_front() {
-                        Some(frame) => this.socket.start_send_unpin(frame)?,
-                        None => this.state = State::ClosingStreamReceiver,
-                    }
-                }
                 State::ClosingSocket => {
                     ready!(this.socket.poll_close_unpin(cx))?;
 
@@ -93,8 +93,8 @@ where
 }
 
 enum State {
+    FlushingPendingFrames,
     ClosingStreamReceiver,
     DrainingStreamReceiver,
-    FlushingPendingFrames,
     ClosingSocket,
 }