golang_Pool: go实现协程池

最新推荐文章于 2023-06-24 22:16:10 发布

Grayan

最新推荐文章于 2023-06-24 22:16:10 发布

阅读量1k

点赞数

分类专栏： golang 文章标签： pool 资源池协程池 golang

本文链接：https://blog.csdn.net/weixin_43851310/article/details/89428938

版权

本文介绍了如何在 Go 语言中初始化并使用协程池。通过维护一个First In Last Out (FILO) 队列来管理Worker，当有任务请求时，会检查队列中是否有空闲的Worker执行任务。如果没有空闲Worker且未超过Pool容量，则创建新的Worker处理任务，否则阻塞等待。每个Worker完成任务后返回Pool等待新的任务。

摘要由CSDN通过智能技术生成

初始化一个 Goroutine Pool 池，这个Pool维护了一个类似栈的FILO队列，里面存放负责处理任务的Worker，然后每有一个请求就开启一个协程。
之后

检查当前Worker队列中是否有空闲的Worker，如果有，取出执行当前的task；
没有空闲Worker，判断当前在运行的Worker是否已超过该Pool的容量，是 — 阻塞等待直至有Worker被放回Pool；否 — 新开一个Worker（goroutine）处理；
每个Worker执行完任务之后，放回Pool的队列中等待。

参考资料：《Go语言实战》7.2章 Pool

package main

import (
	"io"
	"log"
	"sync"
	"errors"
)

// Pool管理一组可以安全在多个goroutine间共享的资源
// 被管理的资源必须实现io.Closer接口
type Pool struct {
   
	m         sync.Mutex
	resources chan io.Closer
	factory   func() (io.Closer, error)
	closed    bool
}

// ErrPoolClosed表示请求了一个已经关闭的池
var ErrPoolClosed = errors.New("Pool has been closed.")

//New创建一个用来管理资源的池
func New(fn func