Golanggoroutine池与worker模式应用示例-Golang-PHP中文网

Golanggoroutine池与worker模式应用示例

P粉602998670

发布： 2025-09-16 13:38:01

原创

614人浏览过

答案：Go中通过goroutine池结合worker模式控制并发，示例创建固定worker从任务channel获取并处理任务，使用WaitGroup等待完成，可限制并发数、减少开销、提升稳定性，适用于批量请求、文件处理等场景，并可扩展支持动态调整worker数、优先级、超时和错误处理。

golanggoroutine池与worker模式应用示例

在Go语言中，goroutine虽然轻量，但如果无限制地创建，仍可能导致资源耗尽或调度开销增加。为高效管理并发任务，常采用goroutine池结合worker模式，控制并发数量，复用执行单元。下面通过一个简单示例说明其基本实现方式。

1. 模式设计思路

核心思想是预先启动固定数量的worker（即goroutine），这些worker从一个共享的任务队列中不断获取任务并执行。任务通过channel提交，避免动态创建大量goroutine。

使用task channel传递待处理任务
启动固定数量的worker监听该channel
主程序将任务发送到channel，由空闲worker接收处理
通过WaitGroup等待所有任务完成

2. 基础代码实现

以下是一个简单的worker池示例，模拟处理整数任务（如计算平方）：

package main
<p>import (
"fmt"
"sync"
"time"
)</p><p>// Task 表示一个待处理的任务
type Task struct {
ID   int
Data int
}</p><p>// WorkerPool 管理worker和任务分发
type WorkerPool struct {
tasks   chan Task
workers int
wg      sync.WaitGroup
}</p><p>// NewWorkerPool 创建一个新的worker池
func NewWorkerPool(workers, queueSize int) *WorkerPool {
return &WorkerPool{
tasks:   make(chan Task, queueSize),
workers: workers,
}
}</p><p>// Start 启动所有worker
func (wp <em>WorkerPool) Start() {
for i := 0; i < wp.workers; i++ {
wp.wg.Add(1)
go func(workerID int) {
defer wp.wg.Done()
for task := range wp.tasks {
fmt.Printf("Worker %d processing task %d with data %d\n", workerID, task.ID, task.Data)
time.Sleep(time.Millisecond </em> 100) // 模拟处理时间
}
}(i)
}
}</p><p>// Submit 提交任务到池
func (wp *WorkerPool) Submit(task Task) {
wp.tasks <- task
}</p><p>// Stop 关闭任务通道，等待所有worker结束
func (wp *WorkerPool) Stop() {
close(wp.tasks)
wp.wg.Wait()
}</p><p>func main() {
pool := NewWorkerPool(3, 10) // 3个worker，队列最多10个任务
pool.Start()</p><pre class='brush:php;toolbar:false;'>// 提交一批任务
for i := 1; i <= 5; i++ {
    pool.Submit(Task{ID: i, Data: i * 10})
}

pool.Stop()
fmt.Println("All tasks completed.")

登录后复制

}

立即学习“go语言免费学习笔记（深入）”；