01 介紹

Go 語言中函數(shù)的最后一個參數(shù)可以是變長參數(shù)，細心的讀者朋友們可能已經(jīng)發(fā)現(xiàn)，在 Go 語言標準庫 fmt?包中就有使用變長參數(shù)函數(shù)，比如 Println? 和 Printf。

我們在使用 Go 語言開發(fā)應(yīng)用程序時，也可以在合適的場景使用變長參數(shù)函數(shù)，使我們的代碼更優(yōu)雅。

02 變長參數(shù)

顧名思義，變長參數(shù)是指參數(shù)的數(shù)量不固定，可以是 0 個，1 個或多個。變長參數(shù)的格式是 ...T?，在參數(shù)的類型前面有 3 個 .，表示該參數(shù)是變長參數(shù)。

變長參數(shù)在函數(shù)體內(nèi)是切片類型的參數(shù)，也就是說 ...T? 等價于 []T。

變長參數(shù)在函數(shù)外部可匹配的參數(shù)類型有兩種，分別是一個 []T? 切片類型的變量，和多個 T 類型的變量，并且二者不可以同時使用，也就是說它們不能同時出現(xiàn)在函數(shù)的參數(shù)列表中。

03 變長參數(shù)函數(shù)

我們已經(jīng)知道什么是變長參數(shù)，自然我們也就可以想到接受 ...T 形式的形參的函數(shù)就是變長參數(shù)函數(shù)。

一個變長參數(shù)函數(shù)只能有一個 ...T 形式的形參，并且該形參必須是函數(shù)參數(shù)列表中的最后一個形參。

需要注意的是，變長參數(shù)函數(shù)最容易踩的“坑”就是形參和實參的類型不匹配，例如以下這段代碼：

func sum(args ...interface{}) {
    res := 0
    for _, v := range args {
        res += v.(int)
    }
    fmt.Println(res)
}

func main() {
    num := []int{1, 2, 3}
    sum(num...)
}

輸出結(jié)果：

./prog.go:18:6: cannot use num (variable of type []int) as type []interface{} in argument to sum

閱讀上面這段代碼，我們發(fā)現(xiàn)程序運行錯誤的原因是實參類型和形參類型不一致，導(dǎo)致編譯錯誤。可能有讀者朋友們感到疑惑，int? 類型的變量可以直接賦值給 interface{} 類型的變量，為什么會報錯呢？

這是因為實參的類型是 []int?，它不能匹配形參 []interface{}?。修改該錯誤也簡單，只需將實參的類型修改為 []interface{}。

num := []interface{}{1, 2, 3}

04 使用場景

我們了解完變長參數(shù)和變長參數(shù)函數(shù)，再通過一個示例代碼加深一下讀者朋友們的理解。

該使用場景是通過調(diào)用下游方法，輸入用戶的個人資料。但是，后期調(diào)用的下游方法的入?yún)l(fā)生變化，新增了一個或多個請求參數(shù)。

如果不使用變長參數(shù)，我們原來調(diào)用該下游方法的代碼都需要隨之修改。以下是示例代碼：

變更前的示例代碼：

func CallUserCenter(name string, age int, gender string) (detail *User, err error) {
 detail, err = NewUserUsecase().Create(name, age, gender)
 if err != nil {
  return
 }
 return
}

變更后的示例代碼：

func CallUserCenter(name string, age int, gender string, args ...interface{}) (detail *User, err error) {
 if len(args) == 0 {
  detail, err = NewUserUsecase().Create(name, age, gender)
 } else {
  detail, err = NewUserUsecase().Create(name, age, gender, args[0])
 }
 if err != nil {
  return
 }
 return
}

調(diào)用函數(shù)的示例代碼：

func main() {
 name := "frank"
 age := 18
 gender := "male"
 detail, err := CallUserCenter(name, age, gender)
 if err != nil {
  fmt.Printf("err=%v\n", err)
  return
 }
 fmt.Printf("name:%s\nage:%d\ngender:%s\n", detail.name, detail.age, detail.gender)
 fmt.Printf("%s\n", "********************")
 name2 := "lucy"
 age2 := 17
 gender2 := "female"
 salary2 := 5000

 detail2, err := CallUserCenter(name2, age2, gender2, salary2)
 if err != nil {
  fmt.Printf("err=%v\n", err)
  return
 }
 fmt.Printf("name:%s\nage:%d\ngender:%s\nsalary:%d\n", detail2.name, detail2.age, detail2.gender, detail2.salary)
}

閱讀上面這段代碼，因為我們使用變長參數(shù)的形式，修改調(diào)用的下游函數(shù)的入?yún)ⅲ簿褪菍⒃瓉碚{(diào)用的下游函數(shù)由普通函數(shù)改為變長參數(shù)函數(shù)。

通過該方式變更代碼，不僅實現(xiàn)了函數(shù)的預(yù)期功能，還不會入侵之前的調(diào)用代碼。限于篇幅，示例完整代碼請查閱 Github「閱讀原文」。

05 總結(jié)

本文我們主要介紹在 Go 語言中怎么使用變長參數(shù)函數(shù)，先是介紹變長參數(shù)和變長參數(shù)函數(shù)的相關(guān)知識，然后列舉了一個簡單示例，通過示例代碼，加深讀者朋友們的理解。

感興趣的讀者朋友們，不妨檢查一下自己的項目中是否也有適合使用變長參數(shù)函數(shù)的場景，并嘗試重構(gòu)一下相關(guān)代碼。