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

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值