rust - 如何在Vec上更新或插入？

﹏ヽ暗。殇╰゛Y 2023-01-24 02:17 240阅读 0赞

我正在用Rust编写数据结构。它包含键值对的`Vec`。当插入结构时，我需要找到一个匹配的键并同时更新键和值(实际上是一个子指针)。代码看起来像这样，其中`pivots`是`ref mut`的`Vec<Pivot>`，而`Pivot`只是具有两个字段的结构:

match pivots.iter_mut().find(|ref p| key <= p.min_key) { // first mutable borrow Some(ref mut pivot) => { // If there is one, insert into it and update the pivot key pivot.min_key = key; pivot.child.insert(key, value) // recursive call }, // o/w, insert a new leaf at the end None => pivots.push(Pivot /* ... */) // second mutable borrow }

但是有一个问题。即使我没有在`match`的第二个分支中使用可变的迭代器，借阅检查器仍提示我“一次不能借用`*pivots`可变为多次”。  
  
这对我来说很有意义，因为即使在`match`的情况下不使用第一个借位，它仍然在范围内。这有点不方便:聪明的检查员当然可以说这些借贷是不重叠的。我见过有人在网上建议使用早退以避免这种问题，例如:

match pivots.iter_mut().find(|ref p| key <= p.min_key) { Some(ref mut pivot) => { pivot.min_key = key; pivot.child.insert(key, value); return }, None => () }; pivots.push(Pivot /* ... */)

但这似乎很难理解，尤其是当它意味着将此代码分解为自己的函数以允许`return`时。有没有更惯用的方式来执行更新或插入操作？

**最佳答案**

从长远来看，有一个合并的RFC ["non-lexical lifetimes"][non-lexical lifetimes]可以解决这个问题。使用Rust 1.31中可用的Rust 2018中的非词汇生存期，您的代码将按原样运行:  
  
[Playground][]

use std::collections::HashMap; pub struct Pivot { pub min_key: u64, pub child: HashMap<u64, ()>, } fn update_or_append(pivots: &mut Vec<Pivot>, key: u64, value: ()) { match pivots.iter_mut().find(|ref p| key <= p.min_key) { Some(pivot) => { // If there is one, insert into it and update the pivot key pivot.min_key = key; pivot.child.insert(key, value); return; } // o/w insert a new leaf at the end None => { let mut m = HashMap::new(); m.insert(key, value); pivots.push(Pivot { min_key: key, child: m, }); } } } fn main() { let mut pivots = Vec::new(); update_or_append(&mut pivots, 100, ()); }

如果此**不适用于您的代码，请 checkout**

*  **[Returning a reference from a HashMap or Vec causes a borrow to last beyond the scope it's in?][Returning a reference from a HashMap or Vec causes a borrow to last beyond the scope it_s in]**  
      
      
     
 *  **在Rust 2018之前，您可以通过一些其他控制流处理来解决它。  
      
    无论更新是否发生，您都可以让您的匹配产生一个`bool`值，并在下面使用该值附加条件块。我考虑将“update-or-append”逻辑放入单独的函数中(在更新后使用`return`)是更惯用的方法:  
      
    [Playground][Playground 1]**
    
        use std::collections::HashMap; pub struct Pivot { pub min_key: u64, pub child: HashMap<u64, ()>, } fn update_or_append(pivots: &mut Vec<Pivot>, key: u64, value: ()) { if let Some(pivot) = pivots.iter_mut().find(|ref p| key <= p.min_key) { // If there is one, insert into it and update the pivot key pivot.min_key = key; pivot.child.insert(key, value); return; } // otherwise insert a new leaf at the end let mut m = HashMap::new(); m.insert(key, value); pivots.push(Pivot { min_key: key, child: m, }); } fn main() { let mut pivots = Vec::new(); update_or_append(&mut pivots, 100, ()); }
    
      
    **使用`bool`跟踪更新是否发生:  
      
    [Playground][Playground 2]**
    
        use std::collections::HashMap; pub struct Pivot { pub min_key: u64, pub child: HashMap<u64, ()>, } fn update_or_append(pivots: &mut Vec<Pivot>, key: u64, value: ()) { let updated = match pivots.iter_mut().find(|ref p| key <= p.min_key) { Some(pivot) => { // If there is one, insert into it and update the pivot key pivot.min_key = key; pivot.child.insert(key, value); true } // o/w insert a new leaf at the end below None => false, }; if !updated { let mut m = HashMap::new(); m.insert(key, value); pivots.push(Pivot { min_key: key, child: m, }); } } fn main() { let mut pivots = Vec::new(); update_or_append(&mut pivots, 100, ()); }
    
    **关于rust - 如何在Vec上更新或插入？，我们在Stack Overflow上找到一个类似的问题： [https://stackoverflow.com/questions/47395171/][https_stackoverflow.com_questions_47395171]**

[non-lexical lifetimes]: https://github.com/rust-lang/rfcs/blob/master/text/2094-nll.md#problem-case-2-conditional-control-flow
[Playground]: https://play.rust-lang.org/?version=stable&mode=debug&edition=2018&gist=f11a2a6272e18648e9aa6965bdf89c84
[Returning a reference from a HashMap or Vec causes a borrow to last beyond the scope it_s in]: https://stackoverflow.com/questions/38023871/returning-a-reference-from-a-hashmap-or-vec-causes-a-borrow-to-last-beyond-the-s
[Playground 1]: https://play.rust-lang.org/?gist=e66e877cc3b900c2babfae721b9b4a61&version=stable
[Playground 2]: https://play.rust-lang.org/?gist=d53208db6b9ccd2c2fd95c002226f5b8&version=stable
[https_stackoverflow.com_questions_47395171]: https://stackoverflow.com/questions/47395171/