速率限制功能 40/second with "golang.org/x/time/rate"

Question

我正在尝试使用“golang.org/x/time/rate”来构建一个函数，该函数在令牌可用之前一直阻塞。这是使用库将代码块速率限制为每秒 40 个请求，存储桶大小为 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)
    }
}

对我调用的代码块进行速率限制

RateLimitFunc()

我不想使用自动收报机，因为我希望速率限制器考虑调用代码运行的时间长度。

Answer 1

在这里阅读文档；关联

您可以看到第一个参数 toNewLimiter的类型为rate.Limit。

如果您想要 40 个请求/秒，那么这转化为每 25 毫秒 1 个请求的速率。

您可以通过执行以下操作来创建它：

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

边注：

• 在生成上下文中ctx不应存储在结构中，而应按请求进行。看起来Client会被重用，因此您可以将上下文传递给RateLimitFunc()或 任何合适的地方，而不是在客户端结构上存储单个上下文。