5.7. 可變參數

參數數量可變的函數稱為可變參數函數。典型的例子就是fmt.Printf和類似函數。Printf首先接收一個必備的參數,之後接收任意個數的後續參數。

在聲明可變參數函數時,需要在參數列表的最後一個參數類型之前加上省略符號“...”,這表示該函數會接收任意數量的該類型參數。

gopl.io/ch5/sum

func sum(vals ...int) int {
    total := 0
    for _, val := range vals {
        total += val
    }
    return total
}

sum函數返回任意個int型參數的和。在函數體中,vals被看作是類型為[] int的切片。sum可以接收任意數量的int型參數:

fmt.Println(sum())           // "0"
fmt.Println(sum(3))          // "3"
fmt.Println(sum(1, 2, 3, 4)) // "10"

在上面的代碼中,調用者隱式的創建一個數組,並將原始參數複製到數組中,再把數組的一個切片作為參數傳給被調用函數。如果原始參數已經是切片類型,我們該如何傳遞給sum?只需在最後一個參數後加上省略符。下面的代碼功能與上個例子中最後一條語句相同。

values := []int{1, 2, 3, 4}
fmt.Println(sum(values...)) // "10"

雖然在可變參數函數內部,...int 型參數的行為看起來很像切片類型,但實際上,可變參數函數和以切片作為參數的函數是不同的。

func f(...int) {}
func g([]int) {}
fmt.Printf("%T\n", f) // "func(...int)"
fmt.Printf("%T\n", g) // "func([]int)"

可變參數函數經常被用於格式化字符串。下面的errorf函數構造了一個以行號開頭的,經過格式化的錯誤信息。函數名的後綴f是一種通用的命名規範,代表該可變參數函數可以接收Printf風格的格式化字符串。

func errorf(linenum int, format string, args ...interface{}) {
    fmt.Fprintf(os.Stderr, "Line %d: ", linenum)
    fmt.Fprintf(os.Stderr, format, args...)
    fmt.Fprintln(os.Stderr)
}
linenum, name := 12, "count"
errorf(linenum, "undefined: %s", name) // "Line 12: undefined: count"

interface{}表示函數的最後一個參數可以接收任意類型,我們會在第7章詳細介紹。

練習5.15: 編寫類似sum的可變參數函數max和min。考慮不傳參時,max和min該如何處理,再編寫至少接收1個參數的版本。

練習5.16:編寫多參數版本的strings.Join。

練習5.17:編寫多參數版本的ElementsByTagName,函數接收一個HTML結點樹以及任意數量的標籤名,返回與這些標籤名匹配的所有元素。下面給出了2個例子:

func ElementsByTagName(doc *html.Node, name...string) []*html.Node
images := ElementsByTagName(doc, "img")
headings := ElementsByTagName(doc, "h1", "h2", "h3", "h4")