24 KiB
闭包closure
关于闭包这个词语由来已久，自上世纪60年代就由Scheme语言引进，之后，被广泛用于函数式编程语言中，进入21世纪后，各种现代化的编程语言也都不约而同的把闭包作为核心特性纳入到语言设计中来。那么到底何为闭包？
闭包是一种匿名函数，它可以赋值给变量也可以作为参数传递给其它函数，不同于函数的是，它允许捕获调用者作用域中的值，例如：
fn main() {
let x = 1;
let sum = |y| x + y;
assert_eq!(3, sum(2));
}
上面的代码展示了非常简单的闭包sum，它拥有一个入参y，同时捕获了作用域中的x的值，因此调用sum(2)意味着将2(参数y)跟1（x）进行相加,最终返回它们的和:3。
可以看到sum非常符合闭包的定义：可以赋值给变量，允许捕获调用者作用域中的值。
使用闭包来简化代码
传统函数实现
想象一下，我们要进行健身，用代码怎么实现(写代码什么鬼，健身难道不应该去健身房嘛？答曰：健身太累了，还是虚拟健身好，点到为止)？这里是我的想法：
use std::thread;
use std::time::Duration;
// 开始健身，好累，我得发出声音：muuuu...
fn muuuuu(intensity: u32) -> u32 {
println!("muuuu.....");
thread::sleep(Duration::from_secs(2));
intensity
}
fn workout(intensity: u32, random_number: u32) {
if intensity < 25 {
println!(
"今天活力满满, 先做 {} 个俯卧撑!",
muuuuu(intensity)
);
println!(
"旁边有妹子在看，俯卧撑太low, 再来 {} 组卧推!",
muuuuu(intensity)
);
} else {
if random_number == 3 {
println!("昨天练过度了，今天还是休息下吧！");
} else {
println!(
"昨天练过度了，今天干干有氧, 跑步 {} 分钟!",
muuuuu(intensity)
);
}
}
}
fn main() {
// 强度
let intensity = 10;
// 随机值用来决定某个选择
let random_number = 7;
// 开始健身
workout(intensity, random_number);
}
可以看到，在健身时我们根据想要的强度来调整具体的动作，然后调用muuuuu函数来开始健身。这个程序本身很简单，没啥好说的，但是假如未来不用muuuu函数了，是不是得把所有muuuu都替换成，比如说woooo? 如果muuuu出现了几十次，那意味着我们要修改几十处地方。
函数变量实现
一个可行的办法是，把函数赋值给一个变量，然后通过变量调用:
fn workout(intensity: u32, random_number: u32) {
let action = muuuuu;
if intensity < 25 {
println!(
"今天活力满满, 先做 {} 个俯卧撑!",
action(intensity)
);
println!(
"旁边有妹子在看，俯卧撑太low, 再来 {} 组卧推!",
action(intensity)
);
} else {
if random_number == 3 {
println!("昨天练过度了，今天还是休息下吧！");
} else {
println!(
"昨天练过度了，今天干干有氧, 跑步 {} 分钟!",
action(intensity)
);
}
}
}
经过上面修改后，所有的调用都通过action来完成，若未来声(动)音(作)变了，只要修改为let action = woooo即可。
但是问题又来了,若intensity也变了怎么办？例如变成action(intensity + 1)，那你又得哐哐哐修改几十处调用。
该怎么办？没太好的办法了，只能祭出大杀器：闭包。
闭包实现
上面提到intensity要是变化怎么办，简单，使用闭包来捕获它，这是我们的拿手好戏:
fn workout(intensity: u32, random_number: u32) {
let action = || {
println!("muuuu.....");
thread::sleep(Duration::from_secs(2));
intensity
};
if intensity < 25 {
println!(
"今天活力满满, 先做 {} 个俯卧撑!",
action()
);
println!(
"旁边有妹子在看，俯卧撑太low, 再来 {} 组卧推!",
action()
);
} else {
if random_number == 3 {
println!("昨天练过度了，今天还是休息下吧！");
} else {
println!(
"昨天练过度了，今天干干有氧, 跑步 {} 分钟!",
action()
);
}
}
}
fn main() {
// 动作次数
let intensity = 10;
// 随机值用来决定某个选择
let random_number = 7;
// 开始健身
workout(intensity, random_number);
}
在上面代码中，无论你要修改什么，只要修改闭包action的实现即可，其它地方只负责调用，完美解决了我们的问题！
Rust闭包在形式上借鉴了Smalltalk和Ruby语言，与函数最大的不同就是它的参数是通过|parm1|的形式进行声明，如果是多个参数就|param1, param2,...|, 下面给出闭包的形式定义：
|param1, param2,...| {
语句1;
语句2;
返回表达式
}
如果只有一个返回表达式的话，定义可以简化为：
|param1| 返回表达式
上例中还有两点值得注意:
- 闭包中最后一行表达式返回的值，就是闭包执行后的返回值，因此
action()调用返回了intensity的值10 let action = ||...只是把闭包赋值给变量action，并不是把闭包执行后的结果赋值给action，因此这里action就相当于闭包函数，可以跟函数一样进行调用：action()
闭包的类型推导
Rust是静态语言，因此所有的变量都具有类型，但是得益于编译器的强大类型推导能力，在很多时候我们并不需要显式的去声明类型，但是显然函数并不在此列，必须手动为函数的所有参数和返回值指定类型，原因在于函数往往会作为API提供给你的用户，因此你的用户必须在使用时知道传入参数的类型和返回值类型。
与函数相反，闭包并不会作为API对外提供，因此它可以享受编译器的类型推导能力，无需标注参数和返回值的类型。
为了增加代码可读性，有时候我们会显式的给类型进行标注，出于同样的目的，也可以给闭包标注类型：
let sum = |x: i32, y: i32| -> i32 {
x + y
}
与之相比，不标注类型的闭包声明会更简洁些: let sum = |x, y| x + y, 需要注意的是，针对sum闭包，如果你不在后续代码中使用它，编译器会提示你为x,y添加类型标注，因为它缺乏必要的上下文：
let sum = |x, y| x + y;
let v = sum(1,2);
这里我们用使用了sum，同时把1传给了x，2传给了y，因此编译器才可以推导出x,y的类型为i32。
下面展示了同一个功能的函数和闭包实现形式:
fn add_one_v1 (x: u32) -> u32 { x + 1 }
let add_one_v2 = |x: u32| -> u32 { x + 1 };
let add_one_v3 = |x| { x + 1 };
let add_one_v4 = |x| x + 1 ;
可以看出第一行的函数和后面的闭包其实在形式上是非常接近的，同时三种不同的闭包也展示了三种不同的使用方式：省略参数、返回值和花括号对。
虽然类型推导很好用，但是它不是泛型，当编译器推导出一种类型后，它就会一直使用该类型:
let example_closure = |x| x;
let s = example_closure(String::from("hello"));
let n = example_closure(5);
首先，在s中，编译器为x推导出类型String，但是紧接着n试图用5这个整型去调用闭包，跟编译器之前推导的String类型不符，因此报错：
error[E0308]: mismatched types
--> src/main.rs:5:29
|
5 | let n = example_closure(5);
| ^
| |
| expected struct `String`, found integer // 期待String类型，却发现一个整数
| help: try using a conversion method: `5.to_string()`
结构体中的闭包
假设我们要实现一个简易缓存，功能是获取一个值，然后将其缓存起来，那么可以这样设计：
- 一个闭包用于获取值
- 一个变量，用于存储该值
可以使用结构体来代表缓存对象，最终设计如下：
struct Cacher<T>
where
T: Fn(u32) -> u32,
{
query: T,
value: Option<u32>,
}
等等，我都跟着这本教程学完Rust基础了，为何还有我不认识的东东？Fn(u32) -> u32是什么鬼？别急，先回答你第一个问题：骚年，too young too naive，你以为Rust的语法特性就基础入门那一些嘛？太年轻了！如果是长征，你才刚到赤水河.
其实，可以看的出这一长串是T的特征约束，再结合之前的已知信息：query是一个闭包，大概可以推测出，Fn(u32) -> u32是一个特征，用来表示T是一个闭包类型？Bingo，恭喜你，答对了！
那为什么不用具体的类型来标注query呢？原因很简单，每一个闭包实例都有独属于自己的类型，甚至于两个签名一模一样的闭包，它们的类型都是不同的，因此你无法用一个统一的类型来标注query闭包。
而标准库提供的Fn系列特征，再结合特征约束，就很好的解决了这个问题. T: Fn(u32) -> u32意味着query的类型是T，该类型必须实现了相应的闭包特征Fn(u32) -> u32。从特征的角度来看它长得非常反直觉，但是如果从闭包的角度来看又极其符合直觉，不得不佩服Rust团队的鬼才设计。。。
特征Fn(u32) -> u32从表面来看，就对闭包形式进行了显而易见的限制：该闭包拥有一个u32类型的参数，同时返回一个u32类型的值.
需要注意的是，其实Fn特征不仅仅适用于闭包，还适用于函数，因此上面的
query字段除了使用闭包作为值外，还能使用一个具名的函数来作为它的值
接着，为缓存实现方法:
impl<T> Cacher<T>
where
T: Fn(u32) -> u32,
{
fn new(query: T) -> Cacher<T> {
Cacher {
query,
value: None,
}
}
// 先查询缓存值`self.value`，若不存在，则调用`query`加载
fn value(&mut self, arg: u32) -> u32 {
match self.value {
Some(v) => v,
None => {
let v = (self.query)(arg);
self.value = Some(v);
v
}
}
}
}
上面的缓存有一个很大的问题：只支持u32类型的值，若我们想要缓存String类型，显然就行不通了，因此需要将u32替换成泛型E，该练习就留给读者自己完成，具体代码可以参考这里
捕获作用域中的值
在之前代码中，我们一直在用闭包的匿名函数特性(赋值给变量)，然而闭包还拥有一项函数所不具备的特性: 捕获作用域中的值。
fn main() {
let x = 4;
let equal_to_x = |z| z == x;
let y = 4;
assert!(equal_to_x(y));
}
上面代码中，x并不是闭包equal_to_x的参数，但是它依然可以去使用x，因为x在equal_to_x的作用域范围内。
对于函数来说，就算你把函数定义在main函数体中，它也不能访问x:
fn main() {
let x = 4;
fn equal_to_x(z: i32) -> bool {
z == x
}
let y = 4;
assert!(equal_to_x(y));
}
报错如下：
error[E0434]: can't capture dynamic environment in a fn item // 在函数中无法捕获动态的环境
--> src/main.rs:5:14
|
5 | z == x
| ^
|
= help: use the `|| { ... }` closure form instead // 使用闭包替代
如上所示，编译器准确的告诉了我们错误，同时甚至给出了提示：使用闭包来替代函数，这种聪明令我有些无所适从，总感觉会显得我很笨。
闭包对内存的影响
当闭包从环境中捕获一个值时，会分配内存去存储这些值。对于有些场景来说，这种额外的内存分配会成为一种负担。与之相比，函数就不会去捕获这些环境值，因此定义和使用函数不会拥有这种内存负担。
三种Fn特征
闭包捕获变量有三种途径，恰好对应函数参数的三种传入方式：转移所有权、可变借用、不可变借用，因此相应的Fn特征也有三种:
FnOnce, 该类型的闭包会拿走被捕获变量的所有权。Once顾名思义，说明该闭包只能拿走所有权一次:
fn main() {
let x = vec![1,2,3];
let len_is = move |z| z == x.len();
let len_is_not = move |z| z != x.len();
println!("{}",len_is(3));
println!("{}",len_is_not(4));
}
move关键字用来告诉闭包，捕获变量的所有权而不是进行借用，因此x变量的所有权将被首选转移到len_is中，紧接着len_is_not又试图获取x的所有权，此时自然会报错：
error[E0382]: use of moved value: `x` // 使用已经没有所有权的x
--> src/main.rs:5:22
|
2 | let x = vec![1,2,3];
| - move occurs because `x` has type `Vec<i32>`, which does not implement the `Copy` trait
3 | - 发生所有权转移因为x的类型是Vec<i32>,它没有实现Copy特征
4 | let len_is = move |z| z == x.len();
| -------- - variable moved due to use in closure // 在此处x所有权被转移
| |
| value moved into closure here
5 | let len_is_not = move |z| z != x.len();
| ^^^^^^^^ - use occurs due to use in closure
| |
| value used here after move // 试图再次转移所有权
这里面有一个很重要的提示，因为Vec<i32>没有实现Copy特征，所以会报错，那么我们试试有实现Copy的类型：
fn main() {
let x = 3;
let equal_to = move |z| z == x;
let not_equal_to = move |z| z != x;
println!("{}",equal_to(3));
println!("{}",not_equal_to(4));
}
上面代码中，x的类型是i32，该类型实现了Copy特征，因此虽然我们使用了move关键字让闭包拿走x的所有权，但是由于x是可复制的，闭包仅仅是复制了x的值，编译后，顺利通过:
true
true
FnMut, 它以可变借用的方式捕获了环境中的值，因此可以修改该值：
fn main() {
let mut s = String::new();
let update_string = |str| s.push_str(str);
update_string("hello");
println!("{:?}",s);
}
在闭包中，我们调用s.push_str去改变外部s的字符串值，因此这里捕获了它的可变借用，运行下试试：
error[E0596]: cannot borrow `update_string` as mutable, as it is not declared as mutable
--> src/main.rs:5:5
|
4 | let update_string = |str| s.push_str(str);
| ------------- - calling `update_string` requires mutable binding due to mutable borrow of `s`
| |
| help: consider changing this to be mutable: `mut update_string`
5 | update_string("hello");
| ^^^^^^^^^^^^^ cannot borrow as mutable
虽然报错了，但是编译器给出了非常清晰的提示，想要在闭包内部捕获可变借用，需要把该闭包声明为可变类型，也就是update_string要修改为mut update_string:
fn main() {
let mut s = String::new();
let mut update_string = |str| s.push_str(str);
update_string("hello");
println!("{:?}",s);
}
这种写法有点反直觉，相比起来前面的move更符合使用和阅读习惯。但是如果你忽略update_string的类型，仅仅把它当成一个普通变量，那么这种声明就比较合理了。
再来看一个复杂点的：
fn main() {
let mut s = String::new();
let update_string = |str| s.push_str(str);
exec(update_string);
println!("{:?}",s);
}
fn exec<'a, F: FnMut(&'a str)>(mut f: F) {
f("hello")
}
这段代码非常清晰的说明了update_string实现了FnMut特征
Fn特征，它以不可变借用的方式捕获环境中的值 让我们把上面的代码中exec的F泛型参数类型修改为Fn(&'a str)，然后运行看看结果:
error[E0525]: expected a closure that implements the `Fn` trait, but this closure only implements `FnMut`
--> src/main.rs:4:26 // 期望闭包实现的是`Fn`特征，但是它只实现了`FnMut`特征
|
4 | let update_string = |str| s.push_str(str);
| ^^^^^^-^^^^^^^^^^^^^^
| | |
| | closure is `FnMut` because it mutates the variable `s` here
| this closure implements `FnMut`, not `Fn` //闭包实现的是FnMut，而不是Fn
5 |
6 | exec(update_string);
| ---- the requirement to implement `Fn` derives from here
从报错中很清晰的看出，我们的闭包实现的是FnMut特征，但是在exec中却给它标注了Fn特征，因此产生了不匹配，再来看看正确的不可变借用方式：
fn main() {
let s = "hello, ".to_string();
let update_string = |str| println!("{},{}",s,str);
exec(update_string);
println!("{:?}",s);
}
fn exec<'a, F: Fn(String) -> ()>(f: F) {
f("world".to_string())
}
在这里，因为无需改变s，因此闭包中只对s进行了不可变借用，那么在exec中，将其标记为Fn特征就完全正确。
move和Fn
在上面，我们讲到了move关键字对于FnOnce特征的重要性，但是实际上使用了move的闭包依然可能实现了Fn或FnMut特征。
因为，一个闭包实现了哪种Fn特征取决于该闭包如何使用被捕获的变量，而不是取决于闭包如何捕获它们。move本身强调的就是后者：闭包如何捕获变量:
fn main() {
let s = String::new();
let update_string = move || println!("{}",s);
exec(update_string);
}
fn exec<F: Fn()>(f: F) {
f()
}
我们在上面的闭包中使用了move关键字，因此我们的闭包捕获了它，但是由于闭包对s的使用仅仅是不可变借用，因为该闭包实际上还实现了Fn特征，如exec函数所示。
细心的读者肯定发现我在上段中使用了一个还字，这是什么意思呢？因为该闭包不仅仅实现了Fn特征，还实现了FnOnce特征，因此将代码修改成下面这样，依然可以编译：
fn main() {
let s = String::new();
let update_string = move || println!("{}",s);
exec(update_string);
}
fn exec<F: FnOnce()>(f: F) {
f()
}
三种Fn的关系
实际上，一个闭包并不仅仅实现某一种Fn特征，规则如下：
- 所有的闭包都实现了
FnOnce特征，因此任何一个闭包都至少可以被调用一次 - 没有使用
move的闭包实现了FnMut特征 - 不需要对捕获变量进行改变的闭包实现了
Fn特征
用一段代码来简单诠释上述规则:
fn main() {
let s = String::new();
let update_string = || println!("{}",s);
exec(update_string);
exec1(update_string);
exec2(update_string);
}
fn exec<F: FnOnce()>(f: F) {
f()
}
fn exec1<F: FnMut()>(mut f: F) {
f()
}
fn exec2<F: Fn()>(f: F) {
f()
}
虽然，闭包只是对s进行了不可变借用，实际上，它可以适用于任何一种Fn特征：三个exec函数说明了一切。强烈建议读者亲自动手试试各种情况下使用的Fn特征，更有助于加深这方面的理解。
如果还是有疑惑？没关系，我们来看看这三个特征的简化版源码：
pub trait Fn<Args> : FnMut<Args> {
extern "rust-call" fn call(&self, args: Args) -> Self::Output;
}
pub trait FnMut<Args> : FnOnce<Args> {
extern "rust-call" fn call_mut(&mut self, args: Args) -> Self::Output;
}
pub trait FnOnce<Args> {
type Output;
extern "rust-call" fn call_once(self, args: Args) -> Self::Output;
}
看到没？从特征约束能看出来Fn的前提是实现FnMut，FnMut的前提是实现FnOne，因此要实现Fn就要同时实现FnMut和FnOnce，这段源码从侧面印证了之前规则的正确性。
从源码中还能看出一点：Fn获取&self，FnMut获取&mut self，而FnOnce获取self.
在实际项目中，建议先使用Fn特征，然后编译器会告诉你正误以及该如何选择。
闭包作为函数返回值
看到这里，相信大家对于如何使用闭包作为函数参数，已经很熟悉了，但是如果要使用闭包作为函数返回值，该如何做？
先来看一段代码：
fn factory() -> Fn(i32) -> i32 {
let num = 5;
|x| x + num
}
let f = factory();
let answer = f(1);
assert_eq!(6, answer);
上面这段代码看起来还是蛮正常的，用Fn(i32) -> i32特征来代表|x| x + num，非常合理嘛，肯定可以编译通过, 可惜理想总是难以照进现实，编译器给我们报了一大堆错误，先挑几个重点来看看：
fn factory<T>() -> Fn(i32) -> i32 {
| ^^^^^^^^^^^^^^ doesn't have a size known at compile-time // 该类型在编译器没有固定的大小
Rust要求函数的参数和返回类型，必须有固定的内存大小，例如i32就是4个字节，引用类型是8个字节，总之，绝大部分类型都有固定的大小，但是不包括特征，因为特征类似接口，对于编译器来说，无法知道它后面藏的真实类型是什么，因为也无法得知具体的大小。
但是我们又无法知道闭包的具体类型，该怎么办呢？再看看报错提示:
help: use `impl Fn(i32) -> i32` as the return type, as all return paths are of type `[closure@src/main.rs:11:5: 11:21]`, which implements `Fn(i32) -> i32`
|
8 | fn factory<T>() -> impl Fn(i32) -> i32 {
嗯，编译器提示我们加一个impl关键字，哦，这样一说，读者可能就想起来了，impl Trait可以用来返回一个实现了指定特征的类型，那么这里impl Fn(i32) -> i32的返回值形式，说明我们要返回一个闭包类型，它实现了Fn(i32) -> i32特征。
完美解决，但是，在[特征]那一章，我们提到过，impl Trait的返回方式有一个非常大的局限，就是你只能返回同样的类型，例如:
fn factory(x:i32) -> impl Fn(i32) -> i32 {
let num = 5;
if x > 1{
move |x| x + num
} else {
move |x| x - num
}
}
运行后，编译器报错:
error[E0308]: `if` and `else` have incompatible types
--> src/main.rs:15:9
|
12 | / if x > 1{
13 | | move |x| x + num
| | ---------------- expected because of this
14 | | } else {
15 | | move |x| x - num
| | ^^^^^^^^^^^^^^^^ expected closure, found a different closure
16 | | }
| |_____- `if` and `else` have incompatible types
|
嗯，提示很清晰：if和else分支中返回了不同的闭包类型，这就很奇怪了，明明这两个闭包长的一样的，好在细心的读者应该回想起来，本章节前面咱们有提到：就算签名一样的闭包，类型也是不同的，因此在这种情况下，就无法再使用impl Trait的方式去返回闭包。
怎么办？再看看编译器提示，里面有这样一行小字:
= help: consider boxing your closure and/or using it as a trait object
哦，相信你已经恍然大悟，可以用特征对象！只需要用Box的方式即可实现：
fn factory(x:i32) -> Box<dyn Fn(i32) -> i32> {
let num = 5;
if x > 1{
Box::new(move |x| x + num)
} else {
Box::new(move |x| x - num)
}
}
至此，闭包作为函数返回值就已完美解决，若以后你再遇到报错时，一定要仔细阅读编译器的提示，很多时候，转角都能遇到爱。
闭包的生命周期
这块儿内容在进阶生命周期章节中有讲，这里就不再赘述，读者可移步此处进行回顾。