本文是阅读GoFrame内grpool包的理解

大致理解

// 这是个Goroutine的 pool type Pool struct { limit int //池子中最大数量 count *gtype.Int // 当前运行的任务计数器，并发安全，郭大nb list *glist.List //异步工作的队列 closed *gtype.Bool // 是否关闭的状态 }

两种工作方式，一种是链式操作，另一种非链式操作，这里只讲链式操作

第一步初始化一个池子，可以设置池子的容量，默认不限制

//初始化并返回池子的指针 func New(limit ...int) *Pool { //默认 p := &Pool{ limit: -1, count: gtype.NewInt(), list: glist.New(true), closed: gtype.NewBool(), } //有传入池子容量，赋值 if len(limit) > 0 && limit[0] > 0 { p.limit = limit[0] } return p }

添加任务到池中

//简单粗暴，直接传入一个闭包方法 func (p *Pool) Add(f func()) error { //判断池子是否关闭 for p.closed.Val() { return errors.New("pool closed") } //添加任务到队列中 p.list.PushFront(f) // 判断是否需要开启新的Goroutine var n int for { //获取池子运行的任务数量 n = p.count.Val() //判断当前运行的协程数量达到池子的最大容量，达到最大数量直接退出 if p.limit != -1 && n >= p.limit { // No need fork new goroutine. return nil } //未到达，跳出循环，进行fork if p.count.Cas(n, n+1) { // Use CAS to guarantee atomicity. break } } p.fork() return nil }

fork

//开启一个协程，完成任务的协程会被go自动回收 func (p *Pool) fork() { go func() { //退出前减少运行协程池子容量 defer p.count.Add(-1) var job interface{} //伪死循环，当清空队列中的所有任务后会退出循环以及协程 for !p.closed.Val() { //抛出队列中的随机一项任务，并执行，如果抛出为空，退出当前循环，以及协程 if job = p.list.PopBack(); job != nil { job.(func())() } else { return } } }() }

好了，一个协程池的最主要内容就讲完了，其他的我就不讲了