我的书签
添加书签
移除书签闭包：可以捕获环境的匿名函数
ch13-01-closures.md commit a2cb72d3ad7584cc1ae3b85f715c877872f5e3ab
Rust 的 闭包（closures）是可以保存在变量中或作为参数传递给其他函数的匿名函数。你可以在一个地方创建闭包，然后在不同的上下文中执行闭包运算。不同于函数，闭包允许捕获其被定义时所在作用域中的值。我们将展示这些闭包特性如何支持代码复用和行为定制。
闭包会捕获其环境
我们首先了解如何通过闭包捕获定义它的环境中的值以便之后使用。考虑如下场景：我们的 T 恤公司偶尔会向邮件列表中的某位成员赠送一件限量版的独家 T 恤作为促销。邮件列表中的成员可以选择将他们的喜爱的颜色添加到个人信息中。如果被选中的成员设置了喜爱的颜色，他们将获得那个颜色的 T 恤。如果他没有设置喜爱的颜色，他们会获赠公司当前库存最多的颜色的款式。
有很多种方式来实现这一点。例如，使用有 Red 和 Blue 两个成员的 ShirtColor 枚举（出于简单考虑限定为两种颜色）。我们使用 Inventory 结构体来代表公司的库存，它有一个类型为 Vec<ShirtColor> 的 shirts 字段表示库存中的衬衫的颜色。Inventory 上定义的 giveaway 方法获取免费衬衫得主所喜爱的颜色（如有），并返回其获得的衬衫的颜色。初始代码如示例 13-1 所示：
文件名：src/main.rs
1. #[derive(Debug, PartialEq, Copy, Clone)]
2. enum ShirtColor {
3. Red,
4. Blue,
5. }
7. struct Inventory {
8. shirts: Vec<ShirtColor>,
9. }
11. impl Inventory {
12. fn giveaway(&self, user_preference: Option<ShirtColor>) -> ShirtColor {
13. user_preference.unwrap_or_else(|| self.most_stocked())
14. }
16. fn most_stocked(&self) -> ShirtColor {
17. let mut num_red = 0;
18. let mut num_blue = 0;
20. for color in &self.shirts {
21. match color {
22. ShirtColor::Red => num_red += 1,
23. ShirtColor::Blue => num_blue += 1,
24. }
25. }
26. if num_red > num_blue {
27. ShirtColor::Red
28. } else {
29. ShirtColor::Blue
30. }
31. }
32. }
34. fn main() {
35. let store = Inventory {
36. shirts: vec![ShirtColor::Blue, ShirtColor::Red, ShirtColor::Blue],
37. };
39. let user_pref1 = Some(ShirtColor::Red);
40. let giveaway1 = store.giveaway(user_pref1);
41. println!(
42. "The user with preference {:?} gets {:?}",
43. user_pref1, giveaway1
44. );
46. let user_pref2 = None;
47. let giveaway2 = store.giveaway(user_pref2);
48. println!(
49. "The user with preference {:?} gets {:?}",
50. user_pref2, giveaway2
51. );
52. }
示例 13-1：衬衫公司赠送场景
main 函数中定义的 store 还剩下两件蓝衬衫和一件红衬衫，可以在限量版促销活动中赠送。我们通过调用 giveaway 方法，为一个期望红衬衫的用户和一个没有特定偏好的用户进行赠送。
再次强调，这段代码有多种实现方式。这里为了专注于闭包，我们继续使用已经学习过的概念，除了 giveaway 方法体中使用了闭包。在 giveaway 方法中，我们将用户偏好作为 Option<ShirtColor> 类型的参数获取，并在 user_preference 上调用 unwrap_or_else 方法。Option 上的 unwrap_or_else 方法 由标准库定义。它接受一个无参闭包作为参数，该闭包返回一个 T 类型的值（与 Option<T> 的 Some 变体中存储的值类型相同，这里是 ShirtColor）。如果 Option<T> 是 Some 成员，则 unwrap_or_else 返回 Some 中的值。如果 Option<T> 是 None 成员，则 unwrap_or_else 调用闭包并返回闭包的返回值。
我们将闭包表达式 || self.most_stocked() 作为 unwrap_or_else 的参数。这是一个本身不获取参数的闭包（如果闭包有参数，它们会出现在两道竖杠之间）。闭包体调用了 self.most_stocked()。我们在这里定义了闭包，而 unwrap_or_else 的实现会在之后需要其结果的时候执行闭包。
运行代码会打印出：
1. $ cargo run
2. Compiling shirt-company v0.1.0 (file:///projects/shirt-company)
3. Finished `dev` profile [unoptimized + debuginfo] target(s) in 0.27s
4. Running `target/debug/shirt-company`
5. The user with preference Some(Red) gets Red
6. The user with preference None gets Blue
这里有一个有趣的地方是，我们传递了一个闭包，该闭包会在当前的 Inventory 实例上调用 self.most_stocked() 方法。标准库不需要了解我们定义的 Inventory 或 ShirtColor 类型，也不需要了解我们在这个场景中要使用的逻辑。闭包捕获了对 self（即 Inventory 实例）的不可变引用，并将其与我们指定的代码一起传递给 unwrap_or_else 方法。相比之下，函数无法以这种方式捕获其环境。
闭包类型推断和注解
函数与闭包还有更多区别。闭包通常不要求像 fn 函数那样对参数和返回值进行类型注解。函数需要类型注解是因为这些类型是暴露给用户的显式接口的一部分。严格定义这些接口对于确保所有人对函数使用和返回值的类型达成一致理解非常重要。与此相比，闭包并不用于这样暴露在外的接口：它们储存在变量中并被使用，不用命名它们或暴露给库的用户调用。
闭包通常较短，并且只与特定的上下文相关，而不是适用于任意情境。在这些有限的上下文中，编译器可以推断参数和返回值的类型，类似于它推断大多数变量类型的方式（尽管在某些罕见的情况下，编译器也需要闭包的类型注解）。
类似于变量，如果我们希望增加代码的明确性和清晰度，可以添加类型注解，但代价是是会使代码变得比严格必要的更冗长。为示例 13-1 中定义的闭包标注类型看起来如示例 13-2 中的定义一样。这个例子中，我们定义了一个闭包并将它保存在变量中，而不是像示例 13-1 那样在传参的地方定义它。
文件名：src/main.rs
1. use std::thread;
2. use std::time::Duration;
4. fn generate_workout(intensity: u32, random_number: u32) {
5. let expensive_closure = |num: u32| -> u32 {
6. println!("calculating slowly...");
7. thread::sleep(Duration::from_secs(2));
8. num
9. };
11. if intensity < 25 {
12. println!("Today, do {} pushups!", expensive_closure(intensity));
13. println!("Next, do {} situps!", expensive_closure(intensity));
14. } else {
15. if random_number == 3 {
16. println!("Take a break today! Remember to stay hydrated!");
17. } else {
18. println!(
19. "Today, run for {} minutes!",
20. expensive_closure(intensity)
21. );
22. }
23. }
24. }
26. fn main() {
27. let simulated_user_specified_value = 10;
28. let simulated_random_number = 7;
30. generate_workout(simulated_user_specified_value, simulated_random_number);
31. }
示例 13-2：为闭包的参数和返回值增加可选的类型注解
有了类型注解，闭包的语法看起来就更像函数的语法了。如下是一个对其参数加一的函数的定义与拥有相同行为闭包语法的纵向对比。这里增加了一些空格来对齐相应部分。这展示了除了使用竖线以及一些可选语法外，闭包语法与函数语法有多么地相似：
1. fn add_one_v1 (x: u32) -> u32 { x + 1 }
2. let add_one_v2 = |x: u32| -> u32 { x + 1 };
3. let add_one_v3 = |x| { x + 1 };
4. let add_one_v4 = |x| x + 1 ;
第一行展示了一个函数定义，第二行展示了一个完整标注的闭包定义。第三行闭包定义中省略了类型注解，而第四行去掉了可选的大括号，因为闭包体只有一个表达式，所以大括号是可选的。这些都是有效的闭包定义，并在调用时产生相同的行为。调用闭包是 add_one_v3 和 add_one_v4 能够编译的必要条件，因为类型将从其用法中推断出来。这类似于 let v = Vec::new();，Rust 需要类型注解或是某种类型的值被插入到 Vec 中，才能推断其类型。
对于闭包定义，编译器会为每个参数和返回值推断出一个具体类型。例如，示例 13-3 展示了一个简短的闭包定义，该闭包仅仅返回作为参数接收到的值。除了作为示例用途外，这个闭包并不是很实用。注意这个定义没有增加任何类型注解。因为没有类型注解，我们可以使用任意类型来调用这个闭包，我们在这里第一次调用时使用了 String 类型。但是如果我们接着尝试使用整数来调用 example_closure，就会得到一个错误。
文件名：src/main.rs
1. fn main() {
2. let example_closure = |x| x;
4. let s = example_closure(String::from("hello"));
5. let n = example_closure(5);
6. }
示例 13-3：尝试调用一个被推断为两个不同类型的闭包
编译器给出如下错误：
1. $ cargo run
2. Compiling closure-example v0.1.0 (file:///projects/closure-example)
3. error[E0308]: mismatched types
4. --> src/main.rs:5:29
5. |
6. 5 | let n = example_closure(5);
7. | --------------- ^- help: try using a conversion method: `.to_string()`
8. | | |
9. | | expected `String`, found integer
10. | arguments to this function are incorrect
11. |
12. note: expected because the closure was earlier called with an argument of type `String`
13. --> src/main.rs:4:29
14. |
15. 4 | let s = example_closure(String::from("hello"));
16. | --------------- ^^^^^^^^^^^^^^^^^^^^^ expected because this argument is of type `String`
17. | |
18. | in this closure call
19. note: closure parameter defined here
20. --> src/main.rs:2:28
21. |
22. 2 | let example_closure = |x| x;
23. | ^
25. For more information about this error, try `rustc --explain E0308`.
26. error: could not compile `closure-example` (bin "closure-example") due to 1 previous error
第一次使用 String 值调用 example_closure 时，编译器推断出 x 的类型以及闭包的返回类型为 String。接着这些类型被锁定进闭包 example_closure 中，如果尝试对同一闭包使用不同类型则就会得到类型错误。
捕获引用或者移动所有权
闭包可以通过三种方式捕获其环境中的值，它们直接对应到函数获取参数的三种方式：不可变借用、可变借用和获取所有权。闭包将根据函数体中对捕获值的操作来决定使用哪种方式。
在示例 13-4 中定义了一个捕获名为 list 的 vector 的不可变引用的闭包，因为只需不可变引用就能打印其值：
文件名：src/main.rs
1. fn main() {
2. let list = vec![1, 2, 3];
3. println!("Before defining closure: {list:?}");
5. let only_borrows = || println!("From closure: {list:?}");
7. println!("Before calling closure: {list:?}");
8. only_borrows();
9. println!("After calling closure: {list:?}");
10. }
示例 13-4：定义并调用一个捕获不可变引用的闭包
这个示例也展示了变量可以绑定一个闭包定义，并且我们可以像使用函数名一样，使用变量名和括号来调用该闭包。
因为同时可以有多个 list 的不可变引用，所以在闭包定义之前，闭包定义之后调用之前，闭包调用之后代码仍然可以访问 list。该代码可以编译、运行并输出：
1. $ cargo run
2. Locking 1 package to latest compatible version
3. Adding closure-example v0.1.0 (/Users/chris/dev/rust-lang/book/tmp/listings/ch13-functional-features/listing-13-04)
4. Compiling closure-example v0.1.0 (file:///projects/closure-example)
5. Finished `dev` profile [unoptimized + debuginfo] target(s) in 0.43s
6. Running `target/debug/closure-example`
7. Before defining closure: [1, 2, 3]
8. Before calling closure: [1, 2, 3]
9. From closure: [1, 2, 3]
10. After calling closure: [1, 2, 3]
接下来在示例 13-5 中，我们修改闭包体让它向 list vector 增加一个元素。闭包现在捕获一个可变引用：
文件名：src/main.rs
1. fn main() {
2. let mut list = vec![1, 2, 3];
3. println!("Before defining closure: {list:?}");
5. let mut borrows_mutably = || list.push(7);
7. borrows_mutably();
8. println!("After calling closure: {list:?}");
9. }
示例 13-5：定义并调用一个捕获可变引用的闭包
代码可以编译、运行并打印：
1. $ cargo run
2. Locking 1 package to latest compatible version
3. Adding closure-example v0.1.0 (/Users/chris/dev/rust-lang/book/tmp/listings/ch13-functional-features/listing-13-05)
4. Compiling closure-example v0.1.0 (file:///projects/closure-example)
5. Finished `dev` profile [unoptimized + debuginfo] target(s) in 0.43s
6. Running `target/debug/closure-example`
7. Before defining closure: [1, 2, 3]
8. After calling closure: [1, 2, 3, 7]
注意在 borrows_mutably 闭包的定义和调用之间不再有 println!，这是因为当 borrows_mutably 被定义时，它捕获了对 list 的可变引用。闭包在被调用后就不再被使用，这时可变借用结束。因为当可变借用存在时不允许有其它的借用，所以在闭包定义和调用之间不能有不可变引用来进行打印。可以尝试在这里添加 println! 看看你会得到什么报错信息！
即使闭包体不严格需要所有权，如果希望强制闭包获取它在环境中所使用的值的所有权，可以在参数列表前使用 move 关键字。
当将闭包传递到一个新的线程时，这个技巧特别有用，因为它将数据的所有权移动到新线程中。我们将在第十六章讨论并发时详细讨论线程以及为什么你可能需要使用它们。不过现在，我们先简要探索一下如何使用需要 move 关键字的闭包来生成一个新线程。示例 13-6 展示了如何修改示例 13-4，以便在一个新线程中而不是在主线程中打印 vector：
文件名：src/main.rs
1. use std::thread;
3. fn main() {
4. let list = vec![1, 2, 3];
5. println!("Before defining closure: {list:?}");
7. thread::spawn(move || println!("From thread: {list:?}"))
8. .join()
9. .unwrap();
10. }
示例 13-6：使用 move 来强制闭包为线程获取 list 的所有权
我们生成了一个新的线程，并给这个线程传递一个闭包作为参数来运行，闭包体打印出列表。在示例 13-4 中，闭包仅通过不可变引用捕获了 list，因为这是打印列表所需的最少的访问权限。这个例子中，尽管闭包体依然只需要不可变引用，我们还是在闭包定义前写上 move 关键字，以确保 list 被移动到闭包中。新线程可能在主线程剩余部分执行完前执行完，也可能在主线程执行完之后执行完。如果主线程维护了 list 的所有权但却在新线程之前结束并且丢弃了 list，则在线程中的不可变引用将失效。因此，编译器要求 list 被移动到在新线程中运行的闭包中，这样引用就是有效的。试着移除 move 关键字，或者在闭包定义后在主线程中使用 list，看看你会得到什么编译器报错！
将被捕获的值移出闭包和 Fn trait
一旦闭包捕获了定义它的环境中的某个值的引用或所有权（也就影响了什么会被移 进 闭包，如有），闭包体中的代码则决定了在稍后执行闭包时，这些引用或值将如何处理（也就影响了什么会被移 出 闭包，如有）。闭包体可以执行以下任一操作：将一个捕获的值移出闭包，修改捕获的值，既不移动也不修改值，或者一开始就不从环境中捕获任何值。
闭包捕获和处理环境中的值的方式会影响闭包实现哪些 trait，而 trait 是函数和结构体指定它们可以使用哪些类型闭包的方式。根据闭包体如何处理这些值，闭包会自动、渐进地实现一个、两个或全部三个 Fn trait。
FnOnce适用于只能被调用一次的闭包。所有闭包至少都实现了这个 trait，因为所有闭包都能被调用。一个会将捕获的值从闭包体中移出的闭包只会实现FnOncetrait，而不会实现其他Fn相关的 trait，因为它只能被调用一次。FnMut适用于不会将捕获的值移出闭包体，但可能会修改捕获值的闭包。这类闭包可以被调用多次。Fn适用于既不将捕获的值移出闭包体，也不修改捕获值的闭包，同时也包括不从环境中捕获任何值的闭包。这类闭包可以被多次调用而不会改变其环境，这在会多次并发调用闭包的场景中十分重要。
让我们来看示例 13-1 中使用的在 Option<T> 上的 unwrap_or_else 方法的定义：
1. impl<T> Option<T> {
2. pub fn unwrap_or_else<F>(self, f: F) -> T
3. where
4. F: FnOnce() -> T
5. {
6. match self {
7. Some(x) => x,
8. None => f(),
9. }
10. }
11. }
回忆一下，T 是表示 Option 中 Some 成员中的值的类型的泛型。类型 T 也是 unwrap_or_else 函数的返回值类型：举例来说，在 Option<String> 上调用 unwrap_or_else 会得到一个 String。
接着注意到 unwrap_or_else 函数有额外的泛型参数 F。F 是参数 f 的类型，f 是调用 unwrap_or_else 时提供的闭包。
泛型 F 的 trait bound 是 FnOnce() -> T，这意味着 F 必须能够被调用一次，没有参数并返回一个 T。在 trait bound 中使用 FnOnce 表示 unwrap_or_else 最多只会调用 f 一次。在 unwrap_or_else 的函数体中可以看到，如果 Option 是 Some，f 不会被调用。如果 Option 是 None，f 将会被调用一次。由于所有的闭包都实现了 FnOnce，unwrap_or_else 接受所有三种类型的闭包，十分灵活。
注意：函数也可以实现所有的三种
Fntraits。如果我们要做的事情不需要从环境中捕获值，则可以在需要某种实现了Fntrait 的东西时使用函数而不是闭包。举个例子，可以在Option<Vec<T>>的值上调用unwrap_or_else(Vec::new)，以便在值为None时获取一个新的空的 vector。
现在让我们来看定义在 slice 上的标准库方法 sort_by_key，看看它与 unwrap_or_else 的区别，以及为什么 sort_by_key 使用 FnMut 而不是 FnOnce 作为 trait bound。这个闭包以一个 slice 中当前被考虑的元素的引用作为参数，并返回一个可以排序的 K 类型的值。当你想按照 slice 中每个元素的某个属性进行排序时，这个函数非常有用。在示例 13-7 中，我们有一个 Rectangle 实例的列表，并使用 sort_by_key 按 Rectangle 的 width 属性对它们从低到高排序：
文件名：src/main.rs
1. #[derive(Debug)]
2. struct Rectangle {
3. width: u32,
4. height: u32,
5. }
7. fn main() {
8. let mut list = [
9. Rectangle { width: 10, height: 1 },
10. Rectangle { width: 3, height: 5 },
11. Rectangle { width: 7, height: 12 },
12. ];
14. list.sort_by_key(|r| r.width);
15. println!("{list:#?}");
16. }
示例 13-7：使用 sort_by_key 对长方形按宽度排序
代码输出：
1. $ cargo run
2. Compiling rectangles v0.1.0 (file:///projects/rectangles)
3. Finished `dev` profile [unoptimized + debuginfo] target(s) in 0.41s
4. Running `target/debug/rectangles`
5. [
6. Rectangle {
7. width: 3,
8. height: 5,
9. },
10. Rectangle {
11. width: 7,
12. height: 12,
13. },
14. Rectangle {
15. width: 10,
16. height: 1,
17. },
18. ]
sort_by_key 被定义为接收一个 FnMut 闭包的原因是它会多次调用这个闭包：对 slice 中的每个元素调用一次。闭包 |r| r.width 不捕获、修改或将任何东西移出它的环境，所以它满足 trait bound 的要求。
相比之下，示例 13-8 展示了一个只实现了 FnOnce trait 的闭包的例子，因为它从环境中移出了一个值。编译器不允许我们在 sort_by_key 中使用这个闭包：
文件名：src/main.rs
1. #[derive(Debug)]
2. struct Rectangle {
3. width: u32,
4. height: u32,
5. }
7. fn main() {
8. let mut list = [
9. Rectangle { width: 10, height: 1 },
10. Rectangle { width: 3, height: 5 },
11. Rectangle { width: 7, height: 12 },
12. ];
14. let mut sort_operations = vec![];
15. let value = String::from("closure called");
17. list.sort_by_key(|r| {
18. sort_operations.push(value);
19. r.width
20. });
21. println!("{list:#?}");
22. }
示例 13-8：尝试在 sort_by_key 上使用一个 FnOnce 闭包
这是一个刻意构造的、复杂且无效的方式，试图统计在对 list 进行排序时 sort_by_key 调用闭包的次数。该代码试图通过将闭包环境中的 value（一个 String）插入 sort_operations vector 来实现计数。闭包捕获了 value，然后通过将 value 的所有权转移给 sort_operations vector 的方式将其移出闭包。这个闭包只能被调用一次；尝试第二次调用它将无法工作，因为这时 value 已经不在闭包的环境中，无法被再次插入 sort_operations 中！因而，这个闭包只实现了 FnOnce。当我们尝试编译此代码时，会出现错误提示：value 不能从闭包中移出，因为闭包必须实现 FnMut：
1. $ cargo run
2. Compiling rectangles v0.1.0 (file:///projects/rectangles)
3. error[E0507]: cannot move out of `value`, a captured variable in an `FnMut` closure
4. --> src/main.rs:18:30
5. |
6. 15 | let value = String::from("closure called");
7. | ----- captured outer variable
8. 16 |
9. 17 | list.sort_by_key(|r| {
10. | --- captured by this `FnMut` closure
11. 18 | sort_operations.push(value);
12. | ^^^^^ move occurs because `value` has type `String`, which does not implement the `Copy` trait
13. |
14. help: consider cloning the value if the performance cost is acceptable
15. |
16. 18 | sort_operations.push(value.clone());
17. | ++++++++
19. For more information about this error, try `rustc --explain E0507`.
20. error: could not compile `rectangles` (bin "rectangles") due to 1 previous error
报错指向了闭包体中将 value 移出环境的那一行。要修复此问题，我们需要修改闭包体，使其不会将值移出环境。在环境中维护一个计数器，并在闭包体中递增其值，是计算闭包被调用次数的一个更简单直接的方法。示例 13-9 中的闭包可以在 sort_by_key 中使用，因为它只捕获了 num_sort_operations 计数器的可变引用，因此可以被多次调用：
文件名：src/main.rs
1. #[derive(Debug)]
2. struct Rectangle {
3. width: u32,
4. height: u32,
5. }
7. fn main() {
8. let mut list = [
9. Rectangle { width: 10, height: 1 },
10. Rectangle { width: 3, height: 5 },
11. Rectangle { width: 7, height: 12 },
12. ];
14. let mut num_sort_operations = 0;
15. list.sort_by_key(|r| {
16. num_sort_operations += 1;
17. r.width
18. });
19. println!("{list:#?}, sorted in {num_sort_operations} operations");
20. }
示例 13-9：允许在 sort_by_key 上使用一个 FnMut 闭包
当定义或使用涉及闭包的函数或类型时，Fn traits 十分重要。在下个小节中，我们将讨论迭代器。许多迭代器方法都接收闭包参数，因此在继续前，请记住这些闭包的细节！
