2 3 可变参数

2.3 可变参数 #

本节源码位置 https://github.com/golang-minibear2333/golang/blob/master/2.func-containers/2.3-varargs

接连两篇函数专题深度解析,相信大家已经对函数的语法有了深入的了解。

这次小熊给大家带来了一个函数的特性【可变参数】,作为函数专题的结束。

2.3.1 有没有发现? #

我们有时候会用到的输出、错误输出、字符串格式化系统函数,你可以传入任意个数的参数,他全都能处理!

fmt.Println("a","b","c","d")

就像一个任劳任怨的老父亲,那到底是为什么呢?

2.3.2 因为 #

因为在go语言中语言级别自带了一种语法,可以声明可变参数!

func 函数名(固定参数v ...T) (返回参数列表){
函数体
}

2.3.3 怎么用? #

先和小熊一起试试,写一个不定参数累加的函数

func sum(t ...int) (res int) {
	for _, v := range t {
		res += v
	}
	return res
}
fmt.Println(sum(1, 2, 3, 4, 5))

输出结果

15

2.3.4 如果连参数类型都不知道,怎么办? #

上一节我们已知参数类型是int,还记得我们前面说过的switch判断类型做处理的例子吗? switch和type switch

参考这个例子重写下函数,让他可以接收任意类型的参数。


func sumNum(t ...interface{}) (res float64){
	for _,tmp := range t{
		switch v :=tmp.(type) {
		case int:
			res += float64(v)
		case float64:
			res+= v
		case float32:
			res += float64(v)
		}
	}
	return res
}

测试下

fmt.Println(sumNum(1,2.1,"asd",true))

因为忽略了输出

3.1

但是上面的例子并没有覆盖全部的数字,如果一个一个类型的匹配会疯掉的。有没有更好的方法,可以一下子匹配到所有的数字?


func sumNum(t ...interface{}) (res float64) {
	for _, tmp := range t {
		switch v := tmp.(type) {
		case int, int8, int16, 
    int32, int64, uint, 
    uint8, uint16, uint32,
    uint64, float32, float64, 
    complex64, complex128:
			convertStr := fmt.Sprintf("%v", v)
			convertFloat64, _ := strconv.ParseFloat(convertStr, 64)
			res += convertFloat64
		}
	}
	return res
}

上面的代码在case里一下子匹配了所有可能的数字类型,再用格式化输出转换成字符串,最后转换成float64来使用,这是一种折中的办法,速度可能会比较慢(因为格式化内部逻辑比较复杂消耗速度)。

为了保证速度还是不要省掉一大堆的case,明确了类型再做强转。

2.3.5 留给你的寻找的答案 #

有没有一种只留一个case,同时处理速度又快的方法呢?

—— 爱你们的小熊

2.3.6 小结 #

本节介绍了不定参数,知道参数类型与不知道参数类型的处理方法,你可以把这种模式用于类型转换、以及策略模式。

fmt包中,有很多Print就是使用了不定参数,有兴趣可以看一下源码。



本图书由小熊©2021 版权所有,所有文章采用知识署名-非商业性使用-禁止演绎 4.0 国际进行许可。