mirror of
https://github.com/KaiserY/trpl-zh-cn
synced 2025-04-07 04:18:04 +08:00
141 lines
9.4 KiB
Markdown
141 lines
9.4 KiB
Markdown
## 使用迭代器处理元素序列
|
||
|
||
> [ch13-02-iterators.md](https://github.com/rust-lang/book/blob/main/src/ch13-02-iterators.md)
|
||
> <br>
|
||
> commit eabaaaa90ee6937db3690dc56f739116be55ecb2
|
||
|
||
迭代器模式允许你对一个序列的项进行某些处理。**迭代器**（*iterator*）负责遍历序列中的每一项和决定序列何时结束的逻辑。当使用迭代器时，我们无需重新实现这些逻辑。
|
||
|
||
在 Rust 中，迭代器是 **惰性的**（*lazy*），这意味着在调用方法使用迭代器之前它都不会有效果。例如，示例 13-10 中的代码通过调用定义于 `Vec` 上的 `iter` 方法在一个 vector `v1` 上创建了一个迭代器。这段代码本身没有任何用处：
|
||
|
||
```rust
|
||
{{#rustdoc_include ../listings/ch13-functional-features/listing-13-10/src/main.rs:here}}
|
||
```
|
||
|
||
<span class="caption">示例 13-10：创建一个迭代器</span>
|
||
|
||
迭代器被储存在 `v1_iter` 变量中。一旦创建迭代器之后，可以选择用多种方式利用它。在第三章的示例 3-5 中，我们使用 `for` 循环来遍历一个数组并在每一个项上执行了一些代码。在底层它隐式地创建并接着消费了一个迭代器，不过直到现在我们都一笔带过了它具体是如何工作的。
|
||
|
||
|
||
示例 13-11 中的例子将迭代器的创建和 `for` 循环中的使用分开。迭代器被储存在 `v1_iter` 变量中，而这时没有进行迭代。一旦 `for` 循环开始使用 `v1_iter`，接着迭代器中的每一个元素被用于循环的一次迭代，这会打印出其每一个值：
|
||
|
||
```rust
|
||
{{#rustdoc_include ../listings/ch13-functional-features/listing-13-11/src/main.rs:here}}
|
||
```
|
||
|
||
<span class="caption">示例 13-11：在一个 `for` 循环中使用迭代器</span>
|
||
|
||
在标准库中没有提供迭代器的语言中，我们可能会使用一个从 0 开始的索引变量，使用这个变量索引 vector 中的值，并循环增加其值直到达到 vector 的元素数量。
|
||
|
||
迭代器为我们处理了所有这些逻辑，这减少了重复代码并消除了潜在的混乱。另外，迭代器的实现方式提供了对多种不同的序列使用相同逻辑的灵活性，而不仅仅是像 vector 这样可索引的数据结构。让我们看看迭代器是如何做到这些的。
|
||
|
||
### `Iterator` trait 和 `next` 方法
|
||
|
||
迭代器都实现了一个叫做 `Iterator` 的定义于标准库的 trait。这个 trait 的定义看起来像这样：
|
||
|
||
```rust
|
||
pub trait Iterator {
|
||
type Item;
|
||
|
||
fn next(&mut self) -> Option<Self::Item>;
|
||
|
||
// 此处省略了方法的默认实现
|
||
}
|
||
```
|
||
|
||
注意这里有一个我们还未讲到的新语法：`type Item` 和 `Self::Item`，它们定义了 trait 的 **关联类型**（*associated type*）。第十九章会深入讲解关联类型，不过现在只需知道这段代码表明实现 `Iterator` trait 要求同时定义一个 `Item` 类型，这个 `Item` 类型被用作 `next` 方法的返回值类型。换句话说，`Item` 类型将是迭代器返回元素的类型。
|
||
|
||
`next` 是 `Iterator` 实现者被要求定义的唯一方法。`next` 一次返回迭代器中的一个项，封装在 `Some` 中，当迭代器结束时，它返回 `None`。
|
||
|
||
可以直接调用迭代器的 `next` 方法；示例 13-12 有一个测试展示了重复调用由 vector 创建的迭代器的 `next` 方法所得到的值：
|
||
|
||
<span class="filename">文件名：src/lib.rs</span>
|
||
|
||
```rust,noplayground
|
||
{{#rustdoc_include ../listings/ch13-functional-features/listing-13-12/src/lib.rs:here}}
|
||
```
|
||
|
||
<span class="caption">示例 13-12：在迭代器上（直接）调用 `next` 方法</span>
|
||
|
||
注意 `v1_iter` 需要是可变的：在迭代器上调用 `next` 方法改变了迭代器中用来记录序列位置的状态。换句话说，代码 **消费**（consume）了，或使用了迭代器。每一个 `next` 调用都会从迭代器中消费一个项。使用 `for` 循环时无需使 `v1_iter` 可变因为 `for` 循环会获取 `v1_iter` 的所有权并在后台使 `v1_iter` 可变。
|
||
|
||
另外需要注意到从 `next` 调用中得到的值是 vector 的不可变引用。`iter` 方法生成一个不可变引用的迭代器。如果我们需要一个获取 `v1` 所有权并返回拥有所有权的迭代器，则可以调用 `into_iter` 而不是 `iter`。类似的，如果我们希望迭代可变引用，则可以调用 `iter_mut` 而不是 `iter`。
|
||
|
||
### 消费迭代器的方法
|
||
|
||
`Iterator` trait 有一系列不同的由标准库提供默认实现的方法；你可以在 `Iterator` trait 的标准库 API 文档中找到所有这些方法。一些方法在其定义中调用了 `next` 方法，这也就是为什么在实现 `Iterator` trait 时要求实现 `next` 方法的原因。
|
||
|
||
这些调用 `next` 方法的方法被称为 **消费适配器**（*consuming adaptors*），因为调用它们会消耗迭代器。一个消费适配器的例子是 `sum` 方法。这个方法获取迭代器的所有权并反复调用 `next` 来遍历迭代器，因而会消费迭代器。当其遍历每一个项时，它将每一个项加总到一个总和并在迭代完成时返回总和。示例 13-13 有一个展示 `sum` 方法使用的测试：
|
||
|
||
<span class="filename">文件名：src/lib.rs</span>
|
||
|
||
```rust,noplayground
|
||
{{#rustdoc_include ../listings/ch13-functional-features/listing-13-13/src/lib.rs:here}}
|
||
```
|
||
|
||
<span class="caption">示例 13-13：调用 `sum` 方法获取迭代器所有项的总和</span>
|
||
|
||
调用 `sum` 之后不再允许使用 `v1_iter` 因为调用 `sum` 时它会获取迭代器的所有权。
|
||
|
||
### 产生其他迭代器的方法
|
||
|
||
`Iterator` trait 中定义了另一类方法，被称为 **迭代器适配器**（*iterator adaptors*），它们允许我们将当前迭代器变为不同类型的迭代器。可以链式调用多个迭代器适配器。不过因为所有的迭代器都是惰性的，必须调用一个消费适配器方法以便获取迭代器适配器调用的结果。
|
||
|
||
示例 13-14 展示了一个调用迭代器适配器方法 `map` 的例子，该 `map` 方法使用闭包来调用每个元素以生成新的迭代器。这里的闭包创建了一个新的迭代器，对其中 vector 中的每个元素都被加 1。：
|
||
|
||
<span class="filename">文件名：src/main.rs</span>
|
||
|
||
```rust,not_desired_behavior
|
||
{{#rustdoc_include ../listings/ch13-functional-features/listing-13-14/src/main.rs:here}}
|
||
```
|
||
|
||
<span class="caption">示例 13-14：调用迭代器适配器 `map` 来创建一个新迭代器</span>
|
||
|
||
不过这些代码会产生一个警告
|
||
|
||
```console
|
||
{{#include ../listings/ch13-functional-features/listing-13-14/output.txt}}
|
||
```
|
||
|
||
示例 13-14 中的代码实际上并没有做任何事；所指定的闭包从未被调用过。警告提醒了我们为什么：迭代器适配器是惰性的，而这里我们需要消费迭代器。
|
||
|
||
为了修复这个警告并消费迭代器获取有用的结果，我们将使用第十二章示例 12-1 结合 `env::args` 使用的 `collect` 方法。这个方法消费迭代器并将结果收集到一个数据结构中。
|
||
|
||
在示例 13-15 中，我们将遍历由 `map` 调用生成的迭代器的结果收集到一个 vector 中，它将会含有原始 vector 中每个元素加 1 的结果：
|
||
|
||
<span class="filename">文件名：src/main.rs</span>
|
||
|
||
```rust
|
||
{{#rustdoc_include ../listings/ch13-functional-features/listing-13-15/src/main.rs:here}}
|
||
```
|
||
|
||
<span class="caption">示例 13-15：调用 `map` 方法创建一个新迭代器，接着调用 `collect` 方法消费新迭代器并创建一个 vector</span>
|
||
|
||
因为 `map` 获取一个闭包，可以指定任何希望在遍历的每个元素上执行的操作。这是一个展示如何使用闭包来自定义行为同时又复用 `Iterator` trait 提供的迭代行为的绝佳例子。
|
||
|
||
可以链式调用多个迭代器适配器来以一种可读的方式进行复杂的操作。不过因为所有的迭代器都是惰性的，你需要调用一个消费适配器方法从迭代器适配器调用中获取结果。
|
||
|
||
### 使用捕获其环境的闭包
|
||
|
||
很多迭代器适配器接受闭包作为参数，而通常指定为迭代器适配器参数的闭包会是捕获其环境的闭包。
|
||
|
||
作为一个例子，我们使用 `filter` 方法来获取一个闭包。该闭包从迭代器中获取一项并返回一个 `bool`。如果闭包返回 `true`，其值将会包含在 `filter` 提供的新迭代器中。如果闭包返回 `false`，其值不会被包含。
|
||
|
||
示例 13-16 中使用 `filter` 和一个捕获环境中变量 `shoe_size` 的闭包来遍历一个 `Shoe` 结构体集合。它只会返回指定大小的鞋子。
|
||
|
||
<span class="filename">文件名：src/lib.rs</span>
|
||
|
||
```rust,noplayground
|
||
{{#rustdoc_include ../listings/ch13-functional-features/listing-13-16/src/lib.rs}}
|
||
```
|
||
|
||
<span class="caption">示例 13-16：使用 `filter` 方法和一个捕获 `shoe_size` 的闭包</span>
|
||
|
||
`shoes_in_my_size` 函数获取一个鞋子 vector 的所有权和一个鞋子大小作为参数。它返回一个只包含指定大小鞋子的 vector。
|
||
|
||
`shoes_in_my_size` 函数体中调用了 `into_iter` 来创建一个获取 vector 所有权的迭代器。接着调用 `filter` 将这个迭代器适配成一个只含有那些闭包返回 `true` 的元素的新迭代器。
|
||
|
||
闭包从环境中捕获了 `shoe_size` 变量并使用其值与每一只鞋的大小作比较，只保留指定大小的鞋子。最终，调用 `collect` 将迭代器适配器返回的值收集进一个 vector 并返回。
|
||
|
||
这个测试展示当调用 `shoes_in_my_size` 时，我们只会得到与指定值相同大小的鞋子。
|