Rollup merge of #78083 - ChaiTRex:master, r=m-ou-se

Stabilize or_insert_with_key

Stabilizes the `or_insert_with_key` feature from https://github.com/rust-lang/rust/issues/71024. This allows inserting key-derived values when a `HashMap`/`BTreeMap` entry is vacant.

The difference between this and  `.or_insert_with(|| ... )` is that this provides a reference to the key to the closure after it is moved with `.entry(key_being_moved)`, avoiding the need to copy or clone the key.
This commit is contained in:
Yuki Okushi 2020-12-19 15:15:57 +09:00 committed by GitHub
commit 0765536c0b
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
3 changed files with 15 additions and 11 deletions

View file

@ -40,7 +40,7 @@ const SSO_ARRAY_SIZE: usize = 8;
// into_keys/into_values (unstable) // into_keys/into_values (unstable)
// all raw_entry-related // all raw_entry-related
// PartialEq/Eq (requires sorting the array) // PartialEq/Eq (requires sorting the array)
// Entry::or_insert_with_key (unstable) // Entry::or_insert_with_key
// Vacant/Occupied entries and related // Vacant/Occupied entries and related
// //
// FIXME: In HashMap most methods accepting key reference // FIXME: In HashMap most methods accepting key reference

View file

@ -116,15 +116,16 @@ impl<'a, K: Ord, V> Entry<'a, K, V> {
} }
} }
#[unstable(feature = "or_insert_with_key", issue = "71024")] /// Ensures a value is in the entry by inserting, if empty, the result of the default function.
/// Ensures a value is in the entry by inserting, if empty, the result of the default function, /// This method allows for generating key-derived values for insertion by providing the default
/// which takes the key as its argument, and returns a mutable reference to the value in the /// function a reference to the key that was moved during the `.entry(key)` method call.
/// entry. ///
/// The reference to the moved key is provided so that cloning or copying the key is
/// unnecessary, unlike with `.or_insert_with(|| ... )`.
/// ///
/// # Examples /// # Examples
/// ///
/// ``` /// ```
/// #![feature(or_insert_with_key)]
/// use std::collections::BTreeMap; /// use std::collections::BTreeMap;
/// ///
/// let mut map: BTreeMap<&str, usize> = BTreeMap::new(); /// let mut map: BTreeMap<&str, usize> = BTreeMap::new();
@ -134,6 +135,7 @@ impl<'a, K: Ord, V> Entry<'a, K, V> {
/// assert_eq!(map["poneyland"], 9); /// assert_eq!(map["poneyland"], 9);
/// ``` /// ```
#[inline] #[inline]
#[stable(feature = "or_insert_with_key", since = "1.50.0")]
pub fn or_insert_with_key<F: FnOnce(&K) -> V>(self, default: F) -> &'a mut V { pub fn or_insert_with_key<F: FnOnce(&K) -> V>(self, default: F) -> &'a mut V {
match self { match self {
Occupied(entry) => entry.into_mut(), Occupied(entry) => entry.into_mut(),

View file

@ -2219,14 +2219,16 @@ impl<'a, K, V> Entry<'a, K, V> {
} }
} }
/// Ensures a value is in the entry by inserting, if empty, the result of the default function, /// Ensures a value is in the entry by inserting, if empty, the result of the default function.
/// which takes the key as its argument, and returns a mutable reference to the value in the /// This method allows for generating key-derived values for insertion by providing the default
/// entry. /// function a reference to the key that was moved during the `.entry(key)` method call.
///
/// The reference to the moved key is provided so that cloning or copying the key is
/// unnecessary, unlike with `.or_insert_with(|| ... )`.
/// ///
/// # Examples /// # Examples
/// ///
/// ``` /// ```
/// #![feature(or_insert_with_key)]
/// use std::collections::HashMap; /// use std::collections::HashMap;
/// ///
/// let mut map: HashMap<&str, usize> = HashMap::new(); /// let mut map: HashMap<&str, usize> = HashMap::new();
@ -2236,7 +2238,7 @@ impl<'a, K, V> Entry<'a, K, V> {
/// assert_eq!(map["poneyland"], 9); /// assert_eq!(map["poneyland"], 9);
/// ``` /// ```
#[inline] #[inline]
#[unstable(feature = "or_insert_with_key", issue = "71024")] #[stable(feature = "or_insert_with_key", since = "1.50.0")]
pub fn or_insert_with_key<F: FnOnce(&K) -> V>(self, default: F) -> &'a mut V { pub fn or_insert_with_key<F: FnOnce(&K) -> V>(self, default: F) -> &'a mut V {
match self { match self {
Occupied(entry) => entry.into_mut(), Occupied(entry) => entry.into_mut(),