move the Action enum declaration outside the loop

twilight-rs · Jun 24, 2023 · 90422cb · 90422cb
1 parent e4829a5
commit 90422cb
Showing 1 changed file with 14 additions and 13 deletions.
diff --git a/twilight-gateway/src/shard.rs b/twilight-gateway/src/shard.rs
@@ -577,20 +577,21 @@ impl Shard {
             _ => {}
         }
 
+        /// Actions the shard might take.
+        enum Action {
+            /// Close the gateway connection with this close frame.
+            Close(CloseFrame<'static>),
+            /// Send this command to the gateway.
+            Command(String),
+            /// Send a heartbeat command to the gateway.
+            Heartbeat,
+            /// Identify with the gateway.
+            Identify,
+            /// Handle this incoming gateway message.
+            Message(Option<Result<TungsteniteMessage, TungsteniteError>>),
+        }
+
         let message = loop {
-            /// Actions the shard might take.
-            enum Action {
-                /// Close the gateway connection with this close frame.
-                Close(CloseFrame<'static>),
-                /// Send this command to the gateway.
-                Command(String),
-                /// Send a heartbeat command to the gateway.
-                Heartbeat,
-                /// Identify with the gateway.
-                Identify,
-                /// Handle this incoming gateway message.
-                Message(Option<Result<TungsteniteMessage, TungsteniteError>>),
-            }
             let next_action = |cx: &mut Context<'_>| {
                 if !(self.status.is_disconnected() || self.status.is_fatally_closed()) {
                     if let Poll::Ready(frame) = self.user_channel.close_rx.poll_recv(cx) {