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

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

到此這篇關於Go 語言怎麼使用變長參數函數的文章就介紹到這瞭,更多相關Go 變長參數函數內容請搜索WalkonNet以前的文章或繼續瀏覽下面的相關文章希望大傢以後多多支持WalkonNet!

推薦閱讀: