函数
函数就是一段基本的代码块,一般用来对需要重复执行的代码进行复用。在 go 中,函数是『一等公民』,这与 js 类似,也就是可以将函数当做一个变量进行传递。
函数声明
由于是强类型语言,与 js 不同,在函数声明的过程中,需要指定参数与返回值的类型。
- func max (n1, n2 int) int {
- var result int
- if n1 >= n2 {
- result = n1
- }
- if n1 < n2 {
- result = n2
- }
- return result
- }
在声明函数参数和类型的时候,与声明变量类似,可以一次性指定多个参数的类型,也可以分别指定多个参数为不同类型。
- func max (n1 int, n2 int) int {
- ……
- }
如果函数返回值有多个,在指定返回类型的时候,需要指定每个返回值的类型。
- func max (n1 int, n2 int) (error, int) {
- ……
- return errors.New(""), result
- }
上面的代码,表示返回的时候需要返回两个值,第一个值为 error 对象,用来表示执行期间是否出现异常。这种方式也是 Node.js 中常见的 error-first callback 的写法。
特殊函数
在 go 中,有两个特殊函数:main、init,这两个函数声明之后,一般不需要主动调用,会有自动执行的机制。
func main()
main 函数是 go 语言中默认的入口函数,只能应用于 package main 中,如果在其他的 package 中不会执行。main 函数有如下几点需要注意:
- 不能定义参数;
- 不能定义返回值;
- 必须在 package main 中声明;
func init()
init 函数所有的包启动的时候都会执行,执行时机比 main 函数早,与 main 函数一样,不能定义参数和返回值。
- package main
- import "fmt"
- func init() {
- fmt.Println("执行 init 函数\n")
- }
- func main() {
- fmt.Println("执行 main 函数\n")
- }
函数调用
函数的调用比较简单,和其他编程语言类似,只需要将函数需要接受的参数传入其中,在执行结束后,就能得到对应的返回值。
- // 定义 max 函数
- func max (n1, n2 int) int {
- var result int
- if n1 >= n2 {
- result = n1
- }
- if n1 < n2 {
- result = n2
- }
- return result
- }
- func main () {
- var result = max(5, 100)
- fmt.Println("max return", result)
- }
匿名函数
匿名函数就是一个没有定义函数名的函数,匿名函数可以当成一个值,将其赋值放到某个变量中。这也是之前为什么说函数是『一等公民』,就是可以将函数当成一个变量。
- var max = func (n1, n2 int) int {
- var result int
- if n1 >= n2 {
- result = n1
- }
- if n1 < n2 {
- result = n2
- }
- return result
- }
- var result = max(5, 100)
- fmt.Println("max return", result)
立即执行函数
由于 go 中的函数是 『一等公民』,可以在声明之后立即执行,就是在函数声明结束后,直接加上一个括号,表示该函数会立即执行,执行之后的结果可以通过变量进行接收。
- import "math"
- var Pi = func () float64 {
- return math.Pi
- }()
- fmt.Println("PI =",Pi)
闭包
“闭包就是能够读取其他函数内部变量的函数。在本质上,闭包是将函数内部和函数外部连接起来的桥梁。 ——百度百科
上面的描述来自百度百科,初次看概念比较难理解,如果站在使用的角度来说,闭包就是在一个函数调用后,返回另一个匿名函数,并保持当前函数内的局部变量,可以给匿名函数引用。
下面我们可以简单实现一个迭代器函数,该函数接受一个切片,返回一个匿名函数,该匿名函数每次执行,都会取出切片的一个值,直到全部读取。
- func generate(slice []int) func() (bool, int) {
- i := 0
- length := len(slice)
- return func () (bool, int) {
- if i >= length {
- return true, 0
- }
- var result = slice[i]
- i++
- return false, result
- }
- }
- func main() {
- slice := []int{1, 2, 3, 4, 5}
- nextNum := generate(slice)
- done, result := nextNum()
- // 直到 done 不等于 false,才停止
- for done == false {
- fmt.Println(result, done)
- done, result = nextNum()
- }
- fmt.Println(result, done)
- }
指针
我们前面常说的变量指的一般是一个值,指针是指向该变量存储在内存的位置。指针也可以存储在一个变量中,该变量称为『指针变量』。
指针变量声明
声明指针变量时,需要指针指向哪一种类型,因为不同类型的值在内存占用的空间大小不一样,仅仅知道内存地址还是不够,还需要知道该变量在内存中占用多大空间。声明指针变量只需要在类型前,加上 * 即可。
- var point *int // 声明 int 类型的指针
指针变量赋值
给指针变量赋值,需要在对应类型的变量前加上&符号,表示取出该变量的地址。
- var i = 1
- var point *int
- point = &i
值传递与引用传递
一般情况下,我们传入函数的参数仅为变量的值,这样的传递被称为值传递,在函数内对参数修改也不会影响到外部变量。
- func addOne(slice []int, number int) {
- slice = append(slice, number)
- fmt.Println("inner slice =", slice)
- }
- slice := []int{1, 2, 3}
- addOne(slice, 100)
- fmt.Println("outer slice =", slice)
上述代码中,我们写了一个函数,会对传入的切片追加一个值,调用之后,我们会发现外部切片的值并没有发生变量。
如果需要外部变量的值会跟随函数调用发生变化,就需要将变量的指针传入函数中,这样的传递被称为引用传递。这样在函数中修改参数就会影响到外部的变量了。
- // 此时 slice 为指针变量
- func addOne(slice *[]int, number int) {
- // 通过 *slice 可以取出 slice 指针对应的值
- *slice = append(*slice, number)
- fmt.Println("inner slice =", *slice)
- }
- slice := []int{1, 2, 3}
- addOne(&slice, 100)
- fmt.Println("outer slice =", slice)