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

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

P粉602998670

发布： 2025-08-20 13:18:02

原创

632人浏览过

使用带缓冲通道控制并发数，通过信号量机制限制goroutine数量，实现轻量级协程池，适用于需控制并发的任务场景。

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

Go语言中的协程（goroutine）轻量且高效，但无限制地创建协程可能导致资源耗尽。为控制并发数量，常使用协程池配合带缓冲的通道来实现任务调度。这种方案简单、稳定，适合处理大量短期任务。

基本思路：使用缓冲通道控制并发数

协程池的核心是限制同时运行的goroutine数量。通过一个带缓冲的通道作为信号量，每启动一个任务前先向通道写入一个值，任务结束再读出，从而实现并发控制。

这种方式不维护复杂的池结构，而是用通道天然的同步机制管理并发，代码简洁且线程安全。

实现步骤

以下是基于带缓冲通道的协程池实现方法：

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

定义任务类型：使用函数类型表示任务，如
```
type Task func()
```
登录后复制
创建信号量通道：容量为最大并发数，如
```
sem := make(chan struct{}, 5)
```
登录后复制
表示最多5个并发任务
提交任务时获取信号量：在执行任务前尝试向通道写入，若通道满则阻塞等待
任务执行完成后释放信号量：在goroutine末尾从通道读取，释放并发槽位

示例代码：

抖云猫AI论文助手

一款AI论文写作工具，最快 2 分钟，生成 3.5 万字论文。论文可插入表格、代码、公式、图表，依托自研学术抖云猫大模型，生成论文具备严谨的学术专业性。

146

查看详情

package main
<p>import (
"fmt"
"sync"
"time"
)</p><p>type Task func()</p><p>type Pool struct {
sem  chan struct{}
wg   sync.WaitGroup
}</p><p>func NewPool(maxConcurrency int) *Pool {
return &Pool{
sem: make(chan struct{}, maxConcurrency),
}
}</p><p>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;"><pre class="brush:php;toolbar:false;">    task() // 执行任务
}()

登录后复制

}

func (p *Pool) Wait() { p.wg.Wait() }

func main() { pool := NewPool(3) // 最大3个并发

for i := 0; i < 10; i++ {
    i := i
    pool.Submit(func() {
        fmt.Printf("执行任务 %d\n", i)
        time.Sleep(1 * time.Second)
    })
}

pool.Wait()
fmt.Println("所有任务完成")

登录后复制

}