如何传递Rc <refcell <box <mystruct>&gt;&gt;接受Rc的函数<refcell <box <dyn mytrait =“”>&gt;&gt;?

时间:2015-06-16 07:26:30

标签: rust smart-pointers trait-objects

我最初问过这个问题here,但它被标记为重复,虽然它在我看来只复制了它的一部分,所以我创建了一个更具体的一个:

请考虑以下代码:

use std::rc::Rc;

trait MyTrait {
    fn trait_func(&self);
}

struct MyStruct1;

impl MyStruct1 {
    fn my_fn(&self) {
        // do something
    }
}

impl MyTrait for MyStruct1 {
    fn trait_func(&self) {
        // do something
    }
}

fn my_trait_fn(t: Rc<dyn MyTrait>) {
    t.trait_func();
}

fn main() {
    let my_str: Rc<MyStruct1> = Rc::new(MyStruct1);
    my_trait_fn(my_str.clone());
    my_str.my_fn();
}

此代码工作正常。现在我想更改trait_func的定义以接受&mut self,但它不起作用,因为Rc仅适用于不可变数据。我使用的解决方案是将MyTrait包装到RefCell

use std::cell::RefCell;

fn my_trait_fn(t: Rc<RefCell<Box<dyn MyTrait>>>) {
    t.borrow_mut().trait_func();
}

fn main() {
    let my_str: Rc<RefCell<Box<MyStruct1>>> = Rc::new(RefCell::new(Box::new(MyStruct1)));
    my_trait_fn(my_str.clone());
    my_str.my_fn();
}

当我编译它时,我收到一个错误:

error[E0308]: mismatched types
  --> src/main.rs:27:17
   |
27 |     my_trait_fn(my_str.clone());
   |                 ^^^^^^^^^^^^^^ expected trait MyTrait, found struct `MyStruct1`
   |
   = note: expected type `std::rc::Rc<std::cell::RefCell<std::boxed::Box<dyn MyTrait + 'static>>>`
              found type `std::rc::Rc<std::cell::RefCell<std::boxed::Box<MyStruct1>>>`
   = help: here are some functions which might fulfill your needs:
           - .into_inner()

解决这个问题的最佳方法是什么?

1 个答案:

答案 0 :(得分:2)

(这个答案的旧版本基本上建议克隆底层结构并将其放在一个新的Rc<RefCell<Box<MyTrait>>对象中;这在稳定的Rust上是必要的,但是在那之后不久,Rc<RefCell<MyStruct>>会毫不费力地强迫Rc<RefCell<MyTrait>>。)

放弃Box<>包装,您可以轻松地将Rc<RefCell<MyStruct>>强制转换为Rc<RefCell<MyTrait>>。回顾克隆一个Rc<T>只会产生另一个Rc<T>,将refcount增加一个,你可以这样做:

use std::rc::Rc;
use std::cell::RefCell;

trait MyTrait {
    fn trait_func(&self);
}

#[derive(Clone)]
struct MyStruct1;
impl MyStruct1 {
    fn my_fn(&self) {
        // do something
    }
}

impl MyTrait for MyStruct1 {
    fn trait_func(&self) {
        // do something
    }
}

fn my_trait_fn(t: Rc<RefCell<MyTrait>>) {
    t.borrow_mut().trait_func();
}

fn main() {
    // (The type annotation is not necessary here, but helps explain it.
    // If the `my_str.borrow().my_fn()` line was missing, it would actually
    // be of type Rc<RefCell<MyTrait>> instead of Rc<RefCell<MyStruct1>>,
    // essentially doing the coercion one step earlier.)
    let my_str: Rc<RefCell<MyStruct1>> = Rc::new(RefCell::new(MyStruct1));
    my_trait_fn(my_str.clone());
    my_str.borrow().my_fn();
}

作为一般规则,看看你是否可以通过引用来获取所包含的值,理想情况下甚至通常是fn my_trait_fn<T: MyTrait>(t: &T)和类似的,通常可以通过自动引用和解引用来调用my_str.borrow()关心其余的事情 - 而不是整个Rc<RefCell<MyTrait>>事。