函数的参数需要标注类型。
函数最后的表达式将作为返回值。也可在函数内使用 return
语句来提前返回值。
// 和 C/C++ 不一样,Rust 的函数定义位置是没有限制的
fn main() {
// 我们可以在这里使用函数,后面再定义它
fizzbuzz(100);
}
// 一个 “不” 返回值的函数。实际上会返回一个单元类型 `()`。
fn fizzbuzz(n: u32) -> () {
println!("fizzbuzz");
}
// 当函数返回 `()` 时,函数签名可以省略返回类型
方法
方法(method)是依附于对象的函数。这些方法通过关键字 self
来访问对象中的数据和其他。方法在 impl
代码块中定义。
静态方法
struct Point {
x: f64,
y: f64,
}
// 实现的代码块,`Point` 的所有方法都在这里给出
impl Point {
// 这是一个静态方法(static method)
// 静态方法不需要被实例调用
// 这类方法一般用作构造器(constructor)
fn origin() -> Point {
Point { x: 0.0, y: 0.0 }
}
// 另外一个静态方法,需要两个参数:
fn new(x: f64, y: f64) -> Point {
Point { x: x, y: y }
}
}
实例方法
struct Rectangle {
p1: Point,
p2: Point,
}
impl Rectangle {
// 这是一个实例方法(instance method)
// `&self` 是 `self: &Self` 的语法糖(sugar),其中 `Self` 是方法调用者的
// 类型。在这个例子中 `Self` = `Rectangle`
fn area(&self) -> f64 {
// `self` 通过点运算符来访问结构体字段
let Point { x: x1, y: y1 } = self.p1;
let Point { x: x2, y: y2 } = self.p2;
// `abs` 是一个 `f64` 类型的方法,返回调用者的绝对值
((x1 - x2) * (y1 - y2)).abs()
}
fn perimeter(&self) -> f64 {
let Point { x: x1, y: y1 } = self.p1;
let Point { x: x2, y: y2 } = self.p2;
2.0 * ((x1 - x2).abs() + (y1 - y2).abs())
}
// 这个方法要求调用者是可变的
// `&mut self` 为 `self: &mut Self` 的语法糖
fn translate(&mut self, x: f64, y: f64) {
}
}
非引用的实例方法
// `Pair` 拥有资源:两个堆分配的整型
struct Pair(Box<i32>, Box<i32>);
impl Pair {
// 这个方法会 “消耗” 调用者的资源
// `self` 为 `self: Self` 的语法糖
fn destroy(self) {
// 解构 `self`
let Pair(first, second) = self;
println!("Destroying Pair({}, {})", first, second);
// `first` 和 `second` 离开作用域后释放
}
}
闭包
闭包(closure),也叫 lambda,是一类能够捕获周围作用中变量的函数。调用闭包时,输入和返回类型两者都可以自动推导,而输入变量名必须指明。
特点:
- 声明时使用
||
替代()
将输入参数括起来。 - 函数体定界符(
{}
)对于单个表达式是可选的,其他情况必须加上。 有能力捕获外部环境的变量。
``rust fn main() { // 闭包是匿名的,这里我们将它们绑定到引用。 // 类型标注和函数的一样,不过类型标注和使用
{}` 来围住函数体都是可选的。 // 这些匿名函数(nameless function)被赋值给合适地命名的变量。 let closure_annotated = |i: i32| -> i32 { i + 1 }; let closure_inferred = |i | i + 1 ;// 译注:将闭包绑定到引用的说法可能不准。 // 据语言参考 // 闭包表达式产生的类型就是 “闭包类型”,不属于引用类型,而且确实无法对上面两个 //
closure_xxx
变量解引用。let i = 1; // 调用闭包。 println!(“closure_annotated: {}”, closure_annotated(i)); println!(“closure_inferred: {}”, closure_inferred(i));
// 没有参数的闭包,返回一个
i32
类型。 // 返回类型是自动推导的。 let one = || 1; println!(“closure returning one: {}”, one()); }
<a name="wF3dh"></a>
### 捕获
闭包中变量捕获(capture)很灵活,既可移动(move),又可借用(borrow)变量。<br />可以通过以下手段捕获变量:
- 通过引用: `&T`
- 通过可变引用: `&mut T`
- 通过值: `T`
闭包更倾向于通过引用来捕获变量,并且只在被要求时才使用其他手段。
```rust
fn main() {
use std::mem;
let color = "green";
// 这个闭包打印 `color`。它会立即借用(通过引用,`&`)`color` 并将该借用和
// 闭包本身存储到 `print` 变量中。`color` 会一直保持被借用状态直到
// `print` 离开作用域。
// `println!` 只需传引用就能使用,而这个闭包捕获的也是变量的引用,因此无需
// 进一步处理就可以使用 `println!`。
let print = || println!("`color`: {}", color);
// 调用闭包,闭包又借用 `color`。
print();
print();
let mut count = 0;
// 这个闭包使 `count` 值增加。要做到这点,它需要得到 `&mut count` 或者
// `count` 本身,但 `&mut count` 的要求没那么严格,所以我们采取这种方式。
// 该闭包立即借用 `count`。
//
// `inc` 前面需要加上 `mut`,因为闭包里存储着一个 `&mut` 变量。调用闭包时,
// 该变量的变化就意味着闭包内部发生了变化。因此闭包需要是可变的。
let mut inc = || {
count += 1;
println!("`count`: {}", count);
};
// 不可复制类型(non-copy type)。
let movable = Box::new(3);
// `mem::drop` 要求 `T` 类型本身,所以闭包将会捕获变量的值。这种情况下,
// 可复制类型将会复制给闭包,从而原始值不受影响。不可复制类型必须移动
// (move)到闭包中,因而 `movable` 变量在这里立即移动到了闭包中。
let consume = || {
println!("`movable`: {:?}", movable);
mem::drop(movable);
};
// `consume` 消耗了该变量,所以该闭包只能调用一次。
consume();
}
在竖线 |
之前使用 move
会强制闭包取得被捕获变量的所有权:
fn main() {
// `Vec` 在语义上是不可复制的。
let haystack = vec![1, 2, 3];
let contains = move |needle| haystack.contains(needle);
println!("{}", contains(&1));
//println!("There're {} elements in vec", haystack.len());
// ^ 取消上面一行的注释将导致编译时错误,因为借用检查不允许在变量被移动走
// 之后继续使用它。
// 在闭包的签名中删除 `move` 会导致闭包以不可变方式借用 `haystack`,因此之后
// `haystack` 仍然可用,取消上面的注释也不会导致错误。
}
作为输入参数
当以闭包作为输入参数时,必须指出闭包的完整类型,它是通过使用以下 trait
中的一种来指定的。其受限制程度按以下顺序递减:
Fn
表示捕获方式为通过引用(&T
)的闭包FnMut
表示捕获方式为通过可变引用(&mut T
)的闭包FnOnce
表示捕获方式为通过值(T
)的闭包顺序之所以是这样,是因为
&T
只是获取了不可变的引用,&mut T
则可以改变 变量,T
则是拿到了变量的所有权而非借用。
事实上是在满足使用需求的前提下尽量以限制最多的方式捕获。
如果一个类型说明为 FnOnce
的闭包作为参数。这说明闭包可能采取 &T
,&mut T
或 T
中的一种捕获方式;
如果参数的类型说明是 Fn
,那么不允许该闭包通过 &mut T
或 T
捕获变量。
// 该函数将闭包作为参数并调用它。
fn apply<F>(f: F) where
// 闭包没有输入值和返回值。
F: FnOnce() {
// ^ 试一试:将 `FnOnce` 换成 `Fn` 或 `FnMut`。
f();
}
// 输入闭包,返回一个 `i32` 整型的函数。
fn apply_to_3<F>(f: F) -> i32 where
// 闭包处理一个 `i32` 整型并返回一个 `i32` 整型。
F: Fn(i32) -> i32 {
f(3)
}
fn main() {
use std::mem;
let greeting = "hello";
// 不可复制的类型。
// `to_owned` 从借用的数据创建有所有权的数据。
let mut farewell = "goodbye".to_owned();
// 捕获 2 个变量:通过引用捕获 `greeting`,通过值捕获 `farewell`。
let diary = || {
// `greeting` 通过引用捕获,故需要闭包是 `Fn`。
println!("I said {}.", greeting);
// 下文改变了 `farewell` ,因而要求闭包通过可变引用来捕获它。
// 现在需要 `FnMut`。
farewell.push_str("!!!");
println!("Then I screamed {}.", farewell);
println!("Now I can sleep. zzzzz");
// 手动调用 drop 又要求闭包通过值获取 `farewell`。
// 现在需要 `FnOnce`。
mem::drop(farewell);
};
// 以闭包作为参数,调用函数 `apply`。
apply(diary);
// 闭包 `double` 满足 `apply_to_3` 的 trait 约束。
let double = |x| 2 * x;
println!("3 doubled: {}", apply_to_3(double));
}
类型匿名
使用闭包作为函数参数,这要求闭包是泛型的。
// `F` 必须是泛型的。
fn apply<F>(f: F) where
F: FnOnce() {
f();
}
指明为该结构体实现的是 Fn
、FnMut
、或 FnOnce
中的哪种 trait
,对于约束该结构体的类型而言就已经足够了。
// `F` 必须为一个没有输入参数和返回值的闭包实现 `Fn`,这和对 `print` 的
// 要求恰好一样。
fn apply<F>(f: F) where
F: Fn() {
f();
}
fn main() {
let x = 7;
// 捕获 `x` 到匿名类型中,并为它实现 `Fn`。
// 将闭包存储到 `print` 中。
let print = || println!("{}", x);
apply(print);
}
输入函数
如果你声明一个接受闭包作为参数的函数,那么任何满足该闭包的 trait 约束的函数都可以作为其参数。
// 定义一个函数,可以接受一个由 `Fn` 限定的泛型 `F` 参数并调用它。
fn call_me<F: Fn()>(f: F) {
f()
}
// 定义一个满足 `Fn` 约束的封装函数(wrapper function)。
fn function() {
println!("I'm a function!");
}
fn main() {
// 定义一个满足 `Fn` 约束的闭包。
let closure = || println!("I'm a closure!");
call_me(closure);
call_me(function);
}
Fn
、FnMut
和 FnOnce
这些 trait
明确了闭包如何从周围的作用域中捕获变量。
作为输出参数
目前 Rust 只支持返回具体(非泛型)的类型。匿名闭包的类型是未知的,所以只有使用 impl Trait
才能返回一个闭包。
返回值的合法 trait:
Fn
:和前面的一样FnMut
:和前面的一样FnOnce
:不平常的事情正是发生在这里。总之现在你需要返回 [FnBox
][fnbox] 类型,目前该类型还是不稳定的。这个情况估计将来会改进。
除此之外,还必须使用 move
关键字,它表明所有的捕获都是通过值进行的。
fn create_fn() -> impl Fn() {
let text = "Fn".to_owned();
move || println!("This is a: {}", text)
}
fn create_fnmut() -> impl FnMut() {
let text = "FnMut".to_owned();
move || println!("This is a: {}", text)
}
fn main() {
let fn_plain = create_fn();
let mut fn_mut = create_fnmut();
fn_plain();
fn_mut();
}
std
中的例子
Iterator::any
Iterator::any
是一个函数,若传给它一个迭代器(iterator),当其中任一元素满足谓词(predicate)时它将返回 true
,否则返回 false
。
谓词是闭包规定的,
true
/false
是闭包作用在元素上的返回值。
签名:
pub trait Iterator {
// 被迭代的类型。
type Item;
// `any` 接受 `&mut self` 参数(译注:回想一下,这是 `self: &mut Self` 的简写)
// 表明函数的调用者可以被借用和修改,但不会被消耗。
fn any<F>(&mut self, f: F) -> bool where
// `FnMut` 表示被捕获的变量最多只能被修改,而不能被消耗。
// `Self::Item` 指明了被捕获变量的类型(译注:是迭代器的元素本身的类型)
F: FnMut(Self::Item) -> bool {}
// 译注:原文说 `Self::Item` 表明变量是通过值传递给闭包的,这是说错了。
// `FnMut` 就表示闭包只能通过引用捕获变量。把类型为 `T` 的变量作为闭包
// 的参数不代表闭包会拿走它的值,也可能是拿走它的引用。
}
fn main() {
let vec1 = vec![1, 2, 3];
let vec2 = vec![4, 5, 6];
// 对 vec 的 `iter()` 举出 `&i32`。(通过用 `&x` 匹配)把它解构成 `i32`。
// 译注:注意 `any` 方法会自动地把 `vec.iter()` 举出的迭代器的元素一个个地
// 传给闭包。因此闭包接收到的参数是 `&i32` 类型的。
println!("2 in vec1: {}", vec1.iter() .any(|&x| x == 2));
// 对 vec 的 `into_iter()` 举出 `i32` 类型。无需解构。
println!("2 in vec2: {}", vec2.into_iter().any(| x| x == 2));
let array1 = [1, 2, 3];
let array2 = [4, 5, 6];
// 对数组的 `iter()` 举出 `&i32`。
println!("2 in array1: {}", array1.iter() .any(|&x| x == 2));
// 对数组的 `into_iter()` 通常举出 `&i32`。
println!("2 in array2: {}", array2.into_iter().any(|&x| x == 2));
}
Iterator::find
Iterator::find
是一个函数,在传给它一个迭代器时,将用 Option
类型返回第一个满足谓词的元素。
pub trait Iterator {
// 被迭代的类型。
type Item;
// `find` 接受 `&mut self` 参数,表明函数的调用者可以被借用和修改,
// 但不会被消耗。
fn find<P>(&mut self, predicate: P) -> Option<Self::Item> where
// `FnMut` 表示被捕获的变量最多只能被修改,而不能被消耗。
// `&Self::Item` 指明了被捕获变量的类型(译注:是对迭代器元素的引用类型)
P: FnMut(&Self::Item) -> bool {}
}
fn main() {
let vec1 = vec![1, 2, 3];
let vec2 = vec![4, 5, 6];
// 对 vec1 的 `iter()` 举出 `&i32` 类型。
let mut iter = vec1.iter();
// 对 vec2 的 `into_iter()` 举出 `i32` 类型。
let mut into_iter = vec2.into_iter();
// 对迭代器举出的元素的引用是 `&&i32` 类型。解构成 `i32` 类型。
// 译注:注意 `find` 方法会把迭代器元素的引用传给闭包。迭代器元素自身
// 是 `&i32` 类型,所以传给闭包的是 `&&i32` 类型。
println!("Find 2 in vec1: {:?}", iter .find(|&&x| x == 2));
// 对迭代器举出的元素的引用是 `&i32` 类型。解构成 `i32` 类型。
println!("Find 2 in vec2: {:?}", into_iter.find(| &x| x == 2));
let array1 = [1, 2, 3];
let array2 = [4, 5, 6];
// 对数组的 `iter()` 举出 `&i32`。
println!("Find 2 in array1: {:?}", array1.iter() .find(|&&x| x == 2));
// 对数组的 `into_iter()` 通常举出 `&i32``。
println!("Find 2 in array2: {:?}", array2.into_iter().find(|&&x| x == 2));
}
高阶函数
Rust 提供了高阶函数(Higher Order Function, HOF),指那些输入一个或多个函数,并且/或者产生一个更有用的函数的函数。
HOF 和惰性迭代器(lazy iterator)给 Rust 带来了函数式(functional)编程的风格。
fn is_odd(n: u32) -> bool {
n % 2 == 1
}
fn main() {
println!("Find the sum of all the squared odd numbers under 1000");
let upper = 1000;
// 命令式(imperative)的写法
// 声明累加器变量
let mut acc = 0;
// 迭代:0,1, 2, ... 到无穷大
for n in 0.. {
// 数字的平方
let n_squared = n * n;
if n_squared >= upper {
// 若大于上限则退出循环
break;
} else if is_odd(n_squared) {
// 如果是奇数就计数
acc += n_squared;
}
}
println!("imperative style: {}", acc);
// 函数式的写法
let sum_of_squared_odd_numbers: u32 =
(0..).map(|n| n * n) // 所有自然数取平方
.take_while(|&n| n < upper) // 取小于上限的
.filter(|&n| is_odd(n)) // 取奇数
.fold(0, |sum, i| sum + i); // 最后加起来
println!("functional style: {}", sum_of_squared_odd_numbers);
}
发散函数
发散函数(diverging function)绝不会返回。 它们使用 !
标记,这是一个空类型。
fn foo() -> ! {
panic!("This call never returns.");
}
和所有其他类型相反,这个类型无法实例化,因为此类型可能具有的所有可能值的集合为空。 注意,它与 ()
类型不同,后者只有一个可能的值。
fn main() {
fn sum_odd_numbers(up_to: u32) -> u32 {
let mut acc = 0;
for i in 0..up_to {
// 注意这个 match 表达式的返回值必须为 u32,
// 因为 “addition” 变量是这个类型。
let addition: u32 = match i%2 == 1 {
// “i” 变量的类型为 u32,这毫无问题。
true => i,
// 另一方面,“continue” 表达式不返回 u32,但它仍然没有问题,
// 因为它永远不会返回,因此不会违反匹配表达式的类型要求。
false => continue,
};
acc += addition;
}
acc
}
println!("Sum of odd numbers up to 9 (excluding): {}", sum_odd_numbers(9));
}