golang defer

不要在for循环直接用defer

https://studygolang.com/articles/12061

panic

Golang里比较常见的错误处理方法是返回error给调用者,但如果是无法恢复的错误,返回error也没有意义,此时可以选择go die: 主动触发panic。

除了代码中主动触发的panic,程序运行过程中也会因为出现某些错误而触发panic,例如数组越界。

panic 会停掉当前正在执行的程序 (注意,不只是协程) ,但是与os.Exit(-1)这种直愣愣的退出不同,panic的撤退比较有秩序,他会先处理完当前 goroutine 已经 defer 挂上去的任务,执行完毕后再退出整个程序。

而defer的存在,让我们有更多的选择,比如在defer中通过recover截取panic,从而达到try..catch的效果。

panic允许传递一个参数给他,参数通常是将出错的信息以字符串的形式来表示。panic会打印这个字符串,以及触发panic的调用栈。

1
2
3
4
5
6
7
8
defer func() {
        if e := recover(); e != nil {
            var buf [4096]byte
            n := runtime.Stack(buf[:], false)
            fmt.Printf("==> %s\n", string(buf[:n]))
            os.Exit(1)
        }
}()

defer

defer 和 go一样都是Go语言提供的关键字。defer用于资源的释放,会在函数返回之前进行调用。一般采用如下模式:

1
2
3
4
5
f,err := os.Open(filename)
if err != nil {
  panic(err)
}
defer f.Close()

如果有多个defer表达式,调用顺序类似于栈,越后面的defer表达式越先被调用。

不过如果对defer的了解不够深入,使用起来可能会踩到一些坑,尤其是跟带命名的返回参数一起使用时。在讲解defer的实现之前先看一看使用defer容易遇到的问题。

defer使用时的坑

先来看看几个例子。例1:

func f() (result int) {

defer func() {

result++

}()

return 0

}

例2:

func f() (r int) {

t := 5

defer func() {

t = t + 5

}()

return t

}

例3:

func f() (r int) {

defer func(r int) {

r = r + 5

}(r)

return 1

}

请读者先不要运行代码,在心里跑一遍结果,然后去验证。

例1的正确答案不是0,例2的正确答案不是10,如果例3的正确答案不是6……

defer是在return之前执行的。这个在官方文档中是明确说明了的。要使用defer时不踩坑,最重要的一点就是要明白,return xxx这一条语句并不是一条原子指令!

函数返回的过程是这样的: 先给返回值赋值,然后调用defer表达式,最后才是返回到调用函数中。

defer表达式可能会在设置函数返回值之后,在返回到调用函数之前,修改返回值,使最终的函数返回值与你想象的不一致。

其实使用defer时,用一个简单的转换规则改写一下,就不会迷糊了。改写规则是将return语句拆成两句写,return xxx会被改写成:

返回值 = xxx

调用defer函数

空的return

先看例1,它可以改写成这样:

func f() (result int) {

result = 0 //return语句不是一条原子调用,return xxx其实是赋值+ret指令

func() { //defer被插入到return之前执行,也就是赋返回值和ret指令之间

result++

}()

return

}

所以这个返回值是1。

再看例2,它可以改写成这样:

func f() (r int) {

t := 5

r = t //赋值指令

func() { //defer被插入到赋值与返回之间执行,这个例子中返回值r没被修改过

t = t + 5

}

return //空的return指令

}

所以这个的结果是5。

最后看例3,它改写后变成:

func f() (r int) {

r = 1 //给返回值赋值

func(r int) { //这里改的r是传值传进去的r,不会改变要返回的那个r值

r = r + 5

}(r)

return //空的return

}

所以这个例子的结果是1。

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

defer的实现

defer关键字的实现跟go关键字很类似,不同的是它调用的是runtime.deferproc而不是runtime.newproc。

在defer出现的地方,插入了指令call runtime.deferproc,然后在函数返回之前的地方,插入指令call runtime.deferreturn。

普通的函数返回时,汇编代码类似:

add xx SP

return

如果其中包含了defer语句,则汇编代码是:

call runtime.deferreturn,

add xx SP

return

goroutine 的控制结构中,有一张表记录defer,调用runtime.deferproc时会将需要defer的表达式记录在表中,而在调用runtime.deferreturn的时候,则会依次从defer表中出栈并执行。

https://ieevee.com/tech/2017/11/23/go-panic.html

https://tiancaiamao.gitbooks.io/go-internals/content/zh/03.4.html

https://blog.csdn.net/fengfengdiandia/article/details/80058805

Contents

golang defer

defer