Rust中如何实现基于旧值的原地内存更新（无需Option或额外枚举）

你需要的是取出可变引用的旧值，基于它生成新值后直接替换原位置的操作，标准库中没有直接对应的std::mem::update函数，但可以通过两种方式实现，且不需要修改结构体定义、添加默认枚举变体或包装Option：

方法一：安全临时占位法 #

利用std::mem::replace，先将原state替换为一个临时构造的有效State实例（使用现有枚举变体），再用旧值生成新值替换回去。这种方法完全安全，临时实例的开销极小：

impl Doer {
    pub fn do_thing(&mut self) {
        // 用StateA的空字符串作为临时占位，取出旧state
        let old_state = std::mem::replace(&mut self.state, State::A(StateA { s: String::new() }));
        // 根据旧值生成新值并替换回原位置
        self.state = match old_state {
            State::A(state_a) => State::B(do_a_thing(state_a)),
            State::B(state_b) => State::A(do_b_thing(state_b)),
        };
    }
}

方法二：Unsafe直接内存读写 #

如果不想构造任何临时实例，可以使用std::ptr::read和std::ptr::write直接操作内存。这是标准库底层的unsafe操作，但只要遵守规则就是安全的：

use std::ptr;

impl Doer {
    pub fn do_thing(&mut self) {
        // 读取旧值并转移所有权，此时原内存处于未初始化状态
        let old_state = unsafe { ptr::read(&self.state) };
        // 基于旧值生成新值
        let new_state = match old_state {
            State::A(state_a) => State::B(do_a_thing(state_a)),
            State::B(state_b) => State::A(do_b_thing(state_b)),
        };
        // 将新值写入原内存位置，恢复内存有效性
        unsafe {
            ptr::write(&mut self.state, new_state);
        }
    }
}

安全说明： #

• ptr::read会读取内存中的值并转移所有权，原内存暂时变为未初始化，但我们紧接着用ptr::write覆盖，不会出现未初始化内存被访问的情况。
• 旧值的所有权完全转移到old_state，处理后新值写入，不会发生双重释放或内存泄漏。

原代码无法编译的原因 #

标准库中不存在std::mem::update函数：Rust的借用规则默认不允许同时获取旧值的所有权并保持原引用的有效性，直接取出旧值会导致原引用悬空，必须通过临时占位或unsafe操作绕过这一限制。

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