Provide insert() and remove() on SparseSecondaryEntry

Signed-off-by: Ian Jackson <ijackson@chiark.greenend.org.uk>

diff --git a/src/utils.rs b/src/utils.rs
index 5aeb296a4c40d4fd8b17c2ab5a5fb7f7958fac38..853be507b3d0e733c9a65263f7b9a9129565fab0 100644 (file)
--- a/src/utils.rs
+++ b/src/utils.rs
@@ -491,3 +491,38 @@ macro_rules! want_let {
     want_let!{ $($variant(y))::+ = $input; ?; $($otherwise)* }
   };
 }
+
+macro_rules! entry_define_insert_remove {
+  { $name:ident, $name_mod:ident, $entry:path, $into_key:ident } =>
+  {
+    #[allow(non_snake_case)]
+    mod $name_mod {
+      use $crate::imports::extend::ext;
+      use $entry as Entry;
+      use Entry::{Occupied, Vacant};
+      #[ext(pub, name=EntryExt)]
+      impl<'e,K,V> Entry<'e,K,V> where K: slotmap::Key {
+        fn insert(self, v: V) {
+          match self {
+            Vacant(ve)   => { ve.insert(v); }
+            Occupied(mut oe) => { oe.insert(v); }
+          }
+        }
+        fn remove(self) -> (K, Option<V>) {
+          match self {
+            Vacant(ve)   => { let k = ve.$into_key();        (k, None)    }
+            Occupied(oe) => { let (k,v) = oe.remove_entry(); (k, Some(v)) }
+          }
+        }
+      }
+    }
+    pub use $name_mod::EntryExt as $name;
+  }
+}
+
+entry_define_insert_remove!{
+  SlotmapSparseSecondaryEntryExt,
+  SlotmapSparseSecondaryEntryExt_mod,
+  slotmap::sparse_secondary::Entry,
+  key
+}