在 Go 語言高效能實踐中，合理使用棧記憶體可以顯著減少堆分配，從而最佳化程式效能。透過避免變數逃逸、精簡結構體使用、選擇高效資料結構，並藉助工具分析逃逸情況，可以有效降低垃圾回收壓力，提升執行效率。這些最佳化方法簡單而高效，是 Go 開發者不可忽視的關鍵技巧。
理解 Go 語言中的棧與堆
在 Go 語言中，棧記憶體分配速度快，主要用於儲存短期存活的小型區域性變數。當函式執行結束時，棧記憶體會自動清理，無需額外操作，因此效率極高，非常適合用於臨時資料的處理。
相比之下，堆記憶體的分配和釋放速度較慢，通常用於儲存較大或長期存活的資料。堆上的物件由 Go 的垃圾回收器（GC）負責管理，這一過程會增加額外的時間開銷。因此，在效能最佳化中，儘量減少堆分配，讓短期資料優先留在棧上，是提升執行效率的重要策略。
謹慎傳遞指標
為了減少堆分配，應儘量避免不必要的指標傳遞。傳遞指標通常會導致 Go 在堆上為指標指向的資料分配記憶體。如果函式不需要修改資料，優先選擇值傳遞而非指標傳遞，這不僅能避免逃逸，還能提高程式的記憶體效率。
// 由於指標導致的堆分配
func example(p *int) {
*p = 10
}
// 棧分配，避免堆
func example(p int) {
p = 10
}
使用區域性變數
函式內定義的變數通常會優先分配在棧上，這使得記憶體管理更高效。為了避免變數逃逸到堆，儘量減少返回會在其他地方使用的變數，除非確有必要。這種做法有助於降低垃圾回收的壓力，從而提升程式效能。
// 這個區域性變數很可能保留在棧上
func doWork() {
value := 10 // 棧分配
fmt.Println(value)
}
限制變數作用域
變數的作用域越小，越有可能被分配在棧上，從而提高效能。應儘量在使用變數的地方附近宣告它們，避免擴大作用域。除非必要，儘量減少使用全域性或包級變數，以降低逃逸到堆的風險並最佳化記憶體管理。
預分配記憶體
在 Go 中，當切片或對映的大小已知時，提前分配足夠的記憶體能夠避免頻繁的記憶體分配和重新分配，從而減少堆分配的負擔。若每次操作都需要動態調整記憶體，Go 的垃圾回收器（GC）會為這些資料分配和釋放堆記憶體，這可能導致效能下降。透過預先分配記憶體空間，可以顯著減少記憶體分配和垃圾回收的開銷，從而提高程式的執行效率。對於切片和對映來說，建議使用 make 函式時指定足夠的容量，避免容量擴充套件導致資料遷移到堆上。此外，合理使用容量和長度來管理切片和對映，不僅能減少記憶體開銷，還能提高程式效能和穩定性。
// 預分配切片記憶體以避免重新分配
numbers := make([]int, 0, 100) // len=0, cap=100
for i := 0; i < 100; i++ {
numbers = append(numbers, i)
}
避免閉包逃逸
閉包有時會導致變數逃逸到堆上，因為它們捕獲了外部作用域中的變數。如果閉包長期持有這些變數，Go 的垃圾回收器可能會將這些變數遷移到堆上，從而增加記憶體開銷。因此，在建立捕獲區域性變數的閉包時，需要謹慎處理，避免將臨時資料放在堆上，確保高效的記憶體使用。
// 由於閉包捕獲 `num` 導致的堆分配
func example() func() {
num := 10
return func() {
fmt.Println(num)
}
}
在這個例子中，num 逃逸到堆上，因為它需要在函式的生命週期之外存活更久，正是由於閉包的捕獲作用。閉包會持有函式外部的變數，導致這些變數的生命週期延長，從而導致堆分配的記憶體開銷增加。
剖析你的程式碼
使用 Go 編譯器的逃逸分析工具來檢視你的變數是否以及在哪裡逃逸到堆上。你可以執行：
go build -gcflags="-m"
這將告訴你哪些變數逃逸到了堆上。它將產生如下輸出：
example.go:6:9: moved to heap: num
這個輸出告訴你 num 已經在堆上分配了。
使用 Sync.Pool 重用物件
如果你有頻繁分配和釋放的大物件，使用 sync.Pool 是一種有效的策略。sync.Pool 允許重用物件，從而減少不必要的堆分配和回收，提升記憶體使用效率。透過將物件儲存在池中，當需要時可重複使用，減少了頻繁的分配和釋放操作，進而改善程式效能。
import "sync"
var pool = sync.Pool{
New: func() interface{} {
return new(bigStruct) // 大物件
},
}
// 從池中借用物件
obj := pool.Get().(*bigStruct)
// 將其歸還到池中
pool.Put(obj)
避免不必要的介面使用
介面會導致額外的記憶體分配，因為它們包含了型別資訊，可能會引發動態型別轉換。為了減少堆分配，儘量避免在介面中儲存具體型別，特別是當你只使用單一具體型別時。使用具體型別代替介面可以有效減少記憶體開銷，提高程式的效能。
// 透過使用具體型別避免堆分配
func processData(data MyStruct) {
fmt.Println(data)
}
// 這可能會因為介面而強制堆分配
func processData(data interface{}) {
fmt.Println(data)
}
保持 Goroutine 輕量級
Go 語言的 Goroutines 擁有自己的棧，Go 會動態調整這些棧的大小。然而，為了減少堆記憶體使用，建議：
- 避免建立大量具有大初始棧大小的 Goroutines，因為過大的棧分配可能導致不必要的堆使用。
- 確保 Goroutines 儘快結束，如果它們只是短期任務，這可以有效減少堆分配和記憶體開銷，提高效能。
FunTester 原創精華
【連載】從 Java 開始效能測試
- 混沌工程、故障測試、Web 前端
- 服務端功能測試
- 效能測試專題
- Java、Groovy、Go
- 白盒、工具、爬蟲、UI 自動化
- 理論、感悟、影片