Golang中defer、return、返回值之間執行順序的坑

Go語言中延遲函數defer充當着 try...catch 的重任，使用起來也很是簡便，然而在實際應用中，不少gopher並無真正搞明白defer、return和返回值之間的執行順序，從而掉進坑中，今天咱們就來揭開它的神祕面紗！

先來運行下面兩段代碼：

A. 匿名返回值的狀況

package main

import (
	"fmt"
)

func main() {
	fmt.Println("a return:", a()) // 打印結果爲 a return: 0
}

func a() int {
	var i int
	defer func() {
		i++
		fmt.Println("a defer2:", i) // 打印結果爲 a defer2: 2
	}()
	defer func() {
		i++
		fmt.Println("a defer1:", i) // 打印結果爲 a defer1: 1
	}()
	return i
}

 

B. 有名返回值的狀況

package main

import (
	"fmt"
)

func main() {
	fmt.Println("b return:", b()) // 打印結果爲 b return: 2
}

func b() (i int) {
	defer func() {
		i++
		fmt.Println("b defer2:", i) // 打印結果爲 b defer2: 2
	}()
	defer func() {
		i++
		fmt.Println("b defer1:", i) // 打印結果爲 b defer1: 1
	}()
	return i // 或者直接 return 效果相同
}

 

先來假設出結論（這是正確結論），幫助你們理解緣由：

1. 多個defer的執行順序爲「後進先出」；

2. 全部函數在執行RET返回指令以前，都會先檢查是否存在defer語句，若存在則先逆序調用defer語句進行收尾工做再退出返回；

3. 匿名返回值是在return執行時被聲明，有名返回值則是在函數聲明的同時被聲明，所以在defer語句中只能訪問有名返回值，而不能直接訪問匿名返回值；

4. return其實應該包含先後兩個步驟：第一步是給返回值賦值（若爲有名返回值則直接賦值，若爲匿名返回值則先聲明再賦值）；第二步是調用RET返回指令並傳入返回值，而RET則會檢查defer是否存在，若存在就先逆序插播defer語句，最後RET攜帶返回值退出函數；

‍‍所以，‍‍defer、return、返回值三者的執行順序應該是：return最早給返回值賦值；接着defer開始執行一些收尾工做；最後RET指令攜帶返回值退出函數。

 

 

如何解釋兩種結果的不一樣：

上面兩段代碼的返回結果之因此不一樣，其實從上面的結論中已經很好理解了。

a()int 函數的返回值沒有被提早聲明，其值來自於其餘變量的賦值，而defer中修改的也是其餘變量（其實該defer根本沒法直接訪問到返回值），所以函數退出時返回值並無被修改。

b()(i int) 函數的返回值被提早聲明，這使得defer能夠訪問該返回值，所以在return賦值返回值 i 以後，defer調用返回值 i 並進行了修改，最後導致return調用RET退出函數後的返回值纔會是defer修改過的值。

 

C. 下面咱們再來看第三個例子，驗證上面的結論：

package main

import (
	"fmt"
)

func main() {
	c:=c()
	fmt.Println("c return:", *c, c) // 打印結果爲 c return: 2 0xc082008340
}

func c() *int {
	var i int
	defer func() {
		i++
		fmt.Println("c defer2:", i, &i) // 打印結果爲 c defer2: 2 0xc082008340
	}()
	defer func() {
		i++
		fmt.Println("c defer1:", i, &i) // 打印結果爲 c defer1: 1 0xc082008340
	}()
	return &i
}

 

雖然 c()*int 的返回值沒有被提早聲明，可是因爲 c()*int 的返回值是指針變量，那麼在return將變量 i 的地址賦給返回值後，defer再次修改了 i 在內存中的實際值，所以return調用RET退出函數時返回值雖然依舊是原來的指針地址，可是其指向的內存實際值已經被成功修改了。

即，咱們假設的結論是正確的！

------------------------------------補充--------------------------------------

D.defer聲明時會先計算肯定參數的值，defer推遲執行的僅是其函數體。

package main

import (
	"fmt"
	"time"
)

func main() {
	defer P(time.Now())
	time.Sleep(5e9)
	fmt.Println("1", time.Now())
}
func P(t time.Time) {
	fmt.Println("2", t)
	fmt.Println("3", time.Now())
}

// 輸出結果：
// 1 2017-08-01 14:59:47.547597041 +0800 CST
// 2 2017-08-01 14:59:42.545136374 +0800 CST
// 3 2017-08-01 14:59:47.548833586 +0800 CST

 

E. defer的做用域

1. defer只對當前協程有效（main能夠看做是主協程）；

2. 當panic發生時依然會執行當前（主）協程中已聲明的defer，但若是全部defer都未調用recover()進行異常恢復，則會在執行完全部defer後引起整個進程崩潰；

3. 主動調用os.Exit(int)退出進程時，已聲明的defer將再也不被執行。