RustBook——Cacher学习示例+注释

最新推荐文章于 2025-08-14 12:19:48 发布

原创最新推荐文章于 2025-08-14 12:19:48 发布 · 778 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#RustBook #Rust #缓存 #泛型 #测试

Rust 专栏收录该内容

3 篇文章

订阅专栏

本文详细介绍了 Rust 语言中缓存器 Cacher 的实现原理与使用方法。通过具体示例，展示了如何利用 Cacher 结构体进行高效的数据存储与计算，避免重复运算，提高程序性能。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

文章目录

RustBook——Cacher学习示例+注释
- 代码
- 测试

RustBook——Cacher学习示例+注释

代码

use std::collections::HashMap;
use std::cmp::Eq;
use std::hash::Hash;

/// # 缓存器结构体
/// 
/// # Arguments
/// * calculation - 用于计算value的函数
/// * map - 用于存储数据的容器HashMap
///
/// # Examples
/// ```
/// use rust_demo::Cacher;
///
/// let mut cache = Cacher::new(|x: u32| x * x);
///
/// assert_eq!(25, *cache.value(5));
/// assert_eq!(100, *cache.value(10));
/// assert_eq!(25, *cache.value(5));
/// ```
pub struct Cacher<T, K, V>
where
    T: Fn(K) -> V,
    K: Eq + Hash + Copy,
{
    calculation: T,
    map: HashMap<K, V>,
}

impl<T, K, V> Cacher<T, K, V>
where
    T: Fn(K) -> V,
    K: Eq + Hash + Copy,
{
    /// # 工厂方法
    /// 用于初次创建Cacher
    ///
    /// # Arguments
    /// * calculation - 用于计算value的函数
    pub fn new(calculation: T) -> Cacher<T, K, V> {
        Cacher {
            calculation,
            map: HashMap::new(),
        }
    }

    /// # 取值的方法
    /// 当map中没有key对应的value时，调用calculation生成value
    ///
    /// * arg - 用于取值的key
    pub fn value(&mut self, arg: K) -> &V {
        self.map.entry(arg).or_insert((self.calculation)(arg))
    }
}

测试

#[cfg(test)]
mod tests {

    use super::*;

    #[test]
    fn test_cache() {
        let mut cache = Cacher::new(|x: u32| x * x);

        assert_eq!(25, *cache.value(5));
        assert_eq!(100, *cache.value(10));
        assert_eq!(25, *cache.value(5));
    }
}