golang dynamic rate limiter

gistfile1.txt

package main

import (
	"fmt"
	"sync"
	"time"

	"go.uber.org/ratelimit"
)

type limiter struct {
	l   ratelimit.Limiter
	m   sync.RWMutex
	qps int
}

func (l *limiter) ReSet(qps int) {
	l.m.Lock()
	l.qps = qps
	l.l = ratelimit.New(qps, ratelimit.WithoutSlack)
	l.m.Unlock()
}

func (l *limiter) Take() time.Time {
	l.m.RLock()
	now := l.l.Take()
	l.m.RUnlock()
	return now
}

func main() {
	rl := &limiter{}
	rl.ReSet(10)

	prev := time.Now()
	for i := 0; i < 10; i++ {
		now := rl.Take()
		if i > 0 {
			fmt.Println(i, now.Sub(prev))
		}
		prev = now
		if i == 3 {
			rl.ReSet(5)
		}
		if i == 7 {
			rl.ReSet(3)
		}
	}

}