dongxiaoxing3058
2019-02-27 07:25
浏览 192
已采纳

限速功能40 /秒,带有“ golang.org/x/time/rate”

I'm trying to use "golang.org/x/time/rate" to build a function which blocks until a token is free. Is this the correct way to use the library to rate limit blocks of code to 40 requests per second, with a bucket size of 2.

type Client struct {
    limiter        *rate.Limiter
    ctx context.Context
}

func NewClient() *Client {
    c :=Client{}
    c.limiter = rate.NewLimiter(40, 2)
    c.ctx = context.Background()
    return &c
}

func (client *Client) RateLimitFunc() {

    err := client.limiter.Wait(client.ctx)
    if err != nil {
        fmt.Printf("rate limit error: %v", err)
    }
}

To rate limit a block of code I call

RateLimitFunc()

I don't want to use a ticker as I want the rate limiter to take into account the length of time the calling code runs for.

  • 点赞
  • 写回答
  • 关注问题
  • 收藏
  • 邀请回答

1条回答 默认 最新

  • doulu1945 2019-02-27 07:58
    已采纳

    Reading the docs here; link

    You can see that the first param to NewLimiter is of type rate.Limit.

    If you want 40 requests / second then that translates into a rate of 1 requests every 25ms.

    You can create that by doing:

    limiter := rate.NewLimiter(rate.Every(25 * time.Millisecond), 2)
    

    Side note:

    • In generate a context ctx should not be stored on a struct and should be per-request. It would appear that Client will be reused, thus you could pass a context to the RateLimitFunc() or where ever appropriate instead of storing a single context on the client struct.
    点赞 打赏 评论

相关推荐 更多相似问题