函数有自己的类型,可以像使用基础类型一样使用函数,包括将函数保存在变量中、保存在 vec 中、声明在结构体成员字段中。闭包函数也是函数,也有自己的类型定义。不过,函数实际上是指针类型,在 rust 所有权中属于借用的关系。
我们声明一个 Vec 对象集,并使用闭包函数做排序。接下来我们自定义一个 City 结构体类型,Vec 中的每个元素都是 City 类型,后面会根据结构体中的 population 属性做排序。
#[derive(Debug)]
struct City {
name: String,
population: i64,
country: String,
}
我们初始化 cities 变量,并向其中依次追加 3 个元素。vec 追加元素使用 push 函数,push 过程也会触发底层数组的扩容。rust 也提供了初始化指定 vec 容量的函数 with_capacity。
fn main() {
let mut cities: Vec<City> = vec![];
cities.push(City {
name: ("上海".to_string()),
population: (123i64),
country: ("China".to_string()),
});
cities.push(City {
name: ("北京".to_string()),
population: (12i64),
country: ("China".to_string()),
});
cities.push(City {
name: ("广州".to_string()),
population: (124i64),
country: ("China".to_string()),
});
sort_cities(&mut cities);
println!("{:?}", cities)
}
sort_cities 算是这篇文章的关键函数了,函数内部使用到了闭包。想要彻底明白下面的逻辑,首先要了解 sort_by_key 函数,这个函数的入参是一个闭包函数。
fn sort_cities(cities: &mut Vec<City>) {
cities.sort_by_key(|city| -city.population);
}
声明中的 &mut self 表示方法能够可变地借用 self 实例,支持对它进行修改。闭包函数的类型是 FnMut(&T),返回的类型需要满足 Ord 约束。从声明中可以看出,闭包函数是有类型的,同时,返回的类型必须是可排序的。
代码部分,闭包函数的写法特别简单,两个竖线是闭包函数识别的标志。尽管闭包函数要求有返回值,但代码中并没有体现,都是 rust 自己默认推断实现的。
我们尝试对代码部分的闭包声明做写法上的调整,下面的闭包声明都是相同的。
// 加了代码块
fn sort_cities(cities: &mut Vec<City>) {
cities.sort_by_key(|city| {-city.population});
}
// 加了入参声明
fn sort_cities(cities: &mut Vec<City>) {
cities.sort_by_key(|city: &City| -city.population);
}
// 加了返回值声明
fn sort_cities(cities: &mut Vec<City>) {
cities.sort_by_key(|city| -> i64 { -city.population });
}
闭包引用对象的所有权
在闭包函数体中使用外部的变量,会导致外部变量的所有权发生变更吗?如果是 go 语言,外部变量在闭包中是通过引用来访问的。具体到 rust 语言上,我们可以通过自定义闭包函数来做验证。
我们在原代码基础上做了简化,is_long_name 函数体中直接访问外部变量 sh,并将 sh 的名字重新赋值一个新的变量 name。编译器检测到 sh.name 发生了所有权转移,编译失败。
fn main() {
let sh = City {
name: ("上海".to_string()),
population: (123i64),
country: ("China".to_string()),
};
let is_long_name = || -> bool {
let name = sh.name;
if name.len() > 2 {
true
} else {
false
}
};
is_long_name();
println!("{:?}", sh);
}
在闭包声明的开头,大家也经常看到过 move 关键字。那么,这个 move 是用来解决什么问题的呢?
针对这个例子,我们专门在闭包内对 sh.name 重新做了赋值运算,变量所有权发生转移也符合认知。那么,我们现在直接使用 sh.name 进行长度判断,对应截图中的第 13 行代码。左右差异点只有:右边的闭包声明中追加了 move 关键字。
文章来源:https://www.toymoban.com/news/detail-616001.html
rust 中闭包取得数据有两种方式:转移和借用。move 会导致使用到的外部变量发生所有权转移。右边的例子中,闭包中的变量 sh.name 的所有权就发生了转移,导致编译失败。文章来源地址https://www.toymoban.com/news/detail-616001.html
到了这里,关于rust 闭包函数的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!