go語(yǔ)言中的defer
在Go語(yǔ)言中,defer語(yǔ)句用于在函數(shù)返回之前執(zhí)行一段代碼,通常用于資源清理或錯(cuò)誤處理。你提供的代碼片段是一個(gè)defer函數(shù)defer語(yǔ)句可以出現(xiàn)在函數(shù)的任何地方,但通常放在函數(shù)的開頭,以確保資源清理或錯(cuò)誤處理的邏輯不會(huì)被遺漏。
1.defer 的特點(diǎn)
延遲執(zhí)行:defer語(yǔ)句中的函數(shù)調(diào)用將在當(dāng)前函數(shù)返回之前執(zhí)行。后進(jìn)先出:如果有多個(gè)defer語(yǔ)句,它們將按照后進(jìn)先出(LIFO)的順序執(zhí)行。
錯(cuò)誤處理:defer語(yǔ)句可以捕獲并處理函數(shù)返回時(shí)的錯(cuò)誤。
資源清理:defer語(yǔ)句常用于關(guān)閉文件、釋放鎖、關(guān)閉網(wǎng)絡(luò)連接等資源清理操作。
defer func() { ... }():
- 確保在函數(shù)返回時(shí)執(zhí)行清理操作。
- 刷新緩沖區(qū)并關(guān)閉連接,即使發(fā)生錯(cuò)誤也不會(huì)影響程序的邏輯。
- 忽略Flush方法的返回值,因?yàn)榧词顾⑿率∫膊粫?huì)影響后續(xù)操作。
- 如果存在錯(cuò)誤,關(guān)閉連接以釋放資源。
- 忽略Close方法的返回值,因?yàn)榧词龟P(guān)閉失敗也不會(huì)影響程序的邏輯。
defer func() {
_ = c.buf.Flush()
if err != nil {
_ = c.Close()
}
}()
2.作用
是確保在函數(shù)返回時(shí):
- 刷新緩沖區(qū),將所有未寫入的數(shù)據(jù)寫入底層的io.Writer。
- 如果存在錯(cuò)誤,關(guān)閉連接以釋放資源。
3.defer的原理
`defer` 是 Go 語(yǔ)言中用于延遲執(zhí)行函數(shù)的機(jī)制,其原理依賴于 Go 的運(yùn)行時(shí)系統(tǒng)。每次遇到 `defer` 語(yǔ)句時(shí),編譯器會(huì)將其轉(zhuǎn)換為對(duì) `runtime.deferproc` 的調(diào)用,該函數(shù)創(chuàng)建一個(gè) `_defer` 結(jié)構(gòu)體對(duì)象并將其鏈接到當(dāng)前 goroutine 的 `_defer` 鏈表中。`_defer` 結(jié)構(gòu)體包含關(guān)鍵字段:`siz`(參數(shù)和結(jié)果大?。sp` 和 `pc`(棧指針和程序計(jì)數(shù)器)、`fn`(延遲函數(shù)指針)和 `link`(指向下一個(gè) `_defer` 對(duì)象)。當(dāng)函數(shù)返回時(shí),`runtime.deferreturn` 按后進(jìn)先出順序執(zhí)行延遲函數(shù)并返回。
4.defer 在資源清理中的應(yīng)用
defer 常用于關(guān)閉文件、釋放鎖、關(guān)閉網(wǎng)絡(luò)連接等資源清理操作,確保資源在函數(shù)退出時(shí)被正確釋放。
func readFile(filename string) error {
file, err := os.Open(filename) if err != nil { return err
} defer file.Close() // 讀取文件內(nèi)容 return nil }