一、為什么需要內(nèi)部可變性？

通常，Rust 編譯器通過靜態(tài)分析確保：

• 同一時刻只能存在一個可變引用，或任意多個不可變引用；
• 引用始終保持有效。

這種嚴格的借用規(guī)則使得許多內(nèi)存錯誤在編譯階段就能被捕獲，但也因此在某些場景下過于保守。

例如，當我們需要在不可變對象的內(nèi)部修改狀態(tài)時（比如記錄日志、計數(shù)等），就需要借助內(nèi)部可變性。通過內(nèi)部可變性，我們可以在外部保持不可變的同時，通過封裝的方式實現(xiàn)內(nèi)部數(shù)據(jù)的變更，而這些變更的安全性則由運行時檢查保證。

二、RefCell<T>：運行時借用規(guī)則的守護者

與 Box<T> 和 Rc<T> 不同，RefCell<T> 使用運行時而非編譯時來檢查借用規(guī)則。它提供了兩個核心方法：

• borrow() 返回一個 Ref<T> 智能指針，相當于不可變引用。
• borrow_mut() 返回一個 RefMut<T> 智能指針，相當于可變引用。

每當調(diào)用 borrow 或 borrow_mut 時，RefCell<T> 都會在內(nèi)部記錄當前的借用狀態(tài)。如果試圖同時獲取多個可變引用，或者在已有可變引用的情況下獲取不可變引用，RefCell<T> 將在運行時觸發(fā) panic，從而防止數(shù)據(jù)競爭。

例如，下述代碼嘗試在同一作用域內(nèi)創(chuàng)建兩個可變借用，就會觸發(fā) panic：

let cell = RefCell::new(5);
let _borrow1 = cell.borrow_mut();
let _borrow2 = cell.borrow_mut(); // 此處將 panic: already borrowed: BorrowMutError

這種設(shè)計的優(yōu)點在于，它允許我們在某些靜態(tài)檢查無法覆蓋的場景下依然保證數(shù)據(jù)安全；缺點則是這些檢查會帶來一定的運行時開銷，同時可能將錯誤暴露在生產(chǎn)環(huán)境中。

三、實際案例：使用 RefCell<T> 編寫 Mock 對象

在測試代碼中，我們常常需要模擬一些真實對象的行為（即所謂的“測試替身”或 mock 對象），以驗證代碼邏輯是否正確。

假設(shè)我們有一個 Messenger 接口，其 send 方法只接受不可變引用。這在編寫 mock 對象時會帶來問題：我們希望在調(diào)用 send 時記錄下發(fā)送的信息，但由于方法簽名只接受 &self，直接修改內(nèi)部狀態(tài)會違反 Rust 的借用規(guī)則。

解決方案是使用 RefCell<T> 來包裝內(nèi)部的可變狀態(tài)。

例如，我們可以這樣定義一個 MockMessenger：

struct MockMessenger {
    sent_messages: RefCell<Vec<String>>,
}

impl MockMessenger {
    fn new() -> MockMessenger {
        MockMessenger {
            sent_messages: RefCell::new(vec![]),
        }
    }
}

impl Messenger for MockMessenger {
    fn send(&self, message: &str) {
        // 雖然 `self` 是不可變引用，但我們可以通過 `RefCell<T>` 在運行時獲取可變引用
        self.sent_messages.borrow_mut().push(String::from(message));
    }
}

這樣，在測試中，我們可以通過調(diào)用 borrow() 來檢查內(nèi)部保存的消息，而無需修改 Messenger trait 的定義。

RefCell<T> 的內(nèi)部借用計數(shù)確保了我們在使用時不會違反借用規(guī)則。

四、結(jié)合 Rc<T> 實現(xiàn)多所有權(quán)的可變數(shù)據(jù)

有時我們希望多個所有者可以共享同一份數(shù)據(jù)，并且能夠修改其中的值。這時可以結(jié)合使用 Rc<T> 和 RefCell<T>。Rc<T> 允許多個所有者共享數(shù)據(jù)，而 RefCell<T> 則允許我們在不可變引用的上下文中修改數(shù)據(jù)。

例如，下例展示了如何創(chuàng)建一個共享的可變值，并通過多個所有者修改它：

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

enum List {
    Cons(Rc<RefCell<i32>>, Rc<List>),
    Nil,
}

use List::{Cons, Nil};

fn main() {
    let value = Rc::new(RefCell::new(5));
    let a = Rc::new(Cons(Rc::clone(&value), Rc::new(Nil)));
    let b = Cons(Rc::clone(&value), Rc::clone(&a));
    let c = Cons(Rc::clone(&value), Rc::clone(&a));

    // 修改內(nèi)部值
    *value.borrow_mut() += 10;

    // 輸出 a, b, c 中存儲的值都會反映內(nèi)部值的改變
    println!("a after modification: {:?}", a);
}

通過這種方式，我們既能享受多所有權(quán)的便利，又能保持內(nèi)部數(shù)據(jù)的可變性。這在需要共享狀態(tài)的場景下非常有用，但需要注意的是，這種模式僅適用于單線程場景；如果在多線程環(huán)境中，則應(yīng)使用 Mutex<T> 等線程安全的數(shù)據(jù)結(jié)構(gòu)。

五、總結(jié)

內(nèi)部可變性：允許在不可變引用中修改內(nèi)部數(shù)據(jù)。通過封裝 unsafe 代碼，將運行時檢查借用規(guī)則的責任交給 RefCell<T>。

RefCell 的特點：在運行時記錄不可變與可變借用的狀態(tài)，一旦違反借用規(guī)則會導(dǎo)致 panic。這為某些靜態(tài)檢查無法覆蓋的場景提供了解決方案。

應(yīng)用場景：

• Mock 對象：在測試中記錄調(diào)用信息，滿足接口要求而無需修改方法簽名。
• 多所有權(quán)與可變性結(jié)合：結(jié)合 Rc<T> 和 RefCell<T>，可以實現(xiàn)多個所有者共享并修改數(shù)據(jù)，但僅適用于單線程環(huán)境。

內(nèi)部可變性為 Rust 程序員提供了一種在嚴格的編譯時借用檢查之外，依然保持內(nèi)存安全的靈活方案。只需謹慎使用，理解其運行時檢查的局限性，即可在設(shè)計上更好地解決某些復(fù)雜場景的問題。

希望這篇博客能夠幫助你更好地理解 RefCell<T> 及其在 Rust 中的實際應(yīng)用。

以上為個人經(jīng)驗，希望能給大家一個參考，也希望大家多多支持腳本之家。

最新評論