0

0

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

P粉602998670

P粉602998670

发布时间:2025-08-28 14:11:01

|

882人浏览过

|

来源于php中文网

原创

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

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

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

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

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

优点:

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

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

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

立即学习go语言免费学习笔记(深入)”;

示例代码:

package main

import ( "fmt" "sync" "time" )

type Task func()

type Pool struct { capacity int // 最大并发数 sem chan struct{} // 信号量通道 wg sync.WaitGroup }

// NewPool 创建协程池,指定最大并发数 func NewPool(capacity int) *Pool { return &Pool{ capacity: capacity, sem: make(chan struct{}, capacity), } }

// Submit 提交任务 func (p *Pool) Submit(task Task) { p.wg.Add(1) go func() { defer p.wg.Done() p.sem <- struct{}{} // 获取令牌 defer func() { <-p.sem }() // 释放令牌

    task() // 执行任务
}()

}

Bing图像创建器
Bing图像创建器

必应出品基于DALL·E的AI绘图工具

下载

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

使用示例

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

func main() {
    pool := NewPool(3) // 最多3个并发
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个任务并发执行,符合预期。

适用场景与注意事项

该方案适用于任务数量较多但希望限制并发数的场景,如爬虫、批量IO操作、并发请求处理等。

注意事项:

  • 缓冲通道的大小决定了最大并发数,应根据系统负载合理设置
  • 任务函数内部应避免panic导致协程退出,建议在task外层加recover
  • Submit是非阻塞的,任务提交后立即返回,适合异步处理
  • 如需获取任务返回值,可扩展Task类型为 func() interface{} 并配合channel返回结果

基本上就这些。这种带缓冲通道的协程池实现简单、高效,适合大多数并发控制需求,是Go中推荐的轻量级协程管理方式。

相关专题

更多
golang如何定义变量
golang如何定义变量

golang定义变量的方法:1、声明变量并赋予初始值“var age int =值”;2、声明变量但不赋初始值“var age int”;3、使用短变量声明“age :=值”等等。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

178

2024.02.23

golang有哪些数据转换方法
golang有哪些数据转换方法

golang数据转换方法:1、类型转换操作符;2、类型断言;3、字符串和数字之间的转换;4、JSON序列化和反序列化;5、使用标准库进行数据转换;6、使用第三方库进行数据转换;7、自定义数据转换函数。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

226

2024.02.23

golang常用库有哪些
golang常用库有哪些

golang常用库有:1、标准库;2、字符串处理库;3、网络库;4、加密库;5、压缩库;6、xml和json解析库;7、日期和时间库;8、数据库操作库;9、文件操作库;10、图像处理库。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

339

2024.02.23

golang和python的区别是什么
golang和python的区别是什么

golang和python的区别是:1、golang是一种编译型语言,而python是一种解释型语言;2、golang天生支持并发编程,而python对并发与并行的支持相对较弱等等。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

209

2024.03.05

golang是免费的吗
golang是免费的吗

golang是免费的。golang是google开发的一种静态强类型、编译型、并发型,并具有垃圾回收功能的开源编程语言,采用bsd开源协议。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

391

2024.05.21

golang结构体相关大全
golang结构体相关大全

本专题整合了golang结构体相关大全,想了解更多内容,请阅读专题下面的文章。

196

2025.06.09

golang相关判断方法
golang相关判断方法

本专题整合了golang相关判断方法,想了解更详细的相关内容,请阅读下面的文章。

191

2025.06.10

golang数组使用方法
golang数组使用方法

本专题整合了golang数组用法,想了解更多的相关内容,请阅读专题下面的文章。

192

2025.06.17

高德地图升级方法汇总
高德地图升级方法汇总

本专题整合了高德地图升级相关教程,阅读专题下面的文章了解更多详细内容。

43

2026.01.16

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
golang socket 编程
golang socket 编程

共2课时 | 0.1万人学习

nginx浅谈
nginx浅谈

共15课时 | 0.8万人学习

golang和swoole核心底层分析
golang和swoole核心底层分析

共3课时 | 0.1万人学习

关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号 技术交流群
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号