Go 请求处理频率控制

频率控制是控制资源利用和保证服务高质量的重要机制。Go可以使用goroutine，channel和ticker来以优雅的方式支持频率控制。

  1. package main
  3. import "time"
  4. import "fmt"
  6. func main() {
  8.     // 首先我们看下基本的频率限制。假设我们得控制请求频率，
  9.     // 我们使用一个通道来处理所有的这些请求，这里向requests
  10.     // 发送5个数据，然后关闭requests通道
  11.     requests := make(chan int, 5)
  12.     for i := 1; i <= 5; i++ {
  13.         requests <- i
  14.     }
  15.     close(requests)
  17.     // 这个limiter的Ticker每隔200毫秒结束通道阻塞
  18.     // 这个limiter就是我们频率控制处理器
  19.     limiter := time.Tick(time.Millisecond * 200)
  21.     // 通过阻塞从limiter通道接受数据，我们将请求处理控制在每隔200毫秒
  22.     // 处理一个请求，注意`<-limiter`的阻塞作用。
  23.     for req := range requests {
  24.         <-limiter
  25.         fmt.Println("request", req, time.Now())
  26.     }
  28.     // 我们可以保持正常的请求频率限制，但也允许请求短时间内爆发
  29.     // 我们可以通过通道缓存来实现，比如下面的这个burstyLimiter
  30.     // 就允许同时处理3个事件。
  31.     burstyLimiter := make(chan time.Time, 3)
  33.     // 填充burstyLimiter，先发送3个数据
  34.     for i := 0; i < 3; i++ {
  35.         burstyLimiter <- time.Now()
  36.     }
  38.     // 然后每隔200毫秒再向burstyLimiter发送一个数据，这里是不断地
  39.     // 每隔200毫秒向burstyLimiter发送数据
  40.     go func() {
  41.         for t := range time.Tick(time.Millisecond * 200) {
  42.             burstyLimiter <- t
  43.         }
  44.     }()
  46.     // 这里模拟5个请求，burstyRequests的前面3个请求会连续被处理，
  47.     // 因为burstyLimiter被先连续发送3个数据的的缘故，而后面两个
  48.     // 则每隔200毫秒处理一次
  49.     burstyRequests := make(chan int, 5)
  50.     for i := 1; i <= 5; i++ {
  51.         burstyRequests <- i
  52.     }
  53.     close(burstyRequests)
  54.     for req := range burstyRequests {
  55.         <-burstyLimiter
  56.         fmt.Println("request", req, time.Now())
  57.     }
  58. }

运行结果

  1. request 1 2014-02-21 14:20:05.2696437 +0800 CST
  2. request 2 2014-02-21 14:20:05.4696637 +0800 CST
  3. request 3 2014-02-21 14:20:05.6696837 +0800 CST
  4. request 4 2014-02-21 14:20:05.8697037 +0800 CST
  5. request 5 2014-02-21 14:20:06.0697237 +0800 CST
  6. request 1 2014-02-21 14:20:06.0697237 +0800 CST
  7. request 2 2014-02-21 14:20:06.0697237 +0800 CST
  8. request 3 2014-02-21 14:20:06.0707238 +0800 CST
  9. request 4 2014-02-21 14:20:06.2707438 +0800 CST
  10. request 5 2014-02-21 14:20:06.4707638 +0800 CST

我们从输出的结果上可以看出最后的5个输出结果中，前三个的时间是连续的，而后两个的时间是隔了200毫秒。