面：go能不能手写一个简单的协程池？

bugplus-free

于 2024-04-22 00:11:36 发布

阅读量575

点赞数 2

分类专栏：面经文章标签： golang 开发语言后端

本文链接：https://blog.csdn.net/m0_61682542/article/details/138050542

版权

面经专栏收录该内容

5 篇文章

订阅专栏

本文介绍了如何通过chan管道和sync.WaitGroup构建一个无缓冲的协程池，包括新Pool的创建、submit任务和shutdown停止机制。作者详细展示了如何定义Job类型、Pool结构以及关键函数的实现过程。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

最简单的想法是使用chan管道和wg也就是sync.WaitGroup这两个搭档来进行实现协程池

那么在chan管道中最简单的就是无缓冲的了

对于协程池有啥操作？

第一个要想到newPool

并且要明白，这个协程池是不是你叫他停他才能停，如果没说，就让他一直死循环做他应该做的事情

对于这个东西，我们肯定要提交任务给他这是一个submit功能函数

之后是不是还要有一个shutdown函数告诉waitgroup要停止了，结束

有这些之后，那就开始书写

package main

type Job fun()//抽象一点点，就说他是一个函数

type Pool struct{//抽象出来一个池子
    jobChan chan Job
    wg sync.WaitGroup
}
func NewPool(size int)* Pool{
      p:=&Pool{
        jobChan: make(chan Job)
        }
      P.wg.Add(size)
      for i:=0;i<size;i++{
      go func(){
        defer p.wg.Done()
        for job:=range p.jobChan{//管道只要没关闭，你一直读的话，没东西就阻塞，一直等待
            job()
        }
    }()
  }  
}

func (p *Pool)submit(job Job){
    p.jobChan<-job
}
func (p *Pool)shutdown(){
    close(p.jobChan)
    p.wg.Wait()
}