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用于初始化限流器，指定限流的速率