Go defer使用時的兩個常見陷阱與避免方法

更新時間：2025年03月14日 09:23:27 作者：Ai 編碼

這篇文章主要將帶大家一起深入探討 Go 1.20 中 defer 的優(yōu)化機制,并揭示在使用 defer 時需要避免的兩個常見陷阱,有需要的可以了解下

1. Go 1.20 中的 defer 優(yōu)化

在 Go 1.13 中，defer 的性能得到了顯著提升，主要得益于編譯器對 defer 的堆棧分配優(yōu)化。而在 Go 1.20 中，defer 的優(yōu)化進一步得到了增強，特別是在處理循環(huán)中的 defer 時，編譯器能夠更智能地決定 defer 對象的分配方式。

1.1 堆棧分配優(yōu)化

在 Go 1.20 中，編譯器會根據 defer 的使用場景，自動選擇將其分配在棧上還是堆上。對于大多數簡單的 defer 調用，編譯器會優(yōu)先將其分配在棧上，從而避免了堆分配帶來的性能開銷。

package main

import "fmt"

func main() {
    defer fmt.Println("Go 1.20 defer 優(yōu)化")
    fmt.Println("開始執(zhí)行")
}

輸出結果：

開始執(zhí)行
Go 1.20 defer 優(yōu)化

在這個例子中，defer 語句被分配在棧上，執(zhí)行效率更高。

1.2 循環(huán)中的 defer 優(yōu)化

在 Go 1.20 中，編譯器對循環(huán)中的 defer 進行了更智能的處理。如果編譯器能夠確定循環(huán)的迭代次數較少，它會將 defer 分配在棧上，從而避免頻繁的堆分配。

package main

import "fmt"

func main() {
    for i := 0; i < 3; i++ {
        defer fmt.Println("迭代次數:", i)
    }
    fmt.Println("循環(huán)結束")
}

輸出結果：

循環(huán)結束
迭代次數: 2
迭代次數: 1
迭代次數: 0

在這個例子中，由于循環(huán)次數較少，編譯器將 defer 分配在棧上，避免了堆分配的開銷。

2. 使用 defer 時需要避免的兩個陷阱

盡管 Go 1.20 對 defer 進行了優(yōu)化，但在某些情況下，不當使用 defer 仍然會導致性能問題。以下是兩個常見的陷阱：

2.1 顯式循環(huán)中的 defer

在顯式循環(huán)中使用 defer 可能會導致 defer 鏈表過長，從而影響性能。特別是在循環(huán)次數較多的情況下，defer 鏈表會變得非常龐大，導致內存占用增加和性能下降。

package main

import "fmt"

func main() {
    for i := 0; i < 10000; i++ {
        defer fmt.Println("顯式循環(huán)中的 defer:", i)
    }
    fmt.Println("顯式循環(huán)結束")
}

在這個例子中，defer 鏈表會包含 10000 個節(jié)點，導致內存占用增加和性能下降。

2.2 隱式循環(huán)中的 defer

隱式循環(huán)中的 defer 同樣會導致性能問題。例如，使用 goto 語句實現(xiàn)的隱式循環(huán)會導致 defer 鏈表不斷增長，從而影響性能。

package main

import "fmt"
func main() {
    i := 1
food:
    defer func() {
        fmt.Println("隱式循環(huán)中的 defer")
    }()
    if i == 1 {
        i -= 1
        goto food
    }
    fmt.Println("隱式循環(huán)結束")
}

在這個例子中，goto 語句會導致 defer 鏈表不斷增長，最終影響性能。