快捷導(dǎo)航

一文詳解Go語言中的Defer機制

更新時間：2024年11月21日 10:38:08 作者：好奇的菜鳥

在Go語言中,defer是一個關(guān)鍵字,用于確保資源的清理和釋放,特別是在函數(shù)中創(chuàng)建的資源,下面就跟隨小編一起來了解下Defer機制的具體使用吧

在Go語言中，defer是一個關(guān)鍵字，用于確保資源的清理和釋放，特別是在函數(shù)中創(chuàng)建的資源。defer語句會將其后的函數(shù)調(diào)用推遲到包含它的函數(shù)即將返回時執(zhí)行。這使得defer成為處理文件關(guān)閉、數(shù)據(jù)庫連接釋放、解鎖等資源清理操作的理想選擇。

Defer的工作原理

defer語句會將其函數(shù)調(diào)用放入一個延遲調(diào)用棧中。當函數(shù)執(zhí)行完畢，開始退出時，這些被推遲的函數(shù)會按照后進先出（LIFO）的順序執(zhí)行。這意味著最后被defer的函數(shù)會最先被執(zhí)行。

Defer的特點

延遲執(zhí)行：defer后的函數(shù)調(diào)用會延遲到包含它的函數(shù)即將返回時執(zhí)行。

后進先出：如果有多個defer語句，它們的執(zhí)行順序是后進先出。

參數(shù)評估：defer語句的參數(shù)在defer時就已評估，而不是在執(zhí)行時。

Defer的示例

以下是一個使用defer的示例，展示了如何確保文件在函數(shù)退出前被正確關(guān)閉，即使在寫入文件時發(fā)生錯誤。

package main

import (
	"fmt"
	"os"
)

func main() {
	// 打開一個文件
	file, err := os.OpenFile("example.txt", os.O_CREATE|os.O_WRONLY|os.O_TRUNC, 0666)
	if err != nil {
		fmt.Println("Error opening file:", err)
		return
	}

	// 使用 defer 確保文件在函數(shù)退出前關(guān)閉
	defer file.Close()

	// 寫入數(shù)據(jù)到文件
	_, err = file.WriteString("Hello, World!\n")
	if err != nil {
		fmt.Println("Error writing to file:", err)
		return
	}

	fmt.Println("Data written to file successfully.")
}

代碼解釋

打開文件：使用os.OpenFile打開或創(chuàng)建一個文件，如果打開失敗，打印錯誤并返回。

Defer語句：defer file.Close()確保文件在函數(shù)退出前被關(guān)閉。無論文件寫入操作是否成功，file.Close()都會被執(zhí)行。

寫入文件：使用file.WriteString寫入數(shù)據(jù)到文件。如果寫入失敗，打印錯誤并返回。

Defer的多個調(diào)用

如果有多個defer語句，它們的執(zhí)行順序是后進先出。以下是一個展示多個defer語句執(zhí)行順序的示例：

package main

import "fmt"

func main() {
	fmt.Println("main start")

	defer func() {
		fmt.Println("defer 1")
	}()

	defer func() {
		fmt.Println("defer 2")
	}()

	fmt.Println("main end")
}

輸出