golang实现限制每秒多少次的限频操作

前言

一些函数的执行可能会限制频率，比如某个api接口要求每秒最大请求30次。下面记录了自己写的限频和官方的限频

代码

// 加锁限频，输出次数大概率小于最大值
func ExecLimit(lastExecTime *time.Time, l *sync.RWMutex ,maxTimes int, perDuration time.Duration, f func()) {
    l.Lock()
    defer l.Unlock()
	// per times cost time(s)
	SecondsPerTimes := float64(perDuration) / float64(time.Second) / float64(maxTimes)

	now := time.Now()
	interval := now.Sub(*lastExecTime).Seconds()
	if interval < SecondsPerTimes {
		time.Sleep(time.Duration(int64((SecondsPerTimes-interval)*1000000000)) * time.Nanosecond)
	}
	f()
	*lastExecTime = time.Now()
}

// 官方的，需要引用	"golang.org/x/time/rate"
// 基本上可以达到满值，比自己写的更优
func ExecLimit2(l *rate.Limiter, f func()) {
	go func() {
		l.Wait(context.Background())
		f()
	}()
}

使用

func TestExecLimit(t *testing.T) {
	runtime.GOMAXPROCS(runtime.NumCPU())
	go func() {
		var lastExecTime time.Time
		var l sync.RWMutex
		for {
			ExecLimit(&lastExecTime, &l, 10, time.Second, func() {
				fmt.Println("do")
			})
		}
	}()

	select {
	case <-time.After(1 * time.Second):
		fmt.Println("1秒到时")
	}
}

func TestExecLimit2(t *testing.T) {
	runtime.GOMAXPROCS(runtime.NumCPU())

	l := rate.NewLimiter(1, 30)

	go func() {
		for {
            ExecLimit2(l, func() {
				fmt.Println("do")
			})
		}
	}()

	select {
	case <-time.After(1 * time.Second):
		fmt.Println("1秒到时")
	}
}

输出:

一秒内输出了<=10次 "do"

如何在多节点服务中限制频

上述使用，定义在某个服务节点的全局变量lastExecTime仅仅会对该服务的函数f()操作限频，如果在负载均衡后，多个相同服务的节点，对第三方的接口累计限频，比如三个服务共同拉取第三方接口，合计限频为30次/s.
则，必须将lastExecTime的获取，从redis等共享中间件中获取，而不应该从任何一个单点服务获取。