[[409993]]

大家好，我是站長(zhǎng) polarisxu。

這是 Rust 勸退系列的第 9 個(gè)教程，探討 Rust 中的函數(shù)。

Rust 是支持函數(shù)式編程的語(yǔ)言。Rust 中，函數(shù)作為一等公民，本身就是一種類型。函數(shù)類型變量可以作為其他函數(shù)的參數(shù)或返回值，也可以賦值給別的變量，還可以直接調(diào)用執(zhí)行。

01 函數(shù)定義

在 Rust 中，函數(shù)使用 fn 關(guān)鍵字定義(Go 中使用 func，想著夠省略了，沒想到 Rust 來個(gè)更省略!)。

和大部分其他語(yǔ)言的語(yǔ)法類似，Rust 中的函數(shù)簽名同樣包括函數(shù)名、函數(shù)參數(shù)類型和返回值類型。和 Go 語(yǔ)言的函數(shù)比，Rust 的函數(shù)定義有些不同：

• 除了上面說的關(guān)鍵字不同外，Rust 函數(shù)名一般建議使用下劃線風(fēng)格(小寫字母)，即 蛇形命名法(snake_case)，否則編譯器會(huì)警告;而 Go 使用駝峰風(fēng)格;
• 返回值類型使用 -> 標(biāo)識(shí);
• 大括號(hào)雖然沒有嚴(yán)格要求怎么放，但一般建議緊接函數(shù)簽名之后，而不是另起一行(Go 是只能這樣);

函數(shù)聲明如下：

fn functionname(parametername: type) -> returntype {   
    // 函數(shù)體（具體實(shí)現(xiàn)的功能） 
} 

函數(shù)示例：

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

02 函數(shù)參數(shù)

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

和 Go 語(yǔ)言一樣，Rust 中的函數(shù)也沒有不能指定默認(rèn)值。但 Go 支持不定參數(shù)，Rust目前不支持。

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

函數(shù)參數(shù)和變量一樣，默認(rèn)是不可變的，當(dāng)需要可變參數(shù)時(shí)，一樣得使用 mut 關(guān)鍵字?？匆粋€(gè)例子：(來自 《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); 
} 

按值傳遞參數(shù)，函數(shù)里需要對(duì)傳入其中的動(dòng)態(tài)數(shù)組進(jìn)行修改，因此在參數(shù)前加上了 mut 關(guān)鍵字。正因?yàn)楹瘮?shù)參數(shù)前使用了 mut(這叫做可變修飾)，因此在 main 里的聲明和調(diào)用處并沒有使用 mut。

再看另一種情況：

fn modify(v: &mut Vec<u32>) { 
  v.push(4); 
} 
 
fn main() { 
  let mut v = vec![1, 2, 3]; 
  modify(&mut v); 
  println!("{:?}", v); 
} 

這是按引用傳遞參數(shù)。注意 mut 的位置：按值傳遞，參數(shù)可變，mut 放在參數(shù)前，這和定義可變變量是一樣的寫法;但按引用傳遞，&mut Vec 這是一個(gè)整體，是可變引用類型，因此參數(shù)前面不再需要 mut。(mut 位置不一樣，著實(shí)容易暈!)

此外，函數(shù)參數(shù)可以使用 _ 忽略該它。

03 函數(shù)返回值

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

雖然 Rust 不支持多返回值，但因?yàn)橛性M類型，因此返回元組相當(dāng)于支持多返回值。

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

不知道大家是否注意到，swap 函數(shù)要求返回一個(gè)元組，但我們并沒有使用 return 語(yǔ)句，但要特別注意結(jié)尾是沒有分號(hào)的。

前面說過，Rust 中一切都是表達(dá)式。if 表達(dá)式、循環(huán)表達(dá)式等，它們的值是最后一個(gè)表達(dá)式的值。對(duì)于函數(shù)，它的返回值是最后一個(gè)表達(dá)式的值。所以，Rust 中經(jīng)常不通過 return 來返回值。當(dāng)然，用 return 也是可以的，但需要注意加上分號(hào)。

04 特殊函數(shù) main

這個(gè)不用多說，是 Rust 程序的入口函數(shù)，簽名如下：

fn main() 

沒有參數(shù)、沒有返回值。

05 總結(jié)

關(guān)于函數(shù)還有很多其他知識(shí)點(diǎn)，比如高階函數(shù)、方法、閉包等，下一節(jié)就介紹高階函數(shù)和閉包。

本文轉(zhuǎn)載自微信公眾號(hào)「polarisxu」，可以通過以下二維碼關(guān)注。轉(zhuǎn)載本文請(qǐng)聯(lián)系polarisxu公眾號(hào)。