快捷導(dǎo)航

一文詳解Go語(yǔ)言中的Defer機(jī)制

更新時(shí)間：2024年11月21日 10:38:08 作者：好奇的菜鳥(niǎo)

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

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

Defer的工作原理

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

Defer的特點(diǎn)

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

后進(jìn)先出：如果有多個(gè)defer語(yǔ)句，它們的執(zhí)行順序是后進(jìn)先出。

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

Defer的示例

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

package main

import (
	"fmt"
	"os"
)

func main() {
	// 打開(kāi)一個(gè)文件
	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()

	// 寫(xiě)入數(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.")
}

代碼解釋

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

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

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

Defer的多個(gè)調(diào)用

如果有多個(gè)defer語(yǔ)句，它們的執(zhí)行順序是后進(jìn)先出。以下是一個(gè)展示多個(gè)defer語(yǔ)句執(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")
}

輸出