多核處理:使用Go WaitGroup實(shí)現(xiàn)Golang并發(fā)計(jì)算

小云
102
2023-10-08 16:35:27

使用Go語(yǔ)言的WaitGroup可以很方便地實(shí)現(xiàn)并發(fā)計(jì)算的多核處理。下面是一個(gè)示例代碼,演示了如何使用WaitGroup實(shí)現(xiàn)并發(fā)計(jì)算。

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.")
}

在上面的代碼中,首先創(chuàng)建了一個(gè)WaitGroup對(duì)象wg,用于等待所有并發(fā)計(jì)算的完成。然后定義了一個(gè)calculate函數(shù),用于執(zhí)行具體的計(jì)算任務(wù)。在calculate函數(shù)中,使用defer wg.Done()標(biāo)記函數(shù)執(zhí)行完成,然后通過(guò)wg.Done()通知WaitGroup計(jì)算任務(wù)已完成。

main函數(shù)中,首先定義了一個(gè)待計(jì)算的數(shù)字列表nums。然后,使用wg.Add(1)增加計(jì)數(shù)器的值,表示有一個(gè)計(jì)算任務(wù)需要等待完成。接著,使用go calculate(num, &wg)并發(fā)地執(zhí)行計(jì)算任務(wù)。

最后,使用wg.Wait()等待所有并發(fā)計(jì)算的完成。當(dāng)所有計(jì)算任務(wù)都完成時(shí),會(huì)通過(guò)fmt.Println輸出"All calculations completed."。

0