Golang协程池怎么实现与应用

   2025-02-04 5810
核心提示:Golang中的协程池可以通过使用goroutine和channel实现。下面是一个简单的示例,展示了如何使用协程池来处理任务:package mainim

Golang中的协程池可以通过使用goroutinechannel实现。下面是一个简单的示例,展示了如何使用协程池来处理任务:

package mainimport ("fmt""sync")type Worker struct {ID     intTask   chan intDone   chan boolWorkerPool chan chan int}func NewWorker(id int, workerPool chan chan int) *Worker {return &Worker{ID:         id,Task:       make(chan int),Done:       make(chan bool),WorkerPool: workerPool,}}func (w *Worker) Start() {go func() {for {// 把自己的任务通道注册到工作池w.WorkerPool <- w.Taskselect {case task := <-w.Task:// 处理任务fmt.Printf("Worker %d processing task %d\n", w.ID, task)case <-w.Done:// 任务完成fmt.Printf("Worker %d stopping\n", w.ID)return}}}()}func (w *Worker) Stop() {go func() {w.Done <- true}()}type Pool struct {WorkerPool  chan chan intTasks       chan intMaxWorkers  intWaitGroup   sync.WaitGroup}func NewPool(maxWorkers, maxTasks int) *Pool {return &Pool{WorkerPool:  make(chan chan int, maxWorkers),Tasks:       make(chan int, maxTasks),MaxWorkers:  maxWorkers,}}func (p *Pool) Start() {// 启动协程池中的工作协程for i := 0; i < p.MaxWorkers; i++ {worker := NewWorker(i, p.WorkerPool)worker.Start()}go p.dispatch()}func (p *Pool) dispatch() {for {select {case task := <-p.Tasks:workerTask := <-p.WorkerPool// 分发任务给空闲的工作协程workerTask <- task}}}func main() {pool := NewPool(3, 10)pool.Start()// 添加任务到任务队列for i := 0; i < 10; i++ {pool.Tasks <- i}pool.WaitGroup.Wait()}

在上面的示例中,我们定义了一个Worker结构体,其中包含了一个任务通道Task和一个完成通道Done。当Worker启动时,它会把自己的任务通道注册到工作池中,并等待任务的到来。在任务到来时,它会从任务通道中接收任务并处理。当任务完成时,它会通过完成通道通知主线程。

Pool结构体包含了一个工作协程池和一个任务通道。在Start方法中,我们创建了maxWorkers个工作协程,并启动它们。同时,我们还启动了一个dispatch协程,该协程用于从任务通道中接收任务,并将其分发给空闲的工作协程。

main函数中,我们创建了一个协程池,并向任务通道中添加了10个任务。最后,我们使用WaitGroup等待所有任务完成。

这就是一个简单的Golang协程池的实现和应用。你可以根据自己的需求进行扩展和修改。

 
 
更多>同类维修知识
推荐图文
推荐维修知识
点击排行
网站首页  |  关于我们  |  联系方式  |  用户协议  |  隐私政策  |  网站留言