go 并发模式之一，协程池(pool)

贵哥的编程之路

发布于 2024-03-23 08:02:11

810

发布于 2024-03-23 08:02:11

协程池就是提前创建一些协程（goroutine），当有任务来时，从这些协程中选择一个空闲的协程来执行任务，任务执行完后继续保持这个协程，以便下次任务到来时复用，避免频繁地创建和销毁协程，提高程序性能和效率。

package main

import (
	"fmt"
	"sync"
)
//
func worker(id int, jobs <-chan int, results chan<- int) {//id 表示工作者的编号，jobs 是一个只接收整数的通道（用于接收任务），results 是一个只发送整数的通道（用于发送处理结果）。
	for j := range jobs {//遍历(接收所有任务)
		fmt.Printf("工作者正在处理特定 %d 编号的任务 %d\n", id, j)
		results <- j * 2//处理任务，返回结果到results中
	}
}

func main() {
	const numJobs = 5//任务数量
	const numWorkers = 3//工作的人数数量

//创建了两个通道
	jobs := make(chan int, numJobs)//传递任务
	results := make(chan int, numJobs)//获取结果

	// 创建并启动多个工作者
	for w := 1; w <= numWorkers; w++ {//在一个 for 循环中创建并启动了多个工作者协程
		go worker(w, jobs, results)//job:jobs 通道用于向工作者发送任务,esults 通道用于接收工作者处理任务后的结果。
	}//带缓冲区的。可以遍历:for w := 1; w <= numWorkers; w++ 

	// 提交任务到任务通道
	for j := 1; j <= numJobs; j++ {
		jobs <- j
	}

	close(jobs)//调用 close(jobs) 关闭了 jobs 通道，表示不再向其中提交新的任务

	// 获取处理结果
	var wg sync.WaitGroup
	wg.Add(numJobs)//表示需要等待 numJobs 个任务完成后才能继续执行后面的代码。
	//通过匿名函数启动了一个新的协程，用于等待所有任务完成后关闭 results 通道：
	go func() {
		wg.Wait()//等待所有的协程任务完毕后，就执行以下的.关闭 results 通道：
		close(results)
	}()
//执行到这里证明表示所有结果都已经发送完毕。
//通过 for range 循环遍历 results 通道中的结果，并输出每个结果：
	for r := range results {
		fmt.Println("Result:", r)
		wg.Done()//每次执行这个代表一个任务已经完成
	}
}

/*简单点来说就是首先，通过 worker 函数定义了工作者的行为：从 jobs 通道接收任务，并将处理结果发送到 results 通道。、
没有任务提交后，关闭任务jobs通道.
等待所有的任务完成后关闭results通道
wg.Done() 减少等待组的计数器，表示有一个任务已经完成。

*/