如何正确简化Rust中修改Mutex包裹向量的重复代码？

错误原因 #

核心问题出在lock().unwrap()的临时值生命周期上：调用lock().unwrap()会返回一个MutexGuard对象，它是保护Vec<Task>的智能指针。当你试图把&mut这个临时生成的MutexGuard赋值给tasks时，临时的MutexGuard会在if/else语句块结束后立即被销毁，导致tasks变成指向已销毁数据的悬垂引用，Rust的借用检查器会直接拦截这种不安全行为。

修复方案 #

不要存储MutexGuard的引用，直接把MutexGuard本身绑定到tasks变量上。因为MutexGuard实现了Deref和DerefMut trait，它会被自动解引用成内部的Vec<Task>，后续的iter_mut()等操作完全不受影响。

修改后的代码如下：

impl Coordinator {
    pub fn update_status(&self, task_type: TaskType, task_id: u8) {
        let mut tasks = match task_type {
            TaskType::Map => self.map_tasks.lock().unwrap(),
            TaskType::Reduce => self.reduce_tasks.lock().unwrap(),
        };

        for task in tasks.iter_mut() {
            if task.task_id == task_id {
                task.status = TaskStatus::Success;
            }
        }
    }
}

额外优化（可选） #

如果只是修改单个任务，没必要遍历整个向量，用iter_mut().find()可以直接定位目标任务，减少不必要的遍历开销：

impl Coordinator {
    pub fn update_status(&self, task_type: TaskType, task_id: u8) {
        let mut tasks = match task_type {
            TaskType::Map => self.map_tasks.lock().unwrap(),
            TaskType::Reduce => self.reduce_tasks.lock().unwrap(),
        };

        if let Some(task) = tasks.iter_mut().find(|t| t.task_id == task_id) {
            task.status = TaskStatus::Success;
        }
    }
}

内容的提问来源于stack exchange，提问作者Danii672