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

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

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

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

package mainimport ("fmt""sync")func main() {var wg sync.WaitGroupwg.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 mainimport ("fmt")func worker(id int, jobs <-chan int, results chan<- int) {for j := range jobs {// 处理任务的逻辑fmt.Println("worker", id, "开始处理任务", j)results <- j * 2fmt.Println("worker", id, "完成任务", j)}}func main() {const numJobs = 5jobs := 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 开始处理任务 1worker 1 开始处理任务 2worker 2 开始处理任务 3worker 1 完成任务 2worker 1 开始处理任务 4worker 2 完成任务 3worker 3 完成任务 1worker 1 完成任务 4worker 2 开始处理任务 5worker 2 完成任务 5246810

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

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