使用Go語言的WaitGroup可以很方便地實現并發計算的多核處理。下面是一個示例代碼,演示了如何使用WaitGroup實現并發計算。
package main
import (
"fmt"
"sync"
)
func calculate(num int, wg *sync.WaitGroup) {
defer wg.Done()
result := num * num
fmt.Printf("Result for %d: %d\n", num, result)
}
func main() {
var wg sync.WaitGroup
nums := []int{1, 2, 3, 4, 5}
for _, num := range nums {
wg.Add(1)
go calculate(num, &wg)
}
wg.Wait()
fmt.Println("All calculations completed.")
}
在上面的代碼中,首先創建了一個WaitGroup對象wg
,用于等待所有并發計算的完成。然后定義了一個calculate
函數,用于執行具體的計算任務。在calculate
函數中,使用defer wg.Done()
標記函數執行完成,然后通過wg.Done()
通知WaitGroup計算任務已完成。
在main
函數中,首先定義了一個待計算的數字列表nums
。然后,使用wg.Add(1)
增加計數器的值,表示有一個計算任務需要等待完成。接著,使用go calculate(num, &wg)
并發地執行計算任務。
最后,使用wg.Wait()
等待所有并發計算的完成。當所有計算任務都完成時,會通過fmt.Println
輸出"All calculations completed."。