Rust 勸退系列 09:函數

大家好,我是站長 polarisxu。

這是 Rust 勸退系列的第 9 個教程,探討 Rust 中的函數。

Rust 是支持函數式編程的語言。Rust 中,函數作爲一等公民,本身就是一種類型。函數類型變量可以作爲其他函數的參數或返回值,也可以賦值給別的變量,還可以直接調用執行。

01 函數定義

在 Rust 中,函數使用 fn 關鍵字定義(Go 中使用 func,想着夠省略了,沒想到 Rust 來個更省略!)。

和大部分其他語言的語法類似,Rust 中的函數簽名同樣包括函數名、函數參數類型和返回值類型。和 Go 語言的函數比,Rust 的函數定義有些不同:

函數聲明如下:

fn functionname(parametername: type) -> returntype {  
    // 函數體(具體實現的功能)
}

函數示例:

fn max(x: i32, y: i32) -> i32 {
  if x > y {
    return x;
  }
  return y;
}

02 函數參數

注意,和 Go 不同,雖然上面示例中,x、y 的類型相同,但 x 的類型不能省略。

和 Go 語言一樣,Rust 中的函數也沒有不能指定默認值。但 Go 支持不定參數,Rust 目前不支持。

// Go 中這樣的函數,Rust 不支持
func Sum(x, y int, z ...int) int {
  // 函數體
}

函數參數和變量一樣,默認是不可變的,當需要可變參數時,一樣得使用 mut 關鍵字。看一個例子:(來自 《Rust 編程之道》)

fn modify(mut v: Vec<u32>) -> Vec<u32> {
  v.push(4);
  v
}

fn main() {
  let v = vec![1, 2, 3];
  let v = modify(v);
  println!("{:?}", v);
}

按值傳遞參數,函數里需要對傳入其中的動態數組進行修改,因此在參數前加上了 mut 關鍵字。正因爲函數參數前使用了 mut(這叫做可變修飾),因此在 main 裏的聲明和調用處並沒有使用 mut。

再看另一種情況:

fn modify(v: &mut Vec<u32>) {
  v.push(4);
}

fn main() {
  let mut v = vec![1, 2, 3];
  modify(&mut v);
  println!("{:?}", v);
}

這是按引用傳遞參數。注意 mut 的位置:按值傳遞,參數可變,mut 放在參數前,這和定義可變變量是一樣的寫法;但按引用傳遞,&mut Vec<u32> 這是一個整體,是可變引用類型,因此參數前面不再需要 mut。(mut 位置不一樣,着實容易暈!)

此外,函數參數可以使用 _ 忽略該它。

03 函數返回值

如果函數沒有返回值,-> 不寫。其實,根據前面的介紹,沒有返回值的函數,返回的類型是單元值 ()。所以,也可以顯示返回該類型。

雖然 Rust 不支持多返回值,但因爲有元組類型,因此返回元組相當於支持多返回值。

fn main() {
  let (x, y) = swap(2, 3);
  println!("x={},y={}", x, y);
}

fn swap(x: i32, y: i32) -> (i32, i32) {
    (y, x)
}

不知道大家是否注意到,swap 函數要求返回一個元組,但我們並沒有使用 return 語句,但要特別注意結尾是沒有分號的。

前面說過,Rust 中一切都是表達式。if 表達式、循環表達式等,它們的值是最後一個表達式的值。對於函數,它的返回值是最後一個表達式的值。所以,Rust 中經常不通過 return 來返回值。當然,用 return 也是可以的,但需要注意加上分號。

04 特殊函數 main

這個不用多說,是 Rust 程序的入口函數,簽名如下:

fn main()

沒有參數、沒有返回值。

05 總結

關於函數還有很多其他知識點,比如高階函數、方法、閉包等,下一節就介紹高階函數和閉包。

我是 polarisxu,北大碩士畢業,曾在 360 等知名互聯網公司工作,10 多年技術研發與架構經驗!2012 年接觸 Go 語言並創建了 Go 語言中文網!著有《Go 語言編程之旅》、開源圖書《Go 語言標準庫》等。

堅持輸出技術(包括 Go、Rust 等技術)、職場心得和創業感悟!歡迎關注「polarisxu」一起成長!也歡迎加我微信好友交流:gopherstudio

本文由 Readfog 進行 AMP 轉碼,版權歸原作者所有。
來源https://mp.weixin.qq.com/s/nrGagd49cIucvq-ULR9Oow