Go基础之函数

Posted on | In

定义函数

使用 func 关键字

1
2
3
func 函数名(参数) (返回值) {
    函数体
}

例如:

1
2
3
func add(x int, y int) int {
    return x + y
}

调用函数

1
2
sum := add(1, 2)
fmt.Println(sum)

函数的参数

上方的 add 方法的参数可以简写

1
2
3
func add(x, y int) int {
    return x + y
}

可变参数

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
// 可变参数
func compute(params ...int) int {
    // params 是一个int型的切片
    fmt.Printf("x: %T %v \n", params, params)
    sum := 0
    for _, v := range params {
        sum += v
    }
    return sum
}

func main() {
    sum2 := compute(1, 2, 3)
    fmt.Println(sum2)
}

函数返回值

多返回值

1
2
3
4
5
6
7
8
9
10
func compute2(x, y int) (int, int) {
    sum := x + y
    sub := x - y
    return sum, sub
}

func main() {
    sum3, sub3 := compute2(10, 5)
    fmt.Println(sum3, sub3)
}

返回值命名

1
2
3
4
5
6
7
8
9
10
11
func compute3(x, y int) (sum int, sub int) {
    sum = x + y
    sub = x - y
    // 这里只需要return即可
    return
}

func main() {
    sum4, sub4 := compute3(10, 5)
	fmt.Println(sum4, sub4)
}

变量作用域

全局变量

定义在函数之外的变量, 在函数中可以访问

局部变量

定义在函数体内的变量, 函数体外无法访问

语句块中定义的变量, 例如 if for switch, 只在定义的语句块中生效

defer 语句

延迟处理，多用于资源释放

先被 defer 的语句最后被执行，最后被 defer 的语句，最先被执行

1
2
3
4
5
fmt.Println("begin")
defer fmt.Println(1)
defer fmt.Println(2)
defer fmt.Println(3)
fmt.Println("end")

输出

1
2
3
4
5
begin
end
3
2
1

和 return 结合

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
package main

import "fmt"

// 在Go语言的函数中return语句在底层并不是原子操作，它分为给返回值赋值和RET指令两步。
// 而defer语句执行的时机就在返回值赋值操作后，RET指令执行前。
func f1() int {
    x := 5
    defer func() {
        x++
    }()
    return x
}

func f2() (x int) {
    defer func() {
        x++
    }()
    return 5
}

func f3() (y int) {
    x := 5
    defer func() {
        x++
    }()
    return x
}
func f4() (x int) {
    defer func(x int) {
        x++
    }(x)
    return 5
}

func calc5(index string, a, b int) int {
    ret := a + b
    fmt.Println(index, a, b, ret)
    return ret
}

func main() {
    // 5
    fmt.Println(f1())
    // 6
    fmt.Println(f2())
    // 5
    fmt.Println(f3())
    // 5
    fmt.Println(f4())

    // A 1 2 3
    // B 10 2 12
    // BB 10 12 22
    // AA 1 3 4
    x := 1
    y := 2
    defer calc5("AA", x, calc5("A", x, y))
    x = 10
    defer calc5("BB", x, calc5("B", x, y))
    y = 20
}

学习资料

  1. https://www.bilibili.com/video/BV1fz4y1m7Pm/?p=35
  2. https://www.liwenzhou.com/posts/Go/09_function/

我的环境

  • Mac
  • go1.19
  • GoLand