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