mirror of
https://github.com/overleaf/overleaf.git
synced 2024-11-07 20:31:06 -05:00
27 lines
822 B
CoffeeScript
27 lines
822 B
CoffeeScript
settings = require("settings-sharelatex")
|
|
RedisWrapper = require('./RedisWrapper')
|
|
rclient = RedisWrapper.client('ratelimiter')
|
|
|
|
|
|
module.exports = RateLimiter =
|
|
|
|
_buildKey: (endpoint, subject) ->
|
|
return "RateLimiter:#{endpoint}:{#{subject}}"
|
|
|
|
addCount: (opts, callback = (err, shouldProcess)->)->
|
|
k = RateLimiter._buildKey(opts.endpointName, opts.subjectName)
|
|
multi = rclient.multi()
|
|
multi.incr(k)
|
|
multi.get(k)
|
|
multi.expire(k, opts.timeInterval)
|
|
multi.exec (err, results)->
|
|
count = results[1]
|
|
# account for the different results from `multi` when using cluster
|
|
if count instanceof Object
|
|
count = count[1]
|
|
allow = count < opts.throttle
|
|
callback err, allow
|
|
|
|
clearRateLimit: (endpointName, subject, callback) ->
|
|
k = RateLimiter._buildKey(endpointName, subject)
|
|
rclient.del k, callback
|