Golang time/rate 限速器

Go 提供了一个package(golang.org/x/time/rate) 用来方便的对速度进行限制,下面就用示例来说明下如何使用。

首先创建一个rate.Limiter,其有两个参数,第一个参数为每秒发生多少次事件,第二个参数是其缓存最大可存多少个事件。

rate.Limiter提供了三类方法用来限速

  • Wait/WaitN 当没有可用或足够的事件时,将阻塞等待 推荐实际程序中使用这个方法
  • Allow/AllowN 当没有可用或足够的事件时,返回false
  • Reserve/ReserveN 当没有可用或足够的事件时,返回 Reservation,和要等待多久才能获得足够的事件。

Wait/WaitN 示例

package main

import (
    "fmt"
    "time"

    "golang.org/x/net/context"
    "golang.org/x/time/rate"
)

func main() {
    l := rate.NewLimiter(20, 5)
    c, _ := context.WithCancel(context.TODO())
    fmt.Println(l.Limit(), l.Burst())
    for {
        l.Wait(c)
        time.Sleep(200 * time.Millisecond)
        fmt.Println(time.Now().Format("2016-01-02 15:04:05.000"))
    }
}

Allow/AllowN 示例

func main() {
    l := rate.NewLimiter(1, 3)
    for {
        if l.AllowN(time.Now(), 2) {
            fmt.Println(time.Now().Format("2016-01-02 15:04:05.000"))
        } else {
            time.Sleep(6     * time.Second)
        }
    }
}

Reserve/ReserveN 示例

func main() {
    l := rate.NewLimiter(1, 3)
    for {
        r := l.ReserveN(time.Now(), 3)
        time.Sleep(r.Delay())
        fmt.Println(time.Now().Format("2016-01-02 15:04:05.000"))
    }
}
Reserve/ReserveN 还有更多其他的用法,需要的就根据需要查阅文档吧:)

推荐阅读更多精彩内容