s_cherry_e/sr/internal/component/rate/func_limiter_test.go

package rate

import (
	"fmt"
	"testing"
	"time"
)

func TestFuncLimiter_Check(t *testing.T) {
	limiter := NewFuncLimiter(1000, 2, 666)
	limiter.Add("default", 1000, 5, 666)

	fmt.Println("default")
	for range 6 {
		errCode := limiter.Check(time.Now().UnixMilli(), "default")
		fmt.Println(errCode)
	}

	fmt.Println("other")
	for range 5 {
		errCode := limiter.Check(time.Now().UnixMilli(), "other")
		fmt.Println(errCode)
	}
}