Golang基础--defer的用法

发布时间 2023-04-09 16:37:59作者: 99号的格调

defer语句用于延迟函数的调用,每次defer都会把一个函数压入栈中,函数返回前再把延迟的函数取出执行。

三个示例:

import "fmt"

func main() {
    var init = 1
    defer fmt.Println(init)
    init = 2
}

输出1。延迟函数fmt.Println(aInt)的参数在defer语句出现时就已经确定了,所以无论后面如何修改 aInt变量都不会影响延迟函数。

eg2:

import "fmt"
func Print(array *[3]int) {
    for i := range array {
        fmt.Println(array[i])
    }
}

func DeferFuncArray() {
    var array = [3]int{
        1,
        2,
        3,
    }
    defer Print(&array)
    array[0] = 10
}
func main() {
    DeferFuncArray()
}

输出1023三个值。延迟函数printArray()的参数在defer语句出现时就已经确定了,即数组的地 址,由于延迟函数执行时机是在return语句之前,所以对数组的最终修改值会被打印出来。

eg3:

import "fmt"

func deferFunc() (result int) {
    i := 1
    defer func() {
        result++
    }()
    return i
}

func main() {
    fmt.Println(deferFunc())
}

函数输出2。函数的return语句并不是原子的,实际执行分为设置返回值—>retdefer语句实际执行在 返回前,即拥有defer的函数返回过程是:设置返回值—>执行defer—>ret。所以return语句先把result设置为i 的值,即1defer语句中又把result递增1,所以最终返回2

defer规则

  规则一:延迟函数的参数在defer语句出现时就已经确定

    对于指针类型参数,规则仍然适用,只不过延迟函数的参数是一个地址值,这种情况下,defer后面的语句对变量的修改可能会影响到延迟函数

  规则二:延迟函数执行按后进先出顺序执行,即先出现的defer最后执行

  规则三:延迟函数可能操作住函数的具名返回值

    定义defer函数,即住函数可能有返回值,返回值有没有名字没有关系,defer所作用的函数,即延迟函数可能会影响返回值

defer实现原理

  defer数据结构

  type _defer struct{

    sp  uintptr //函数栈指针

    pc  uintptr //函数计数器

    fn  *funcval //函数地址

    link  *_defer //指向自身结构的指针,用于链接多个defer

  }

  与函数不同的一点是它含有一个指针,可用于指向另一个defer,每个goroutine数据结构中实际上也有一个defer 指针,该指针指向一个defer的单链表,每次声明一个defer时就将defer插入到单链表表头,每次执行defer时就 从单链表表头取出一个defer执行。一个goroutine可能连续调用多个函数,defer添加过程跟上述一致,进入函数时添加defer,离开函数时取出defer,所以即便调用多个函数,也总是能保证defer是按照先进先出方式进心的