rust - 我怎样才能同时迭代 Rust HashMap 并修改它的一些值？

Question

我今年正在尝试使用 Rust 代码的出现，作为学习这门语言的一种方式。我已将输入(从第 7 天开始)解析为以下结构:

struct Process {
    name: String,
    weight: u32,
    children: Vec<String>,
    parent: Option<String>
}

这些存储在 HashMap<String, Process> 中.现在，我想根据我在父级“子级”向量中找到的内容迭代 map 中的值并更新父级值。

没用的是

for p in self.processes.values() {
    for child_name in p.children {
        let mut child = self.processes.get_mut(child_name).expect("Child not found.");
        child.parent = p.name;
    }
}

我不能同时拥有对 HashMap 的可变引用( self.processes ) 和一个非可变引用，或两个可变引用。

那么，在 Rust 中实现这一点的最惯用的方法是什么？我可以看到的两个选项是:

1. 一次性将父/子关系复制到新的临时数据结构中，然后在不可变引用超出范围后，在第二遍中更新 Process 结构。
2. 更改我的数据结构，将“parent”放在它自己的 HashMap 中。

还有第三种选择吗？

Answer 1

是的，您可以使用 RefCell 将内部可变性授予 HashMap 的值:

struct ProcessTree {
    processes: HashMap<String, RefCell<Process>>,  // change #1
}

impl ProcessTree {
    fn update_parents(&self) {
        for p in self.processes.values() {
            let p = p.borrow();                    // change #2
            for child_name in &p.children {
                let mut child = self.processes
                    .get(child_name)               // change #3
                    .expect("Child not found.")
                    .borrow_mut();                 // change #4
                child.parent = Some(p.name.clone());
            }
        }
    }
}

borrow_mut 如果 child 已经被 borrow 借用，将在运行时 panic。如果一个进程是它自己的父进程，就会发生这种情况(这应该永远不会发生，但在一个更健壮的程序中，你会希望给出一个有意义的错误消息，而不仅仅是 panic )。

我发明了一些名称并做了一些小改动(除了特别指出的那些)以使这段代码可以编译。值得注意的是，p.name.clone() 制作了 p.name 的完整副本。这是必要的，因为 name 和 parent 都拥有 String。