0

0

如何在 Golang 中控制 Goroutine 数量_Golang 并发任务限流与资源控制策略

P粉602998670

P粉602998670

发布时间:2025-11-16 18:01:49

|

1002人浏览过

|

来源于php中文网

原创

使用带缓冲的channel可限制Goroutine数量,通过容量为N的struct{}类型channel作为信号量,每启动一个goroutine写入值,结束时读出,确保最多N个并发执行。

如何在 golang 中控制 goroutine 数量_golang 并发任务限流与资源控制策略

在 Golang 中,Goroutine 虽然轻量,但如果无限制地创建,可能导致系统资源耗尽、调度开销增加甚至程序崩溃。因此,控制 Goroutine 数量是编写稳定并发程序的关键。常见的做法是通过限流机制来管理并发任务的数量,确保系统在可控负载下运行。

使用带缓冲的 Channel 实现最大并发控制

最常见且简单的方式是使用带缓冲的 channel 作为信号量,控制同时运行的 Goroutine 数量。

定义一个容量为 N 的 channel,在每个 Goroutine 启动前写入一个值,执行完后读出,从而保证最多只有 N 个 Goroutine 并发运行。

func worker(id int, task string, sem chan struct{}) {
    defer func() { <-sem }() // 任务完成释放信号
    fmt.Printf("Worker %d 执行任务: %s\n", id, task)
    time.Sleep(time.Second) // 模拟处理时间
}

func main() { tasks := []string{"task1", "task2", "task3", "task4", "task5"} maxConcurrency := 3 sem := make(chan struct{}, maxConcurrency)

for i, task := range tasks {
    sem <- struct{}{} // 占用一个槽位,超了会阻塞
    go worker(i+1, task, sem)
}

// 等待所有任务完成(这里简化处理,实际可用 WaitGroup)
time.Sleep(6 * time.Second)

}

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

这种方式简洁有效,适合大多数场景下的并发数限制。

结合 WaitGroup 精确控制生命周期

上面的例子用了 Sleep 来等待,不精确。更规范的做法是配合 sync.WaitGroup 使用,确保主协程能准确等待所有任务结束。

func main() {
    tasks := []string{"task1", "task2", "task3", "task4", "task5"}
    maxConcurrency := 2
    sem := make(chan struct{}, maxConcurrency)
    var wg sync.WaitGroup
for i, task := range tasks {
    wg.Add(1)
    go func(id int, t string) {
        defer wg.Done()
        sem <- struct{}{}
        defer func() { <-sem }()

        fmt.Printf("Worker %d 处理: %s\n", id, t)
        time.Sleep(time.Second)
    }(i+1, task)
}

wg.Wait() // 等待全部完成
fmt.Println("所有任务结束")

}

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

WaitGroup 确保了主流程不会提前退出,channel 控制了并发度,二者结合是生产级常用模式。

羚珑
羚珑

京东推出的一站式AI图像处理平台

下载

使用第三方库进行高级限流(如 semaphore 或 runner)

对于更复杂的场景,比如需要超时控制、优先级或动态调整并发数,可以使用 golang.org/x/sync/semaphore

它提供了加权信号量,支持上下文超时,更适合精细控制。

import "golang.org/x/sync/semaphore"

func main() { tasks := []string{"t1", "t2", "t3", "t4", "t5"} sem := semaphore.NewWeighted(2) // 最大并发 2 var wg sync.WaitGroup

for i, task := range tasks {
    wg.Add(1)
    go func(id int, t string) {
        defer wg.Done()
        sem.Acquire(context.Background(), 1) // 获取许可
        defer sem.Release(1)                // 释放

        fmt.Printf("Worker %d 执行: %s\n", id, t)
        time.Sleep(time.Second)
    }(i+1, task)
}

wg.Wait()

}

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

semaphore 支持上下文,可设置 Acquire 超时,适合网络请求等可能卡住的场景。

避免 Goroutine 泄露的关键点

无论用哪种方式,都要注意防止 Goroutine 泄露:

  • 确保每个启动的 Goroutine 都能正常退出,尤其是在错误路径上也要释放信号量。
  • 避免因 channel 阻塞导致 Goroutine 无法结束。
  • 对可能 panic 的逻辑使用 defer 恢复,并释放资源。

例如,在 worker 中加入 recover:

defer func() {
    if r := recover(); r != nil {
        fmt.Printf("Goroutine panic: %v\n", r)
    }
    <-sem
}()

基本上就这些。通过 channel 信号量或 x/sync/semaphore,配合 WaitGroup 和 defer,就能安全有效地控制 Goroutine 数量,实现稳定的并发任务调度。关键是不让并发失控,也不让主流程提前退出。不复杂但容易忽略细节。

相关专题

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

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

173

2024.02.23

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

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

224

2024.02.23

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

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

334

2024.02.23

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

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

204

2024.03.05

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

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

387

2024.05.21

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

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

193

2025.06.09

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

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

184

2025.06.10

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

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

191

2025.06.17

苹果官网入口直接访问
苹果官网入口直接访问

苹果官网直接访问入口是https://www.apple.com/cn/,该页面具备0.8秒首屏渲染、HTTP/3与Brotli加速、WebP+AVIF双格式图片、免登录浏览全参数等特性。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

10

2025.12.24

热门下载

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

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
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-2025 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号