Golang限流器rate使用

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.youkuaiyun.com/article/details/110239775