需要参考时有条件地克隆

Question

我有一个结构的引用。

我需要一个结构的引用。有时是一样的，但有时是克隆然后修改的。

现在，我愿意

if condition {
    let mut copy = original.clone();
    copy.select();
    do_big_thing(&copy);
} else {
    do_big_thing(original);
}

但do_big_thing 接受其他参数，我不喜欢重复这一行。

有没有办法

• 首先构建引用，可以是原始引用，也可以是对本地修改克隆的引用
• 然后在do_big_thing中使用它（它不必活得更久）

?

这当然行不通，因为copy 的寿命不够长：

let mut reference = if condition {
    let mut copy = original.clone();
    copy.select();
    &copy
} else {
    original
};
do_big_thing(reference);

由于这个问题是关于找到一种更简洁、更简洁的方式来编写相同的东西（使用原始引用或克隆），我不能接受会在运行时增加开销或一些 unsafe 的解决方案。

playground

Answer 1

由于您不打算从您的函数中返回值（如Is it possible to return either a borrowed or owned type in Rust? 所示），您不需要使用Cow。相反，您可以在条件之外声明 copy 变量：

fn do_huge_thing_stack(original: &Big, a: usize, condition: bool) {
    let mut copy;

    let r = if condition {
        copy = original.clone();
        copy.select();
        &copy
    } else {
        original
    };

    do_big_thing(r, a);
}

编译器将确保仅在值有效时才进行引用。这会生成与Cow 版本相同的程序集，我认为the Cow version is more obvious，但它仍然是一个有用的技术。

另见：

• Are polymorphic variables allowed?

Answer 2

您可以使用std::borrow::Cow 对借用或拥有的值进行抽象：

fn do_big_thing(b: Cow<Big>, a: usize) {
    println!("{}", a + b.c);
}

fn do_huge_thing(original: &Big, a: usize, condition: bool) {
    if condition {
        let mut copy = original.clone();
        copy.select();
        do_big_thing(Cow::Owned(copy), a);
    } else {
        do_big_thing(Cow::Borrowed(original), a);
    }
}

然后您可以将其重构为：

pub fn do_huge_thing_cow(original: &Big, a: usize, condition: bool) {
    let mut r = Cow::Borrowed(original);

    if condition {
        r.to_mut().select();
    }

    do_big_thing(&r, a);
}

作为you can see，这已被优化掉。