解密：Go語言中主函數的等待策略，需要具體代碼示例
Go語言作為一種并發編程語言，其主函數的等待策略尤為重要。主函數需要保證在所有goroutine執行完畢之后再退出，否則可能導致程序提前終止。本文將介紹幾種常見的主函數等待策略，并提供具體的代碼示例。
在Go語言中，通常使用sync包中的WaitGroup或者channel來實現主函數的等待。下面我們將分別介紹這兩種方式的具體應用。
- 使用sync包中的WaitGroup
WaitGroup是一種同步機制，可以用來等待一組goroutine的結束。主要通過Add方法增加等待的goroutine數量，Done方法減少數量，Wait方法等待所有的goroutine執行完畢。下面是一個示例代碼：
package main
import (
"fmt"
"sync"
)
func worker(id int, wg *sync.WaitGroup) {
defer wg.Done()
fmt.Printf("Worker %d is working
", id)
}
func main() {
var wg sync.WaitGroup
for i := 1; i <= 3; i++ {
wg.Add(1)
go worker(i, &wg)
}
wg.Wait()
fmt.Println("All workers have finished")
}
登錄后復制
在上面的代碼中，我們定義了一個worker函數來模擬一個需要執行的goroutine，然后在主函數中啟動了3個worker goroutine，并通過Wait方法等待它們執行完畢。
- 使用channel
另一種常見的主函數等待策略是使用channel。我們可以通過創建一個channel，讓每個goroutine在結束時往這個channel發送一個信號，主函數則可以通過接收這個信號來判斷是否所有的goroutine都已經執行完畢。以下是一個示例代碼：
package main
import "fmt"
func worker(id int, ch chan bool) {
fmt.Printf("Worker %d is working
", id)
ch <- true
}
func main() {
numWorkers := 3
ch := make(chan bool, numWorkers)
for i := 1; i <= numWorkers; i++ {
go worker(i, ch)
}
for i := 1; i <= numWorkers; i++ {
<-ch
}
fmt.Println("All workers have finished")
}
登錄后復制
在這個示例中，我們創建了一個容量為numWorkers的channel，并讓每個worker goroutine在結束時往這個channel發送一個值。主函數通過接收這些值來判斷是否所有的goroutine都已經執行完畢。
總結
通過上面的兩種具體代碼示例，我們了解了在Go語言中實現主函數的等待策略的兩種常見方式：使用sync包中的WaitGroup和使用channel。在實際開發中，根據具體情況選擇合適的等待策略是非常重要的，這樣可以保證程序在并發執行時能夠正確地等待所有goroutine執行完畢后再退出。
