如何在Rust中为HashSet的每个元素应用返回Result的函数？

你的代码报错是因为for_each要求闭包返回()，但elem.check()返回的是Result<(), Error>，类型不匹配。要实现“所有元素check成功则返回Ok，否则返回第一个错误”的需求，Rust迭代器提供了专门的适配器try_for_each来处理这种场景。

步骤1：为泛型T添加约束 #

首先需要确保泛型类型T拥有check方法，所以需要定义一个trait并添加到State的泛型约束中：

// 定义包含check方法的trait
pub trait Checkable {
    fn check(&self) -> Result<(), Error>;
}

// 更新State的泛型约束，要求T实现Checkable
pub struct State<T: Clone + Eq + Checkable> {
    pub db: HashSet<T>,
}

步骤2：使用try_for_each替代for_each #

try_for_each会遍历迭代器中的每个元素，调用闭包并处理返回的Result：如果所有闭包调用都返回Ok(())，则最终返回Ok(())；一旦遇到第一个Err，就会立即终止遍历并返回该错误。修改后的check函数如下：

fn check(&self) -> Result<(), Error> {
    self.db.iter().try_for_each(|elem| elem.check())
}

替代实现：使用fold手动处理 #

如果你想手动控制错误传播，也可以用fold方法实现相同逻辑：

fn check(&self) -> Result<(), Error> {
    self.db.iter().fold(Ok(()), |acc, elem| {
        // 先检查之前的结果是否成功，失败则直接返回
        acc?;
        // 调用当前元素的check方法
        elem.check()
    })
}

两种实现效果一致，但try_for_each是Rust为这种场景设计的专用方法，代码更简洁直观。

内容的提问来源于stack exchange，提问作者terett