在Go語言中,可以使用sync.WaitGroup
來實現大規模任務的并發優化。WaitGroup
是一個計數器,它等待一組goroutine完成工作。
下面是使用sync.WaitGroup
的并發優化方法的示例代碼:
package main
import (
"fmt"
"sync"
)
func main() {
// 創建WaitGroup
var wg sync.WaitGroup
// 設置要處理的任務數量
taskCount := 10
// 增加WaitGroup的計數器
wg.Add(taskCount)
// 啟動goroutine執行任務
for i := 0; i < taskCount; i++ {
go func(taskID int) {
// 執行任務的邏輯
fmt.Printf("Task %d started\n", taskID)
// 模擬任務的耗時操作
for j := 0; j < 100000000; j++ {
}
fmt.Printf("Task %d finished\n", taskID)
// 減少WaitGroup的計數器
wg.Done()
}(i)
}
// 等待所有任務完成
wg.Wait()
fmt.Println("All tasks finished")
}
在上述示例代碼中,我們首先創建了一個WaitGroup
,并設置要處理的任務數量為10。然后,我們使用wg.Add(taskCount)
增加WaitGroup
的計數器。
在啟動每個任務的goroutine之前,我們使用匿名函數傳遞任務ID,并在該goroutine中執行任務的邏輯。在任務完成后,我們使用wg.Done()
減少WaitGroup
的計數器。
最后,我們使用wg.Wait()
等待所有任務完成。一旦所有任務完成,程序會繼續執行,并輸出"All tasks finished"。
使用sync.WaitGroup
可以方便地管理大規模任務的并發執行,確保所有任務完成后再繼續執行后續的代碼。