Golang中的并發控制和Go WaitGroup
在Golang中，我們可以使用goroutine實現并發執行任務。然而，在某些情況下，我們需要控制并發執行的數量，以避免資源過度消耗或者出現并發競爭的問題。Golang中提供了一些方法來實現并發控制，其中最常用的是使用Go WaitGroup。
Go WaitGroup是一個計數信號量，用于等待一組goroutine完成執行。當我們啟動一組goroutine時，我們可以使用WaitGroup來追蹤這些goroutine的狀態，并在所有goroutine完成后進行下一步的操作。
下面我們通過一個具體的代碼示例來演示Golang中的并發控制和Go WaitGroup的使用。
package main
import (
"fmt"
"sync"
"time"
)
func main() {
numWorkers := 5
var wg sync.WaitGroup
for i := 0; i < numWorkers; i++ {
wg.Add(1) // 每啟動一個goroutine，等待組加1
go worker(i, &wg)
}
wg.Wait() // 等待所有goroutine完成
fmt.Println("All workers have finished")
}
func worker(id int, wg *sync.WaitGroup) {
defer wg.Done() // goroutine執行完畢，等待組減1
fmt.Printf("Worker %d started
", id)
time.Sleep(time.Second) // 模擬耗時操作
fmt.Printf("Worker %d finished
", id)
}
登錄后復制
在這個示例中，我們有5個goroutine，每個執行的任務是調用worker函數。首先，我們定義了一個WaitGroup變量wg，并在主函數中啟動了5個goroutine。每個goroutine都會調用worker函數，并通過傳遞指向wg的指針來操作WaitGroup。
在worker函數中，我們使用defer wg.Done()來在goroutine執行完畢后減少等待組的計數。這意味著每個goroutine在完成后都會調用Done()函數來通知WaitGroup。然后，我們在每個worker函數中加入了一些模擬的耗時操作，以便觀察并發執行的效果。
在主函數中，我們調用wg.Wait()來等待所有goroutine完成。這將導致主函數在所有goroutine執行完畢后阻塞，直到WaitGroup的計數歸零。
運行上述代碼，你將看到類似以下的輸出：
Worker 0 started Worker 1 started Worker 2 started Worker 3 started Worker 4 started Worker 3 finished Worker 2 finished Worker 0 finished Worker 1 finished Worker 4 finished All workers have finished
登錄后復制
從輸出結果可以看出，所有的goroutine都按照并發的方式啟動和運行，并在完成后通知主函數。
通過使用Go WaitGroup，我們可以很方便地控制并發執行的數量，并在所有goroutine完成后進行后續操作。這對于處理大規模并發任務或限制資源消耗非常有幫助。
總結起來，Golang中的并發控制和Go WaitGroup是實現并發編程的重要工具。我們可以使用WaitGroup來追蹤和控制一組goroutine的執行，確保并發操作的正確性和穩定性。通過這種方式，我們可以更好地利用多核處理器和資源，并提高程序的執行效率。
以上就是Golang中的并發控制和Go WaitGroup的詳細內容，更多請關注www.92cms.cn其它相關文章！
