使用Go WaitGroup實現高效并發處理任務
在Go語言中，使用WaitGroup可以實現高效的并發處理任務。WaitGroup是一個計數信號量，用于等待一組goroutine完成它們的工作后再繼續執行下一步操作。
WaitGroup有三個主要的方法：Add()、Done()和Wait()。Add()方法用于增加等待的goroutine的數量，Done()方法表示一個goroutine已經完成，Wait()方法用于阻塞主goroutine，直到所有的goroutine都已完成。
下面將通過一個簡單的示例來說明如何使用WaitGroup并發處理任務。
首先，我們需要創建一個WaitGroup實例：
var wg sync.WaitGroup
登錄后復制
然后，我們可以使用Add()方法增加等待的goroutine數量，并在每個goroutine的最后調用Done()方法：
func main() {
// 設置等待的goroutine數量
wg.Add(2)
// 啟動第一個goroutine
go doTask1()
// 啟動第二個goroutine
go doTask2()
// 等待所有goroutine完成
wg.Wait()
}
func doTask1() {
// 模擬任務1
time.Sleep(time.Second * 1)
fmt.Println("Task 1 is done!")
// 調用Done()方法表示任務完成
wg.Done()
}
func doTask2() {
// 模擬任務2
time.Sleep(time.Second * 2)
fmt.Println("Task 2 is done!")
// 調用Done()方法表示任務完成
wg.Done()
}
登錄后復制
在上面的示例中，我們創建了兩個goroutine來執行doTask1()和doTask2()函數，每個函數都會模擬一個耗時的任務。主goroutine在啟動完所有的goroutine后調用Wait()方法阻塞自己，直到所有的任務都完成。
運行上面的代碼，可以得到類似以下的輸出：
Task 1 is done! Task 2 is done!
登錄后復制
可以看到，兩個任務是并行執行的，并在完成后分別打印出了相應的信息。使用WaitGroup可以方便地進行并發處理，并在所有任務完成后恢復主goroutine的執行。
需要注意的是，在使用WaitGroup時，必須保證每個任務都會調用Done()方法，否則主goroutine將一直阻塞在Wait()方法處，導致程序無法繼續執行。
總結起來，Go WaitGroup是一種簡單有效的并發控制工具，能夠幫助我們實現高效的并發處理任務。通過調用Add()、Done()和Wait()方法，我們可以控制goroutine的執行順序，以及在所有任務完成后繼續執行后續操作。
希望本文對您理解并發處理任務和使用Go WaitGroup有所幫助！
以上就是使用Go WaitGroup實現高效并發處理任務的詳細內容，更多請關注www.xfxf.net其它相關文章！
