使用goroutine池可控制并发数量,避免内存暴涨和调度开销。通过第三方库ants或手动实现基于channel的worker池,能有效管理任务执行,适用于大量短时任务或受限外部服务调用,提升系统稳定性与性能。
在Go语言中,goroutine虽然轻量,但如果无限制地创建,仍可能导致内存暴涨或调度开销增大。使用goroutine池可以有效控制并发数量,复用执行单元,提升系统稳定性与性能。下面介绍几种在Golang中实现和使用goroutine池的常见方式。
Go的goroutine创建成本低,但不代表可以无限创建。大量goroutine会带来以下问题:
goroutine池通过预设最大并发数,从队列中消费任务,避免上述问题。
ants 是目前最流行的Go goroutine池库,功能完整,性能优秀。
安装:
go get github.com/panjf2000/ants/v2
基本使用示例:
package mainimport ( "fmt" "sync" "time" "gith
ub.com/panjf2000/ants/v2" )
func worker(task int) { fmt.Printf("处理任务: %d, 协程ID: %d\n", task, ants.GetWorkerPoolSize()) time.Sleep(time.Second) }
func main() { // 创建一个容量为10的协程池 pool, _ := ants.NewPool(10) defer pool.Release()
var wg sync.WaitGroup for i := 0; i < 100; i++ { wg.Add(1) _ = pool.Submit(func() { defer wg.Done() worker(i) }) } wg.Wait()}
特点:
如果不想引入第三方依赖,可以基于channel和worker模型手写一个简单池。
package mainimport ( "fmt" "sync" "time" )
type Task func()
type Pool struct { workers int tasks chan Task wg sync.WaitGroup }
func NewPool(workers, queueSize int) *Pool { return &Pool{ workers: workers, tasks: make(chan Task, queueSize), } }
func (p *Pool) Start() { for i := 0; i < p.workers; i++ { p.wg.Add(1) go func() { defer p.wg.Done() for task := range p.tasks { if task != nil { task() } } }() } }
func (p *Pool) Submit(task Task) { p.tasks <- task }
func (p *Pool) Close() { close(p.tasks) p.wg.Wait() }
// 使用示例 func main() { pool := NewPool(5, 100) pool.Start()
for i := 0; i < 30; i++ { taskID := i pool.Submit(func() { fmt.Printf("执行任务 %d, 时间: %v\n", taskID, time.Now().Format("15:04:05")) time.Sleep(time.Millisecond * 300) }) } pool.Close() fmt.Println("所有任务完成")}
核心逻辑:
goroutine池适用于以下场景:
不建议使用的情况:
基本上就这些。选择ants可以快速落地,手动实现则更灵活可控。关键是根据业务需求设定合理的池大小和队列长度,避免阻塞或资源浪费。