cartodb/Windshaft-cartodb
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

lua script and rate limit process

Browse Source
This commit is contained in:
Simon Martín 2018-02-14 18:39:57 +01:00
parent 5fca005a3f
commit c059f44bf1
1 changed files with 31 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

33
lib/cartodb/middleware/rate-limit.js
View File

@ -37,7 +37,7 @@ const ENDPOINT_KEY_REDIS = 'rate-limit:store:';
const USER_KEY_REDIS = 'rate-limit:status:';
module.exports = function rateLimitMiddleware (metadataBackend) {
module.exports = function rateLimitMiddleware(metadataBackend) {
return function rateLimit(req, res, next) {
const user = 'cdb';
@ -46,16 +46,26 @@ module.exports = function rateLimitMiddleware (metadataBackend) {
const redisParams = [
getLuaScript(),
3,
2,
getStoreKey(user, endpointGroup), // KEY[1]
getStatusKey(user, endpointGroup) // KEY[2]
];
metadataBackend.redisCmd(REDIS_DB, 'EVAL', redisParams, function(err, {isBloqued, limit, remaining, retry, reset}) {
metadataBackend.redisCmd(REDIS_DB, 'EVAL', redisParams, function(err, data) {
if (err) {
return next(err);
}
if (!data || !Array.isArray(data)) {
return next();
}
const isBloqued = data[0];
const limit = data[1];
const remaining = data[2];
const retry = data[3];
const reset = data[4];
res.set({
'X-Rate-Limit-Limit': limit,
'X-Rate-Limit-Remaining': remaining,
@ -63,13 +73,13 @@ module.exports = function rateLimitMiddleware (metadataBackend) {
'X-Rate-Limit-Reset': reset
});
if(isBloqued) {
if (isBloqued) {
const err = new Error('You are over the limits.');
err.http_status = 429;
return next(err);
}
next();
return next();
});
};
};
@ -110,7 +120,16 @@ function getStatusKey(user, endpointGroup) {
function getLuaScript() {
return `
local userValues = redis.call("HGETALL", KEYS[1])
return redis.call("CL.THROTTLE", KEYS[2], userValues[b], userValues[c], userValues[p])
local limmits = {}
local limmitsArray = redis.call("HGETALL", KEYS[1])
if table.getn(limmitsArray) == 4 then
limmits[limmitsArray[1]] = limmitsArray[2]
limmits[limmitsArray[3]] = limmitsArray[4]
limmits[limmitsArray[5]] = limmitsArray[6]
return redis.call("CL.THROTTLE", KEYS[2], limmits['b'], limmits['c'], limmits['p'])
else
return nil
end
`;
}