Go中可用带缓冲channel实现令牌桶限流:初始化容量为maxTokens的channel并预填满,请求时select尝试取令牌,成功则处理,失败则拒绝;后台goroutine定期补充令牌。

在 Go 中实现请求限流,用 Channel 和计数器是最轻量、最易理解的方式之一。核心思路是:用一个带缓冲的 channel 当作“令牌桶”,每次请求先尝试获取一个令牌(从 channel 取值),成功则处理请求,失败则拒绝;后台 goroutine 定期往 channel 里放令牌(恢复配额)。配合简单计数器还能做更细粒度的窗口限流(比如每秒最多 N 次)。
用 buffered Channel 实现令牌桶(Token Bucket)
这是最典型的 channel 限流方式,适合控制平均速率和应对突发流量。
初始化一个容量为 maxTokens 的 channel,初始填满每次请求执行 select { case 尝试非阻塞取令牌另起 goroutine 每隔 interval 时间向 channel 放一个令牌(不超过容量)
示例代码片段:
func NewTokenBucket(maxTokens int, interval time.Duration) *TokenBucket { ch := make(chan struct{}, maxTokens) for i := 0; i < maxTokens; i++ { ch <- struct{}{} } tb := &TokenBucket{ch: ch} go func() { ticker := time.NewTicker(interval) defer ticker.Stop() for range ticker.C { select { case ch <- struct{}{}: default: // 已满,不覆盖 } } }() return tb}func (tb *TokenBucket) Allow() bool {select {case <-tb.ch:return truedefault:return false}}
用原子计数器 + 时间窗口实现滑动窗口限流
如果需要严格限制“最近 1 秒内最多 100 次”,单纯 channel 不够精确,这时用 sync/atomic 配合时间戳分片更合适。
立即学习“go语言免费学习笔记(深入)”;
把 1 秒切分成多个 slot(如 10 个 100ms slot),每个 slot 记录该时间段请求数用 atomic.Int64 存储当前总请求数,同时定期清理过期 slot或更简单:只维护当前窗口开始时间 + 计数器,超时就重置(适合低精度场景)
简易版(固定窗口):
type FixedWindowLimiter struct { mu sync.RWMutex count int64 lastReset time.Time limit int64 window time.Duration}func (l *FixedWindowLimiter) Allow() bool {l.mu.Lock()defer l.mu.Unlock()
now := time.Now()if now.Sub(l.lastReset) >= l.window { l.count = 0 l.lastReset = now}if l.count >= l.limit { return false}l.count++return true
}
组合使用:Channel 控制并发数 + 计数器控制频次
真实服务常需双重防护:既不让瞬时并发打爆 CPU(channel 控制 goroutine 数量),也不让单 IP 狂刷接口(计数器按 key 限频)。
用 channel 控制全局最大并发处理数(类似 http.Server 的 MaxConns)对每个 client IP 或 user ID 维护独立的计数器(可用 map + sync.Map + 定时清理)中间件中先检查 IP 频率,再尝试获取全局处理许可
注意点:
避免 map 并发写,用 sync.Map 或加锁计数器要带 TTL,否则内存泄漏;可用 expirable.Map 或自己定时扫描channel 方式天然支持超时:select { case
生产建议:优先用成熟库,而非手写
虽然手写能帮你深入理解,但线上建议用经过压测的库:
uber-go/ratelimit:基于 token bucket,高性能,无锁juju/ratelimit:经典 leaky bucket 实现gin-contrib/limiter:Gin 生态集成方便
它们已处理好时钟漂移、高并发竞争、内存回收等细节,自研容易踩坑。
基本上就这些。channel 适合快速控并发,计数器适合按维度控频次,两者不冲突,可以叠加用。关键不是选哪种,而是清楚你要防的是什么——是雪崩?是爬虫?还是资源耗尽?想明白这点,方案自然就清晰了。
以上就是如何在Golang中实现请求限流_使用Channel和计数器控制并发的详细内容,更多请关注创想鸟其它相关文章!
版权声明:本文内容由互联网用户自发贡献,该文观点仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。
如发现本站有涉嫌抄袭侵权/违法违规的内容, 请发送邮件至 chuangxiangniao@163.com 举报,一经查实,本站将立刻删除。
发布者:程序猿,转转请注明出处:https://www.chuangxiangniao.com/p/1428864.html
微信扫一扫
支付宝扫一扫