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 語言中怎麼使用變長參數函數,先是介紹變長參數和變長參數函數的相關知識,然後列舉了一個簡單示例,通過示例代碼,加深讀者朋友們的理解。

感興趣的讀者朋友們,不妨檢查一下自己的項目中是否也有適合使用變長參數函數的場景,並嘗試重構一下相關代碼。

參考資料:

  1. https://gobyexample.com/variadic-functions

  2. https://www.geeksforgeeks.org/golang-program-that-uses-func-with-variable-argument-list/

  3. https://medium.com/rungo/variadic-function-in-go-5d9b23f4c01a

  4. https://golangbot.com/variadic-functions/

  5. https://www.digitalocean.com/community/tutorials/how-to-use-variadic-functions-in-go

  6. https://blog.learngoprogramming.com/golang-variadic-funcs-how-to-patterns-369408f19085

本文由 Readfog 進行 AMP 轉碼,版權歸原作者所有。
來源https://mp.weixin.qq.com/s/fhqtRTGgVu6aSC581luvCg