rust - 如何在 Rust 中编写自己的可链接迭代器函数？

Question

<分区>

Answer 1

首先，您必须创建一个新类型来实现您的迭代器。最简单的方法是包装一个你想要定位的类型的迭代器:

struct Double<T: ?Sized> {
    inner: T,
}

现在，如果内部类型 T 是正确的类型，您只需为该类型实现 Iterator 特性。像这样的东西:

impl<T> Iterator for Double<T>
where T: Iterator<Item = i32> {
    type Item = i32;
    fn next(&mut self) -> Option<i32> {
        self.inner.next().map(|x| 2*x)
    }
}

就是这样!您只需要添加一个 Double 构造函数。最符合人体工程学的解决方案是扩展 Iterator。像这样:

trait Doubler : Iterator<Item = i32> {
    fn into_double(self) -> Double<Self>;
}

impl<T> Doubler for T
where T: Iterator<Item = i32> {
    fn into_double(self) -> Double<Self> {
        Double { inner: self }
    }
}

用法示例(playground):

fn main() {
    vec![1, 2, 3]
        .into_iter()
        .into_double()
        .for_each(|x| println!("{}", x));
}