package main
import (
"context"
"fmt"
"golang.org/x/time/rate"
"log"
"time"
)
func main() {
r := rate.NewLimiter(1, 5) //1表示每次放进筒内的数量,桶内的令牌数是5,最大令牌数也是5,这个筒子是自动补充的,你只要取了令牌不管你取多少个,这里都会在每次取完后自动加1个进来,因为我们设置的是1
ctx := context.Background()
for {
err := r.WaitN(ctx, 2)//每次消耗2个,因为之前放进去一个,所以桶里面只有一个,消耗完了还会放进去,所以这段代码再执行到第4次的时候筒里面就空了
if err != nil {
log.Fatal(err)
}
fmt.Println(time.Now().Format("2016-01-02 15:04:05"))
time.Sleep(time.Second)
}
}
原文:https://www.cnblogs.com/hualou/p/12081458.html