当前位置：首页 > article >正文

【Rust设计模式之Fold模式】

article 2024/11/16 14:36:23

Rust设计模式之Fold

Fold （折叠）
如Rust Collection中的fold方法，是消耗迭代器适配器，将闭包应用于每一个元素，并将结果返回一样。Fold模式的中心思想也是如此，将元素折叠处理，最终计算出新的元素。

代码示例：
iterator fold:

fn main() {
    let even_sum = (1..=10).fold(0, |acc, num| if num % 2 == 0 { acc + num } else { acc });
    println!("{even_sum:?}");
}

而我们自己实现的Fold模式，一般用于映射数据结构，不同于面向对象的原地修改，由于Rust的引用和借用以及不可变性，生成新的结构从而替代老的结构调理更为清晰。
代码示例：

mod ast {
    pub enum Stmt {
        Node(Box<Node>),
        Let(Box<Info>, Box<Node>),
    }

    pub struct Info {
        value: String,
    }

    pub enum Node {
        IntLit(i64),
    }
}
mod fold {

    use crate::ast::{self, Info, Node, Stmt};

    pub trait Folder {
        // A leaf node just returns the node itself. In some cases, we can do this
        // to inner nodes too.
        fn fold_Info(&mut self, n: Box<Info>) -> Box<Info> {
            n
        }
        // Create a new inner node by folding its children.
        fn fold_stmt(&mut self, s: Box<Stmt>) -> Box<Stmt> {
            match *s {
                Stmt::Node(e) => Box::new(Stmt::Node(self.fold_Node(e))),
                Stmt::Let(n, e) => Box::new(Stmt::Let(self.fold_Info(n), self.fold_Node(e))),
            }
        }
        fn fold_Node(&mut self, e: Box<Node>) -> Box<Node> {
            e
        }
    }
}