Как реализовать трейты Iterator и IntoIterator для простой структуры в Rust?

Понимание проблемы

Ваша структура BookCollection не может быть использована в цикле for, потому что цикл for в Rust требует, чтобы выражение реализовывало трейт IntoIterator. Трейт IntoIterator преобразует значение в итератор. А сам итератор должен реализовывать трейт Iterator. Давайте разберёмся по шагам.

Разница между Iterator и IntoIterator

• Iterator — это трейт для типа, который уже является итератором. Он содержит метод next(), который возвращает Option.
• IntoIterator — это трейт для типа, который можно превратить в итератор. Он содержит метод into_iter(), который потребляет исходное значение (или заимствует его) и возвращает итератор.

Решение: три шага

Вам нужно создать отдельную структуру-итератор и реализовать для неё Iterator, а затем реализовать IntoIterator для BookCollection.

Шаг 1: Создаём структуру-итератор

struct BookIterator {
    books: Vec<Book>,
    index: usize,
}

impl Iterator for BookIterator {
    type Item = Book;

    fn next(&mut self) -> Option<Self::Item> {
        if self.index < self.books.len() {
            let book = self.books.swap_remove(self.index);
            // swap_remove удаляет элемент и возвращает его, 
            // заменяя его последним элементом. Эффективно, но меняет порядок.
            // Альтернатива: self.index += 1; return Some(self.books[self.index - 1].clone());
            // Но clone() требует Clone.
            Some(book)
        } else {
            None
        }
    }
}

Важно: swap_remove не сохраняет порядок. Если порядок важен, используйте clone() (тогда Book должен реализовывать Clone) или берите по ссылке (см. альтернативный ответ).

Шаг 2: Реализуем IntoIterator для BookCollection

impl IntoIterator for BookCollection {
    type Item = Book;
    type IntoIter = BookIterator;

    fn into_iter(self) -> Self::IntoIter {
        BookIterator {
            books: self.books,
            index: 0,
        }
    }
}

Шаг 3: Используем в main

fn main() {
    let mut collection = BookCollection::new();
    collection.add_book(Book { title: "1984".to_string(), author: "Orwell".to_string() });
    collection.add_book(Book { title: "Dune".to_string(), author: "Herbert".to_string() });

    for book in collection {
        println!("{} by {}", book.title, book.author);
    }
    // После цикла collection больше недоступен, так как он был consumed.
}

Полный рабочий код

struct Book {
    title: String,
    author: String,
}

struct BookCollection {
    books: Vec<Book>,
}

impl BookCollection {
    fn new() -> Self {
        BookCollection { books: Vec::new() }
    }

    fn add_book(&mut self, book: Book) {
        self.books.push(book);
    }
}

struct BookIterator {
    books: Vec<Book>,
    index: usize,
}

impl Iterator for BookIterator {
    type Item = Book;

    fn next(&mut self) -> Option<Self::Item> {
        if self.index < self.books.len() {
            let book = self.books.swap_remove(self.index);
            Some(book)
        } else {
            None
        }
    }
}

impl IntoIterator for BookCollection {
    type Item = Book;
    type IntoIter = BookIterator;

    fn into_iter(self) -> Self::IntoIter {
        BookIterator {
            books: self.books,
            index: 0,
        }
    }
}

fn main() {
    let mut collection = BookCollection::new();
    collection.add_book(Book { title: "1984".to_string(), author: "Orwell".to_string() });
    collection.add_book(Book { title: "Dune".to_string(), author: "Herbert".to_string() });

    for book in collection {
        println!("{} by {}", book.title, book.author);
    }
}

Ключевые моменты

• Итератор потребляет исходную коллекцию. После цикла collection будет перемещена.
• Если нужно итерироваться по ссылкам (не потребляя коллекцию), реализуйте IntoIterator для &BookCollection.
• Всегда указывайте ассоциированный тип Item — это тип элементов, которые возвращает итератор.

impl<'a> IntoIterator for &'a BookCollection {
    type Item = &'a Book;
    type IntoIter = std::slice::Iter<'a, Book>;

    fn into_iter(self) -> Self::IntoIter {
        self.books.iter()
    }
}

for book in &collection {
    println!("{} by {}", book.title, book.author);
}
// collection всё ещё доступна!

impl IntoIterator for BookCollection {
    type Item = Book;
    type IntoIter = std::vec::IntoIter<Book>;

    fn into_iter(self) -> Self::IntoIter {
        self.books.into_iter()
    }
}