温馨提示×

多核处理:使用Go WaitGroup实现Golang并发计算

小云
101
2023-10-08 16:35:27
栏目: 编程语言

使用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."。

0