Go 专栏｜函数那些事

fmt.Println(“i am funcA”) // i am funcA

}

函数的类型称作函数签名，当两个函数的参数列表和返回列表相同时，则两个函数的类型或签名就相同。

func add(x int, y int) int {

return x + y

}

func sub(x int, y int) (z int) {

z = x - y

return

}

fmt.Printf(“%T\n”, add) // func(int, int) int

fmt.Printf(“%T\n”, sub) // func(int, int) int

参数

多个相邻类型的参数可以使用简写模式，所以刚才的 add 和 sub 函数还可以这样写：

func add(x, y int) int {

return x + y

}

func sub(x, y int) (z int) {

z = x - y

return

}

支持不定参数，使用 ...type 语法。注意不定参数必须是函数的最后一个参数。

func funcSum(args …int) (ret int) {

for _, arg := range args {

ret += arg

}

return

}

// 不定参数

fmt.Println(funcSum(1, 2)) // 3

fmt.Println(funcSum(1, 2, 3)) // 6

也可以使用 slice 作为实参传入，需要使用 ... 将 slice 展开：

// slice 参数

s := []int{1, 2, 3, 4}

fmt.Println(funcSum(s…)) // 10

其实，使用 slice 作为形参同样可以达到相同的效果，但区别就是传参的时候，必须要构造出来一个 slice 才行，没有不定参数使用起来方便。

func funcSum1(args []int) (ret int) {

for _, arg := range args {

ret += arg

}

return

}

fmt.Println(funcSum1(s)) // 10

返回值

函数可以返回一个值，也可以返回多个值。

// 多返回值

func swap(x, y int) (int, int) {

return y, x

}

// 多返回值

fmt.Println(swap(1, 2)) // 2 1

如果有不需要的返回值，使用 _ 将其忽略：

x, _ := swap(1, 2)

fmt.Println(x) // 2

支持命名返回值。使用命名返回值的话，直接使用 return 即可，后面不用跟返回值名。

前面不定参数的例子就是通过这种方式来写的：

func funcSum(args …int) (ret int) {

for _, arg := range args {

ret += arg

}

return

}

再来对比一下，如果不是采用命名返回值，应该怎么写：

func funcSum(args …int) int {

ret := 0

for _, arg := range args {

ret += arg

}

return ret

}

匿名函数

匿名函数是指不需要定义函数名的一种函数实现方式。可以直接赋值给函数变量，可以当作实参，也可以作为返回值，还可以直接调用。

// 匿名函数

sum := func(a, b int) int { return a + b }

fmt.Println(sum(1, 2)) // 3

作为参数：

// 匿名函数作为参数

func funcSum2(f func(int, int) int, x, y int) int {

return f(x, y)

}

fmt.Println(funcSum2(sum, 3, 5)) // 8

作为返回值：

// 匿名函数作为返回值

func wrap(op string) func(int, int) int {

switch op {

case “add”:

return func(a, b int) int {

return a + b

}

case “sub”:

return func(a, b int) int {

return a + b

}

default:

return nil

}

}

f := wrap(“add”)

fmt.Println(f(2, 4)) // 6

直接调用：

// 直接调用

fmt.Println(func(a, b int) int { return a + b }(4, 5)) // 9

总结

函数在之前的文章中已经使用过了，这篇再系统全面总结一下都有哪些需要注意的点。