Added create_with_id method.

stalwartlabs · Jul 5, 2023 · a55af18 · a55af18
1 parent ab6a9e5
commit a55af18
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/src/core/set.rs b/src/core/set.rs
@@ -189,6 +189,14 @@ impl<O: SetObject> SetRequest<O> {
             .unwrap()
     }
 
+    pub fn create_with_id(&mut self, create_id: impl Into<String>) -> &mut O {
+        let create_id = create_id.into();
+        self.create
+            .get_or_insert_with(AHashMap::new)
+            .insert(create_id.clone(), O::new(0.into()));
+        self.create.as_mut().unwrap().get_mut(&create_id).unwrap()
+    }
+
     pub fn create_item(&mut self, item: O) -> String {
         let create_id = self.create.as_ref().map_or(0, |c| c.len());
         let create_id_str = format!("c{}", create_id);