Go by Example 中文版: 速率限制
 
package main |
|
import ( "fmt" "time" ) |
|
func main() { |
|
|
首先，我们将看一个基本的速率限制。 假设我们想限制对收到请求的处理，我们可以通过一个渠道处理这些请求。 |
requests := make(chan int, 5)
for i := 1; i <= 5; i++ {
requests <- i
}
close(requests)
|
|
|
limiter := time.Tick(200 * time.Millisecond)
|
|
通过在每次请求前阻塞 |
for req := range requests {
<-limiter
fmt.Println("request", req, time.Now())
}
|
|
有时候我们可能希望在速率限制方案中允许短暂的并发请求，并同时保留总体速率限制。
我们可以通过缓冲通道来完成此任务。
|
burstyLimiter := make(chan time.Time, 3)
|
|
填充通道，表示允许的爆发（bursts）。 |
for i := 0; i < 3; i++ {
burstyLimiter <- time.Now()
}
|
|
每 200ms 我们将尝试添加一个新的值到 |
go func() {
for t := range time.Tick(200 * time.Millisecond) {
burstyLimiter <- t
}
}()
|
|
现在，模拟另外 5 个传入请求。
受益于 |
burstyRequests := make(chan int, 5)
for i := 1; i <= 5; i++ {
burstyRequests <- i
}
close(burstyRequests)
for req := range burstyRequests {
<-burstyLimiter
fmt.Println("request", req, time.Now())
}
}
|
|
运行程序，我们看到第一批请求意料之中的大约每 200ms 处理一次。 |
$ go run rate-limiting.go request 1 2012-10-19 00:38:18.687438 +0000 UTC request 2 2012-10-19 00:38:18.887471 +0000 UTC request 3 2012-10-19 00:38:19.087238 +0000 UTC request 4 2012-10-19 00:38:19.287338 +0000 UTC request 5 2012-10-19 00:38:19.487331 +0000 UTC |
|
第二批请求，由于爆发（burstable）速率控制，我们直接连续处理了 3 个请求， 然后以大约每 200ms 一次的速度，处理了剩余的 2 个请求。 |
request 1 2012-10-19 00:38:20.487578 +0000 UTC request 2 2012-10-19 00:38:20.487645 +0000 UTC request 3 2012-10-19 00:38:20.487676 +0000 UTC request 4 2012-10-19 00:38:20.687483 +0000 UTC request 5 2012-10-19 00:38:20.887542 +0000 UTC |
下一个例子: 原子计数器