🦔

0023-RefCell

に公開

RefCellはCellと同じく構造体メンバーにアクセスして参照を取得し処理することができる
RefCell<T>で定義されているときに参照を借用(borrow)したときはRef<T>を、可変参照を借用したときはRefMut<T>を処理する
借用チェックはコンパイル時でなく、実行時に実施される

pub trait Messenger {
    fn send(&self, msg: &str);
}

pub struct LimitTracker<'a, T: 'a + Messenger> {
    messenger: &'a T,
    value: usize,
    max: usize,
}

impl <'a, T> LimitTracker<'a, T> where T: Messenger {
    pub fn new(messenger: &T, max: usize) -> LimitTracker<T>{
        LimitTracker{messenger,value: 0, max}
    }

    pub fn set_value(&mut self, value: usize) {
        self.value = value;

        let percentage_of_max = self.value as f64 / self.max as f64;

        if percentage_of_max >= 0.75 && percentage_of_max < 0.9 {
            // 警告: 割り当ての75%以上を使用してしまいました
            self.messenger.send("Warning: You've used up over 75% of your quota!");
        } else if percentage_of_max >= 0.9 && percentage_of_max < 1.0 {
            // 切迫した警告: 割り当ての90%以上を使用してしまいました
            self.messenger.send("Urgent warning: You've used up over 90% of your quota!");
        } else if percentage_of_max >= 1.0 {
            // エラー: 割り当てを超えています
            self.messenger.send("Error: You are over your quota!");
        }
    }
}

#[cfg(test)]
mod tests {
    use super::*;
    use std::cell::RefCell;
    #[allow(dead_code)]
    struct MockMessenger {
        sent_messages: RefCell<Vec<String>>,
        val: u32,
    }
    
    impl MockMessenger {
        fn new() -> MockMessenger {
            MockMessenger { sent_messages: RefCell::new(vec![]), val: 0 }
        }
    }

    impl Messenger for MockMessenger {
        fn send(&self, message: &str) {
            self.sent_messages.borrow_mut().push(String::from(message));
        }
    }

    #[test]
    fn it_sends_an_over_75_percent_warning_message() {
        let mock_messenger = MockMessenger::new();
        let mut limit_tracker = LimitTracker::new(&mock_messenger, 100);

        limit_tracker.set_value(80);

        assert_eq!(mock_messenger.sent_messages.borrow().len(), 1);
        assert_eq!(mock_messenger.sent_messages.borrow()[0], "Warning: You've used up over 75% of your quota!");
    }
}

このコードはコンパイルできるがs1を借用しようとしたところでpanic!になる

use std::cell::RefCell;
fn main() {
    let s = RefCell::new(String::from("hello"));

    let s0 = s.borrow();
    let s1 = s.borrow_mut(); // ここで実行時エラー
}

Discussion