Go语言 可变参数
最近與同事討論時(shí),提到Go語言的可變參數(shù),之前沒有總結(jié)過相關(guān)知識(shí)點(diǎn),今天我們介紹一下Go語言的可變參數(shù)。
可變參數(shù)(Variable Parameters):參數(shù)數(shù)量可變的函數(shù)稱之為可變參數(shù)函數(shù),主要是在使用語法糖(syntactic sugar)。最經(jīng)典的例子就是fmt.Printf()和類似的函數(shù),fmt.Printf首先接收一個(gè)參數(shù),后面可接收若干個(gè)參數(shù)。
在開始Go語言實(shí)例之前,我們先看一下在C語言里面是如何實(shí)現(xiàn)可變參數(shù)的,如示例:
#include <stdio.h> #include <stdarg.h>int sum(int count, ...) {int sum=0;int val=0;// 定義一個(gè)可變參數(shù)列表,可以看成是一種特殊的指針類型,list指向的對(duì)象是棧上的數(shù)據(jù)。va_list list;// 初始化list,list指向第一個(gè)被壓棧的參數(shù),即函數(shù)的最后一個(gè)參數(shù),而count則是棧上最后一個(gè)參數(shù),系統(tǒng)由此確定棧上參數(shù)內(nèi)存的范圍。va_start(list, count);while(count--){// 通過va_arg依次獲取參數(shù)值,兩個(gè)參數(shù),一個(gè)是指向棧上參數(shù)的指針list,這個(gè)指針每取出一個(gè)數(shù)據(jù)移動(dòng)一次,總是指向棧上第一個(gè)未取出的參數(shù),int指需要取出的參數(shù)的類型,CPU根據(jù)這個(gè)類型所占的地址空間來進(jìn)行尋址。val = va_arg(list,int);// printf("%d at %X\n", val, &val);sum += val;}//釋放list,釋放后list置為空。va_end(list);return sum; }int main() {printf("Sum of 1,2,3,4,5 = %d\n", sum(5, 1, 2, 3, 4, 5));printf("Sum of 10,20,30 = %d\n", sum(3, 10,20,30)); }運(yùn)行結(jié)果:
我們可以看到,在C語言里面,需要指定參數(shù)個(gè)數(shù)和若干個(gè)參數(shù),下面我們介紹一下Go語言中的實(shí)現(xiàn)。
實(shí)例一:
?
func1使用的是Go語言的語法糖,按照內(nèi)部機(jī)制來說,...type本質(zhì)是一個(gè)切片,也就是[]type,params被看作是類型為[] int的切片傳入func1中,func1可接收任意個(gè)int值,返回sum結(jié)果。
雖然在可變參數(shù)函數(shù)內(nèi)部,...int型參數(shù)的行為看起來類似slice,實(shí)際上,可變參數(shù)函數(shù)和切片作為參數(shù)的函數(shù)是不相同的。
// 可變參數(shù) func func1(params ...int) {sum := 0for _, param := range params {sum += param}fmt.Println("params : ", params, "\tsum : ", sum) }調(diào)用一:
func TestFunc1(t *testing.T) {var params = []int{1, 2, 3}func1(params...)func1(2, 5)func1(2, 5, 8) }結(jié)果一:
=== RUN TestFunc1 params : [1 2 3] sum : 6 params : [2 5] sum : 7 params : [2 5 8] sum : 15 --- PASS: TestFunc1 (0.00s)實(shí)例二:
func2雖然同樣實(shí)現(xiàn)了不定參數(shù)的功能,但是使用起來比較繁瑣,需要[]type{}來構(gòu)造切片實(shí)例。我們可以看到傳遞的數(shù)據(jù)是slice,但是在參數(shù)傳遞的時(shí)候,我們需要手工初始化slice再傳入函數(shù)。
// 切片 func func2(params []int) {sum := 0for _, param := range params {sum += param}fmt.Println("params : ", params, "\tsum : ", sum) }調(diào)用二:
func TestFunc2(t *testing.T) {func2([]int{3})func2([]int{3, 6})func2([]int{3, 6, 9}) }結(jié)果二:
=== RUN TestFunc2 params : [3] sum : 3 params : [3 6] sum : 9 params : [3 6 9] sum : 18 --- PASS: TestFunc2 (0.00s)綜上兩例,我們可以看出語法糖實(shí)現(xiàn)更簡(jiǎn)潔方便。
實(shí)例三:
我們?cè)倏匆幌驴勺冾愋偷目勺儏?shù),見func3:
// 可變類型的可變參數(shù) func func3(params ...interface{}) {for _, param := range params {switch reflect.TypeOf(param).Kind().String() {case "int":fmt.Printf("param:%d is an int value!\n", param)case "int32":fmt.Printf("param:%v is an int32 value!\n", param)case "int64":fmt.Printf("param:%v is an int64 value!\n", param)case "float32":fmt.Printf("param:%v is an float32 value!\n", param)case "float64":fmt.Printf("param:%v is an float64 value!\n", param)case "string":fmt.Printf("param:%s is an string value!\n", param)case "func":fmt.Printf("param:%v is an func value!\n", param)case "map":fmt.Printf("param:%v is an map value!\n", param)default:fmt.Printf("param:%v is an unknown type.\n", param)}} }調(diào)用三:
func TestFunc3(t *testing.T) {var p1 int = 100 //傳遞int值func3(p1)var p2 int32 = 200 //傳遞int32func3(p2)var p3 int64 = 300 //傳遞int64func3(p3)var p4 = "test string" //傳遞stringfunc3(p4)var p5 float32 = 1.11 //傳遞float32func3(p5)var p6 float64 = 2.22 //傳遞float64func3(p6)var p7 = func(a, b int) int { return a + b } //傳遞funcfunc3(p7)var p8 = map[string]string{} 傳遞mapfunc3(p8) }結(jié)果三:
=== RUN TestFunc3 param:100 is an int value! param:200 is an int32 value! param:300 is an int64 value! param:test string is an string value! param:1.11 is an float32 value! param:2.22 is an float64 value! param:0x506b50 is an func value! param:map[] is an map value! --- PASS: TestFunc3 (0.00s) PASS總結(jié):
可變參數(shù),主要是Go語言的語法糖之"...type"的使用。
參數(shù)個(gè)數(shù)靈活。
總結(jié)
- 上一篇: Hive的列分隔符和行分隔符
- 下一篇: Go语言 命令行解析(一)