在Go语言中,可以使用WaitGroup和协程池来实现高效的并发编程。
- 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完成 所有协程完成
- 协程池:协程池是一组可重用的协程,用于执行并发任务。通过限制协程的数量,可以避免创建过多的协程导致系统资源耗尽。
下面是一个简单的协程池的实现示例:
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和协程池,我们可以高效地实现并发编程,并充分利用多核处理器的性能。