rate包是Google基于令牌桶的算法实现的限流器,可以在服务的限流中使用。
我们通过一个例子来使用rate
package main
import (
"fmt"
"time"
"context"
"golang.org/x/time/rate"
)
func main() {
//初始化 limiter 每秒10个令牌,令牌桶容量为20
limiter := rate.NewLimiter(rate.Every(time.Millisecond*100),20)
for i := 0 ; i < 25 ; i++ {
if limiter.Allow() {
fmt.Println("success") //do something
}else {
fmt.Println("busy")
}
}
//阻塞直到获取足够的令牌或者上下文取消
ctx,_ := context.WithTimeout(context.Background(),time.Second*10)
fmt.Println("start get token",time.Now())
err := limiter.WaitN(ctx,20)
if err != nil {
fmt.Println("error",err)
}
fmt.Println("success get token",time.Now())
}
说明:rate.NewLimiter用于初始化限流器,指定限流的速率及令牌桶的最大容量
Limiter有三个主要的方法 Allow、Reserve和Wait,最常用的是Wait和Allow方法
这三个方法每调用一次都会消耗一个令牌,这三个方法的区别在于没有令牌时,他们的处理方式不同
Allow: 如果没有令牌,则直接返回false
Reserve:如果没有令牌,则返回一个reservation,
Wait:如果没有令牌,则等待直到获取一个令牌或者其上下文被取消。
想要深入学习rate,可以学习rate的源码:https://skaygo.blog.csdn.net/article/details/110239775