Как правильно вернуть итератор (или любой другой трейт) из функции в Rust?

Проблема: трейты не имеют известного размера на этапе компиляции

Вы абсолютно правы: трейты в Rust — это не конкретные типы, а контракты, описывающие поведение. Компилятору нужно знать точный размер возвращаемого значения, чтобы выделить память на стеке. Поскольку Iterator может быть реализован тысячами разных типов (каждый со своим размером), вернуть Iterator напрямую невозможно.

Решение 1: impl Trait — самый идиоматичный способ

Начиная с Rust 1.26, вы можете использовать impl Trait в возвращаемом типе. Это говорит компилятору: "Я верну какой-то конкретный тип, реализующий трейт, но я не хочу называть его имя". Компилятор сам выведет тип.

fn get_even_numbers(numbers: Vec<i32>) -> impl Iterator<Item = i32> {
    numbers.into_iter().filter(|&x| x % 2 == 0)
}

Как это работает: Компилятор видит, что возвращается Filter<IntoIter<i32>, closure> — конкретный тип, который реализует Iterator<Item = i32>. Он подставляет этот тип неявно. Это статическая диспетчеризация — никаких накладных расходов на виртуальные вызовы.

Ограничения impl Trait

• Вы можете вернуть только один конкретный тип. Нельзя вернуть Vec::into_iter() в одной ветке if и std::iter::empty() в другой.
• Нельзя использовать в трейтах (для методов трейтов используйте associated types или Box).
• Работает только для одного трейта. Если нужно вернуть тип, реализующий два трейта, используйте Box или обобщения.

Решение 2: Box<dyn Iterator> — когда нужна динамическая диспетчеризация

Если вам нужно вернуть разные типы итераторов (например, в зависимости от условия), используйте Box<dyn Iterator>:

fn get_iterator(flag: bool) -> Box<dyn Iterator<Item = i32>> {
    if flag {
        Box::new(0..5) // Range — один тип
    } else {
        Box::new(vec![1, 2, 3].into_iter()) // IntoIter — другой тип
    }
}

Недостатки:

• Динамическая диспетчеризация через vtable — небольшой оверхед (один косвенный вызов).
• Выделение памяти в куче (хотя для итераторов это обычно не критично).
• Нужно импортировать Box и писать dyn.

Решение 3: Конкретный тип (когда вы знаете, что возвращаете)

Если вы точно знаете, какой тип итератора возвращаете, можно указать его явно. Это редкость, но для учебных целей:

use std::iter::Filter;
use std::vec::IntoIter;

fn get_even_numbers(numbers: Vec<i32>) -> Filter<IntoIter<i32>, fn(&i32) -> bool> {
    numbers.into_iter().filter(|&x| x % 2 == 0)
}

Проблема: тип замыкания анонимен, поэтому пришлось бы использовать fn указатель, что не всегда удобно.

Общее правило для возврата трейтов

• Используйте impl Trait для возврата одного конкретного типа (95% случаев).
• Используйте Box<dyn Trait> когда нужно вернуть разные типы или когда impl Trait нельзя использовать (например, в рекурсивных типах).
• Используйте конкретные типы только если вы мазохист или пишете библиотеку, где важна каждая микросекунда.

Пример с двумя трейтами

Если нужно вернуть тип, реализующий и Iterator, и Debug:

fn get_debug_iterator() -> impl Iterator<Item = i32> + Debug {
    // ... возвращаем что-то, что реализует оба трейта
}

Надеюсь, теперь стало понятнее! Rust даёт гибкость: статическая диспетчеризация для производительности, динамическая — для гибкости. Выбирайте по ситуации.

fn get_even_numbers(numbers: Vec<i32>) -> Vec<i32> {
    numbers.into_iter().filter(|&x| x % 2 == 0).collect()
}