歡迎來到 Golang 系列教程第 12 部分。

什么是可變參數(shù)函數(shù)

可變參數(shù)函數(shù)是一種參數(shù)個數(shù)可變的函數(shù)。

語法

如果函數(shù)最后一個參數(shù)被記作 ...T，這時函數(shù)可以接受任意個 T 類型參數(shù)作為最后一個參數(shù)。

請注意只有函數(shù)的最后一個參數(shù)才允許是可變的。

通過一些例子理解可變參數(shù)函數(shù)如何工作

你是否曾經(jīng)想過 append 函數(shù)是如何將任意個參數(shù)值加入到切片中的。這樣 append 函數(shù)可以接受不同數(shù)量的參數(shù)。

func?append(slice?[]Type,?elems?...Type)?[]Type

上面是 append 函數(shù)的定義。在定義中 elems 是可變參數(shù)。這樣 append 函數(shù)可以接受可變化的參數(shù)。

讓我們創(chuàng)建一個我們自己的可變參數(shù)函數(shù)。我們將寫一段簡單的程序，在輸入的整數(shù)列表里查找某個整數(shù)是否存在。

package?main

import?(
?"fmt"
)

func?find(num?int,?nums?...int)?{
?fmt.Printf("type?of?nums?is?%T\n",?nums)
?found?:=?false
?for?i,?v?:=?range?nums?{
??if?v?==?num?{
???fmt.Println(num,?"found?at?index",?i,?"in",?nums)
???found?=?true
??}
?}
?if?!found?{
??fmt.Println(num,?"not?found?in?",?nums)
?}
?fmt.Printf("\n")
}
func?main()?{
?find(89,?89,?90,?95)
?find(45,?56,?67,?45,?90,?109)
?find(78,?38,?56,?98)
?find(87)
}

在線運行代碼^[1]

在上面程序中 func find(num int, nums ...int) ?中的 nums 可接受任意數(shù)量的參數(shù)。在 find 函數(shù)中，參數(shù) nums 相當于一個整型切片。

可變參數(shù)函數(shù)的工作原理是把可變參數(shù)轉換為一個新的切片。以上面程序中的第 22 行為例，find 函數(shù)中的可變參數(shù)是 89，90，95 。find 函數(shù)接受一個 int 類型的可變參數(shù)。因此這三個參數(shù)被編譯器轉換為一個 int 類型切片 int []int{89, 90, 95} 然后被傳入 find 函數(shù)。

在第 10 行， for 循環(huán)遍歷 nums 切片,如果 num 在切片中，則打印 num 的位置。如果 num 不在切片中,則打印提示未找到該數(shù)字。

上面代碼的輸出值如下,

type?of?nums?is?[]int
89?found?at?index?0?in?[89?90?95]

type?of?nums?is?[]int
45?found?at?index?2?in?[56?67?45?90?109]

type?of?nums?is?[]int
78?not?found?in??[38?56?98]

type?of?nums?is?[]int
87?not?found?in??[]

在上面程序的第 25 行，find 函數(shù)僅有一個參數(shù)。我們沒有給可變參數(shù) nums ...int 傳入任何參數(shù)。這也是合法的，在這種情況下 nums 是一個長度和容量為 0 的 nil 切片。

給可變參數(shù)函數(shù)傳入切片

下面例子中，我們給可變參數(shù)函數(shù)傳入一個切片，看看會發(fā)生什么。

package?main

import?(
?"fmt"
)

func?find(num?int,?nums?...int)?{
?fmt.Printf("type?of?nums?is?%T\n",?nums)
?found?:=?false
?for?i,?v?:=?range?nums?{
??if?v?==?num?{
???fmt.Println(num,?"found?at?index",?i,?"in",?nums)
???found?=?true
??}
?}
?if?!found?{
??fmt.Println(num,?"not?found?in?",?nums)
?}
?fmt.Printf("\n")
}
func?main()?{
?nums?:=?[]int{89,?90,?95}
?find(89,?nums)
}

在線運行代碼^[2]

在第 23 行中，我們將一個切片傳給一個可變參數(shù)函數(shù)。

這種情況下無法通過編譯，編譯器報出錯誤 main.go:23: cannot use nums (type []int) as type int in argument to find 。

為什么無法工作呢？原因很直接，find 函數(shù)的說明如下，

func?find(num?int,?nums?...int)

由可變參數(shù)函數(shù)的定義可知，nums ...int 意味它可以接受 int 類型的可變參數(shù)。

在上面程序的第 23 行，nums 作為可變參數(shù)傳入 find 函數(shù)。前面我們知道，這些可變參數(shù)參數(shù)會被轉換為 int 類型切片然后在傳入 find 函數(shù)中。但是在這里 nums 已經(jīng)是一個 int 類型切片，編譯器試圖在 nums 基礎上再創(chuàng)建一個切片，像下面這樣

find(89,?[]int{nums})

這里之所以會失敗是因為 nums 是一個 []int類型 而不是 int類型。

那么有沒有辦法給可變參數(shù)函數(shù)傳入切片參數(shù)呢？答案是肯定的。

有一個可以直接將切片傳入可變參數(shù)函數(shù)的語法糖，你可以在在切片后加上 ... 后綴。如果這樣做，切片將直接傳入函數(shù)，不再創(chuàng)建新的切片

在上面的程序中，如果你將第 23 行的 find(89, nums) 替換為 find(89, nums...) ，程序將成功編譯并有如下輸出

type?of?nums?is?[]int
89?found?at?index?0?in?[89?90?95]

下面是完整的程序供您參考。

package?main

import?(
?"fmt"
)

func?find(num?int,?nums?...int)?{
?fmt.Printf("type?of?nums?is?%T\n",?nums)
?found?:=?false
?for?i,?v?:=?range?nums?{
??if?v?==?num?{
???fmt.Println(num,?"found?at?index",?i,?"in",?nums)
???found?=?true
??}
?}
?if?!found?{
??fmt.Println(num,?"not?found?in?",?nums)
?}
?fmt.Printf("\n")
}
func?main()?{
?nums?:=?[]int{89,?90,?95}
?find(89,?nums...)
}

在線運行代碼^[3]

不直觀的錯誤

當你修改可變參數(shù)函數(shù)中的切片時，請確保你知道你正在做什么。

下面讓我們來看一個簡單的例子。

package?main

import?(
?"fmt"
)

func?change(s?...string)?{
?s[0]?=?"Go"
}

func?main()?{
?welcome?:=?[]string{"hello",?"world"}
?change(welcome...)
?fmt.Println(welcome)
}

在線運行代碼^[4]

你認為這段代碼將輸出什么呢？如果你認為它輸出 [Go world] 。恭喜你！你已經(jīng)理解了可變參數(shù)函數(shù)和切片。如果你猜錯了，那也不要緊，讓我來解釋下為什么會有這樣的輸出。

在第 13 行，我們使用了語法糖 ... 并且將切片作為可變參數(shù)傳入 change 函數(shù)。

正如前面我們所討論的，如果使用了 ... ，welcome 切片本身會作為參數(shù)直接傳入，不需要再創(chuàng)建一個新的切片。這樣參數(shù) welcome 將作為參數(shù)傳入 change 函數(shù)

在 change 函數(shù)中，切片的第一個元素被替換成 Go，這樣程序產(chǎn)生了下面的輸出值

[Go?world]

這里還有一個例子來理解可變參數(shù)函數(shù)。

package?main

import?(
?"fmt"
)

func?change(s?...string)?{
?s[0]?=?"Go"
?s?=?append(s,?"playground")
?fmt.Println(s)
}

func?main()?{
?welcome?:=?[]string{"hello",?"world"}
?change(welcome...)
?fmt.Println(welcome)
}

在線運行代碼^[5]

我將把它作為一個練習留個你，請你指出上面的程序是如何運行的 :) 。

以上就是關于可變參數(shù)函數(shù)的介紹。感謝閱讀。歡迎您留下有價值的反饋和意見。祝您生活愉快。

上一教程 - Array 和 Slice

下一教程 - Maps^[6]

via: https://golangbot.com/variadic-functions/

作者：Nick Coghlan^[7]譯者：fengchunsgit^[8]校對：Noluye^[9]

本文由 GCTT^[10] 原創(chuàng)編譯，Go 中文網(wǎng)^[11] 榮譽推出

參考資料