如何在Golang中实现请求限流_使用Channel和计数器控制并发-Golang-PHP中文网

如何在Golang中实现请求限流_使用Channel和计数器控制并发

P粉602998670

发布： 2025-12-17 01:01:18

原创

198人浏览过

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

如何在golang中实现请求限流_使用channel和计数器控制并发

在 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
}
<p>func (tb *TokenBucket) Allow() bool {
select {
case <-tb.ch:
return true
default:
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
}
<p>func (l *FixedWindowLimiter) Allow() bool {
l.mu.Lock()
defer l.mu.Unlock()</p><pre class="brush:php;toolbar:false;">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

登录后复制

}