go get golang.org/x/time/rate

package main

import (
    "fmt"
    "golang.org/x/time/rate"
    "time"
)

func main() {
    // 每秒生成3個(gè)令牌，桶的容量為10個(gè)令牌
    // 相當(dāng)于每秒最多能處理三個(gè)請(qǐng)求，順時(shí)并發(fā)最多能處理10個(gè)請(qǐng)求
    limiter := rate.NewLimiter(3, 10)

    fmt.Println("Limiter created with rate 3 tokens per second and burst size of 10")
}

if limiter.Allow() {
    fmt.Println("Request allowed")
} else {
    fmt.Println("Request denied")
}

reservation := limiter.Reserve()
if reservation.OK() {
    fmt.Println("Request reserved, delay:", reservation.Delay())
} else {
    fmt.Println("Request cannot be reserved")
}

ctx := context.Background()
if err := limiter.Wait(ctx); err == nil {
    fmt.Println("Request allowed after wait")
} else {
    fmt.Println("Request denied:", err)
}

// Demo_CurrentHandleWorks 模擬并發(fā)處理任務(wù)
func Demo_CurrentHandleWorks() {
	var wg sync.WaitGroup
	numWorkers := 5 // 模擬5個(gè)并發(fā)請(qǐng)求
	// 構(gòu)造限流器：每10s向桶中新增一個(gè)令牌，桶里最多能存放2個(gè)令牌 => 每10s能處理一個(gè)任務(wù)，一定時(shí)間內(nèi)最多能處理2個(gè)任務(wù)（令牌有剩余）
	var l = rate.NewLimiter(rate.Every(time.Second*10), 2)
	for i := 1; i <= numWorkers; i++ {
		wg.Add(1)
		go worker(i, &wg, l)
	}

	wg.Wait()
}

func worker(id int, wg *sync.WaitGroup, limiter *rate.Limiter) {
	defer wg.Done()
	if limiter.Allow() {
		fmt.Printf("Worker %d processed at %s\n", id, time.Now().Format("15:04:05.000"))
	} else {
		fmt.Printf("Worker %d rejected at %s\n", id, time.Now().Format("15:04:05.000"))
	}
}

// Demo_CurrentHandleWithHTTP 模擬HTTP請(qǐng)求限流
func Demo_CurrentHandleWithHTTP() {
	// 每秒最多處理 1 個(gè)請(qǐng)求，允許突發(fā) 2 個(gè)請(qǐng)求
	limiter := rate.NewLimiter(1, 2)
	http.HandleFunc("/", func(w http.ResponseWriter, request *http.Request) {
		if limiter.Allow() {
			fmt.Fprintln(w, "Request allowed")
		} else {
			http.Error(w, "Too Many Requests", http.StatusTooManyRequests)
		}
	})
	fmt.Println("Server started at :8080")
	_ = http.ListenAndServe(":8080", nil)
}

func Demo_HandleWorkWithWait() {
	// 創(chuàng)建一個(gè)限速器，每3秒允許1個(gè)事件
	limiter := rate.NewLimiter(rate.Every(3*time.Second), 1)

	// 模擬10次對(duì)資源的訪問
	for i := 0; i < 10; i++ {
		// 使用limiter.Wait(ctx)等待，直到可以訪問資源（訪問文件/執(zhí)行數(shù)據(jù)庫(kù)查詢等）
		if err := limiter.Wait(context.Background()); err != nil {
			log.Fatalf("Failed to wait for rate limiter: %v", err)
		}
		// 訪問資源
		fmt.Printf("Accessing resource at %v\n", time.Now())
	}
}