Go語言怎么使用變長參數(shù)函數(shù)

更新時間：2022年07月15日 11:46:33 作者：frank

本文主要介紹了Go語言怎么使用變長參數(shù)函數(shù)，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧

01 介紹

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

我們在使用 Go 語言開發(fā)應用程序時，也可以在合適的場景使用變長參數(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)程序運行錯誤的原因是實參類型和形參類型不一致，導致編譯錯誤?？赡苡凶x者朋友們感到疑惑，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ù)的預期功能，還不會入侵之前的調(diào)用代碼。限于篇幅，示例完整代碼請查閱 Github。