01 介紹

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

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

02 變長(zhǎng)參數(shù)

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

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

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

03 變長(zhǎng)參數(shù)函數(shù)

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

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

需要注意的是，變長(zhǎng)參數(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)程序運(yùn)行錯(cuò)誤的原因是實(shí)參類型和形參類型不一致，導(dǎo)致編譯錯(cuò)誤?？赡苡凶x者朋友們感到疑惑，int? 類型的變量可以直接賦值給 interface{} 類型的變量，為什么會(huì)報(bào)錯(cuò)呢？

這是因?yàn)閷?shí)參的類型是 []int?，它不能匹配形參 []interface{}?。修改該錯(cuò)誤也簡(jiǎn)單，只需將實(shí)參的類型修改為 []interface{}。

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

04 使用場(chǎng)景

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

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

如果不使用變長(zhǎng)參數(shù)，我們?cè)瓉?lái)調(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)
}

閱讀上面這段代碼，因?yàn)槲覀兪褂米冮L(zhǎng)參數(shù)的形式，修改調(diào)用的下游函數(shù)的入?yún)ⅲ簿褪菍⒃瓉?lái)調(diào)用的下游函數(shù)由普通函數(shù)改為變長(zhǎng)參數(shù)函數(shù)。

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

05 總結(jié)

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

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