0

0

Golang如何在多goroutine中实现限流器

P粉602998670

P粉602998670

发布时间:2025-10-25 16:22:02

|

260人浏览过

|

来源于php中文网

原创

使用channel、rate包或自定义计数器实现Go限流。1. buffered channel控制并发数,通过信号量限制goroutine数量;2. golang.org/x/time/rate基于令牌桶限制请求速率;3. sync.Mutex配合计数器实现简单QPS控制,适用于轻量场景但需注意锁竞争。

golang如何在多goroutine中实现限流器

在Go语言中,多goroutine环境下实现限流器的常见方式是使用 channel标准库中的 sync.RWMutex 配合计数器,也可以借助第三方库如 golang.org/x/time/rate。下面介绍几种实用且高效的限流实现方法。

使用 buffered channel 实现并发数限制

这是最简单直观的方式:通过一个有缓冲的channel来控制同时运行的goroutine数量。

示例代码:

package main

import ( "fmt" "time" )

func worker(id int, sem chan struct{}) { defer func() { <-sem }() // 任务结束释放信号 fmt.Printf("Worker %d starting\n", id) time.Sleep(1 * time.Second) // 模拟工作 fmt.Printf("Worker %d done\n", id) }

func main() { const maxConcurrency = 3 sem := make(chan struct{}, maxConcurrency)

for i := 1; i <= 10; i++ {
    sem <- struct{}{} // 获取令牌
    go worker(i, sem)
}

// 等待所有任务完成(简化处理)
time.Sleep(5 * time.Second)

}

这个方法限制了最多只有3个goroutine同时运行。每当启动一个goroutine前先向channel写入数据,worker结束后从channel读出,从而实现“信号量”机制。

使用 golang.org/x/time/rate 实现速率限流

该包提供了基于令牌桶算法的限流器,适合控制请求频率,比如每秒最多N次调用。

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

示例代码:

package main

import ( "fmt" "sync" "time" "golang.org/x/time/rate" )

func main() { limiter := rate.NewLimiter(2, 5) // 每秒2个令牌,最多积压5个 var wg sync.WaitGroup

for i := 0; i < 10; i++ {
    wg.Add(1)
    go func(id int) {
        defer wg.Done()
        if err := limiter.Wait(nil); err != nil {
            fmt.Printf("Request %d failed: %v\n", id, err)
            return
        }
        fmt.Printf("Request %d processed at %v\n", id, time.Now())
    }(i)
    time.Sleep(100 * time.Millisecond) // 模拟请求到来
}

wg.Wait()

}

这种方式适用于接口级或API调用的速率控制,能平滑地限制单位时间内的请求数量。

TextIn Tools
TextIn Tools

是一款免费在线OCR工具,包含文字识别、表格识别,PDF转文件,文件转PDF、其他格式转换,识别率高,体验好,免费。

下载

自定义计数器 + Mutex 实现简单QPS控制

如果不想引入外部依赖,可以自己实现一个简单的每秒请求数(QPS)限制器。

package main

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

type RateLimiter struct { mu sync.Mutex count int limit int interval time.Duration lastReset time.Time }

func NewRateLimiter(qps int, interval time.Duration) *RateLimiter { return &RateLimiter{ limit: qps, interval: interval, lastReset: time.Now(), } }

func (r *RateLimiter) Allow() bool { r.mu.Lock() defer r.mu.Unlock()

now := time.Now()
if now.Sub(r.lastReset) > r.interval {
    r.count = 0
    r.lastReset = now
}

if r.count < r.limit {
    r.count++
    return true
}
return false

}

func main() { limiter := NewRateLimiter(3, time.Second) // 每秒最多3次 var wg sync.WaitGroup

for i := 0; i < 10; i++ {
    wg.Add(1)
    go func(id int) {
        defer wg.Done()
        for !limiter.Allow() {
            time.Sleep(10 * time.Millisecond)
        }
        fmt.Printf("Processed request %d at %v\n", id, time.Now())
    }(i)
    time.Sleep(200 * time.Millisecond)
}

wg.Wait()

}

这种方法适合轻量级场景,但要注意锁竞争在高并发下可能成为瓶颈。

基本上就这些。根据实际需求选择合适的方式:channel适合控制并发数,rate.Limiter适合精确控制速率,自定义方案则灵活但需注意性能和正确性。不复杂但容易忽略的是资源释放和时钟漂移问题,尤其是长时间运行的服务。

相关专题

更多
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、图像处理库。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

335

2024.02.23

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

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

206

2024.03.05

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

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

388

2024.05.21

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

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

193

2025.06.09

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

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

187

2025.06.10

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

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

191

2025.06.17

桌面文件位置介绍
桌面文件位置介绍

本专题整合了桌面文件相关教程,阅读专题下面的文章了解更多内容。

0

2025.12.30

热门下载

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

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
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号