使用带缓冲channel实现并发限流,通过固定容量channel作为许可证池控制最大并发数;2. 采用golang.org/x/sync/semaphore的WeightedSemaphore支持超时和上下文取消,适合复杂场景;3. 基于rate.Limiter控制请求速率,适用于API调用等需限频的场景;4. 根据需求选择合适方法:限并发用channel或信号量,限频率用rate.Limiter,可组合使用。

在Golang中实现并发限流,核心目标是控制同时运行的协程数量,防止系统资源被耗尽。常用方法包括使用带缓冲的channel、semaphore(信号量)或第三方库如golang.org/x/sync/semaphore。下面介绍几种实用且清晰的实现方式。
使用Buffered Channel控制并发数
通过一个固定容量的channel来充当“许可证池”,每个goroutine执行前需先获取一个许可,执行完后归还。这种方式简单直观,适合大多数场景。
示例代码:
func main() { maxConcurrency := 3 sem := make(chan struct{}, maxConcurrency)for i := 0; i < 10; i++ { sem <- struct{}{} // 获取许可 go func(id int) { defer func() { <-sem }() // 释放许可 fmt.Printf("处理任务 %dn", id) time.Sleep(2 * time.Second) // 模拟工作 fmt.Printf("完成任务 %dn", id) }(i)}// 等待所有任务完成(这里可用WaitGroup更精确)time.Sleep(10 * time.Second)
}
立即学习“go语言免费学习笔记(深入)”;
这种方式利用channel的阻塞特性自动实现限流,当channel满时,新的goroutine会等待,直到有空位。
使用golang.org/x/sync/semaphore
标准库没有原生信号量,但golang.org/x/sync包提供了WeightedSemaphore,支持异步获取,还能处理超时和上下文取消,更适合复杂控制。
安装:
go get golang.org/x/sync/semaphore
使用示例:
sem := semaphore.NewWeighted(3) // 最多3个并发for i := 0; i < 10; i++ {if err := sem.Acquire(context.Background(), 1); err != nil {log.Printf("获取信号量失败: %v", err)continue}
go func(id int) { defer sem.Release(1) fmt.Printf("处理任务 %dn", id) time.Sleep(2 * time.Second) fmt.Printf("完成任务 %dn", id)}(i)
}
立即学习“go语言免费学习笔记(深入)”;
优势在于可结合context实现超时控制,例如:
ctx, cancel := context.WithTimeout(context.Background(), 500*time.Millisecond)
基于Ticker的速率限流(RPS控制)
如果关注的是请求频率(如每秒最多10次),可以用time.Ticker或golang.org/x/time/rate实现。
使用rate.Limiter示例:
import "golang.org/x/time/rate"limiter := rate.NewLimiter(10, 1) // 每秒10个,突发1
for i := 0; i < 100; i++ {limiter.Wait(context.Background())go func(id int) {fmt.Printf("执行任务 %dn", id)}(i)}
注意:这种方式控制的是请求发起速率,不直接限制并发数,适合API调用等场景。
选择哪种方式取决于需求:控制最大并发数用channel或WeightedSemaphore;控制请求频率用rate.Limiter。实际项目中常组合使用,比如先限频再限并发。
基本上就这些,关键点是理解不同工具的适用场景,避免过度设计。
以上就是如何在Golang中实现并发限流的详细内容,更多请关注创想鸟其它相关文章!
版权声明:本文内容由互联网用户自发贡献,该文观点仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。
如发现本站有涉嫌抄袭侵权/违法违规的内容, 请发送邮件至 chuangxiangniao@163.com 举报,一经查实,本站将立刻删除。
发布者:程序猿,转转请注明出处:https://www.chuangxiangniao.com/p/1413928.html
微信扫一扫
支付宝扫一扫