温馨提示×

高效并发编程:使用Go WaitGroup和协程池

小云
101
2023-10-08 14:29:28
栏目: 编程语言

在Go语言中,可以使用WaitGroup和协程池来实现高效的并发编程。

  1. WaitGroup:WaitGroup是一个计数器,用于等待一组协程的完成。在主协程中添加计数器的数量,然后在每个协程中完成任务后减少计数器的数量。主协程可以使用Wait方法等待所有协程完成任务。

下面是一个使用WaitGroup的示例代码:

package main
import (
"fmt"
"sync"
)
func main() {
var wg sync.WaitGroup
wg.Add(2)
go func() {
defer wg.Done()
// 第一个协程的任务
fmt.Println("协程1完成")
}()
go func() {
defer wg.Done()
// 第二个协程的任务
fmt.Println("协程2完成")
}()
wg.Wait()
fmt.Println("所有协程完成")
}

输出结果为:

协程2完成
协程1完成
所有协程完成
  1. 协程池:协程池是一组可重用的协程,用于执行并发任务。通过限制协程的数量,可以避免创建过多的协程导致系统资源耗尽。

下面是一个简单的协程池的实现示例:

package main
import (
"fmt"
)
func worker(id int, jobs <-chan int, results chan<- int) {
for j := range jobs {
// 处理任务的逻辑
fmt.Println("worker", id, "开始处理任务", j)
results <- j * 2
fmt.Println("worker", id, "完成任务", j)
}
}
func main() {
const numJobs = 5
jobs := make(chan int, numJobs)
results := make(chan int, numJobs)
// 启动3个协程池
for w := 1; w <= 3; w++ {
go worker(w, jobs, results)
}
// 添加任务到任务通道
for j := 1; j <= numJobs; j++ {
jobs <- j
}
close(jobs)
// 读取任务结果
for r := 1; r <= numJobs; r++ {
fmt.Println(<-results)
}
}

输出结果为:

worker 3 开始处理任务 1
worker 1 开始处理任务 2
worker 2 开始处理任务 3
worker 1 完成任务 2
worker 1 开始处理任务 4
worker 2 完成任务 3
worker 3 完成任务 1
worker 1 完成任务 4
worker 2 开始处理任务 5
worker 2 完成任务 5
2
4
6
8
10

这个示例中,我们启动了3个协程池,每个协程池会不断从任务通道中获取任务并处理。任务结果会被发送到结果通道中,最后读取结果通道中的结果。

通过使用WaitGroup和协程池,我们可以高效地实现并发编程,并充分利用多核处理器的性能。

0