函数

functions.md
commit ccb1d87d6faa9ff528d22b96595a0e2cbb16c0f2

到目前为止你应该见过一个函数,main函数:

  1. fn main() {
  2. }

这可能是最简单的函数声明。就像我们之前提到的,fn表示“这是一个函数”,后面跟着名字,一对括号因为这函数没有参数,然后是一对大括号代表函数体。下面是一个叫foo的函数:

  1. fn foo() {
  2. }

那么有参数是什么样的呢?下面这个函数打印一个数字:

  1. fn print_number(x: i32) {
  2. println!("x is: {}", x);
  3. }

下面是一个使用了print_number函数的完整的程序:

  1. fn main() {
  2. print_number(5);
  3. }
  4. fn print_number(x: i32) {
  5. println!("x is: {}", x);
  6. }

如你所见,函数参数与let声明非常相似:参数名加上冒号再加上参数类型。

下面是一个完整的程序,它将两个数相加并打印结果:

  1. fn main() {
  2. print_sum(5, 6);
  3. }
  4. fn print_sum(x: i32, y: i32) {
  5. println!("sum is: {}", x + y);
  6. }

在调用函数和声明函数时,你需要用逗号分隔多个参数。

let不同,你必须为函数参数声明类型。下面代码将不能工作:

  1. fn print_sum(x, y) {
  2. println!("sum is: {}", x + y);
  3. }

你会获得如下错误:

  1. expected one of `!`, `:`, or `@`, found `)`
  2. fn print_sum(x, y) {

这是一个有意为之的设计决定。即使像 Haskell 这样的能够全程序推断的语言,注明类型也经常作为一个最佳实践而被建议。我们认为即使允许在在函数体中推断,也要强制函数声明参数类型。这是一个全推断与无推断的最佳平衡。

如果我们要一个返回值呢?下面这个函数给一个整数加一:

  1. fn add_one(x: i32) -> i32 {
  2. x + 1
  3. }

Rust 函数只能返回一个值,并且你需要在一个“箭头”后面声明类型,它是一个破折号(-)后跟一个大于号(>)。

注意这里并没有一个分号。如果你把它加上:

  1. fn add_one(x: i32) -> i32 {
  2. x + 1;
  3. }

你将会得到一个错误:

  1. error: not all control paths return a value
  2. fn add_one(x: i32) -> i32 {
  3. x + 1;
  4. }
  5. help: consider removing this semicolon:
  6. x + 1;
  7. ^

这揭露了关于 Rust 两个有趣的地方:它是一个基于表达式的语言,并且分号与其它基于“大括号和分号”的语言不同。这两个方面是相关的。

表达式 VS 语句

Rust 主要是一个基于表达式的语言。只有两种语句,其它的一切都是表达式。

然而这又有什么区别呢?表达式返回一个值,而语句不是。这就是为什么这里我们以“不是所有控制路径都返回一个值”结束:x + 1;语句不返回一个值。Rust 中有两种类型的语句:“声明语句”和“表达式语句”。其余的一切是表达式。让我们先讨论下声明语句。

在一些语言中,变量绑定可以被写成一个表达式,不仅仅是语句。例如 Ruby:

  1. x = y = 5

然而,在 Rust 中,使用let引入一个绑定并不是一个表达式。下面的代码会产生一个编译时错误:

  1. let x = (let y = 5); // expected identifier, found keyword `let`

编译器告诉我们这里它期望看到表达式的开头,而let只能开始一个语句,不是一个表达式。

注意赋值一个已经绑定过的变量(例如,y = 5)仍是一个表达式,即使它的(返回)值并不是特别有用。不像其它语言中赋值语句返回它赋的值(例如,前面例子中的5),在 Rust 中赋值的值是一个空的元组()

  1. let mut y = 5;
  2. let x = (y = 6); // `x` has the value `()`, not `6`.

Rust中第二种语句是表达式语句。它的目的是把任何表达式变为语句。在实践环境中,Rust 语法期望语句后跟其它语句。这意味着你用分号来分隔各个表达式。这意味着Rust看起来很像大部分其它使用分号做为语句结尾的语言,并且你会看到分号出现在几乎每一行你看到的 Rust 代码。

那么我们说“几乎”的例外是神马呢?你已经见过它了,在这些代码中:

  1. fn add_one(x: i32) -> i32 {
  2. x + 1
  3. }

我们的函数声称它返回一个i32,但是带上个分号,它就会返回一个()。Rust意识到这大概不是我们想要的,并在之前我们看到的错误中建议去掉分号。

提早返回(Early returns)

不过提早返回怎么破?Rust确实有这么一个关键字,return

  1. fn foo(x: i32) -> i32 {
  2. return x;
  3. // we never run this code!
  4. x + 1
  5. }

使用return作为函数的最后一行是可行的,不过被认为是一个糟糕的风格:

  1. fn foo(x: i32) -> i32 {
  2. return x + 1;
  3. }

如果你之前没有使用过基于表达式的语言,那么前面的没有return的定义可能看起来有点奇怪。不过它随着时间的推移它会变得直观。

发散函数(Diverging functions)

Rust有些特殊的语法叫“发散函数”,这些函数并不返回:

  1. fn diverges() -> ! {
  2. panic!("This function never returns!");
  3. }

panic!是一个宏,类似我们已经见过的println!()。与println!()不同的是,panic!()导致当前的执行线程崩溃并返回指定的信息。因为这个函数会崩溃,所以它不会返回,所以它拥有一个类型!,它代表“发散”。

如果你添加一个叫做diverges()的函数并运行,你将会得到一些像这样的输出:

  1. thread main panicked at This function never returns!’, hello.rs:2

如果你想要更多信息,你可以设定RUST_BACKTRACE环境变量来获取 backtrace :

  1. $ RUST_BACKTRACE=1 ./diverges
  2. thread 'main' panicked at 'This function never returns!', hello.rs:2
  3. stack backtrace:
  4. 1: 0x7f402773a829 - sys::backtrace::write::h0942de78b6c02817K8r
  5. 2: 0x7f402773d7fc - panicking::on_panic::h3f23f9d0b5f4c91bu9w
  6. 3: 0x7f402773960e - rt::unwind::begin_unwind_inner::h2844b8c5e81e79558Bw
  7. 4: 0x7f4027738893 - rt::unwind::begin_unwind::h4375279447423903650
  8. 5: 0x7f4027738809 - diverges::h2266b4c4b850236beaa
  9. 6: 0x7f40277389e5 - main::h19bb1149c2f00ecfBaa
  10. 7: 0x7f402773f514 - rt::unwind::try::try_fn::h13186883479104382231
  11. 8: 0x7f402773d1d8 - __rust_try
  12. 9: 0x7f402773f201 - rt::lang_start::ha172a3ce74bb453aK5w
  13. 10: 0x7f4027738a19 - main
  14. 11: 0x7f402694ab44 - __libc_start_main
  15. 12: 0x7f40277386c8 - <unknown>
  16. 13: 0x0 - <unknown>

如果你需要覆盖一个已经设置的RUST_BACKTRACE的值,同时你又不能仅仅 unset 这个变量,这时把它设置成0来避免获得 backtrace。任何其它(非 0 )值将打开 backtrace。

  1. $ export RUST_BACKTRACE=1
  2. ...
  3. $ RUST_BACKTRACE=0 ./diverges
  4. thread '<main>' panicked at 'This function never returns!', hello.rs:2
  5. note: Run with `RUST_BACKTRACE=1` for a backtrace.

RUST_BACKTRACE也可以用于 Cargo 的run命令:

  1. $ RUST_BACKTRACE=1 cargo run
  2. Running `target/debug/diverges`
  3. thread '<main>' panicked at 'This function never returns!', hello.rs:2
  4. stack backtrace:
  5. 1: 0x7f402773a829 - sys::backtrace::write::h0942de78b6c02817K8r
  6. 2: 0x7f402773d7fc - panicking::on_panic::h3f23f9d0b5f4c91bu9w
  7. 3: 0x7f402773960e - rt::unwind::begin_unwind_inner::h2844b8c5e81e79558Bw
  8. 4: 0x7f4027738893 - rt::unwind::begin_unwind::h4375279447423903650
  9. 5: 0x7f4027738809 - diverges::h2266b4c4b850236beaa
  10. 6: 0x7f40277389e5 - main::h19bb1149c2f00ecfBaa
  11. 7: 0x7f402773f514 - rt::unwind::try::try_fn::h13186883479104382231
  12. 8: 0x7f402773d1d8 - __rust_try
  13. 9: 0x7f402773f201 - rt::lang_start::ha172a3ce74bb453aK5w
  14. 10: 0x7f4027738a19 - main
  15. 11: 0x7f402694ab44 - __libc_start_main
  16. 12: 0x7f40277386c8 - <unknown>
  17. 13: 0x0 - <unknown>

发散函数可以被用作任何类型:

  1. # fn diverges() -> ! {
  2. # panic!("This function never returns!");
  3. # }
  4. let x: i32 = diverges();
  5. let x: String = diverges();

函数指针

我们也可以创建指向函数的变量绑定:

  1. let f: fn(i32) -> i32;

f是一个指向一个获取i32作为参数并返回i32的函数的变量绑定。例如:

  1. fn plus_one(i: i32) -> i32 {
  2. i + 1
  3. }
  4. // without type inference
  5. let f: fn(i32) -> i32 = plus_one;
  6. // with type inference
  7. let f = plus_one;

你可以用f来调用这个函数:

  1. # fn plus_one(i: i32) -> i32 { i + 1 }
  2. # let f = plus_one;
  3. let six = f(5);