在Go语言中,可以使用WaitGroup和协程池来实现高效的并发编程。
下面是一个使用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和协程池,我们可以高效地实现并发编程,并充分利用多核处理器的性能。