golang的defer精析

最新推荐文章于 2025-04-08 16:26:31 发布

wxmxl

最新推荐文章于 2025-04-08 16:26:31 发布

阅读量201

点赞数

分类专栏： Golang 文章标签： golang defer

Golang 专栏收录该内容

11 篇文章

订阅专栏

关于golang的defer，先看四个例子，在心里默默计算一下返回值：

example1

func f1() (result int) { 
    defer func() { 
        result++ 
    }() 
    return 0
}

example2

func f2() (r int) { 
    t := 5 
    defer func() { 
        t = t + 5 
    }() 
    return t
}

example3

func f3() (r int) { 
    defer func(r int) { 
        r = r + 5 
    }(r) 
    return 1
}

example4

func f4() (r int) { 
    defer func() { 
        r = r + 5 
    }() 
    return 1
}

先不要运行代码，自己在心里跑一遍结果。然后再去验证。如果四个都做对了并且不是蒙的....好吧，不用往下看了，你已经懂defer了。

如果example1中你算的是0，你就错了

如果example2中你觉得是10，你又错了...蒙对的不算...

如果example3中觉得得6，你又错了...如果你有算对的，也有算错，好吧...你丫的就是在蒙！

如果example4中觉得得6，那有没有发现和example3相比，程序有什么不一样？

不懂的继续往下看啊.....

首先要明确的是：defer是在return之前执行的

这是官方文档中明确说明的http://golang.org/ref/spec#Defer_statements ，知道就行了，可以无视

然后要了解是的defer的实现方式

大意就是在defer出现的地方插入的指令

CALL runtime.deferproc

然后在函数返回之前的地方，插入指令

CALL runtime.deferreturn

再就是明确go返回值的方式跟C是不一样的，为了支持多值返回，go是用栈返回值的，而C是用寄存器。

最最最重要的一点就是：return xxx 这一句语句并不是一条原子指令！

整个return过程，没有defer之前是，先把在栈中写一个值，这个值被会当作返回值。然后再调用RET指令返回。return xxx语句汇编后是先给返回值赋值，再做一个空的return: ( 赋值指令＋ RET指令)

defer的执行是被插入到return指令之前的

有了defer之后，就变成了 (赋值指令 + CALL defer指令 + RET指令)

而在CALL defer函数中，有可能将最终的返回值改写了...也有可能没改写。总之，如果改写了，那么看上去就像defer是在return xxx之后执行的~

这是所有你所想不明白的defer故事发生的根源。

上面的基础知识都有了，然后就可以来说说神奇的defer了。告诉大家一个简单的转换规则大家就再也不为defer迷糊了。

改写规则是将return语句分开成两句写，return xxx会被改写成:

返回值 = xxx

调用defer函数

空的return

先看example1。它可以改写成这样:

func f1() (result int) { 
    result = 0 //return语句不是一条原子调用，return xxx其实是赋值＋RET指令 
    func() { //defer被插入到return之前执行，也就是赋返回值和RET指令之间 
        result++ 
    }() 
    return
}

所以这个返回的是1

再看example2。它可以改写成这样：

func f2() (r int) { 
    t := 5 
    r = t //赋值指令 
    func() { //defer被插入到赋值与返回之间执行，这个例子中返回值r没被修改过 
        t = t + 5 
    } 
    return //空的return指令
}

所以这个的结果是5

最后看example3。它改写后变成：

func f3() (r int) { 
    r = 1 //给返回值赋值 
    func(r int) { //这里改的r是传值传进去的r，不会改变要返回的那个r值 
        r = r + 5 
    }(r) 
    return //空的return
}

所以这个例子结果是1

example4,它改写后变成：

它改写后变成：

func f4() (r int) { 
    r = 1 //给返回值赋值 
    func() { //这里改的r是传值传进去的r，不会改变要返回的那个r值 
        r = r + 5 
    }() 
    return //空的return
}

所以这个例子的结果是6

结论：defer确实是在return之前调用的。但表现形式上却可能不像。本质原因是return xxx语句并不是一条原子指令，defer被插入到了赋值与 RET之前，因此可能有机会改变最终的返回值。

当你觉得迷糊时，可以用我给的这套规则转一下代码。

本文主体参考了文章：https://studygolang.com/articles/742