函数

函数

函数是基于功能或 逻辑进行封装的可复用的代码结构。 将一段功能复杂、很长的一段代码封装成多个代码片段(即函数),有助于提高代码可读性和可维护性。

在 Go 语言中,函数可以分为两种:

  • 带有名字的普通函数

  • 没有名字的匿名函数

函数的声明

函数的声明,使用 func 关键字,后面依次接函数名参数列表返回值列表用{}包裹的代码逻辑体

func 函数名(形式参数列表)(返回值列表){
    函数体
}
  • 形式参数列表描述了函数的参数名以及参数类型,这些参数作为局部变量,其值由参数调用者提供

  • 返回值列表描述了函数返回值的变量名以及类型,如果函数返回一个无名变量或者没有返回值,返回值列表的括号是可以省略的。

定义一个 sum 函数,接收两个 int 类型的参数,在运行中,将其值分别赋值给 a,b,并规定必须返回一个int类型的值 代码示例如下:

func sum(a int, b int) (int){
    return a + b
}
func main() {
    fmt.Println(sum(1,2))
}

函数实现可变参数

Golang 可变参数本质上就是 slice。只能有一个,且必须是最后一个。 可变参数分为几种:

  • 多个类型一致的参数

  • 多个类型不一致的参数

多个类型一致的参数

首先是多个类型一致的参数。

这边定义一个可以对多个数值进行求和的函数,

使用...int,表示一个元素为int类型的切片,用来接收调用者传入的参数。

// 使用 ...类型,表示一个元素为int类型的切片
func sum(args ...int) int {
    var sum int
    for _, v := range args {
        sum += v
    }
    return sum
}
func main() {
    fmt.Println(sum(1, 2, 3))
}

其中...是 Go 语言语法糖,如果该函数下有多个类型的函数,这个语法糖必须得是最后一个参数

同时这个语法糖,只能在定义函数时使用。

多个类型不一致的参数

上面那个例子中,我们的参数类型都是 int,如果传多个参数且这些参数的类型都不一样,可以指定类型为...interface{}

比如下面这段代码,是Go语言标准库中 fmt.Printf() 的函数原型:

import "fmt"
func MyPrintf(args ...interface{}) {
    for _, arg := range args {
        switch arg.(type) {
            case int:
                fmt.Println(arg, "is an int value.")
            case string:
                fmt.Println(arg, "is a string value.")
            case int64:
                fmt.Println(arg, "is an int64 value.")
            default:
                fmt.Println(arg, "is an unknown type.")
        }
    }
}

func main() {
    var v1 int = 1
    var v2 int64 = 234
    var v3 string = "hello"
    var v4 float32 = 1.234
    MyPrintf(v1, v2, v3, v4)
}

在某些情况下,我们需要定义一个参数个数可变的函数,具体传入几个参数,由调用者自己决定,但不管传入几个参数,函数都能够处理。

使用 slice 对象做变参时,必须展开。(slice...)

package main

import (
    "fmt"
)

func test(s string, n ...int) string {
    var x int
    for _, i := range n {
        x += i
    }

    return fmt.Sprintf(s, x)
}

func main() {
    s := []int{1, 2, 3}
    res := test("sum: %d", s...)    // slice... 展开slice
    println(res)
}

多个可变参数函数传递参数

上面提到了可以使用...来接收多个参数,除此之外,它还有一个用法,就是用来解序列,将函数的可变参数(一个切片)一个一个取出来,传递给另一个可变参数的函数,而不是传递可变参数变量本身。

同样这个用法,也只能在给函数传递参数里使用。

代码示例:

import "fmt"

func sum(args ...int) int {
    var result int
    for _, v := range args {
        result += v
    }
    return result
}

func Sum(args ...int) int {
    // 利用 ... 来解序列
    result := sum(args...)
    return result
}
func main() {
    fmt.Println(Sum(1, 2, 3))
}

函数的返回值

Go语言中的函数,在你定义的时候,就规定了此函数

  1. 有没有返回值?

    当没有指明返回值的类型时, 函数体不能有 return

  2. 返回几个值?

    Go 支持一个函数返回多个值

    func double(a int) (int, int) {
     b := a * 2
     return a, b
    }
    func main() {
        // 接收参数用逗号分隔
     a, b := double(2)
     fmt.Println(a, b)
    }
    
  3. 怎么返回值?

    Go支持返回带有变量名的值

    func double(a int) (b int) {
        // 不能使用 := ,因为在返回值哪里已经声明了为int
     b = a * 2
        // 不需要指明写回哪个变量,在返回值类型那里已经指定了
     return
    }
    func main() {
     fmt.Println(double(2))
    }
    // output: 4
    

匿名函数的使用

所谓匿名函数,就是没有名字的函数,它只有函数逻辑体,而没有函数名。

定义的格式如下

func(参数列表)(返回参数列表){
    函数体
}

一个名字实际上并没有多大区别,所有使用匿名函数都可以改成普通有名函数,那么什么情况下,会使用匿名函数呢?

定义变量名,是一个不难但是还费脑子的事情,对于那到只使用一次的函数,是没必要拥有姓名的。这才有了匿名函数。

有了这个背景,决定了匿名函数只有拥有短暂的生命,一般都是定义后立即使用。

就像这样,定义后立马执行(这里只是举例,实际代码没有意义)。

func(data int) {
    fmt.Println("hello", data)
}(100)

亦或是做为回调函数使用

// 第二个参数为函数
func visit(list []int, f func(int)) {
    for _, v := range list {
        // 执行回调函数
        f(v)
    }
}
func main() {
    // 使用匿名函数直接做为参数
    visit([]int{1, 2, 3, 4}, func(v int) {
        fmt.Println(v)
    })
}

显式 return 返回前,会先修改命名返回参数。

package main

func add(x, y int) (z int) {
    defer func() {
        println(z+100) // 输出:303
    }()

    z = x + y
    return z + 200 // 执行顺序: (z = z + 200) -> (call defer) -> (return)
}

func main() {
    println(add(1, 2)) // 输出: 203
}

输出结果:

303
203

Last updated

Was this helpful?