Golang用channel实现令牌桶限流器

本文深入剖析了为何不能直接用 Go 的 time.Ticker 实现真正的令牌桶限流器——因其缺乏容量限制、无法缓冲突发流量、也不支持令牌预存；进而提出一种简洁高效的替代方案：利用带缓冲的 channel（chan struct{}）配合 goroutine 定时注入令牌，并通过 select 非阻塞判断实现容量控制与超限拒绝，从而在保持代码轻量的同时，精准满足令牌桶三大核心特性——突发容忍、恒定速率填充和严格超限拦截。

为什么不用 time.Ticker 做令牌桶？

直接用 time.Ticker 模拟“每秒放一个令牌”看似简单，但会漏掉突发流量的缓冲能力——它本质是固定节奏的生产者，没有容量概念，也不支持预存令牌。真正的令牌桶必须能：允许短时爆发（桶没空）、限制长期平均速率（填充速率恒定）、拒绝超限请求（取令牌失败）。所以得自己维护一个带容量和时间戳的状态。

用 chan struct{} 实现最小可行令牌桶

channel 本身不能“存数字”，但可以用它做信号队列：每个 struct{} 代表一个可用令牌。关键在控制入队节奏和容量上限。

• 初始化时用 make(chan struct{}, capacity) 创建带缓冲的 channel，容量即桶大小
• 启动一个 goroutine，按 time.Duration(1e9 / rate)（纳秒）周期向 channel 发送令牌，但只在未满时发：select { case ch
• 取令牌用 select { case ，非阻塞判断
• 注意：不要用 len(ch) 判断剩余令牌数——它不准确，因并发读写下值可能瞬变；始终以能否成功接收为准

time.Now() 和填充逻辑放哪？别在每次 Take() 里算

常见错误是每次调用 Take() 都算距上次填充过了多久、该补几个令牌——这既慢又难保证线程安全。正确做法是让填充 goroutine 独占状态更新，主逻辑只做原子收发。

• 填充 goroutine 里用 time.Sleep() 或 time.AfterFunc() 控制间隔，避免忙等
• 如果填充速率很低（如每分钟 10 个），别用 time.Sleep() 精确卡点，改用“每次填充后重算下次时间”，防止 drift 积累
• channel 缓冲区满时，填充 goroutine 应跳过本次发送，而不是阻塞——否则整个限流器会卡住

真实场景下必须加锁？不一定，但得看怎么用

纯 channel 方案在高并发下能跑，但有个隐藏前提：所有 Take() 调用都走同一个 channel 的 recv 操作，而 Go runtime 对 buffered channel 的 send/recv 是原子的。所以只要不额外读写共享变量（比如记录 lastFillTime），就无需显式锁。

• 如果需要精确统计剩余令牌数、或支持动态调整速率/容量，则必须引入 sync.Mutex 或 atomic 变量
• 用 sync.Pool 复用 token 结构体毫无意义——struct{} 零开销，且 channel 内部不分配堆内存
• 测试时别用 runtime.Gosched() 模拟并发，直接起 100+ goroutine 并发调 Take() 更靠谱

真正难的是把填充节奏和请求响应耦合对齐——比如填充间隔设成 100ms，但某次网络延迟导致连续 3 个请求在 5ms 内到达，这时桶是否该允许全过？答案取决于你填的是“固定速率”还是“平滑速率”。大多数业务要的是后者，那就得换更稳的实现，比如基于 time.Now() + atomic 的滑动窗口式令牌桶。

以上就是《Golang用channel实现令牌桶限流器》的详细内容，更多关于的资料请关注golang学习网公众号！