Set: add try_insert

minirust · Jun 12, 2024 · 450f57c · 450f57c
1 parent af9ecc4
commit 450f57c
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 1 deletion.
diff --git a/libspecr/src/map/func.rs b/libspecr/src/map/func.rs
@@ -44,7 +44,7 @@ impl<K: Obj, V: Obj> Map<K, V> {
 
     /// Like `insert`, but fails if `k` was already in the map.
     pub fn try_insert(&mut self, k: K, v: V) -> Result<(), ()> {
-        if self.contains_key(k.clone()) {
+        if self.contains_key(k) {
             return Err(());
         }
 

diff --git a/libspecr/src/set/func.rs b/libspecr/src/set/func.rs
@@ -20,6 +20,19 @@ impl<T: Obj> Set<T> {
         });
     }
 
+    /// Like `insert`, but fails if `t` was already in the set.
+    pub fn try_insert(&mut self, t: T) -> Result<(), ()> {
+        if self.contains(t) {
+            return Err(());
+        }
+
+        self.0.mutate(|s| {
+            s.insert(t);
+        });
+
+        Ok(())
+    }
+
     /// Removes `t` from `self`.
     pub fn remove(&mut self, t: T) {
         self.0.mutate(|s| {