Go Rate Limiter

ratelimiter is a multi-ratelimiter for go. It is easily configurable, allows whitelist, and easy to use.

Rate Limiters

typeLimiterinterface {
Limit(string) bool
}

You create a limiter by providing it with a configured Limit which is either manually created or parsed from a string

// Using a parsed limitrl:=ratelimiter.MustParseLimit("10/1m")
limiter:=ratelimiter.NewRedisLimiter(redisPool, rls)
// Manually created limit// one per minute, non-global (IP specific)limit:= ratelimiter.Limit{
Global: false,
Limit: 1,
Dur: time.Minute
}
limiter:=ratelimiter.NewRedisLimiter(redisPool, limit)

Limits

Limits are defined as the following struct

typeLimitstruct {
GlobalboolLimitintDur time.Duration
}

You can either manually create them or you can use the parser. The parser parses strings separated into three sections

• Section 1: How many requests to limit to
• Section 2: For what duration
• Section 3: (optional) is this limit global?

Example limit strings

// create a Limit which limits requests to 1 per second by identifier (IP address)rl:=ratelimiter.MustParseLimit("1/1s")
// create a Limit which limits requests to 10 per 5 minutes by identifier (IP address)rl:=ratelimiter.MustParseLimit("10/5m")
// create a Limit which limits requests to 1000 per day globallyrl:=ratelimiter.MustParseLimit("1000/24h/g")
// create chained Limitsrls:=ratelimiter.MustParseLimits([]string{"1000/24h/g", "1/1s", "10/1m"})

Ratelimits are executed in descending order of duration. If you configure the following two limits:

1. 1/1s
2. 10/1h

They will be executed in reverse order because the larger groups should be checked first. Global rate limits are also executed first.

Whitelist

To whitelist IP addresses use the WhitelistedLimiter which wraps a real limiter

rl:=ratelimiter.MustParseLimit("10/1m")
limiter:=ratelimiter.NewRedisLimiter(redisPool, rls)
whitelist:= []string{"127.0.0.1"}
wl:=ratelimiter.NewWhitelistedLimiter(limiter, whitelist)

Implemented Limiters

For now only RedisRatelimiter is implemented, however more can be added. Feel free to submit a PR.