99% 的程序员都被 defer 坑过
99% 的程序员都被 defer 坑过

99% 的程序员都被 defer 坑过

这个小东西坏的很,一不留神就出错

一、

先来一个简单的例子热热身:

输出:

done
second
first

这个比较简单,defer 语句的执行顺序是按调用 defer 语句的倒序执行。

二、

看看这段代码有什么问题?

这段代码其实很危险,很可能会用尽所有文件描述符。因为 defer 语句不到函数的最后一刻是不会执行的,也就是说文件始终得不到关闭。所以切记,一定不要在 for 循环中使用 defer 语句。

那怎么优化呢?可以将循环体单独写一个函数,这样每次循环的时候都会调用关闭函数。

如下:

三、

看看这三个函数的输出结果是什么?

公布答案:

a =  1
b =  5
c =  1

你答对了吗?

说实话刚开始看到这个结果时,我是相当费解,完全不知道怎么回事。

但可以看到,这三个函数都有一个共同特点,它们都有一个命名返回值,并且都在函数中引用了这个返回值。

引用的方式分两种:分别是闭包和函数参数。

先看 a() 函数:

闭包通过 r++ 修改了外部变量,返回值变成了 1。

相当于:

再看 b() 函数:

闭包内修改的只是局部变量 t,而外部变量 t 不受影响,所以还是返回 5。

相当于:

最后是 c 函数:

参数传递是值拷贝,实参不受影响,所以还是返回 1。

相当于:

那么,为了避免写出这么令人意外的代码,最好在定义函数时就不要使用命名返回值。或者如果使用了,就不要在 defer 中引用。

再看下面两个例子:

返回值符合预期,再也不用绞尽脑汁猜了。

四、

defer 表达式的函数如果在 panic 后面,则这个函数无法被执行。

输出如下,b 没有打印出来。

panic: a

goroutine 1 [running]:
main.main()
    xxx.go:87 +0x4ce
exit status 2

而如果 defer 在前,则可以执行。

输出:

b
panic: a

goroutine 1 [running]:
main.main()
    xxx.go:90 +0x4e7
exit status 2
五、

看看下面这段代码的执行顺序:

顺序如下:

  1. 调用 G() 函数;

  2. 调用 F() 函数;

  3. F() 中遇到 panic,立刻终止,不执行 panic 之后的代码;

  4. 执行 F() 中 defer 函数,遇到 recover 捕获错误,继续执行 defer 中代码,然后返回;

  5. 执行 G() 函数后续代码,最后执行 G() 中 defer 函数。

输出:

捕获异常: a
b
继续执行
c
六、

看看下面这段代码的执行顺序:

顺序如下:

  1. 调用 G() 函数;

  2. 调用 F() 函数;

  3. F() 中遇到 panic,立刻终止,不执行 panic 之后的代码;

  4. 执行 F() 中 defer 函数,由于没有 recover,则将 panic 抛到 G() 中;

  5. G() 收到 panic 则不会执行后续代码,直接执行 defer 函数;

  6. 由于没有 recover,直接抛出 F() 抛过来的异常 a,然后退出。

输出:

b
c
panic: a

goroutine 1 [running]:
main.F()
    xxx.go:90 +0x5b
main.G()
    xxx.go:82 +0x48
main.main()
    xxx.go:107 +0x4a5
exit status 2
七、

看看下面这段代码的执行顺序:

顺序如下:

  1. 调用 G() 函数;

  2. 通过 goroutine 调用 F() 函数;

  3. F() 中遇到 panic,立刻终止,不执行 panic 之后的代码;

  4. 执行 F() 中 defer 函数,由于没有 recover,则将 panic 抛到 G() 中;

  5. 由于 goroutine 内部没有进行 recover,则 goroutine 外部函数,也就是 G() 函数是没办法捕获的,程序直接崩溃退出。

输出:

b
panic: a

goroutine 6 [running]:
main.F()
    xxx.go:96 +0x5b
created by main.G
    xxx.go:87 +0x57
exit status 2
八、

最后再说一个 recover 的返回值问题:

recover 返回的是 interface {} 类型,而不是 error 类型,所以这样使用的话会报错:

err.Error undefined (type interface {} is interface with no methods)

可以这样来转换一下:

或者直接打印结果:

输出:

捕获异常: a

以上就是本文的全部内容,其实写过其他的语言的同学都知道,关闭文件句柄,释放锁等操作是很容易忘的。而 Go 语言通过 defer 很好地解决了这个问题,但在使用过程中还是要小心。

点击数:4

发表回复