simplify enumerate

feschber · Dec 15, 2023 · 48f7ad3 · 48f7ad3
1 parent 5fc02d4
commit 48f7ad3
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 7 deletions.
diff --git a/src/client.rs b/src/client.rs
@@ -189,11 +189,11 @@ impl ClientManager {
         self.clients.get_mut(client as usize)?.as_mut()
     }
 
-    pub fn enumerate(&self) -> Vec<(Client, bool)> {
-        self.clients
-            .iter()
-            .filter_map(|s| s.as_ref())
-            .map(|s| (s.client.clone(), s.active))
-            .collect()
+    pub fn get_client_states(&self) -> impl Iterator<Item = &ClientState> {
+        self.clients.iter().filter_map(|x| x.as_ref())
+    }
+
+    pub fn get_client_states_mut(&mut self) -> impl Iterator<Item = &mut ClientState> {
+        self.clients.iter_mut().filter_map(|x| x.as_mut())
     }
 }
diff --git a/src/server.rs b/src/server.rs
@@ -502,7 +502,10 @@ impl Server {
     }
 
     async fn enumerate(&mut self) {
-        let clients = self.client_manager.enumerate();
+        let clients = self.client_manager
+            .get_client_states()
+            .map(|s| (s.client.clone(), s.active))
+            .collect();
         if let Err(e) = self
             .frontend
             .notify_all(FrontendNotify::Enumerate(clients))