使用goroutine池可控制并发数量,避免内存暴涨和调度开销。通过第三方库ants或手动实现基于channel的worker池,能有效管理任务执行,适用于大量短时任务或受限外部服务调用,提升系统稳定性与性能。

在Go语言中,goroutine虽然轻量,但如果无限制地创建,仍可能导致内存暴涨或调度开销增大。使用goroutine池可以有效控制并发数量,复用执行单元,提升系统稳定性与性能。下面介绍几种在Golang中实现和使用goroutine池的常见方式。
Go的goroutine创建成本低,但不代表可以无限创建。大量goroutine会带来以下问题:
goroutine池通过预设最大并发数,从队列中消费任务,避免上述问题。
ants 是目前最流行的Go goroutine池库,功能完整,性能优秀。
立即学习“go语言免费学习笔记(深入)”;
安装:
go get github.com/panjf2000/ants/v2
基本使用示例:
package main
<p>import (
"fmt"
"sync"
"time"
"github.com/panjf2000/ants/v2"
)</p><p>func worker(task int) {
fmt.Printf("处理任务: %d, 协程ID: %d\n", task, ants.GetWorkerPoolSize())
time.Sleep(time.Second)
}</p><p>func main() {
// 创建一个容量为10的协程池
pool, _ := ants.NewPool(10)
defer pool.Release()</p><pre class="brush:php;toolbar:false;"><pre class="brush:php;toolbar:false;">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 main
<p>import (
"fmt"
"sync"
"time"
)</p><p>type Task func()</p><p>type Pool struct {
workers int
tasks chan Task
wg sync.WaitGroup
}</p><p>func NewPool(workers, queueSize int) *Pool {
return &Pool{
workers: workers,
tasks: make(chan Task, queueSize),
}
}</p><p>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()
}
}
}()
}
}</p><p>func (p *Pool) Submit(task Task) {
p.tasks <- task
}</p><p>func (p *Pool) Close() {
close(p.tasks)
p.wg.Wait()
}</p><p>// 使用示例
func main() {
pool := NewPool(5, 100)
pool.Start()</p><pre class="brush:php;toolbar:false;"><pre class="brush:php;toolbar:false;">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可以快速落地,手动实现则更灵活可控。关键是根据业务需求设定合理的池大小和队列长度,避免阻塞或资源浪费。
以上就是如何在Golang中实现goroutine池_Golang goroutine池使用实践汇总的详细内容,更多请关注php中文网其它相关文章!
每个人都需要一台速度更快、更稳定的 PC。随着时间的推移,垃圾文件、旧注册表数据和不必要的后台进程会占用资源并降低性能。幸运的是,许多工具可以让 Windows 保持平稳运行。
Copyright 2014-2025 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号