Rust
итераторы

Вступление

Адаптеры и потребители

Методы Итератора могут быть разбиты на две различные группы:

адаптеры

Адаптеры принимают итератор и возвращают другой итератор

//         Iterator  Adapter
//             |       |
let my_map = (1..6).map(|x| x * x);
println!("{:?}", my_map);

Map { iter: 1..6 }

Обратите внимание, что значения не были перечислены, что указывает на то, что итераторы не оцениваются с высокой вероятностью - итераторы «ленивы».

Потребители

Потребители берут итератор и возвращают нечто иное, чем итератор, потребляя итератор в процессе.

//                    Iterator  Adapter       Consumer
//                        |       |              |
let my_squares: Vec<_> = (1..6).map(|x| x * x).collect();
println!("{:?}", my_squares);

[1, 4, 9, 16, 25]

Другие примеры потребителей включают find , fold и sum .

let my_squared_sum: u32 = (1..6).map(|x| x * x).sum();
println!("{:?}", my_squared_sum);

55

Короткий тест на первичность

fn is_prime(n: u64) -> bool {
    (2..n).all(|divisor| n % divisor != 0)
}

Конечно, это не быстрый тест. Мы можем остановить тестирование с квадратным корнем из n :

(2..n)
    .take_while(|divisor| divisor * divisor <= n)
    .all(|divisor| n % divisor != 0)

Пользовательский итератор

struct Fibonacci(u64, u64);

impl Iterator for Fibonacci {
    type Item = u64;
    
    // The method that generates each item
    fn next(&mut self) -> Option<Self::Item> {
        let ret = self.0;
        self.0 = self.1;
        self.1 += ret;
        
        Some(ret) // since `None` is never returned, we have an infinite iterator
    }

    // Implementing the `next()` method suffices since every other iterator
    // method has a default implementation
}

Пример использования:

// the iterator method `take()` is an adapter which limits the number of items
// generated by the original iterator
for i in Fibonacci(0, 1).take(10) {
    println!("{}", i);
}