golang 请求处理频率控制

/*
   go请求处理
   频率控制

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

package main

import (
    "fmt"
    "time"
)

func main() {
    /*
       首先我们看下基本的频率限制
       假设我们得控制请求频率
       我们使用一个通道来处理所有的这些请求
       这里向requests发送5个数据 然后关闭requests通道
    */

    requests := make(chan int, 5)
    for i := 1; i <= 5; i++ {
        requests <- i
    }

    close(requests)

    /*
      这个limiter的Ticker每隔200毫秒结束通道阻塞
      这个limiter就是我们的控制器
    */

    limiter := time.Tick(time.Millisecond * 200)

    /*
       通过阻塞从limiter通道接收数据 我们将请求处理控制在每隔200毫秒
       处理一个请求
       注意<-limiter的阻塞作用
    */

    for req := range requests {
        <-limiter
        fmt.Println("request", req, time.Now())
    }

    /*
       我们可以保持正常的请求频率限制 但也允许请求短时间内爆发
       我们可以通过通道缓存来实现
       比如下面这个burstyLimiter 就允许同时处理3个事件
    */

    burstyLimiter := make(chan time.Time, 3)
    for i := 0; i < 3; i++ {
        burstyLimiter <- time.Now()
    }

    /*
       然后每隔200毫秒再向burstyLimiter发送一个数据
       这里不断地每隔200毫秒向burstyLimiter发送数据
    */

    go func() {
        for t := range time.Tick(time.Millisecond * 200) {
            burstyLimiter <- t
        }
    }()

    /*
      这里模拟5个请求 burstyRequest的前面3个请求会连续被处理

    */

    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())
    }
}