Golang协程池构建带缓冲通道方案-Golang-PHP中文网

Golang协程池构建带缓冲通道方案

P粉602998670

发布： 2025-08-28 14:11:01

原创

878人浏览过

该方案通过带缓冲通道控制goroutine并发数，避免资源耗尽。使用信号量模型，以固定容量的channel作为令牌桶，任务执行前获取令牌，完成后释放，实现简单且资源开销小。示例中创建容量为3的协程池，提交10个任务，最多3个并发执行，适用于爬虫、批量IO等场景，需注意合理设置并发数并防止panic导致协程异常退出。

golang协程池构建带缓冲通道方案

在Go语言中，协程（goroutine）轻量且高效，但无限制地创建协程可能导致系统资源耗尽。为控制并发数量、提升性能和资源利用率，通常使用协程池。一种常见且高效的实现方式是带缓冲通道的协程池方案。

核心思路：用带缓冲的通道控制并发数

该方案不预先启动固定数量的goroutine，而是通过一个带缓冲的channel来控制最大并发数。每当启动一个任务时，先向channel写入一个信号，任务完成后再释放该信号。这种方式本质上是信号量模型，简单且易于理解。

优点：

无需维护worker队列
资源开销小，代码简洁
可动态提交任务，无需关闭池后等待

实现方式：使用带缓冲的channel作为并发控制

定义一个缓冲通道作为“令牌桶”，其容量即为最大并发数。每个任务执行前需获取一个令牌，执行完成后归还。

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

示例代码：

package main
<p>import (
"fmt"
"sync"
"time"
)</p><p>type Task func()</p><p>type Pool struct {
capacity int           // 最大并发数
sem      chan struct{} // 信号量通道
wg       sync.WaitGroup
}</p><p>// NewPool 创建协程池，指定最大并发数
func NewPool(capacity int) *Pool {
return &Pool{
capacity: capacity,
sem:      make(chan struct{}, capacity),
}
}</p><p>// Submit 提交任务
func (p *Pool) Submit(task Task) {
p.wg.Add(1)
go func() {
defer p.wg.Done()
p.sem <- struct{}{} // 获取令牌
defer func() { <-p.sem }() // 释放令牌</p><pre class='brush:php;toolbar:false;'>    task() // 执行任务
}()

登录后复制

}

讯飞智作-讯飞配音

讯飞智作是一款集AI配音、虚拟人视频生成、PPT生成视频、虚拟人定制等多功能的AI音视频生产平台。已广泛应用于媒体、教育、短视频等领域。

查看详情

// Close 等待所有任务完成 func (p *Pool) Close() { p.wg.Wait() }

使用示例

下面是一个简单使用示例，提交10个任务，但最多同时运行3个goroutine。

func main() {
    pool := NewPool(3) // 最多3个并发
<pre class='brush:php;toolbar:false;'>for i := 0; i < 10; i++ {
    i := i
    pool.Submit(func() {
        fmt.Printf("执行任务 %d，协程ID: %v\n", i, &i)
        time.Sleep(1 * time.Second) // 模拟耗时操作
    })
}

pool.Close() // 等待所有任务完成
fmt.Println("所有任务已完成")

登录后复制

}

输出会显示每次最多3个任务并发执行，符合预期。