如何在Golang中实现并发限流-Golang-PHP中文网

如何在Golang中实现并发限流

P粉602998670

发布： 2025-10-14 17:12:02

原创

574人浏览过

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

如何在golang中实现并发限流

在Golang中实现并发限流，核心目标是控制同时运行的协程数量，防止系统资源被耗尽。常用方法包括使用带缓冲的channel、semaphore（信号量）或第三方库如golang.org/x/sync/semaphore。下面介绍几种实用且清晰的实现方式。

使用Buffered Channel控制并发数

通过一个固定容量的channel来充当“许可证池”，每个goroutine执行前需先获取一个许可，执行完后归还。这种方式简单直观，适合大多数场景。

示例代码：

func main() {
    maxConcurrency := 3
    sem := make(chan struct{}, maxConcurrency)
<pre class='brush:php;toolbar:false;'>for i := 0; i < 10; i++ {
    sem <- struct{}{} // 获取许可
    go func(id int) {
        defer func() { <-sem }() // 释放许可

        fmt.Printf("处理任务 %d\n", id)
        time.Sleep(2 * time.Second) // 模拟工作
        fmt.Printf("完成任务 %d\n", id)
    }(i)
}

// 等待所有任务完成（这里可用WaitGroup更精确）
time.Sleep(10 * time.Second)

登录后复制

}

如知AI笔记

如知笔记——支持markdown的在线笔记，支持ai智能写作、AI搜索，支持DeepseekR1满血大模型

查看详情

立即学习“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个并发
<p>for i := 0; i < 10; i++ {
if err := sem.Acquire(context.Background(), 1); err != nil {
log.Printf("获取信号量失败: %v", err)
continue
}</p><pre class='brush:php;toolbar:false;'>go func(id int) {
    defer sem.Release(1)

    fmt.Printf("处理任务 %d\n", id)
    time.Sleep(2 * time.Second)
    fmt.Printf("完成任务 %d\n", 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"
<p>limiter := rate.NewLimiter(10, 1) // 每秒10个，突发1</p><p>for i := 0; i < 100; i++ {
limiter.Wait(context.Background())
go func(id int) {
fmt.Printf("执行任务 %d\n", id)
}(i)
}</p>

登录后复制

注意：这种方式控制的是请求发起速率，不直接限制并发数，适合API调用等场景。

选择哪种方式取决于需求：控制最大并发数用channel或WeightedSemaphore；控制请求频率用rate.Limiter。实际项目中常组合使用，比如先限频再限并发。

基本上就这些，关键点是理解不同工具的适用场景，避免过度设计。

以上就是如何在Golang中实现并发限流的详细内容，更多请关注php中文网其它相关文章！