Rust 勸退系列之函數
大家好,我是站長 polarisxu。
這是 Rust 勸退系列的第 9 個教程,探討 Rust 中的函數。
Rust 是支持函數式編程的語言。Rust 中,函數作為一等公民,本身就是一種類型。函數類型變量可以作為其他函數的參數或返回值,也可以賦值給別的變量,還可以直接調用執行。
01 函數定義
在 Rust 中,函數使用 fn 關鍵字定義(Go 中使用 func,想著夠省略了,沒想到 Rust 來個更省略!)。
和大部分其他語言的語法類似,Rust 中的函數簽名同樣包括函數名、函數參數類型和返回值類型。和 Go 語言的函數比,Rust 的函數定義有些不同:
- 除了上面說的關鍵字不同外,Rust 函數名一般建議使用下劃線風格(小寫字母),即 蛇形命名法(snake_case),否則編譯器會警告;而 Go 使用駝峰風格;
- 返回值類型使用 -> 標識;
- 大括號雖然沒有嚴格要求怎么放,但一般建議緊接函數簽名之后,而不是另起一行(Go 是只能這樣);
函數聲明如下:
- 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
此外,函數參數可以使用 _ 忽略該它。
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」,可以通過以下二維碼關注。轉載本文請聯系polarisxu公眾號。