go defer的作用是什么

defer 關(guān)鍵字

很多現(xiàn)代的編程語(yǔ)言中都有 defer 關(guān)鍵字，Go 語(yǔ)言的 defer 會(huì)在當(dāng)前函數(shù)返回前執(zhí)行傳入的函數(shù)，它會(huì)經(jīng)常被用于關(guān)閉文件描述符、關(guān)閉數(shù)據(jù)庫(kù)連接以及解鎖資源。

使用 defer 的最常見(jiàn)場(chǎng)景是在函數(shù)調(diào)用結(jié)束后完成一些收尾工作，例如在 defer 中回滾數(shù)據(jù)庫(kù)的事務(wù)：

func InsertObj() error {
    //獲取一個(gè)數(shù)據(jù)庫(kù)session
    db := db.GetDB().Begin()
    defer db.Rollback()
    
    //do someting
    err := db.Insert()
    
    if err != nil{
     return err
    }
    
    return db.Commit().Error
}

在使用數(shù)據(jù)庫(kù)事務(wù)時(shí)，我們可以使用上面的代碼在創(chuàng)建事務(wù)后就立刻調(diào)用 Rollback 保證事務(wù)一定會(huì)回滾。哪怕事務(wù)真的執(zhí)行成功了，那么調(diào)用 Commit() 之后再執(zhí)行 Rollback() 也不會(huì)影響已經(jīng)提交的事務(wù)。

defer的特性

• defer 的調(diào)用時(shí)機(jī)以及多次調(diào)用     defer 時(shí)執(zhí)行順序
• defer 使用傳值的方式傳遞參數(shù)時(shí)會(huì)進(jìn)行預(yù)計(jì)算，導(dǎo)致不符合預(yù)期的結(jié)果

func main() {

	defer fmt.Println(0)
        defer fmt.Println(1)
        defer fmt.Println(2)
        defer fmt.Println(3)
        	
}

上面的代碼輸出:

3
2
1
0

那么defer的執(zhí)行順序就是先進(jìn)后出

func main() {
    {
        defer fmt.Println("defer runs")
        fmt.Println("block ends")
    }
    
    fmt.Println("main ends")
}

輸出:

block ends
main ends
defer runs

可以看出defer不是在退出代碼塊的作用域時(shí)執(zhí)行的，它只會(huì)在當(dāng)前函數(shù)和方法返回之前被調(diào)用

預(yù)計(jì)算

func main() {
	startedAt := time.Now()
	defer fmt.Println(time.Since(startedAt))

	time.Sleep(time.Second)
}

輸出:

0s

調(diào)用 defer 關(guān)鍵字會(huì)立刻拷貝函數(shù)中引用的外部參數(shù)，所以 time.Since(startedAt) 的結(jié)果不是在 main 函數(shù)退出之前計(jì)算的，而是在 defer 關(guān)鍵字調(diào)用時(shí)計(jì)算的，最終導(dǎo)致上述代碼輸出 0s。

那么如果要解決這個(gè)問(wèn)題可以改寫(xiě)代碼

func main() {
	startedAt := time.Now()
	defer func() { fmt.Println(time.Since(startedAt)) }()

	time.Sleep(time.Second)
}

輸出:

1s

雖然調(diào)用 defer 關(guān)鍵字時(shí)也使用值傳遞，但是因?yàn)榭截惖氖?code>函數(shù)指針

defer 常用的案例

捕獲異常和崩潰日志:

func process() error {
	defer func() {
                //防止意外之外的錯(cuò)誤導(dǎo)致程序崩潰
		if err := recover(); err != nil {
                        //崩潰日志的捕獲
			core.LogPnc.Errorf("panic : %v", err)
			core.LogPnc.Errorf("%s", string(debug.Stack()))
		}
	}()
        //do something
	err := obj.CALL(parseYML)
	if err != nil {
		return err
	}
	return nil
}

釋放鎖:

func runTask() {
        //上鎖
	c.lk.Lock()
        //釋放鎖
	defer c.lk.Unlock()
	//do something
}

在函數(shù)返回之前進(jìn)行業(yè)務(wù)操作

func (c *ProcessEngine) Process(ph *ProcessHook) {
	defer func() {

		c.lk.Lock()
                //邏輯處理
		c.running--
		c.lk.Unlock()


	}()
        //do something
}