Go 专栏｜函数那些事

曾经很长一段时间，我都为自己是互联网科技公司的一员而感到自豪，我觉得我们与众不同。

我们的管理更扁平化，没有那么多官僚主义，充满活力，朝气蓬勃。而且我们的产品正在改变大家的衣食住行，我们正在改变世界。

但近几年发生的一系列事件，都让我的信心产生动摇，不停在捶打我：醒醒吧，兄弟，事实不是你想象的那样。

我能做些什么呢？不知道。

还是努力更文吧，争取早日不做打工人。

函数定义

函数包括以下几个部分：关键词 func，函数名，参数列表，返回列表和函数体。

func name(param-list) ret-list {
    body
}

函数可以没有参数，也可以没有返回值。

func funcA() {
    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

总结

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

包括函数定义，参数，返回和匿名函数。其实还有一个闭包，通过匿名函数来实现。但我感觉闭包使用的并不是很多，就没有写，感兴趣的同学可以自己搜搜看。

函数可以把复杂的程序分成更小的模块，使程序可读性更强，复用性更高，维护性更好。在开发过程中一定要具备将特定功能抽象成函数的能力，而不是将所有代码都写在一起，代码堆成一坨。这样的代码除了不好维护，重点是时间长了自己都不想看。

文章中的脑图和源码都上传到了 GitHub，有需要的同学可自行下载。

地址： https://github.com/yongxinz/gopher/tree/main/sc

关注公众号 AlwaysBeta，回复「goebook」领取 Go 编程经典书籍。

Go 专栏文章列表：