0026-Rust-自实现迭代器

环境

• Time 2022-05-21
• Rust 1.61.0

前言

说明

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

目标

接前一节，实现结构体中的元素不是 Copy 的迭代器。

自定义类型

struct MyData {
    value1: MyValue,
    value2: MyValue,
    value3: MyValue,
    value4: MyValue,
}
#[derive(Debug)]
struct MyValue(usize);

Default

impl Default for MyData {
    fn default() -> Self {
        Self {
            value1: MyValue(88),
            value2: MyValue(44),
            value3: MyValue(66),
            value4: MyValue(22),
        }
    }
}

实现迭代器

#[derive(Debug)]
struct MyIterator(Vec<MyValue>);

impl Iterator for MyIterator {
    type Item = MyValue;

    fn next(&mut self) -> Option<Self::Item> {
        self.0.pop()
    }
}

生成迭代器

impl IntoIterator for MyData {
    type Item = MyValue;
    type IntoIter = MyIterator;

    fn into_iter(self) -> Self::IntoIter {
        let vec = vec![self.value4, self.value3, self.value2, self.value1];
        MyIterator(vec)
    }
}

使用迭代器

fn main() {
    for value in MyData::default() {
        println!("value: {value:?}")
    }
}

总结

针对自定义结构体中的元素不是 Copy 的实现迭代器。

附录