Go语言中通过goroutine池控制并发,避免资源浪费。核心组件包括Worker、任务队列和池大小。任务定义为函数类型Task,通过通道分发,Start方法启动指定数量worker监听任务通道并执行,实现高效并发管理。

在Go语言中,goroutine虽然轻量,但无限制地创建大量goroutine可能导致资源浪费甚至系统崩溃。使用goroutine池可以有效控制并发数量,提升程序稳定性与性能。下面介绍一个简单的goroutine池实现方式及任务分配示例。
goroutine池的基本结构
一个典型的goroutine池包含固定数量的工作协程和一个任务队列。工作协程从队列中不断获取任务并执行,避免频繁创建和销毁goroutine。
核心组件包括:
- Worker:实际执行任务的goroutine
- Task Queue:存放待处理任务的通道
- Pool Size:控制并发执行的任务数
任务函数定义与分配
每个任务通常封装为一个函数类型,通过通道发送给worker处理。以下是一个简单任务类型的定义:
立即学习“go语言免费学习笔记(深入)”;
type Task func()type Pool struct { tasks chan Task workers int }
初始化时启动指定数量的worker,它们监听同一个任务通道:
func (p *Pool) Start() {
for i := 0; i < p.workers; i++ {
go func() {
for task := range p.tasks {
task()
}
}()
}
}
提交任务与关闭池
通过Submit方法向池中添加任务:
func (p *Pool) Submit(task Task) {
p.tasks <- task
}
当所有任务提交完毕,可关闭任务通道以通知worker退出(需确保不再提交新任务):
func (p *Pool) Stop() {
close(p.tasks)
}
完整使用示例
下面是一个完整的例子,创建一个10个worker的池,提交100个打印任务:
func main() {
pool := &Pool{
tasks: make(chan Task, 100),
workers: 10,
}
pool.Start()
// 提交任务
for i := 0; i < 100; i++ {
i := i
pool.Submit(func() {
fmt.Printf("执行任务 %d\n", i)
time.Sleep(time.Second) // 模拟耗时
})
}
// 停止池
pool.Stop()
// 注意:这里可能需要等待所有任务完成,可用sync.WaitGroup进一步控制}
基本上就这些。这个模型简单高效,适合大多数并发任务场景。如果需要更复杂的功能(如超时、优先级),可在基础上扩展任务结构和调度逻辑。关键是控制并发度,避免系统过载。










