Rust写时复制实现疑问：remove方法返回自身或新实例报错

你想要实现的逻辑，正好是std::borrow::Cow（Clone-On-Write）设计的场景——它允许我们在不需要修改时借用原有实例，需要修改时创建新的拥有所有权的实例，完美解决你遇到的所有权冲突问题。

先分析你之前尝试的问题 #

• 返回Self时：&self是共享引用，Rust不允许从共享引用中移动出*self（会破坏引用的不变性），因此return *self报错。
• 返回&Self时：函数内创建的n是局部变量，生命周期仅限于函数内部，返回它的引用会导致悬垂引用，编译器直接报错。
• 获取self所有权时：调用方法后原变量的所有权被转移，自然无法再复用，第二次调用remove会提示使用了已移动的值。

使用Cow的解决方案 #

修改remove方法，将返回类型改为Cow<Self>，利用Cow的枚举变体分别处理“无需修改”和“需要修改”的情况：

use std::borrow::Cow;
use std::collections::HashMap;

#[derive(Clone, Debug)]
struct Something {
    data: HashMap<char, u16>,
}

impl Something {
    fn new() -> Self {
        Something {
            data: HashMap::new()
        }
    }

    fn get(&self, c: char) -> Option<&u16> {
        self.data.get(&c)
    }

    fn add(&self, c: char, value: u16) -> Self {
        let mut n = self.clone();
        n.data.insert(c, value);
        n
    }

    fn remove(&self, c: char) -> Cow<Self> {
        if !self.data.contains_key(&c) {
            // 无需修改，返回原有实例的借用
            Cow::Borrowed(self)
        } else {
            // 需要修改，克隆并修改后返回拥有所有权的实例
            let mut n = self.clone();
            n.data.remove(&c);
            Cow::Owned(n)
        }
    }
}

fn main() {
    let s = Something::new();
    let s = s.add('a', 1);
    let s = s.add('b', 2);

    // 使用Cow，无需修改时复用原有引用
    let s_cow = s.remove('c');
    println!("{:?}", s_cow.get('b')); // 输出 Some(2)

    // 如果需要获取所有权，可以调用into_owned()
    let s_owned = s_cow.into_owned();
    let s_owned = s_owned.remove('a');
    println!("{:?}", s_owned.get('a')); // 输出 None

    // 原s仍然可以使用，因为remove方法只借用了它
    println!("{:?}", s.get('a')); // 输出 Some(1)
}

Cow的实现逻辑 #

Cow是一个枚举，简化定义如下：

enum Cow<'a, B> where B: 'a + ToOwned + ?Sized {
    Borrowed(&'a B),
    Owned(<B as ToOwned>::Owned),
}

它的核心特性：

1. 实现了Deref trait，可直接像使用&B或B一样调用方法，无需手动匹配变体。
2. 调用to_mut()方法时，若为Borrowed变体，会自动克隆出拥有所有权的实例并返回可变引用；若已是Owned变体，直接返回可变引用。
3. into_owned()方法可将Cow转换为拥有所有权的实例，不管当前是借用还是拥有状态。

这种设计既避免了不必要的克隆（无需修改时复用引用），又解决了所有权冲突问题，完全契合你想要的“按需返回引用/新实例”的需求。

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