| 123456789101112131415161718192021222324252627 |
- 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
- 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
|
