subscriptions

1 files changed, 21 insertions, 0 deletions

diff --git a/crates/server/src/main_loop/subscriptions.rs b/crates/server/src/main_loop/subscriptions.rs
new file mode 100644
index 000000000..963096aef
--- /dev/null
+++ b/crates/server/src/main_loop/subscriptions.rs
@@ -0,0 +1,21 @@
+use std::collections::HashSet;
+use libanalysis::FileId;
+
+pub struct Subscriptions {
+    subs: HashSet<FileId>,
+}
+
+impl Subscriptions {
+    pub fn new() -> Subscriptions {
+        Subscriptions { subs: HashSet::new() }
+    }
+    pub fn add_sub(&mut self, file_id: FileId) {
+        self.subs.insert(file_id);
+    }
+    pub fn remove_sub(&mut self, file_id: FileId) {
+        self.subs.remove(&file_id);
+    }
+    pub fn subscriptions(&self) -> Vec<FileId> {
+        self.subs.iter().cloned().collect()
+    }
+}