1
0
mirror of https://github.com/google/comprehensive-rust.git synced 2025-06-15 13:50:27 +02:00
Files
comprehensive-rust/src/iterators/collect.md

49 lines
1.7 KiB
Markdown
Raw Normal View History

---
minutes: 5
---
# `collect`
The [`collect`][3] method lets you build a collection from an [`Iterator`][2].
```rust,editable
fn main() {
let primes = vec![2, 3, 5, 7];
let prime_squares = primes.into_iter().map(|p| p * p).collect::<Vec<_>>();
println!("prime_squares: {prime_squares:?}");
}
```
<details>
- Any iterator can be collected in to a `Vec`, `VecDeque`, or `HashSet`.
Iterators that produce key-value pairs (i.e. a two-element tuple) can also be
collected into `HashMap` and `BTreeMap`.
Show the students the definition for `collect` in the standard library docs.
There are two ways to specify the generic type `B` for this method:
- With the "turbofish": `some_iterator.collect::<COLLECTION_TYPE>()`, as shown.
The `_` shorthand used here lets Rust infer the type of the `Vec` elements.
- With type inference: `let prime_squares: Vec<_> = some_iterator.collect()`.
Rewrite the example to use this form.
## More to Explore
- If students are curious about how this works, you can bring up the
[`FromIterator`][1] trait, which defines how each type of collection gets
built from an iterator.
- In addition to the basic implementations of `FromIterator` for `Vec`,
`HashMap`, etc., there are also more specialized implementations which let you
do cool things like convert an `Iterator<Item = Result<V, E>>` into a
`Result<Vec<V>, E>`.
- The reason type annotations are often needed with `collect` is because it's
generic over its return type. This makes it harder for the compiler to infer
the correct type in a lot of cases.
</details>
[1]: https://doc.rust-lang.org/std/iter/trait.FromIterator.html
[2]: https://doc.rust-lang.org/std/iter/trait.Iterator.html
[3]: https://doc.rust-lang.org/std/iter/trait.Iterator.html#method.collect