Golang之轻松化解defer的温柔陷阱

defer是Go语言提供的一种用于注册延迟调用的机制:让函数或语句可以在当前函数执行完毕后(包括通过return正常结束或者panic导致的异常结束)执行。深受Go开发者的欢迎,但一不小心就会掉进它的温柔陷阱,只有深入理解它的原理,我们才能轻松避开,写出漂亮稳健的代码。


为了更好的阅读体验,按惯例我手动贴上文章目录:


640?wx_fmt=png



什么是defer?

defer是Go语言提供的一种用于注册延迟调用的机制:让函数或语句可以在当前函数执行完毕后(包括通过return正常结束或者panic导致的异常结束)执行。

defer语句通常用于一些成对操作的场景:打开连接/关闭连接;加锁/释放锁;打开文件/关闭文件等。

defer在一些需要回收资源的场景非常有用,可以很方便地在函数结束前做一些清理操作。在打开资源语句的下一行,直接一句defer就可以在函数返回前关闭资源,可谓相当优雅。

  1. f, _ := os.Open("defer.txt")f, _ := os.Open("defer.txt")

  2. defer f.Close()defer f.Close()

注意:以上代码,忽略了err, 实际上应该先判断是否出错,如果出错了,直接return. 接着再判断 f是否为空,如果 f为空,就不能调用 f.Close()函数了,会直接panic的。

为什么需要defer?

程序员在编程的时候,经常需要打开一些资源,比如数据库连接、文件、锁等,这些资源需要在用完之后释放掉,否则会造成内存泄漏。

但是程序员都是人,是人就会犯错。因此经常有程序员忘记关闭这些资源。Golang直接在语言层面提供 defer关键字,在打开资源语句的下一行,就可以直接用 defer语句来注册函数结束后执行关闭资源的操作。因为这样一颗“小小”的语法糖,程序员忘写关闭资源语句的情况就大大地减少了。

怎样合理使用defer?

defer的使用其实非常简单:

  1. f,err := os.Open(filename)f,err := os.Open(filename)

  2. if err != nil {if err != nil {

  3.    panic(err)    panic(err)

  4. }}


  5. if f != nil {if f != nil {

  6.    defer f.Close()    defer f.Close()

  7. }}

在打开文件的语句附近,用defer语句关闭文件。这样,在函数结束之前,会自动执行defer后面的语句来关闭文件。

当然,defer会有小小地延迟,对时间要求特别特别特别高的程序,可以避免使用它,其他一般忽略它带来的延迟。

defer进阶

defer的底层原理是什么?

我们先看一下官方对 defer的解释:

Each time a “defer” statement executes, the function value and parameters to the call are evaluated as usual and saved anew but the actual function is not invoked. Instead, deferred functions are invoked immediately before the surrounding function returns, in the reverse order they were deferred. If a deferred function value evaluates to nil, execution panics when the function is invoked, not when the “defer” statement is executed.

翻译一下:每次defer语句执行的时候,会把函数“压栈”,函数参数会被拷贝下来;当外层函数(非代码块,如一个for循环)退出时,defer函数按照定义的逆序执行;如果defer执行的函数为nil, 那么会在最终调用函数的产生panic.

defer语句并不会马上执行,而是会进入一个栈,函数return前,会按先进后出的顺序执行。也说是说最先被定义的defer语句最后执行。先进后出的原因是后面定义的函数可能会依赖前面的资源,自然要先执行;否则,如果前面先执行,那后面函数的依赖就没有了。

在defer函数定义时,对外部变量的引用是有两种方式的,分别是作为函数参数和作为闭包引用。作为函数参数,则在defer定义时就把值传递给defer,并被cache起来;作为闭包引用的话,则会在defer函数真正调用时根据整个上下文确定当前的值。

defer后面的语句在执行的时候,函数调用的参数会被保存起来,也就是复制了一份。真正执行的时候,实际上用到的是这个复制的变量,因此如果此变量是一个“值”,那么就和定义的时候是一致的。如果此变量是一个“引用”,那么就可能和定义的时候不一致。

举个例子:

  1. func main() {func main() {

  2.    var whatever [3]struct{}    var whatever [3]struct{}


  3.    for i := range whatever {    for i := range whatever {

  4.        defer func() {        defer func() {

  5.            fmt.Println(i)            fmt.Println(i)

  6.        }()        }()

  7.    }    }

  8. }}

执行结果:

  1. 22

  2. 22

  3. 22

defer后面跟的是一个闭包(后面会讲到),i是“引用”类型的变量,最后i的值为2, 因此最后打印了三个2.

有了上面的基础,我们来检验一下成果:

  1. type number inttype number int


  2. func (n number) print()   { fmt.Println(n) }func (n number) print()   { fmt.Println(n) }

  3. func (n *number) pprint() { fmt.Println(*n) }func (n *number) pprint() { fmt.Println(*n) }


  4. func main() {func main() {

  5.    var n number    var n number


  6.    defer n.print()    defer n.print()

  7.    defer n.pprint()    defer n.pprint()

  8.    defer func() { n.print() }()    defer func() { n.print() }()

  9.    defer func() { n.pprint() }()    defer func() { n.pprint() }()


  10.    n = 3    n = 3

  11. }}

执行结果是:

  1. 33

  2. 33

  3. 33

  4. 00

第四个defer语句是闭包,引用外部函数的n, 最终结果是3; 第三个defer语句同第四个; 第二个defer语句,n是引用,最终求值是3. 第一个defer语句,对n直接求值,开始的时候n=0, 所以最后是0;

利用defer原理

有些情况下,我们会故意用到defer的先求值,再延迟调用的性质。想象这样的场景:在一个函数里,需要打开两个文件进行合并操作,合并完后,在函数执行完后关闭打开的文件句柄。

  1. func mergeFile() error {func mergeFile() error {

  2.    f, _ := os.Open("file1.txt")    f, _ := os.Open("file1.txt")

  3.    if f != nil {    if f != nil {

  4.        defer func(f io.Closer) {        defer func(f io.Closer) {

  5.            if err := f.Close(); err != nil {            if err := f.Close(); err != nil {

  6.                fmt.Printf("defer close file1.txt err %v\n", err)                fmt.Printf("defer close file1.txt err %v\n", err)

  7.            }            }

  8.        }(f)        }(f)

  9.    }    }


  10.    // ……    // ……


  11.    f, _ = os.Open("file2.txt")    f, _ = os.Open("file2.txt")

  12.    if f != nil {    if f != nil {

  13.        defer func(f io.Closer) {        defer func(f io.Closer) {

  14.            if err := f.Close(); err != nil {            if err := f.Close(); err != nil {

  15.                fmt.Printf("defer close file2.txt err %v\n", err)                fmt.Printf("defer close file2.txt err %v\n", err)

  16.            }            }

  17.        }(f)        }(f)

  18.    }    }


  19.    return nil    return nil

  20. }}

上面的代码中就用到了defer的原理,defer函数定义的时候,参数就已经复制进去了,之后,真正执行close()函数的时候就刚好关闭的是正确的“文件”了,妙哉!可以想像一下如果不这样将f当成函数参数传递进去的话,最后两个语句关闭的就是同一个文件了,都是最后一个打开的文件。

不过在调用close()函数的时候,要注意一点:先判断调用主体是否为空,否则会panic. 比如上面的代码片段里,先判断 f不为空,才会调用 Close()函数,这样最安全。

defer命令的拆解

如果defer像上面介绍地那样简单(其实也不简单啦),这个世界就完美了。事情总是没这么简单,defer用得不好,是会跳进很多坑的。

理解这些坑的关键是这条语句:

  1. return xxxreturn xxx

上面这条语句经过编译之后,变成了三条指令:

  1. 1. 返回值 = xxx1. 返回值 = xxx

  2. 2. 调用defer函数2. 调用defer函数

  3. 3. 空的return3. 空的return

1,3步才是Return 语句真正的命令,第2步是defer定义的语句,这里可能会操作返回值。

下面我们来看两个例子,试着将return语句和defer语句拆解到正确的顺序。

第一个例子:

  1. func f() (r int) {func f() (r int) {

  2.     t := 5     t := 5

  3.     defer func() {     defer func() {

  4.       t = t + 5       t = t + 5

  5.     }()     }()

  6.     return t     return t

  7. }}

拆解后:

  1. func f() (r int) {func f() (r int) {

  2.     t := 5     t := 5


  3.     // 1. 赋值指令     // 1. 赋值指令

  4.     r = t     r = t


  5.     // 2. defer被插入到赋值与返回之间执行,这个例子中返回值r没被修改过     // 2. defer被插入到赋值与返回之间执行,这个例子中返回值r没被修改过

  6.     func() {             func() {        

  7.         t = t + 5         t = t + 5

  8.     }     }


  9.     // 3. 空的return指令     // 3. 空的return指令

  10.     return     return

  11. }}

这里第二步没有操作返回值r, 因此,main函数中调用f()得到5.

第二个例子:

  1. func f() (r int) {func f() (r int) {

  2.    defer func(r int) {    defer func(r int) {

  3.          r = r + 5          r = r + 5

  4.    }(r)    }(r)

  5.    return 1    return 1

  6. }}

拆解后:

  1. func f() (r int) {func f() (r int) {

  2.     // 1. 赋值     // 1. 赋值

  3.     r = 1     r = 1


  4.     // 2. 这里改的r是之前传值传进去的r,不会改变要返回的那个r值     // 2. 这里改的r是之前传值传进去的r,不会改变要返回的那个r值

  5.     func(r int) {     func(r int) {

  6.          r = r + 5          r = r + 5

  7.     }(r)     }(r)


  8.     // 3. 空的return     // 3. 空的return

  9.     return     return

  10. }}

因此,main函数中调用f()得到1.

defer语句的参数

defer语句表达式的值在定义时就已经确定了。下面展示三个函数:

  1. func f1() {func f1() {

  2.    var err error    var err error


  3.    defer fmt.Println(err)    defer fmt.Println(err)


  4.    err = errors.New("defer error")    err = errors.New("defer error")

  5.    return    return

  6. }}


  7. func f2() {func f2() {

  8.    var err error    var err error


  9.    defer func() {    defer func() {

  10.        fmt.Println(err)        fmt.Println(err)

  11.    }()    }()


  12.    err = errors.New("defer error")    err = errors.New("defer error")

  13.    return    return

  14. }}


  15. func f3() {func f3() {

  16.    var err error    var err error


  17.    defer func(err error) {    defer func(err error) {

  18.        fmt.Println(err)        fmt.Println(err)

  19.    }(err)    }(err)


  20.    err = errors.New("defer error")    err = errors.New("defer error")

  21.    return    return

  22. }}


  23. func main() {func main() {

  24.    f1()    f1()

  25.    f2()    f2()

  26.    f3()    f3()

  27. }}

运行结果:

  1. <nil><nil>

  2. defer errordefer error

  3. <nil><nil>

第1,3个函数是因为作为函数参数,定义的时候就会求值,定义的时候err变量的值都是nil, 所以最后打印的时候都是nil. 第2个函数的参数其实也是会在定义的时候求值,只不过,第2个例子中是一个闭包,它引用的变量err在执行的时候最终变成 defer error了。关于闭包在本文后面有介绍。

第3个函数的错误还比较容易犯,在生产环境中,很容易写出这样的错误代码。最后defer语句没有起到作用。

闭包是什么?

闭包是由函数及其相关引用环境组合而成的实体,即:

  1. 闭包=函数+引用环境闭包=函数+引用环境

一般的函数都有函数名,但是匿名函数就没有。匿名函数不能独立存在,但可以直接调用或者赋值于某个变量。匿名函数也被称为闭包,一个闭包继承了函数声明时的作用域。在Golang中,所有的匿名函数都是闭包。

有个不太恰当的例子,可以把闭包看成是一个类,一个闭包函数调用就是实例化一个类。闭包在运行时可以有多个实例,它会将同一个作用域里的变量和常量捕获下来,无论闭包在什么地方被调用(实例化)时,都可以使用这些变量和常量。而且,闭包捕获的变量和常量是引用传递,不是值传递。

举个简单的例子:

  1. func main() {func main() {

  2.    var a = Accumulator()    var a = Accumulator()


  3.    fmt.Printf("%d\n", a(1))    fmt.Printf("%d\n", a(1))

  4.    fmt.Printf("%d\n", a(10))    fmt.Printf("%d\n", a(10))

  5.    fmt.Printf("%d\n", a(100))    fmt.Printf("%d\n", a(100))


  6.    fmt.Println("------------------------")    fmt.Println("------------------------")

  7.    var b = Accumulator()    var b = Accumulator()


  8.    fmt.Printf("%d\n", b(1))    fmt.Printf("%d\n", b(1))

  9.    fmt.Printf("%d\n", b(10))    fmt.Printf("%d\n", b(10))

  10.    fmt.Printf("%d\n", b(100))    fmt.Printf("%d\n", b(100))



  11. }}


  12. func Accumulator() func(int) int {func Accumulator() func(int) int {

  13.    var x int    var x int


  14.    return func(delta int) int {    return func(delta int) int {

  15.        fmt.Printf("(%+v, %+v) - ", &x, x)        fmt.Printf("(%+v, %+v) - ", &x, x)

  16.        x += delta        x += delta

  17.        return x        return x

  18.    }    }

  19. }}

执行结果:

  1. (0xc420014070, 0) - 1(0xc420014070, 0) - 1

  2. (0xc420014070, 1) - 11(0xc420014070, 1) - 11

  3. (0xc420014070, 11) - 111(0xc420014070, 11) - 111

  4. ------------------------------------------------

  5. (0xc4200140b8, 0) - 1(0xc4200140b8, 0) - 1

  6. (0xc4200140b8, 1) - 11(0xc4200140b8, 1) - 11

  7. (0xc4200140b8, 11) - 111(0xc4200140b8, 11) - 111

闭包引用了x变量,a,b可看作2个不同的实例,实例之间互不影响。实例内部,x变量是同一个地址,因此具有“累加效应”。

defer配合recover

Golang被诟病比较多的就是它的error, 经常是各种error满天飞。编程的时候总是会返回一个error, 留给调用者处理。如果是那种致命的错误,比如程序执行初始化的时候出问题,直接panic掉,省得上线运行后出更大的问题。

但是有些时候,我们需要从异常中恢复。比如服务器程序遇到严重问题,产生了panic, 这时我们至少可以在程序崩溃前做一些“扫尾工作”,如关闭客户端的连接,防止客户端一直等待等等。

panic会停掉当前正在执行的程序,不只是当前协程。在这之前,它会有序地执行完当前协程defer列表里的语句,其它协程里挂的defer语句不作保证。因此,我们经常在defer里挂一个recover语句,防止程序直接挂掉,这起到了 try...catch的效果。

注意,recover()函数只在defer的上下文中才有效(且只有通过在defer中用匿名函数调用才有效),直接调用的话,只会返回 nil.

  1. func main() {func main() {

  2.    defer fmt.Println("defer main")    defer fmt.Println("defer main")

  3.    var user = os.Getenv("USER_")    var user = os.Getenv("USER_")


  4.    go func() {    go func() {

  5.        defer func() {        defer func() {

  6.            fmt.Println("defer caller")            fmt.Println("defer caller")

  7.            if err := recover(); err != nil {            if err := recover(); err != nil {

  8.                fmt.Println("recover success. err: ", err)                fmt.Println("recover success. err: ", err)

  9.            }            }

  10.        }()        }()


  11.        func() {        func() {

  12.            defer func() {            defer func() {

  13.                fmt.Println("defer here")                fmt.Println("defer here")

  14.            }()            }()


  15.            if user == "" {            if user == "" {

  16.                panic("should set user env.")                panic("should set user env.")

  17.            }            }


  18.            // 此处不会执行            // 此处不会执行

  19.            fmt.Println("after panic")            fmt.Println("after panic")

  20.        }()        }()

  21.    }()    }()


  22.    time.Sleep(100)    time.Sleep(100)

  23.    fmt.Println("end of main function")    fmt.Println("end of main function")

  24. }}

上面的panic最终会被recover捕获到。这样的处理方式在一个http server的主流程常常会被用到。一次偶然的请求可能会触发某个bug, 这时用recover捕获panic, 稳住主流程,不影响其他请求。

程序员通过监控获知此次panic的发生,按时间点定位到日志相应位置,找到发生panic的原因,三下五除二,修复上线。一看四周,大家都埋头干自己的事,简直完美:偷偷修复了一个bug, 没有发现!嘿嘿!

后记

defer非常好用,一般情况下不会有什么问题。但是只有深入理解了defer的原理才会避开它的温柔陷阱。掌握了它的原理后,就会写出易懂易维护的代码。

参考资料

【defer那些事】https://xiaozhou.net/something-about-defer-2014-05-25.html 【defer代码案例】https://tiancaiamao.gitbooks.io/go-internals/content/zh/03.4.html 【闭包】https://www.kancloud.cn/liupengjie/go/576456 【闭包】http://blog.51cto.com/speakingbaicai/1703229 【闭包】https://blog.csdn.net/zhangzhebjut/article/details/25181151 【延迟】http://liyangliang.me/posts/2014/12/defer-in-golang/ 【defer三条原则】https://leokongwq.github.io/2016/10/15/golang-defer.html 【defer代码例子】https://juejin.im/post/5b948b3e6fb9a05d3827beda 【defer panic】https://ieevee.com/tech/2017/11/23/go-panic.html 【defer panic】https://zhuanlan.zhihu.com/p/33743255



640?wx_fmt=png

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.mzph.cn/news/242144.shtml

如若内容造成侵权/违法违规/事实不符,请联系多彩编程网进行投诉反馈email:809451989@qq.com,一经查实,立即删除!

相关文章

u-net语义分割_使用U-Net的语义分割

u-net语义分割Picture By Martei Macru On Unsplash图片由Martei Macru On Unsplash拍摄 Semantic segmentation is a computer vision problem where we try to assign a class to each pixel . Unlike the classic image classification task where only one class value is …

我国身家超过亿元的有多少人?

目前我国身家达到亿元以上的人数&#xff0c;从公开数据来看大概有13万人&#xff0c;但如果把那些统计不到的隐形亿万富翁计算在内&#xff0c;我认为至少有20万以上。公开资料显示目前我国亿万富翁人数达到133000人根据胡润2018财富报告显示&#xff0c;目前我国&#xff08;…

地理空间数据

摘要 (Summary) In this article, using Data Science and Python, I will show how different Clustering algorithms can be applied to Geospatial data in order to solve a Retail Rationalization business case.在本文中&#xff0c;我将使用数据科学和Python演示如何将…

嵌入式系统分类及其应用场景_词嵌入及其应用简介

嵌入式系统分类及其应用场景Before I give you an introduction on Word Embeddings, take a look at the following examples and ask yourself what is common between them:在向您介绍Word Embeddings之前&#xff0c;请看一下以下示例并问问自己它们之间的共同点是什么&…

山东男子5个月刷信用卡1800次,被银行处理后他选择29次取款100元

虽然我国实行的是存款自愿&#xff0c;取款自由的储蓄政策&#xff0c;客户想怎么取款&#xff0c;在什么时候取&#xff0c;取多少钱&#xff0c;完全是客户的权利&#xff0c;只要客户的账户上有钱&#xff0c;哪怕他每次取一毛钱取个100次都是客户的权利。但是明明可以一次性…

深发银行为什么要更名为平安银行?

深圳发展银行之所以更名为平安银行&#xff0c;最直接的原因是平安银行收购了深圳发展银行&#xff0c;然后又以平安集团作为主体&#xff0c;以深圳发展银行的名义收购了平安银行&#xff0c;最后两个人合并之后统一命名为平安银行。深圳发展银行更名为平安银行&#xff0c;大…

高斯过程分类和高斯过程回归_高斯过程回归建模入门

高斯过程分类和高斯过程回归Gaussian processing (GP) is quite a useful technique that enables a non-parametric Bayesian approach to modeling. It has wide applicability in areas such as regression, classification, optimization, etc. The goal of this article i…

假如购买的期房不小心烂尾了,那银行贷款是否可以不还了?

如今房价一路高升&#xff0c;再加上开发商融资难度越来越大&#xff0c;现在很多人都开始打期房的主意。期房不论是对开发商还是对购房者来说都是双赢的&#xff0c;开发商可以以较低的融资成本维持楼盘的开发&#xff0c;提高财务杠杆&#xff0c;而购房者可以较低的价格买房…

在银行存款5000万,能办理一张50万额度的信用卡吗?

拥有一张大额信用卡是很多人梦寐以求的事情&#xff0c;大额信用卡不仅实用&#xff0c;在关键时刻可以把钱拿出来刷卡或者取现&#xff0c;这是一种非常方便的融资方式。然而大额信用卡并不是说谁想申请就可以申请下来&#xff0c;正常情况下&#xff0c;10万以上额度以上的信…

hotelling变换_基于Hotelling-T²的偏最小二乘(PLS)中的变量选择

hotelling变换背景 (Background) One of the most common challenges encountered in the modeling of spectroscopic data is to select a subset of variables (i.e. wavelengths) out of a large number of variables associated with the response variable. It is common …

商业银行为什么大量组织高净值小规模活动?

在管理界有一个非常著名的定律叫做二八定律&#xff0c;所谓28定律就是20%的客户贡献了企业80%的利润。虽然这个定律在银行不一定适用&#xff0c;但同样的道理用于银行营销也是合适的。银行之所以经常组织一些高净值小规模的活动&#xff0c;因为这些客户的资产和价值比较高&a…

在县城投资买一辆出租车,一个月能收入多少钱?

在县城投资出租车能赚多少钱具体要看你是什么县城&#xff0c;比如西部的县城勉强能养活自己&#xff0c;中部的县城一个月能赚个5、6千&#xff0c;东部的小县城月赚个万元以上也有可能。具体回报率怎么样可以先算下投资一个出租车的成本投资一个出租车的构成成本比较多&#…

通过ISO镜像文件安装Ubuntu(可实现默认启动Windows的双系统)

解压文件 使用WinRAR等软件&#xff0c;Ubuntu ISO镜像文件中的casper文件夹解压到硬盘中的任意分区根目录&#xff0c;把ISO镜像也放在那个分区根目录。 使用Grub4dos启动Ubuntu 使用grub4dos启动Ubuntu&#xff0c;menu.lst写法如下。其中root命令指定了硬盘分区编号&#xf…

命名实体识别 实体抽取_您的公司为什么要关心命名实体的识别

命名实体识别 实体抽取Named entity recognition is the task of categorizing text into entities, such as people, locations, and dates. For example, for the sentence, On April 30, 1789, George Washington was inaugurated as the first president of the United Sta…

表达式测试

1111 (parameters) -> { statements; }//求平方 (int a) -> {return a * a;}//打印&#xff0c;无返回值 (int a) -> {System.out.println("a " a);}

有关西电的课程学分相关问题:必修课、选修课、补考、重修、学分

注&#xff1a;最近一年多以来学校的政策改动比较大&#xff0c;听说有选修一旦选了就必须通过&#xff0c;否则视为挂科需要重修的&#xff1b;还有的说是选修课学分够了再多选可能要收费&#xff08;未经确认&#xff0c;可能只是误传&#xff09;&#xff1b;等各种说法。本…

银行现在都很缺钱吗,为什么给的利息比以前高了?

目前无论是大银行还是小银行&#xff0c;也不论是国有银行还是民营银行&#xff0c;基本上每个银行都上浮利率&#xff0c;如果不上浮利率&#xff0c;那就只能吃土了&#xff0c;当然加息一般主要针对定期存款以及贷款来说&#xff0c;活期存款利率一般是不会上浮&#xff0c;…

机器学习 异常值检测_异常值是否会破坏您的机器学习预测? 寻找最佳解决方案

机器学习 异常值检测内部AI (Inside AI) In the world of data, we all love Gaussian distribution (also known as a normal distribution). In real-life, seldom we have normal distribution data. It is skewed, missing data points or has outliers.在数据世界中&#…

1000万贷款三年,到期一次性偿还1500万,这个利息算不算高?

1000万的贷款三年期到期还1500万&#xff0c;相当于每一年的利息是166.6万&#xff0c;折算下来年化利率是16.6%。至于这个利率是否划算&#xff0c;要看你在什么金融机构贷款以及你个人的资质来看。如果你个人条件比较好&#xff0c;在银行做的抵押贷款&#xff0c;那我认为16…

Golang之变量去哪儿

写过C/C的同学都知道&#xff0c;调用著名的malloc和new函数可以在堆上分配一块内存&#xff0c;这块内存的使用和销毁的责任都在程序员。一不小心&#xff0c;就会发生内存泄露&#xff0c;搞得胆战心惊。切换到Golang后&#xff0c;基本不会担心内存泄露了。虽然也有new函数&…