0029-Rust-自实现迭代器

环境

• Time 2022-05-21
• Rust 1.61.0

前言

说明

参考：https://doc.rust-lang.org/std/iter/index.html

目标

接前一节，理解 Iterator 自动实现了 IntoIterator。

自定义类型

struct MyData<T: Debug> {
    value1: T,
    value2: T,
    value3: T,
    value4: T,
}

new

impl<T: Debug> MyData<T> {
    fn new(v1: T, v2: T, v3: T, v4: T) -> Self {
        Self {
            value1: v1,
            value2: v2,
            value3: v3,
            value4: v4,
        }
    }
}

实现迭代器

struct IntoIter<T>(Vec<T>);

impl<T> Iterator for IntoIter<T> {
    type Item = T;
    fn next(&mut self) -> Option<Self::Item> {
        self.0.pop()
    }
}

生成迭代器

impl<T: Debug> IntoIterator for MyData<T> {
    type Item = T;
    type IntoIter = IntoIter<T>;
    fn into_iter(self) -> IntoIter<T> {
        let vec = vec![self.value4, self.value3, self.value2, self.value1];
        IntoIter(vec)
    }
}

使用迭代器

因为 for 循环可以迭代 IntoIterator，所以第一个循环好理解。

fn main() {
    for value in MyData::new(88, 44, 66, 22) {
        println!("value: {value:?}")
    }

    for value in MyData::new(88, 44, 66, 22).into_iter() {
        println!("value: {value:?}")
    }
}

自动实现 IntoIterator

第二个循环不出错的原因是标准库对所有实现了 Iterator 的自动实现了 IntoIterator。
标准库实现如下：

impl<I: Iterator> IntoIterator for I {
    type Item = I::Item;
    type IntoIter = I;

    #[inline]
    fn into_iter(self) -> I {
        self
    }
}

总结

理解 for 循环上不调用和调用了 into_iter 的区别，标准库自动对 Iterator 实现了 IntoIterator。

附录