Go: defer与return小记

语鬼_7022 · · 416 次点击 · · 开始浏览    
这是一个创建于 的文章,其中的信息可能已经有所发展或是发生改变。

1 官方定义

A defer statement pushes a function call onto a list. The list of saved calls is executed after the surrounding function returns. Defer is commonly used to simplify functions that perform various clean-up actions.
defer表达式将一个函数调用保存在列表中,当包裹defer的函数"返回"后,列表中的调用会被执行。defer通常用于清理收尾工作。
注意:这里的返回加了引号,原因如下

2 实现逻辑

参考 雨痕大神的读书笔记(https://github.com/qyuhen/book)源码第20章
大致表达为:
step 1 : 在defer表达式的地方,会调用runtime.deferproc(size int32, fn *funcval)保存延时调用,注意这里保存了延时调用的参数
step 2 : 在return时,先将返回值保存起来
step 3 : 按FILO顺序调用runtime.deferreturn,即延时调用
step 4 : RET指令

因此,return并不是一个原子操作,函数返回值可能与你的预期不一样。

3 避坑提示

1. defer的参数在声明时即被确定下来,先看个例子(生产环境这样写估计会被唾沫喷死)
func calc(index string, a, b int) int {
    ret := a + b
    fmt.Println(index, a, b, ret)
    return ret
}

func main() {
    a := 1
    b := 2
    defer calc("1", a, calc("10", a, b))
    a = 0
    defer calc("2", a, calc("20", a, b))
    b = 1
}

输出结果为:

10 1 2 3
20 0 2 2
2 0 2 2
1 1 3 4

原因是defer calc("1", a, calc("10", a, b)) 的第3个参数会在调用runtime.deferproc时确定,并不会在延时调用时才会被计算。

2. 有名与无名返回值
func namedReturn() (r int) {
    defer func() {
        r++
        fmt.Println("defer in namedReturn : r = ", r)
    }()

    return
}

func unnamedReturn() int {
    var r int
    defer func() {
        r++
        fmt.Println("defer in unnamedReturn : r = ", r)
    }()
    return r
}

func main() {

    fmt.Println("namedReturn : r = ", namedReturn())

    fmt.Println("unnamedReturn : r = ", unnamedReturn())
}

输出结果为:

defer in namedReturn : r =  1
namedReturn : r =  1
defer in unnamedReturn : r =  1
unnamedReturn : r =  0

原因就是return会将返回值先保存起来,对于无名返回值来说,保存在一个临时对象中,defer是看不到这个临时对象的;而对于有名返回值来说,就保存在已命名的变量中。

3. 延时参数与有名返回值遮蔽
func ShelteredReturn() (r int) {
    defer func(r int) {
        r++
        fmt.Println("defer in ShelteredReturn : r = ", r)
    }(r)
    return 0
}

func main() {

    fmt.Println("ShelteredReturn : r = ", ShelteredReturn())

}

输出结果为:

defer in ShelteredReturn : r =  1
ShelteredReturn : r =  0

虽然r是有名返回值,但在defer func(r int)中的r是形参,与ShelteredReturn的返回值不是同一个。

参考文献

[1]. http://lib.csdn.net/article/go/33950
[2]. https://blog.golang.org/defer-panic-and-recover
[3]. https://my.oschina.net/henrylee2cn/blog/505535
[4]. https://github.com/qyuhen/book


有疑问加站长微信联系(非本文作者)

本文来自:简书

感谢作者:语鬼_7022

查看原文:Go: defer与return小记

入群交流(和以上内容无关):加入Go大咖交流群,或添加微信:liuxiaoyan-s 备注:入群;或加QQ群:701969077

416 次点击  
加入收藏 微博
暂无回复
添加一条新回复 (您需要 登录 后才能回复 没有账号 ?)
  • 请尽量让自己的回复能够对别人有帮助
  • 支持 Markdown 格式, **粗体**、~~删除线~~、`单行代码`
  • 支持 @ 本站用户;支持表情(输入 : 提示),见 Emoji cheat sheet
  • 图片支持拖拽、截图粘贴等方式上传