原文链接: 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

参数

多个相邻类型的参数可以使用简写模式,所以刚才的 addsub 函数还可以这样写:

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/g...

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

Go 专栏文章列表:

  1. Go 专栏|开发环境搭建以及开发工具 VS Code 配置
  2. Go 专栏|变量和常量的声明与赋值
  3. Go 专栏|基础数据类型:整数、浮点数、复数、布尔值和字符串
  4. Go 专栏|复合数据类型:数组和切片 slice
  5. Go 专栏|复合数据类型:字典 map 和 结构体 struct
  6. Go 专栏|流程控制,一网打尽
  7. Go 专栏|函数那些事
  8. Go 专栏|错误处理:defer,panic 和 recover
  9. Go 专栏|说说方法
  10. Go 专栏|接口 interface

alwaysbeta
84 声望13 粉丝