Golang研学：defer!如何掌握并用好（延迟执行）

defer：在函数A内用defer关键字调用的函数B会在在函数A return后执行。

先看一个基础的例子，了解一下defer的效果

1. func main() {
2. fmt.Println('in main func:', foo())
3. }
4. func foo() int {
5. i := 0
6. defer fmt.Println('in defer :', i)
7. i = 1000
8. fmt.Println('in foo:', i)
9. return i+24
10. }
11. 复制代码

这段代码运行后会打印出

1. in foo: 1000
2. in defer : 0
3. in main func: 1024
4. 复制代码

变量i初始化为0，defer指定fmt.Println函数延迟到return后执行，最后main函数调用foo打印返回值。

有什么用途?

函数中会申明使用很多变量资源，函数结束时，我们通常会对它们做一些处理：销毁、释放（例如数据库链接、文件句柄、流）。

一般情况下，我们会在return语句之前处理这些事情。

但是，如果函数中包含多个return，这些处理我们需要在每个return之前都操作一次，实际工作中经常出现遗漏，代码维护时也很麻烦。

例如，在不用defer的时候，代码可能会这样写：

1. func foo(i int) int {
2. if i > 100 {
3. fmt.Println('不是期待的数字')
4. return 0
5. }
6. if i < 50 {
7. fmt.Println('不是期待的数字')
8. return 0
9. }
10. return i
11. }
12. 复制代码

使用defer后，代码可以这样写

1. func foo(i int) int {
2. defer func() {
3. fmt.Println('不是期待的数字')
4. }()
5. if i > 100 {
6. return 0
7. }
8. if i < 50 {
9. return 0
10. }
11. return i
12. }
13. 复制代码

一个函数中多个defer的执行顺序是什么?

defer在同一个函数中可以使用多次。

多个defer指定的函数执行顺序是'先进后出'。

为什么呢 ？

可以这样理解：defer关键字会使其以下的代码先执行后再执行它指定的函数，包括其下的defer语句也会比其先执行，依此类推。

这个顺序非常必要，因为在函数中，后面定义的对象可能依赖前面的对象，否则如果先出现的defer执行了，很可能造成后面的defer执行的时候出现异常。

所以，Go语言设计defer的时候是按先进后出的顺序执行的。

例子：

1. func foo() {
2. i := 0
3. defer func() {
4. i--
5. fmt.Println('第一个defer', i)
6. }()
7. i++
8. fmt.Println('+1后的i：', i)
9. defer func() {
10. i--
11. fmt.Println('第二个defer', i)
12. }()
13. i++
14. fmt.Println('再+1后的i：', i)
15. defer func() {
16. i--
17. fmt.Println('第三个defer', i)
18. }()
19. i++
20. fmt.Println('再+1后的i：', i)
21. }
22. 复制代码

运行后可以看到

1. +1后的i： 1
2. 再+1后的i： 2
3. 再+1后的i： 3
4. 第三个defer 2
5. 第二个defer 1
6. 第一个defer 0
7. 复制代码

这个过程可以看出函数执行后，先进后出执行defer并逐步处理变量的过程。

当传递参数给defer指定的函数时，函数延迟执行，那么参数值会是多少？

网上有一些总结是说：defer指定的函数的参数在 defer 时确定，但，这只是一个总结，真正的原因是， Go语言除了map、slice、chan都是值传递。

改造一下上面这个例子

1. func foo() {
2. i := 0
3. defer func(k int) {
4. fmt.Println('第一个defer', k)
5. }(i)
6. i++
7. fmt.Println('+1后的i：', i)
8. defer func(k int) {
9. fmt.Println('第二个defer', k)
10. }(i)
11. i++
12. fmt.Println('再+1后的i：', i)
13. defer func(k int) {
14. fmt.Println('第三个defer', k)
15. }(i)
16. i++
17. fmt.Println('再+1后的i：', i)
18. }
19. 复制代码

得到的结果

1. +1后的i： 1
2. 再+1后的i： 2
3. 再+1后的i： 3
4. 第三个defer 2
5. 第二个defer 1
6. 第一个defer 0
7. 复制代码

可能会有人觉得有一点出乎预料，i在return时不是已经被计算到3了吗？，为什么延迟执行的defer指定的函数里的i不是3呢？

defer关键字指定的函数是在return后执行的，这很容易让人想象在return后调用函数。

但是，defer指定的函数是在当前行就调用了的，只是延迟到return后执行，而不等同于“移动”到return后执行，因此调用时传递的是当前的参数的值。

传递指针参数会是什么情况？

那么如果希望defer指定的的函数参数的值是经过后面的代码处理过的，可以传递指针参数给defer指定的函数。

改造一下代码：

1. func foo() {
2. i := 0
3. defer func(k *int) {
4. fmt.Println('第一个defer', *k)
5. }(&i)
6. i++
7. fmt.Println('+1后的i：', i)
8. defer func(k *int) {
9. fmt.Println('第二个defer', *k)
10. }(&i)
11. i++
12. fmt.Println('再+1后的i：', i)
13. defer func(k *int) {
14. fmt.Println('第三个defer', *k)
15. }(&i)
16. i++
17. fmt.Println('再+1后的i：', i)
18. }
19. 复制代码

运行后得到

1. +1后的i： 1
2. 再+1后的i： 2
3. 再+1后的i： 3
4. 第三个defer 3
5. 第二个defer 3
6. 第一个defer 3
7. 复制代码

defer会影响返回值吗？

在开头的第一个例子中可以看到，defer是在foo执行完，main里打印返回值之前执行的，但是没有影响到main里的打印结果。

这还是因为相同的原则 Go语言除了map、slice、chan都是值传递

比较一下foo1和foo2两个函数的结果：

1. func main() {
2. fmt.Println('foo1 return :', foo1())
3. fmt.Println('foot return :', foo2())
4. }
5. func foo1() int {
6. i := 0
7. defer func() {
8. i = 1
9. }()
10. return i
11. }
12. func foo2() map[string]string {
13. m := map[string]string{}
14. defer func() {
15. m['a'] = 'b'
16. }()
17. return m
18. }
19. 复制代码

运行后，打印出

1. foo1 return : 0
2. foot return : map[a:b]
3. 复制代码

两个函数不同之处在于的返回值的类型，foo1中，int类型return后，defer不会影响返回结果，但是在foo2中map类型是引用传递，所以defer会改变返回结果。

这说明，在return时，除了map、slice、chan，其他类型return时是将值拷贝到一个临时变量空间，因此，defer指定的函数内对函数内的变量的操作不会影响返回结果的。

还有一种情况，给函数返回值申明变量名，，这时，变量空间是在函数执行前申明出来，return时只是返回这个变量空间的内容，因此defer能够改变返回值。

例如，改造一下foo1函数，给它的返回值申明一个变量名i：

1. func foo1() (i int) {
2. i = 0
3. defer func() {
4. i = 1
5. }()
6. return i
7. }
8. 复制代码

再运行，可以看到 :

1. foo1 return : 1
2. 复制代码

返回值被defer指定的函数修改了。

defer在panic和recover处理上的使用

在Go语言里，defer有一个经典的使用场景就是recover.

在函数执行过程中，有可能在很多地方都会出现panic，panic后如果不调用recover，程序会退出，为了不让程序退出，我们需要在panic后调用recover，但，panic后的代码不会执行，recover是不可能在panic后调用，然而panic所在的函数内defer指定的函数可以执行，所以recover只能在defer指定的函数中被调用，并且只需要在1个defer指定的函数中处理。

例如:

1. func panicfunc() {
2. defer func() {
3. fmt.Println('before recover')
4. recover()
5. fmt.Println('after recover')
6. }()
7. fmt.Println('before panic')
8. panic(0)
9. fmt.Println('after panic')
10. }
11. 复制代码

运行后，打印出：

1. before panic
2. before recover
3. after recover
4. 复制代码

总结以下

1. defer语句非常重要，非常常用，必须掌握
2. 在统一处理多个return和panic/recover场景下使用defer
3. 谨记“Go语言的函数参数传递的都是值（除了map、slice、chan）”这一重要原则，正确的评估defer指定函数的参数值
4. defer不影响返回值，除非是map、slice和chan，或者返回值定义了变量名
5. 执行顺序：先进后出

欢迎关注公众号，和大家一起学习编程吧