Go 語言怎么使用變長參數函數?
?1.介紹
Go 語言中函數的最后一個參數可以是變長參數,細心的讀者朋友們可能已經發現,在 Go 語言標準庫 fmt?包中就有使用變長參數函數,比如 Println? 和 Printf。
我們在使用 Go 語言開發應用程序時,也可以在合適的場景使用變長參數函數,使我們的代碼更優雅。
2.變長參數
顧名思義,變長參數是指參數的數量不固定,可以是 0 個,1 個或多個。變長參數的格式是 ...T?,在參數的類型前面有 3 個 .,表示該參數是變長參數。
變長參數在函數體內是切片類型的參數,也就是說 ...T? 等價于 []T。
變長參數在函數外部可匹配的參數類型有兩種,分別是一個 []T? 切片類型的變量,和多個 T 類型的變量,并且二者不可以同時使用,也就是說它們不能同時出現在函數的參數列表中。
3.變長參數函數
我們已經知道什么是變長參數,自然我們也就可以想到接受 ...T 形式的形參的函數就是變長參數函數。
一個變長參數函數只能有一個 ...T 形式的形參,并且該形參必須是函數參數列表中的最后一個形參。
需要注意的是,變長參數函數最容易踩的“坑”就是形參和實參的類型不匹配,例如以下這段代碼:
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...)
}
輸出結果:
./prog.go:18:6: cannot use num (variable of type []int) as type []interface{} in argument to sum
閱讀上面這段代碼,我們發現程序運行錯誤的原因是實參類型和形參類型不一致,導致編譯錯誤。可能有讀者朋友們感到疑惑,int? 類型的變量可以直接賦值給 interface{} 類型的變量,為什么會報錯呢?
這是因為實參的類型是 []int?,它不能匹配形參 []interface{}?。修改該錯誤也簡單,只需將實參的類型修改為 []interface{}。
num := []interface{}{1, 2, 3}
4.使用場景
我們了解完變長參數和變長參數函數,再通過一個示例代碼加深一下讀者朋友們的理解。
該使用場景是通過調用下游方法,輸入用戶的個人資料。但是,后期調用的下游方法的入參發生變化,新增了一個或多個請求參數。
如果不使用變長參數,我們原來調用該下游方法的代碼都需要隨之修改。以下是示例代碼:
變更前的示例代碼:
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
}
調用函數的示例代碼:
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)
}
閱讀上面這段代碼,因為我們使用變長參數的形式,修改調用的下游函數的入參,也就是將原來調用的下游函數由普通函數改為變長參數函數。
通過該方式變更代碼,不僅實現了函數的預期功能,還不會入侵之前的調用代碼。
5.總結
本文我們主要介紹在 Go 語言中怎么使用變長參數函數,先是介紹變長參數和變長參數函數的相關知識,然后列舉了一個簡單示例,通過示例代碼,加深讀者朋友們的理解。
感興趣的讀者朋友們,不妨檢查一下自己的項目中是否也有適合使用變長參數函數的場景,并嘗試重構一下相關代碼。