如何扩展其他Crate中定义的Trait？以及如何为Rust标准库的std::iter::Iterator Trait添加自定义迭代器方法？

嘿，刚好这俩问题都是Rust里Trait扩展的常见场景，我给你捋清楚怎么搞～

Rust有个孤儿规则：你不能直接给别人的类型实现别人的Trait（除非你同时拥有类型或者Trait其中之一）。但要扩展第三方Trait的功能，我们可以用**扩展Trait（Extension Trait）**的技巧——说白了就是自己造一个新Trait，把想要加的方法放进去，然后给目标类型实现这个新Trait。

核心思路就是：

• 自定义一个Trait，里面包含你想要添加的方法
• 用泛型实现，让所有实现了目标Trait（比如std::iter::Iterator）的类型，都自动实现你的扩展Trait

按照上面的思路，我们可以轻松实现你想要的链式调用效果。直接上代码：

首先定义扩展Trait和实现：

use std::iter::Iterator;

// 自定义扩展Trait，命名可以随便取，比如IteratorExt就很直观
pub trait IteratorExt: Iterator {
    // 按照你的需求，返回包含符合条件元素索引的Option<Vec<usize>>
    fn find_all<P>(self, predicate: P) -> Option<Vec<usize>>
    where
        P: Fn(&Self::Item) -> bool;
}

// 给所有实现了Iterator的类型，自动实现我们的扩展Trait
impl<I> IteratorExt for I
where
    I: Iterator,
{
    fn find_all<P>(self, predicate: P) -> Option<Vec<usize>>
    where
        P: Fn(&Self::Item) -> bool,
    {
        // 用enumerate拿到索引+元素，过滤出符合条件的，再收集索引
        let indexes: Vec<_> = self
            .enumerate()
            .filter(|(_, item)| predicate(item))
            .map(|(idx, _)| idx)
            .collect();

        // 空的话返回None，否则返回Some，对应你示例里的unwrap
        if indexes.is_empty() {
            None
        } else {
            Some(indexes)
        }
    }
}

然后测试你的示例代码（注意这里要处理引用类型，因为iter()返回的是&i32，所以predicate里要解引用）：

fn main() {
    let test_data = vec![1,2,3,1,1,1,1];
    // 这里用&&element是因为iter()的item是&i32，predicate参数是&&i32，或者写成*element == 1也可以
    let indexes_with_val_1 = test_data.iter().find_all(|&&element| element == 1).unwrap();
    assert_eq!(indexes_with_val_1, vec![0,3,4,5,6]);
}

如果觉得返回Option麻烦，也可以直接返回Vec<usize>（空Vec表示没有匹配项），这样就不用unwrap了，修改方法返回值和最后返回逻辑就行，完全看你的需求。

这种扩展方式不仅适用于Iterator，任何第三方Trait都可以这么搞——只要自定义扩展Trait，然后给目标Trait的所有实现类型做泛型实现就行，完全符合Rust的规则。

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