Go 語言怎麼使用變長參數函數?
01 介紹
Go 語言中函數的最後一個參數可以是變長參數,細心的讀者朋友們可能已經發現,在 Go 語言標準庫 fmt
包中就有使用變長參數函數,比如 Println
和 Printf
。
我們在使用 Go 語言開發應用程序時,也可以在合適的場景使用變長參數函數,使我們的代碼更優雅。
02 變長參數
顧名思義,變長參數是指參數的數量不固定,可以是 0 個,1 個或多個。變長參數的格式是 ...T
,在參數的類型前面有 3 個 .
,表示該參數是變長參數。
變長參數在函數體內是切片類型的參數,也就是說 ...T
等價於 []T
。
變長參數在函數外部可匹配的參數類型有兩種,分別是一個 []T
切片類型的變量,和多個 T
類型的變量,並且二者不可以同時使用,也就是說它們不能同時出現在函數的參數列表中。
03 變長參數函數
我們已經知道什麼是變長參數,自然我們也就可以想到接受 ...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}
04 使用場景
我們瞭解完變長參數和變長參數函數,再通過一個示例代碼加深一下讀者朋友們的理解。
該使用場景是通過調用下游方法,輸入用戶的個人資料。但是,後期調用的下游方法的入參發生變化,新增了一個或多個請求參數。
如果不使用變長參數,我們原來調用該下游方法的代碼都需要隨之修改。以下是示例代碼:
變更前的示例代碼:
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)
}
閱讀上面這段代碼,因爲我們使用變長參數的形式,修改調用的下游函數的入參,也就是將原來調用的下游函數由普通函數改爲變長參數函數。
通過該方式變更代碼,不僅實現了函數的預期功能,還不會入侵之前的調用代碼。限於篇幅,示例完整代碼請查閱 Github「閱讀原文」。
05 總結
本文我們主要介紹在 Go 語言中怎麼使用變長參數函數,先是介紹變長參數和變長參數函數的相關知識,然後列舉了一個簡單示例,通過示例代碼,加深讀者朋友們的理解。
感興趣的讀者朋友們,不妨檢查一下自己的項目中是否也有適合使用變長參數函數的場景,並嘗試重構一下相關代碼。
參考資料:
-
https://gobyexample.com/variadic-functions
-
https://www.geeksforgeeks.org/golang-program-that-uses-func-with-variable-argument-list/
-
https://medium.com/rungo/variadic-function-in-go-5d9b23f4c01a
-
https://golangbot.com/variadic-functions/
-
https://www.digitalocean.com/community/tutorials/how-to-use-variadic-functions-in-go
-
https://blog.learngoprogramming.com/golang-variadic-funcs-how-to-patterns-369408f19085
本文由 Readfog 進行 AMP 轉碼,版權歸原作者所有。
來源:https://mp.weixin.qq.com/s/fhqtRTGgVu6aSC581luvCg