golang頻率限制 rate詳解

Posted on by

golang.org/x/time/rate 提對速度進行限制的算法

l := rate.NewLimiter(1, 3) // 一個參數為每秒發生多少次事件，第二個參數是最大可運行多少個事件(burst)

Limter提供三中主要的函數 Allow, Reserve, Wait. 大部分時候使用Wait

Wait/WaitN 當沒有可用事件時，將阻塞等待

c, _ := context.WithCancel(context.TODO())
for {
 l.Wait(c)
 fmt.Println(time.Now().Format("04:05.000"))
}

輸出

07:35.055
07:35.055
07:35.055
07:36.060
07:37.059
07:38.059

緩存3次後，每秒執行一次

Allow/AllowN 當沒有可用事件時，返回false
for {
 if (l.AllowN(time.Now(), 1)) {
  fmt.Println(time.Now().Format("04:05.000"))
 } else {
  time.Sleep(1 * time.Second / 10)
  fmt.Println(time.Now().Format("Second 04:05.000"))
 }

}

Reserve/ReserveN 當沒有可用事件時，返回 Reservation，和要等待多久才能獲得足夠的事件

for {
  r := l.ReserveN(time.Now(), 1)
  s := r.Delay()
  time.Sleep(s)
   fmt.Println(s, time.Now().Format("04:05.000"))

 }

補充：關於golang.org/x包問題

由於谷歌被墻，跟谷歌相關的模塊無法通過go get來下載

解決方法：

git clone https://github.com/golang/net.git $GOPATH/src/github.com/golang/net

git clone https://github.com/golang/sys.git $GOPATH/src/github.com/golang/sys

git clone https://github.com/golang/tools.git $GOPATH/src/github.com/golang/tools

ln -s $GOPATH/src/github.com/golang $GOPATH/src/golang.org/x

如果是Windows下，最後一條可以替換為mklink命令，或者直接拷貝文件夾修改一下名稱即可。

go的官方包都在這個目錄下：

https://github.com/golang

以上為個人經驗，希望能給大傢一個參考，也希望大傢多多支持WalkonNet。如有錯誤或未考慮完全的地方，望不吝賜教。

推薦閱讀：