返回不可变引用后面的可变引用，传递给函数

Question

如何返回一个位于不可变引用后面的可变引用，作为参数传递给函数，如何处理？

struct Foo { i: i32 }

struct Bar<'b> {
    f: &'b mut Foo
}

impl<'a: 'b, 'b> Bar<'b> {
    fn func(&'a self) -> &'b mut Foo {
         self.f
    }
}

fn main() {
    let mut foo = Foo { i: 1 };

    let bar = Bar { f: &mut foo };
    bar.func(); 
}

给出以下错误：

error[E0389]: cannot borrow data mutably in a `&` reference
 --> src/main.rs:9:14
  |
8 |         fn func(&'a self) -> &'b mut Foo {
  |                 -------- use `&'a mut self` here to make mutable
9 |              self.f
  |              ^^^^^^ assignment into an immutable reference

我（有点）理解编译器在这里试图阻止什么。但我对错误消息 assignment into an immutable reference 感到困惑。 self（或self.f 的内部？）究竟分配了什么？

我写了下面的代码来模拟这个问题，得到了同样的错误信息，与上面的不同，我能够理解。代码如下：

fn main() {
    let mut foo = Foo { i: 1 };

    let bar = Bar { f: &mut foo };
    let pbar = &bar;

    pbar.f.i = 2; // assignment into an immutable reference
}

在第一个示例中，它是否试图将可变引用 f 移出 self（因为 &mut 不是 Copy 类型），将其视为不可变引用 self 中的突变，因此出现错误消息assignment into an immutable reference?

Answer 1

您不能从不可变引用创建可变引用。这意味着您需要将&self更改为&mut self：

impl<'a: 'b, 'b> Bar<'b> {
    fn func(&'a mut self) -> &'b mut Foo {
         self.f
    }
}

现在您的变量需要是可变的，以便您可以为方法获取可变引用：

let mut bar = Bar { f: &mut foo };
bar.func(); 

究竟是什么被分配给自我（或self.x内部？）？

错误消息可能有点不对劲。您的代码中没有赋值，但您返回的是一个 mutable 引用。可变引用允许您在这里做的唯一额外事情是分配 self.f 或 self.f.i。

这个错误消息当然可以改进，但它确实包含一个提示，使&'a self 可变以解决问题。

现在，你原来的问题：

如何返回一个位于不可变引用后面的可变引用，作为参数传递给函数，如何处理？

Rust 为内部可变性提供了多种容器类型，例如 Cell 和 RefCell。这些类型负责确保远离编译器的正确性并使其成为运行时检查。将RefCell 应用于您的代码的一种方法可能是这样的：

use std::cell::RefCell;
use std::ops::DerefMut;

struct Foo { i: i32 }

struct Bar<'b> {
    // store the data in a RefCell for interior mutability
    f: &'b RefCell<Foo>
}

impl<'a: 'b, 'b> Bar<'b> {
    // Return a RefMut smart pointer instead of mutable ref, but hide the implementation,
    // just exposing it as something that can be mutably dereferenced as a Foo
    fn func(&'a self) -> impl DerefMut<Target = Foo> + 'b {
         self.f.borrow_mut()
    }
}

fn main() {
    let foo = RefCell::new(Foo { i: 1 });
    let bar = Bar { f: &foo };

    let mut f = bar.func(); 
    f.i = 3;
}