Go語言怎麼使用變長引數函數

2022-07-15 14:02:50

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=%vn", err)
  return
 }
 fmt.Printf("name:%snage:%dngender:%sn", detail.name, detail.age, detail.gender)
 fmt.Printf("%sn", "********************")
 name2 := "lucy"
 age2 := 17
 gender2 := "female"
 salary2 := 5000

 detail2, err := CallUserCenter(name2, age2, gender2, salary2)
 if err != nil {
  fmt.Printf("err=%vn", err)
  return
 }
 fmt.Printf("name:%snage:%dngender:%snsalary:%dn", detail2.name, detail2.age, detail2.gender, detail2.salary)
}

閱讀上面這段程式碼，因為我們使用變長引數的形式，修改呼叫的下游函數的入參，也就是將原來呼叫的下游函數由普通函數改為變長引數函數。

通過該方式變更程式碼，不僅實現了函數的預期功能，還不會入侵之前的呼叫程式碼。限於篇幅，範例完整程式碼請查閱 Github。

05 總結

到此這篇關於Go 語言怎麼使用變長引數函數的文章就介紹到這了,更多相關Go 變長引數函數內容請搜尋it145.com以前的文章或繼續瀏覽下面的相關文章希望大家以後多多支援it145.com！