fixes #198 - allow removing items from TypeMap

specta-rs · Dec 24, 2023 · cab26b1 · cab26b1
1 parent 6b56127
commit cab26b1
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/src/type/map.rs b/src/type/map.rs
@@ -46,6 +46,14 @@ impl TypeMap {
         self.map.len()
     }
 
+    pub fn contains_key(&self, sid: SpectaID) -> bool {
+        self.map.contains_key(&sid)
+    }
+
+    pub fn remove(&mut self, sid: SpectaID) -> Option<NamedDataType> {
+        self.map.remove(&sid).flatten()
+    }
+
     // TODO: It would be nice if this would a proper `Iterator` or `IntoIterator` implementation!
     pub fn iter(&self) -> impl Iterator<Item = (SpectaID, &NamedDataType)> {
         #[allow(clippy::unnecessary_filter_map)]