从 Rc<RefCell<T>> 映射到 Ref<'_, U>

Question

我有以下结构：

struct Inner;

enum State {
    A(Inner),
    B,
}

struct Owner {
    state: Rc<RefCell<State>>,
}

我想提供如下接口：

impl Owner {
    fn a(&self) -> Option<Ref<'_, Inner>>;
}

owner.a() 将返回的位置：

• 当owner.state 匹配State::A(s)，然后Some([some ref to s])（这会增加owner.state 的强计数并在丢弃时减少它，同时保证RefCell 借用属性）；
• 否则，None。

这有可能吗？我试过查看Ref::map，但似乎无法使其与Rc<RefCell<_>> 和Option 一起使用。

我现在正在做的解决方法是：

impl Owner {
    fn with_a(&self, mut callback: impl FnMut(&Inner)) {
        match *self.state.borrow() {
            State::A(ref inner) => callback(inner),
            _ => {}
        }
    }
}

playground

Answer 1

owner.a() 不需要增加Rc 的强计数，因为返回的Ref 的生命周期已经与Rc 相关联。如果您使用 100% Safe Rust 并且它可以编译，那么您将不会遇到任何内存安全问题，因此您不必担心手动记账，例如显式增加或减少 Rc 的强计数。 Rc 甚至没有公开更改强计数的方法，这是有充分理由的。这是您想要的函数签名的实现：

use std::rc::Rc;
use std::cell::{RefCell, Ref};

struct Inner;

enum State {
    A(Inner),
    B,
}

struct Owner {
    state: Rc<RefCell<State>>,
}

impl Owner {
    fn a(&self) -> Option<Ref<'_, Inner>> {
        let state_ref = self.state.borrow();
        match *state_ref {
            State::B => None,
            _ => Some(Ref::map(state_ref, |state| match state {
                State::A(inner) => inner,
                _ => unreachable!(),
            })),
        }
    }
}

playground