feat: Provide mutable access to keys within entries

toml-rs · Apr 26, 2022 · 4156603 · 4156603
1 parent 23275eb
commit 4156603
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/src/inline_table.rs b/src/inline_table.rs
@@ -509,6 +509,11 @@ impl<'a> InlineOccupiedEntry<'a> {
         self.entry.key().as_str()
     }
 
+    /// Gets a mutable reference to the entry key
+    pub fn key_mut(&mut self) -> KeyMut<'_> {
+        self.entry.get_mut().key.as_mut()
+    }
+
     /// Gets a reference to the value in the entry.
     pub fn get(&self) -> &Value {
         self.entry.get().value.as_value().unwrap()

diff --git a/src/table.rs b/src/table.rs
@@ -621,6 +621,11 @@ impl<'a> OccupiedEntry<'a> {
         self.entry.key().as_str()
     }
 
+    /// Gets a mutable reference to the entry key
+    pub fn key_mut(&mut self) -> KeyMut<'_> {
+        self.entry.get_mut().key.as_mut()
+    }
+
     /// Gets a reference to the value in the entry.
     pub fn get(&self) -> &Item {
         &self.entry.get().value