mirrors/dendrite
1
0
Fork 0
mirror of https://github.com/matrix-org/dendrite.git synced 2025-12-25 07:43:10 -06:00
Code Issues Packages Projects Releases Wiki Activity

Take write lock for rate limit map

Browse source
This commit is contained in:
Neil Alexander 2020-10-16 13:26:04 +01:00
parent 4a7fb9c045
commit d52362a106
No known key found for this signature in database
GPG key ID: A02A2019A2BB0944
1 changed files with 7 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
clientapi/routing/rate_limiting.go
View file

@ -73,8 +73,13 @@ func (l *rateLimits) rateLimit(req *http.Request) *util.JSONResponse {
// let's create one. // let's create one.
rateLimit, ok := l.limits[caller] rateLimit, ok := l.limits[caller]
if !ok { if !ok {
l.limits[caller] = make(chan struct{}, l.requestThreshold) rateLimit = make(chan struct{}, l.requestThreshold)
rateLimit = l.limits[caller]
l.limitsMutex.RUnlock()
l.limitsMutex.Lock()
l.limits[caller] = rateLimit
l.limitsMutex.Unlock()
l.limitsMutex.RLock()
} }
// Check if the user has got free resource slots for this request. // Check if the user has got free resource slots for this request.