1 files changed, 34 insertions, 1 deletions
diff --git a/game_server/src/lobby.rs b/game_server/src/lobby.rs
index c0717d3..8808b7f 100644
--- a/game_server/src/lobby.rs
+++ b/game_server/src/lobby.rs
@@ -1,2 +1,35 @@
-struct Lobby {
+use std::collections::HashMap;
+
+use super::group::{Group, GroupId};
+
+pub struct Lobby {
+    groups: HashMap<GroupId, Group>,
+}
+
+impl Lobby {
+    pub fn new() -> Lobby {
+        Self {
+            groups: HashMap::new(),
+        }
+    }
+
+    pub fn add_group(&mut self, group: Group) {
+        self.groups.insert(group.get_id(), group);
+    }
+
+    pub fn iter<'a>(&'a self) -> GroupIterator<'a> {
+        GroupIterator { groups: self.groups.values() }
+    }
+}
+
+pub struct GroupIterator<'a> {
+    groups: std::collections::hash_map::Values<'a, GroupId, Group>
+}
+
+impl<'a> Iterator for GroupIterator<'a> {
+    type Item = &'a Group;
+
+    fn next(&mut self) -> Option<Self::Item> {
+        self.groups.next()
+    }
 }