Golang BucketLimit Examples

Example #1

File: ratelimite_test.go Project: lysu/go-rate-limiter

func TestRedisLimiter(t *testing.T) {
	clearUpRedis()
	rf := limiter.RateLimiter(limiter.UseRedis(redisPool), limiter.BucketLimit(1000*time.Millisecond, 10))
	rr := rf()
	assertRateLimiter(rr, t)
	assert.Equal(t, 100, size("rl-key1"))
}

Example #2

File: ratelimite_test.go Project: lysu/go-rate-limiter

func TestFloodThreshold(t *testing.T) {
	clearUpRedis()
	rf := limiter.RateLimiter(limiter.UseRedis(redisPool), limiter.BucketLimit(1000*time.Millisecond, 10), limiter.FloodThreshold(5))
	rr := rf()
	assertRateLimiter(rr, t)
	assert.Equal(t, 51, size("rl-key1"))
	time.Sleep(1500 * time.Millisecond)
	assertRateLimiter(rr, t)
}

Example #3

File: ratelimite_test.go Project: lysu/go-rate-limiter

func TestMemoryLimiter(t *testing.T) {
	rf := limiter.RateLimiter(limiter.UseMemory(), limiter.BucketLimit(1000*time.Millisecond, 10))
	mr := rf()
	assertRateLimiter(mr, t)
}