在其 for 循环中访问迭代器

Question

我正在 rust 中测试一个自定义的双向迭代器，但我遇到了错误

error[E0382]: borrow of moved value: `iter`
   --> src/main.rs:40:13
    |
37  |     let mut iter = BiIterator::from(vec![1, 2, 3]);
    |         -------- move occurs because `iter` has type `BiIterator<i32>`, which does not implement the `Copy` trait
38  |     for i in iter {
    |              ----
    |              |
    |              `iter` moved due to this implicit call to `.into_iter()`
    |              help: consider borrowing to avoid moving into the for loop: `&iter`
39  |         if i == 3 {
40  |             iter.position(0);
    |             ^^^^^^^^^^^^^^^^ value borrowed here after move
    |
note: this function takes ownership of the receiver `self`, which moves `iter`

reproducable code

Answer 1

for 循环正在获取迭代器的所有权。要在循环体内使用迭代器，您需要将for 循环脱糖为while let：

while let Some(i) = iter.next() {
    if i == 3 {
        iter.position(0);
    }
    println!("{}", i);
}

如果您想让您的迭代器在 for 循环中可用，您需要付出一些额外的努力。你可以为&BiIterator实现Iterator，为pos使用interior mutability，所以position()可以取&self：

// don't need RefCell because we're just mutating a number
use std::cell::Cell;

struct BiIterator<T> {
    values: Vec<T>,
    pos: Cell<usize>,
}

impl<T: Clone> Iterator for &BiIterator<T> {
    type Item = T;
    fn next(&mut self) -> Option<Self::Item> {
        self.pos.set(self.pos.get() + 1);
        self.values.get(self.pos.get() - 1).cloned()
    }
}

impl<T> BiIterator<T> {
    pub fn position(&self, new_pos: usize) {
        self.pos.set(new_pos);
    }
    pub fn prev(&mut self) {
        self.pos.set(self.pos.get() - 1);
    }
}

impl<T> std::convert::From<Vec<T>> for BiIterator<T> {
    fn from(input: Vec<T>) -> Self {
        Self {
            values: input,
            pos: Cell::new(0),
        }
    }
}

通过这些更改，您终于可以按照编译器的原始建议使用for i in &iter：

fn main() {
    let iter = BiIterator::from(vec![1, 2, 3]);
    for i in &iter {
        if i == 3 {
            iter.position(0);
        }
        println!("{}", i);
    }
}

以上实现了一些额外的变化：

• 不需要绑定在T 上的Copy，因为您只是在克隆它。任何T 即Copy 都会自动成为Clone，并且可以预期克隆它只是执行廉价复制。
• 边界是almost never needed on the struct；把它们放在impl 上。
• 将 if/else if let/else 链替换为 match，或者更好的是，替换为 Option::cloned()。