mirrors/gotosocial
1
0
Fork 0
mirror of https://github.com/superseriousbusiness/gotosocial.git synced 2025-10-30 03:12:24 -05:00
Code Issues Projects Releases Packages Wiki Activity Actions

[performance] simpler throttling logic (#2407)

Browse source 
* reduce complexity of throttling logic to use 1 queue and an atomic int

* use atomic add instead of CAS, add throttling test
This commit is contained in:
kim 2023-12-16 11:53:42 +00:00 • committed by GitHub
commit d56a8d095e
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
4 changed files with 208 additions and 40 deletions
Download patch file Download diff file Expand all files Collapse all files

8
internal/middleware/ratelimit.go
View file

@ -29,6 +29,8 @@ import (
"github.com/superseriousbusiness/gotosocial/internal/util"
"github.com/ulule/limiter/v3"
"github.com/ulule/limiter/v3/drivers/store/memory"
apiutil "github.com/superseriousbusiness/gotosocial/internal/api/util"
)
const rateLimitPeriod = 5 * time.Minute
@ -141,10 +143,12 @@ func RateLimit(limit int, exceptions []string) gin.HandlerFunc {
if context.Reached {
// Return JSON error message for
// consistency with other endpoints.
c.AbortWithStatusJSON(
apiutil.Data(c,
http.StatusTooManyRequests,
gin.H{"error": "rate limit reached"},
apiutil.AppJSON,
apiutil.ErrorRateLimitReached,
)
c.Abort()
return
}