“golang.org/x/time/rate”的速率限制功能 40/秒

怎么入门Golang编程？需要学习哪些知识点？这是新手们刚接触编程时常见的问题；下面golang学习网就来给大家整理分享一些知识点，希望能够给初学者一些帮助。本篇文章就来介绍《“golang.org/x/time/rate”的速率限制功能 40/秒》，涉及到，有需要的可以收藏一下

我正在尝试使用“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()

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

解决方案

在这里阅读文档； link

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

如果您想要每秒 40 个请求，那么这意味着每 25 毫秒 1 个请求。

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

limiter := rate.newlimiter(rate.every(25 * time.millisecond), 2)

旁注：

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

func RateLimit(ctx context.Context) {
    limiter := rate.NewLimiter(40, 10)
    err := limiter.Wait(ctx)
    if err != nil {
        // Log the error and return
    }
    // Do the actual work here
}

如 Zak said 所示，根据 go 文档 context，不要将 context 存储在结构类型中。

今天关于《“golang.org/x/time/rate”的速率限制功能 40/秒》的内容就介绍到这里了，是不是学起来一目了然！想要了解更多关于的内容请关注golang学习网公众号！