diamondburned/arikawa
1
0
Fork 0
mirror of https://github.com/diamondburned/arikawa.git synced 2024-11-27 09:12:53 +00:00
Code Issues Releases Activity

gateway: Fix Send rate limiting

Browse source 
Prior to this commit, gateways can send 120 events before being
throttled at a terribly slow rate of 1 send per minute.

This commit permits gateways to send 5 events before being throttled at
a rate of slightly less than 120 events per minute.
This commit is contained in:
diamondburned 2022-12-03 19:19:07 -08:00
parent 789afaa88f
commit 5811606559
No known key found for this signature in database
GPG key ID: D78C4471CE776659
1 changed files with 17 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

18
utils/ws/throttler.go
View file

@ -6,18 +6,34 @@ import (
"golang.org/x/time/rate"
)
// SendBurst determines the number of gateway commands that can be sent all at
// once before being throttled. The higher the burst, the slower the rate
// limiter recovers.
var SendBurst = 5
// NewSendLimiter returns a rate limiter for throttling gateway commands.
func NewSendLimiter() *rate.Limiter {
return rate.NewLimiter(rate.Every(time.Minute), 120)
const perMinute = 120
return rate.NewLimiter(
// Permit r = minute / (120 - b) commands per second.
rate.Every(time.Minute/(perMinute-time.Duration(SendBurst))),
SendBurst,
)
}
// NewDialLimiter returns a rate limiter for throttling new gateway connections.
func NewDialLimiter() *rate.Limiter {
return rate.NewLimiter(rate.Every(5*time.Second), 1)
}
// NewIdentityLimiter returns a rate limiter for throttling gateway Identify
// commands.
func NewIdentityLimiter() *rate.Limiter {
return NewDialLimiter() // same
}
// NewGlobalIdentityLimiter returns a rate limiter for throttling global
// gateway Identify commands.
func NewGlobalIdentityLimiter() *rate.Limiter {
return rate.NewLimiter(rate.Every(24*time.Hour), 1000)
}