overleaf/services/web/app/coffee/infrastructure/RateLimiter.coffee
Shane Kilkelly 25956d4c62 Fix up tests
2017-01-13 16:04:26 +00:00

28 lines
937 B
CoffeeScript

settings = require("settings-sharelatex")
RedisWrapper = require('./RedisWrapper')
rclient = RedisWrapper.client('ratelimiter')
RollingRateLimiter = require('rolling-rate-limiter')
module.exports = RateLimiter =
addCount: (opts, callback = (err, shouldProcess)->)->
namespace = "RateLimit:#{opts.endpointName}:"
k = "{#{opts.subjectName}}"
limiter = RollingRateLimiter({
redis: rclient,
namespace: namespace,
interval: opts.timeInterval * 1000,
maxInInterval: opts.throttle
})
limiter k, (err, timeLeft, actionsLeft) ->
if err?
return callback(err)
allowed = timeLeft == 0
console.log ">> limit", namespace, k, timeLeft, actionsLeft, ", allowed", allowed
callback(null, allowed)
clearRateLimit: (endpointName, subject, callback) ->
# same as the key which will be built by RollingRateLimiter (namespace+k)
keyName = "RateLimit:#{endpointName}:{#{subject}}"
rclient.del keyName, callback