Rust中如何编写可接收trait引用迭代器的泛型函数？

咱们先来拆解下你遇到的问题：你写的foo函数要求迭代器的元素直接实现trait A，但用iter()遍历容器时，得到的是元素的引用（比如&Something），而Rust不会自动为引用类型实现你的自定义trait——这就是编译器报错的原因。你不想局限于只能传切片，因为那样就没法用map这类迭代器适配器做复杂处理了。

别担心，我给你推荐一个最通用的解决方案，几乎能覆盖所有你需要的场景：为引用类型手动实现你的trait。

具体怎么做？ #

只需要给你的trait A加一个针对&T的实现（当T本身实现A时），这样所有引用迭代器的元素就自动满足A的约束了。代码示例如下：

trait A {
    fn f(&self);
}

// 当T实现A时，自动让&T也实现A
impl<'a, T: A + ?Sized> A for &'a T {
    fn f(&self) {
        // 解引用后调用原类型的方法即可
        (**self).f();
        // 或者更简洁的写法：T::f(*self); 效果完全一样
    }
}

为什么这招管用？ #

加上这个实现后，你原来的foo函数完全不需要修改，就能兼容所有情况：

1. 传值迭代器：foo(aa.into_iter()) 照常工作，因为元素是T，本身实现A
2. 传引用迭代器：foo(aa.iter()) 现在也没问题，因为&T已经实现了A
3. 传复杂迭代器链：比如foo(bars.iter().map(|bar| bar.a))，只要bar.a是impl A或者&impl A，都能满足约束

有没有其他方案？ #

如果你不想为引用实现trait，也可以修改foo的泛型约束，让它接受能解引用到A实现者的迭代器元素，比如用Deref trait：

use std::ops::Deref;

fn foo<It>(it: It)
where
    It: Iterator,
    It::Item: Deref,
    <It::Item as Deref>::Target: A,
{
    it.for_each(|x| x.f());
}

不过这个方案有个小局限：它只支持能解引用到A实现者的类型（比如引用、Box等），没法直接接受值迭代器（比如into_iter()产生的元素是T，而普通T没有实现Deref）。所以如果你需要同时兼容值和引用迭代器，还是第一种方案更靠谱。

再啰嗦两句 #

Rust的trait系统是显式的，不会自动为引用类型推导trait实现——这是为了保证类型安全，但也给我们留了手动扩展的空间。通过为引用实现trait，我们能让泛型函数的适配性大大提升，既保留了泛型的灵活性，又解决了引用迭代器的兼容问题。

备注：内容来源于stack exchange，提问作者petersohn